61 lines
2.3 KiB
Python
61 lines
2.3 KiB
Python
import schedule
|
|
import threading
|
|
from time import sleep
|
|
|
|
from Varken.iniparser import INIParser
|
|
from Varken.sonarr import SonarrAPI
|
|
from Varken.tautulli import TautulliAPI
|
|
from Varken.radarr import RadarrAPI
|
|
from Varken.ombi import OmbiAPI
|
|
from Varken.dbmanager import DBManager
|
|
|
|
def threaded(job):
|
|
thread = threading.Thread(target=job)
|
|
thread.start()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
CONFIG = INIParser()
|
|
DBMANAGER = DBManager(CONFIG.influx_server)
|
|
|
|
if CONFIG.sonarr_enabled:
|
|
for server in CONFIG.sonarr_servers:
|
|
SONARR = SonarrAPI(server, DBMANAGER)
|
|
if server.queue:
|
|
schedule.every(server.queue_run_seconds).seconds.do(threaded, SONARR.get_queue)
|
|
if server.missing_days > 0:
|
|
schedule.every(server.missing_days_run_seconds).seconds.do(threaded, SONARR.get_missing)
|
|
if server.future_days > 0:
|
|
schedule.every(server.future_days_run_seconds).seconds.do(threaded, SONARR.get_future)
|
|
|
|
if CONFIG.tautulli_enabled:
|
|
for server in CONFIG.tautulli_servers:
|
|
TAUTULLI = TautulliAPI(server, DBMANAGER)
|
|
if server.get_activity:
|
|
schedule.every(server.get_activity_run_seconds).seconds.do(threaded, TAUTULLI.get_activity)
|
|
if server.get_sessions:
|
|
schedule.every(server.get_sessions_run_seconds).seconds.do(threaded, TAUTULLI.get_sessions)
|
|
|
|
if CONFIG.radarr_enabled:
|
|
for server in CONFIG.radarr_servers:
|
|
RADARR = RadarrAPI(server, DBMANAGER)
|
|
if server.get_missing:
|
|
schedule.every(server.get_missing_run_seconds).seconds.do(threaded, RADARR.get_missing)
|
|
if server.queue:
|
|
schedule.every(server.queue_run_seconds).seconds.do(threaded, RADARR.get_queue)
|
|
|
|
if CONFIG.ombi_enabled:
|
|
for server in CONFIG.ombi_servers:
|
|
OMBI = OmbiAPI(server, DBMANAGER)
|
|
if server.request_type_counts:
|
|
schedule.every(server.request_type_run_seconds).seconds.do(threaded, OMBI.get_request_counts)
|
|
if server.request_total_counts:
|
|
schedule.every(server.request_total_run_seconds).seconds.do(threaded, OMBI.get_total_requests)
|
|
|
|
# Run all on startup
|
|
schedule.run_all()
|
|
|
|
while True:
|
|
schedule.run_pending()
|
|
sleep(1)
|
|
|