#!/usr/bin/python3 # # Copyright (C) 2020-2021 Soleta Networks # # This program is free software: you can redistribute it and/or modify it under # the terms of the GNU Affero General Public License as published by the # Free Software Foundation; either version 3 of the License, or # (at your option) any later version. import json import logging import argparse import platform import subprocess import shutil try: from signal import SIG_DFL, SIGPIPE except ImportError: from signal import SIG_DFL from src.ogClient import * from src.log import * def send_event_dgram(msg, ip='127.0.0.1', port=55885): sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) sock.sendto(bytes(msg, "utf-8"), (ip, port)) def create_parser(): events = ['login', 'logout'] parser = argparse.ArgumentParser() parser.set_defaults(func=None) subparsers = parser.add_subparsers() parser_event = subparsers.add_parser('event') subparsers_event = parser_event.add_subparsers() parser_event_login = subparsers_event.add_parser('login') parser_event_login.set_defaults(func=lambda x: send_event_dgram(f'session start {x.user}')) parser_event_login.add_argument('user', type=str) parser_event_logout = subparsers_event.add_parser('logout') parser_event_logout.set_defaults(func=lambda x: send_event_dgram(f'session stop {x.user}')) parser_event_logout.add_argument('user', type=str) parser.add_argument('-c', '--config', default="", help='ogClient JSON config file path') parser.add_argument('--debug', default=False, action='store_true', help='enables debug log level') parser.add_argument('--server-ip', default="", help='Server IP address override') return parser def main(): parser = create_parser() args = parser.parse_args(sys.argv[1:]) if args.func: args.func(args) return if args.config: config_path = args.config elif platform.system().lower() == 'linux': config_path = f'{ogClient.OG_PATH}ogclient/cfg/ogclient.json' else: config_path = './cfg/ogclient.json' try: with open(config_path, 'r') as f: CONFIG = json.load(f) except: print('Error: Parsing configuration file') return 0 if args.server_ip: CONFIG['opengnsys']['ip'] = args.server_ip MODE = CONFIG['opengnsys']['mode'] URL = CONFIG['opengnsys']['url'] LOGLEVEL = CONFIG['opengnsys']['log'] if MODE == 'live': if shutil.which('browser'): proc = subprocess.Popen(["browser", "-qws", URL]) if MODE != 'windows': signal.signal(SIGPIPE, SIG_DFL) configure_logging(MODE, LOGLEVEL) if args.debug: logging.getLogger().setLevel('DEBUG') try: client = ogClient(config=CONFIG) client.connect() client.run() except OgError as e: logging.critical(e) if __name__ == "__main__": main()