1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
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(
"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()
|
