fix upgrade_db

This commit is contained in:
Phill Ridout 2015-02-18 21:13:12 +00:00
parent 5e8ad1710b
commit a6dfbdcf71
1 changed files with 5 additions and 1 deletions

View File

@ -145,9 +145,13 @@ def upgrade_db(url, upgrade):
version_meta = session.query(Metadata).get('version')
if version_meta is None:
# Tables have just been created - fill the version field with the most recent version
version = upgrade.__version__
if session.query(Metadata).get('dbversion'):
version = 0
else:
version = upgrade.__version__
version_meta = Metadata.populate(key='version', value=version)
session.add(version_meta)
session.commit()
else:
version = int(version_meta.value)
if version > upgrade.__version__: