Add logger and better exception handling

This commit is contained in:
zewelor
2017-11-24 08:53:21 +01:00
parent d463345707
commit 09d1332b92
3 changed files with 49 additions and 12 deletions
+35 -11
View File
@@ -1,27 +1,51 @@
import logging
import sys
if sys.version_info < (3,4):
print("To use this script you need python 3.4 or newer! got %s" % sys.version_info)
sys.exit(1)
from logger import _LOGGER
import argparse
from queue import Queue
from apscheduler.schedulers.background import BackgroundScheduler
from config import settings
from mqtt import MqttClient
from workers_manager import WorkersManager
from apscheduler.schedulers.background import BackgroundScheduler
parser = argparse.ArgumentParser()
parser.add_argument('-d', '--debug', action='store_true', default=False)
parsed = parser.parse_args()
if parsed.debug:
_LOGGER.setLevel(logging.DEBUG)
else:
_LOGGER.setLevel(logging.INFO)
queue = Queue()
scheduler = BackgroundScheduler()
mqtt = MqttClient(settings['mqtt'])
manager = WorkersManager(settings['workers'])
_LOGGER.info('Starting')
for worker_name, interval in manager.interval_enabled_workers.items():
_LOGGER.debug("Added: %s with %d seconds interval" % (worker_name, interval))
queue.put([worker_name])
scheduler.add_job(lambda x: queue.put([x]), 'interval', seconds=interval, args=[worker_name])
scheduler.start()
while True:
for worker_name in queue.get(block=True):
try:
mqtt.publish(manager.update(worker_name))
except (KeyboardInterrupt, SystemExit):
raise
except:
print('Some error')
running = True
try:
while running:
for worker_name in queue.get(block=True):
try:
mqtt.publish(manager.update(worker_name))
except (KeyboardInterrupt, SystemExit):
raise
except:
_LOGGER.info('Some error')
except KeyboardInterrupt:
running = False
_LOGGER.info('Exiting allowing jobs to finish. If you need force exit use kill')