45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
from logbook import Logger
|
|
import shutil
|
|
import time
|
|
|
|
import config
|
|
from . import migrations
|
|
|
|
pyfalog = Logger(__name__)
|
|
|
|
|
|
def getVersion(db):
|
|
cursor = db.execute('PRAGMA user_version')
|
|
return cursor.fetchone()[0]
|
|
|
|
|
|
def getAppVersion():
|
|
return migrations.appVersion
|
|
|
|
|
|
def update(saveddata_engine):
|
|
dbVersion = getVersion(saveddata_engine)
|
|
appVersion = getAppVersion()
|
|
|
|
if dbVersion == appVersion:
|
|
return
|
|
|
|
if dbVersion < appVersion:
|
|
# Automatically backup database
|
|
toFile = "%s/saveddata_migration_%d-%d_%s.db" % (
|
|
config.savePath,
|
|
dbVersion,
|
|
appVersion,
|
|
time.strftime("%Y%m%d_%H%M%S"))
|
|
|
|
shutil.copyfile(config.saveDB, toFile)
|
|
|
|
for version in range(dbVersion, appVersion):
|
|
func = migrations.updates[version + 1]
|
|
if func:
|
|
pyfalog.info("Applying database update: {0}", version + 1)
|
|
func(saveddata_engine)
|
|
|
|
# when all is said and done, set version to current
|
|
saveddata_engine.execute("PRAGMA user_version = {}".format(appVersion))
|