From 2f990b9de24f87c6724a5167596792c7b7d7a77b Mon Sep 17 00:00:00 2001 From: blitzmann Date: Tue, 30 Sep 2014 13:20:51 -0400 Subject: [PATCH] Add auto backup when migrating data --- eos/db/migration.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/eos/db/migration.py b/eos/db/migration.py index 2d91d805a..92224c34c 100644 --- a/eos/db/migration.py +++ b/eos/db/migration.py @@ -1,4 +1,6 @@ import config +import shutil +import time def getVersion(db): cursor = db.execute('PRAGMA user_version') @@ -11,6 +13,15 @@ def update(saveddata_engine): return if currversion < config.dbversion: + # Automatically backup database + toFile = "%s/saveddata_migration_%d-%d_%s.db"%( + config.savePath, + currversion, + config.dbversion, + time.strftime("%Y%m%d_%H%M%S")) + + shutil.copyfile(config.saveDB, toFile) + for version in xrange(currversion, config.dbversion): module = __import__('eos.db.migrations.upgrade%d'%(version+1), fromlist=True) upgrade = getattr(module, "upgrade", False)