policy_module/osm_policy_module/cmd/dbsync.py

   1 import argparse
   2 import logging
   3 import sys
   4
   5 from osm_policy_module.core.config import Config
   6
   7 from osm_policy_module.core.database import DatabaseManager
   8
   9
  10 def main():
  11     cfg = Config.instance()
  12     parser = argparse.ArgumentParser(prog='pm-dbsync')
  13     parser.add_argument('--config-file', nargs='?', help='Policy module database sync configuration file')
  14     args = parser.parse_args()
  15     if args.config_file:
  16         cfg.load_file(args.config_file)
  17     if cfg.get('policy_module', 'log_dir') == 'stdout':
  18         logging.basicConfig(stream=sys.stdout,
  19                             format='%(asctime)s %(message)s',
  20                             datefmt='%m/%d/%Y %I:%M:%S %p', filemode='a',
  21                             level=logging.INFO)
  22     else:
  23         logging.basicConfig(filename=cfg.get('policy_module', 'log_dir') + 'pm_dbsync.log',
  24                             format='%(asctime)s %(message)s',
  25                             datefmt='%m/%d/%Y %I:%M:%S %p', filemode='a',
  26                             level=logging.INFO)
  27     log = logging.getLogger(__name__)
  28     log.info("Starting database sync...")
  29     db_manager = DatabaseManager()
  30     db_manager.create_tables()