#!/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 try: from signal import SIG_DFL, SIGPIPE except ImportError: from signal import SIG_DFL from src.ogClient import * from src.log import configure_logging 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) return parser def main(): parser = create_parser() args = parser.parse_args(sys.argv[1:]) if args.func: args.func(args) return if 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 MODE = CONFIG['opengnsys']['mode'] URL = CONFIG['opengnsys']['url'] if MODE == 'live': proc = subprocess.Popen(["browser", "-qws", URL]) if MODE != 'windows': signal.signal(SIGPIPE, SIG_DFL) configure_logging(MODE) client = ogClient(config=CONFIG) client.connect() client.run() if __name__ == "__main__": main()