aboutsummaryrefslogtreecommitdiffstats
path: root/python_moondb/python_moondb/db_manager.py
diff options
context:
space:
mode:
Diffstat (limited to 'python_moondb/python_moondb/db_manager.py')
-rw-r--r--python_moondb/python_moondb/db_manager.py82
1 files changed, 0 insertions, 82 deletions
diff --git a/python_moondb/python_moondb/db_manager.py b/python_moondb/python_moondb/db_manager.py
deleted file mode 100644
index c251afbb..00000000
--- a/python_moondb/python_moondb/db_manager.py
+++ /dev/null
@@ -1,82 +0,0 @@
-# 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(
- "python_moondb.migrate_repo.versions.{}".format(filename))
- logger.info("Command is {}".format(command))
- if command in ("upgrade", "u", "up"):
- logger.info(
- "upgrading python_moondb.migrate_repo.versions.{}".format(filename))
- o.upgrade(engine)
- elif command in ("downgrade", "d", "down"):
- logger.info(
- "downgrading python_moondb.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()