60 lines
1.9 KiB
Python
Executable File
60 lines
1.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
|
|
if sys.version_info < (3, 5):
|
|
print("To use this script you need python 3.5 or newer! got %s" % sys.version_info)
|
|
sys.exit(1)
|
|
|
|
import logger
|
|
logger.setup()
|
|
|
|
import logging
|
|
import argparse
|
|
import queue
|
|
|
|
from workers_queue import _WORKERS_QUEUE
|
|
from config import settings
|
|
from mqtt import MqttClient
|
|
from workers_manager import WorkersManager
|
|
|
|
|
|
parser = argparse.ArgumentParser()
|
|
group = parser.add_mutually_exclusive_group()
|
|
group.add_argument('-d', '--debug', action='store_true', default=False, help='Set logging to output debug information')
|
|
group.add_argument('-q', '--quiet', action='store_true', default=False, help='Set logging to just output warnings and errors')
|
|
parser.add_argument('-s', '--suppress-update-failures', dest='suppress', action='store_true', default=False, help='Suppress any errors regarding failed device updates')
|
|
parsed = parser.parse_args()
|
|
|
|
_LOGGER = logger.get()
|
|
if parsed.quiet:
|
|
_LOGGER.setLevel(logging.WARNING)
|
|
elif parsed.debug:
|
|
_LOGGER.setLevel(logging.DEBUG)
|
|
logger.enable_debug_formatter()
|
|
else:
|
|
_LOGGER.setLevel(logging.INFO)
|
|
logger.suppress_update_failures(parsed.suppress)
|
|
|
|
_LOGGER.info('Starting')
|
|
|
|
mqtt = MqttClient(settings['mqtt'])
|
|
manager = WorkersManager(settings['manager'])
|
|
manager.register_workers().start(mqtt)
|
|
|
|
running = True
|
|
|
|
while running:
|
|
try:
|
|
mqtt.publish(_WORKERS_QUEUE.get(timeout=10).execute())
|
|
except queue.Empty: # Allow for SIGINT processing
|
|
pass
|
|
except TimeoutError as e:
|
|
logger.log_exception(_LOGGER, str(e) if str(e) else 'Timeout while executing worker command', suppress=True)
|
|
except (KeyboardInterrupt, SystemExit):
|
|
running = False
|
|
_LOGGER.info('Finish current jobs and shut down. If you need force exit use kill')
|
|
except Exception as e:
|
|
logger.log_exception(_LOGGER, "Fatal error while executing worker command: %s", type(e).__name__)
|
|
raise e
|