diff --git a/Varken.py b/Varken.py index dd6de9a..1e1e94c 100644 --- a/Varken.py +++ b/Varken.py @@ -26,22 +26,12 @@ def threaded(job): if __name__ == "__main__": - # Initiate the logger - vl = VarkenLogger() - vl.logger.info('Starting Varken...') - vl.logger.info(u"{} {} ({}{})".format( - platform.system(), platform.release(), platform.version(), - ' - {}'.format(PLATFORM_LINUX_DISTRO) if PLATFORM_LINUX_DISTRO else '' - )) - vl.logger.info(u"Python {}".format(sys.version)) - - parser = ArgumentParser(prog='varken', description='Command-line utility to aggregate data from the plex ecosystem into InfluxDB', formatter_class=RawTextHelpFormatter) parser.add_argument("-d", "--data-folder", help='Define an alternate data folder location') - parser.add_argument("-l", "--log-level", choices=['info', 'error', 'debug'], help='Not yet implemented') + parser.add_argument("-D", "--debug", action='store_true', help='Use to enable DEBUG logging') opts = parser.parse_args() @@ -57,6 +47,16 @@ if __name__ == "__main__": else: exit("{} does not exist".format(ARG_FOLDER)) + # Initiate the logger + vl = VarkenLogger(data_folder=DATA_FOLDER, debug=opts.debug) + vl.logger.info('Starting Varken...') + + vl.logger.info(u"{} {} ({}{})".format( + platform.system(), platform.release(), platform.version(), + ' - {}'.format(PLATFORM_LINUX_DISTRO) if PLATFORM_LINUX_DISTRO else '' + )) + vl.logger.info(u"Python {}".format(sys.version)) + CONFIG = INIParser(DATA_FOLDER) DBMANAGER = DBManager(CONFIG.influx_server) diff --git a/varken/tautulli.py b/varken/tautulli.py index 9592636..e6f183f 100644 --- a/varken/tautulli.py +++ b/varken/tautulli.py @@ -1,4 +1,5 @@ import logging + from datetime import datetime, timezone from geoip2.errors import AddressNotFoundError from requests import Session, Request