From 1100c66ce03a059ebe7ece9734e799b49b3a5a9e Mon Sep 17 00:00:00 2001
From: WuKong <rebirthmonkey@gmail.com>
Date: Sat, 23 Dec 2017 21:49:35 +0100
Subject: moonv4 cleanup

Change-Id: Icef927f3236d985ac13ff7376f6ce6314b2b39b0
Signed-off-by: WuKong <rebirthmonkey@gmail.com>
---
 python_moondb/python_moondb/db_manager.py | 82 +++++++++++++++++++++++++++++++
 1 file changed, 82 insertions(+)
 create mode 100644 python_moondb/python_moondb/db_manager.py

(limited to 'python_moondb/python_moondb/db_manager.py')

diff --git a/python_moondb/python_moondb/db_manager.py b/python_moondb/python_moondb/db_manager.py
new file mode 100644
index 00000000..c305284d
--- /dev/null
+++ b/python_moondb/python_moondb/db_manager.py
@@ -0,0 +1,82 @@
+# Copyright 2015 Open Platform for NFV Project, Inc. and its contributors
+# This software is distributed under the terms and conditions of the 'Apache-2.0'
+# license which can be found in the file 'LICENSE' in this package distribution
+# or at 'http://www.apache.org/licenses/LICENSE-2.0'.
+"""
+"""
+
+import os
+import glob
+import importlib
+import argparse
+import logging
+from sqlalchemy import create_engine
+from python_moonutilities import configuration
+from python_moondb.migrate_repo import versions
+
+
+def init_args():
+    parser = argparse.ArgumentParser()
+    parser.add_argument('command', help='command (upgrade or downgrade)',
+                        nargs=1)
+    parser.add_argument("--verbose", "-v", action='store_true',
+                        help="verbose mode")
+    parser.add_argument("--debug", "-d", action='store_true',
+                        help="debug mode")
+    args = parser.parse_args()
+
+    FORMAT = '%(asctime)-15s %(levelname)s %(message)s'
+    if args.debug:
+        logging.basicConfig(
+            format=FORMAT,
+            level=logging.DEBUG)
+    elif args.verbose:
+        logging.basicConfig(
+            format=FORMAT,
+            level=logging.INFO)
+    else:
+        logging.basicConfig(
+            format=FORMAT,
+            level=logging.WARNING)
+
+    requests_log = logging.getLogger("requests.packages.urllib3")
+    requests_log.setLevel(logging.WARNING)
+    requests_log.propagate = True
+
+    logger = logging.getLogger("moon.db.manager")
+    return args, logger
+
+
+def init_engine():
+    db_conf = configuration.get_configuration("database")["database"]
+    return create_engine(db_conf['url'])
+
+
+def main(command, logger, engine):
+    files = glob.glob(versions.__path__[0] + "/[0-9][0-9][0-9]*.py")
+    for filename in files:
+        filename = os.path.basename(filename).replace(".py", "")
+        o = importlib.import_module(
+            "moon_db.migrate_repo.versions.{}".format(filename))
+        logger.info("Command is {}".format(command))
+        if command in ("upgrade", "u", "up"):
+            logger.info(
+                "upgrading moon_db.migrate_repo.versions.{}".format(filename))
+            o.upgrade(engine)
+        elif command in ("downgrade", "d", "down"):
+            logger.info(
+                "downgrading moon_db.migrate_repo.versions.{}".format(
+                    filename))
+            o.downgrade(engine)
+        else:
+            logger.critical("Cannot understand the command!")
+
+
+def run():
+    args, logger = init_args()
+    engine = init_engine()
+    main(args.command[0], logger, engine)
+
+
+if __name__ == "__main__":
+    run()
-- 
cgit 1.2.3-korg