d_mcknight/Varken
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use the correct db manager for varken

Browse source
This commit is contained in:
Gabe Revells 2021-02-13 15:50:12 -05:00 committed by Daniel
parent 2273083d00
commit 163798bc20
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
Varken.py
View file

@ -23,6 +23,7 @@ from varken.radarr import RadarrAPI
from varken.lidarr import LidarrAPI from varken.lidarr import LidarrAPI
from varken.iniparser import INIParser from varken.iniparser import INIParser
from varken.dbmanager import DBManager from varken.dbmanager import DBManager
from varken.influxdb2manager import InfluxDB2Manager
from varken.helpers import GeoIPHandler from varken.helpers import GeoIPHandler
from varken.tautulli import TautulliAPI from varken.tautulli import TautulliAPI
from varken.sickchill import SickChillAPI from varken.sickchill import SickChillAPI
@ -92,7 +93,15 @@ if __name__ == "__main__":
vl.logger.info("Varken v%s-%s %s", VERSION, BRANCH, BUILD_DATE) vl.logger.info("Varken v%s-%s %s", VERSION, BRANCH, BUILD_DATE)
CONFIG = INIParser(DATA_FOLDER) CONFIG = INIParser(DATA_FOLDER)
if CONFIG.influx2_enabled:
# Use INFLUX version 2
vl.logger.info('Using INFLUXDBv2')
DBMANAGER = InfluxDB2Manager(CONFIG.influx_server)
else:
vl.logger.info('Using INFLUXDB')
DBMANAGER = DBManager(CONFIG.influx_server) DBMANAGER = DBManager(CONFIG.influx_server)
QUEUE = Queue() QUEUE = Queue()
if CONFIG.sonarr_enabled: if CONFIG.sonarr_enabled: