From 670bef6a91d550b02692ffd4dd6de73566cc5faa Mon Sep 17 00:00:00 2001 From: spl0k Date: Sun, 2 Sep 2018 17:26:18 +0200 Subject: [PATCH] Added support for python migrations --- supysonic/db.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/supysonic/db.py b/supysonic/db.py index 540bd57..78d56f1 100644 --- a/supysonic/db.py +++ b/supysonic/db.py @@ -7,10 +7,11 @@ # # Distributed under terms of the GNU AGPLv3 license. -import time +import importlib import mimetypes import os.path import pkg_resources +import time from datetime import datetime from hashlib import sha1 @@ -535,9 +536,13 @@ def init_database(database_uri): migrations = sorted(pkg_resources.resource_listdir(__package__, 'schema/migration/' + settings['provider'])) for migration in migrations: date, ext = os.path.splitext(migration) - if date <= version.value or ext != '.sql': + if date <= version.value: continue - execute_sql_resource_script('schema/migration/{}/{}'.format(settings['provider'], migration)) + if ext == '.sql': + execute_sql_resource_script('schema/migration/{}/{}'.format(settings['provider'], migration)) + elif ext == '.py': + m = importlib.import_module('.schema.migration.{}.{}'.format(settings['provider'], date), __package__) + m.apply(settings.copy()) version.value = SCHEMA_VERSION # Hack for in-memory SQLite databases (used in tests), otherwise 'db' and 'metadb' would be two distinct databases