Coverage for osm_policy_module/cmd/policy_module_agent.py: 0%

1# -*- coding: utf-8 -*- 

2 

3# Copyright 2018 Whitestack, LLC 

4# ************************************************************* 

5 

6# This file is part of OSM Monitoring module 

7# All Rights Reserved to Whitestack, LLC 

8 

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 

12 

13# http://www.apache.org/licenses/LICENSE-2.0 

14 

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 

19# under the License. 

20 

21# For those usages not covered by the Apache License, Version 2.0 please 

22# contact: bdiaz@whitestack.com or glavado@whitestack.com 

23## 

24import argparse 

25import logging 

26import sys 

27import os 

28 

29from osm_policy_module.core.agent import PolicyModuleAgent 

30from osm_policy_module.core.config import Config 

31from osm_policy_module.core.database import DatabaseManager 

32 

33 

34def main(): 

35 # Cleanup old temp health file 

36 if os.path.exists("/tmp/osm_pol_agent_health_flag"): 

37 os.remove("/tmp/osm_pol_agent_health_flag") 

38 

39 parser = argparse.ArgumentParser(prog="osm-policy-agent") 

40 parser.add_argument("--config-file", nargs="?", help="POL configuration file") 

41 args = parser.parse_args() 

42 cfg = Config(args.config_file) 

43 

44 root = logging.getLogger() 

45 root.setLevel(logging.getLevelName(cfg.get("global", "loglevel"))) 

46 ch = logging.StreamHandler(sys.stdout) 

47 ch.setLevel(logging.getLevelName(cfg.get("global", "loglevel"))) 

48 formatter = logging.Formatter( 

49 "%(asctime)s - %(name)s - %(levelname)s - %(message)s", "%m/%d/%Y %I:%M:%S %p" 

50 ) 

51 ch.setFormatter(formatter) 

52 root.addHandler(ch) 

53 

54 log = logging.getLogger(__name__) 

55 log.debug("Config: %s", cfg.conf) 

56 log.info("Initializing database...") 

57 db_manager = DatabaseManager(cfg) 

58 db_manager.create_tables() 

59 log.info("Database initialized correctly.") 

60 log.info("Starting policy module agent...") 

61 agent = PolicyModuleAgent(cfg) 

62 agent.run() 

63 

64 

65if __name__ == "__main__": 

66 main()