# For those usages not covered by the Apache License, Version 2.0 please
# contact: bdiaz@whitestack.com or glavado@whitestack.com
##
-import argparse
+import asyncio
import logging
import sys
def main():
cfg = Config.instance()
- parser = argparse.ArgumentParser(prog='pm-scaling-config-agent')
- parser.add_argument('--config-file', nargs='?', help='Policy module agent configuration file')
- args = parser.parse_args()
- if args.config_file:
- cfg.load_file(args.config_file)
log_formatter_str = '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
logging.basicConfig(stream=sys.stdout,
format=log_formatter_str,
datefmt='%m/%d/%Y %I:%M:%S %p',
level=logging.getLevelName(cfg.OSMPOL_LOG_LEVEL))
kafka_logger = logging.getLogger('kafka')
- kafka_logger.setLevel(logging.WARN)
+ kafka_logger.setLevel(logging.getLevelName(cfg.OSMPOL_KAFKA_LOG_LEVEL))
kafka_formatter = logging.Formatter(log_formatter_str)
kafka_handler = logging.StreamHandler(sys.stdout)
kafka_handler.setFormatter(kafka_formatter)
kafka_logger.addHandler(kafka_handler)
log = logging.getLogger(__name__)
- log.info("Config: %s", cfg)
+ log.info("Config: %s", vars(cfg))
log.info("Syncing database...")
db_manager = DatabaseManager()
db_manager.create_tables()
log.info("Database synced correctly.")
log.info("Starting policy module agent...")
- agent = PolicyModuleAgent()
+ loop = asyncio.get_event_loop()
+ agent = PolicyModuleAgent(loop)
agent.run()