ac0316750637f09a4c28ed8af48efc6b3446f504
1 # -*- coding: utf-8 -*-
3 # Copyright 2018 Whitestack, LLC
4 # *************************************************************
6 # This file is part of OSM Monitoring module
7 # All Rights Reserved to Whitestack, LLC
9 # Licensed under the Apache License, Version 2.0 (the "License"); you may
10 # not use this file except in compliance with the License. You may obtain
11 # a copy of the License at
13 # http://www.apache.org/licenses/LICENSE-2.0
15 # Unless required by applicable law or agreed to in writing, software
16 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
17 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
18 # License for the specific language governing permissions and limitations
21 # For those usages not covered by the Apache License, Version 2.0 please
22 # contact: bdiaz@whitestack.com or glavado@whitestack.com
29 from osm_policy_module
.core
.agent
import PolicyModuleAgent
30 from osm_policy_module
.core
.config
import Config
31 from osm_policy_module
.core
.database
import DatabaseManager
35 cfg
= Config
.instance()
36 parser
= argparse
.ArgumentParser(prog
='pm-scaling-config-agent')
37 parser
.add_argument('--config-file', nargs
='?', help='Policy module agent configuration file')
38 args
= parser
.parse_args()
40 cfg
.load_file(args
.config_file
)
41 # TODO: Handle different log levels in config
42 if cfg
.get('policy_module', 'log_dir') == 'stdout':
43 logging
.basicConfig(stream
=sys
.stdout
,
44 format
='%(asctime)s %(message)s',
45 datefmt
='%m/%d/%Y %I:%M:%S %p',
48 logging
.basicConfig(filename
=cfg
.get('policy_module', 'log_dir') + 'pm_agent.log',
49 format
='%(asctime)s %(message)s',
50 datefmt
='%m/%d/%Y %I:%M:%S %p', filemode
='a',
52 if cfg
.get('policy_module', 'enable_logstash_handler') == 'true':
53 logstash_host
= cfg
.get('policy_module', 'logstash_host')
54 logstash_port
= int(cfg
.get('policy_module', 'logstash_port'))
55 root_logger
= logging
.getLogger()
56 root_logger
.addHandler(logstash
.TCPLogstashHandler(logstash_host
, logstash_port
, version
=1))
57 root_logger
.info("Logstash handler configured.")
58 kafka_logger
= logging
.getLogger('kafka')
59 kafka_logger
.setLevel(logging
.WARN
)
60 kafka_formatter
= logging
.Formatter(
61 '%(asctime)s - %(name)s - %(levelname)s - %(message)s')
62 kafka_handler
= logging
.StreamHandler(sys
.stdout
)
63 kafka_handler
.setFormatter(kafka_formatter
)
64 kafka_logger
.addHandler(kafka_handler
)
65 log
= logging
.getLogger(__name__
)
66 log
.info("Config: %s", cfg
)
67 log
.info("Syncing database...")
68 db_manager
= DatabaseManager()
69 db_manager
.create_tables()
70 log
.info("Database synced correctly.")
71 log
.info("Starting policy module agent...")
72 agent
= PolicyModuleAgent()
76 if __name__
== '__main__':