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

passed influx_seraver to sonarrapi class

Browse source
This commit is contained in:
Nicholas St. Germain 2018-11-28 22:26:00 -06:00
parent 01370d92dc
commit 7910efc641
2 changed files with 5 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
Varken/sonarr.py
View file

@ -6,19 +6,18 @@ import argparse
from influxdb import InfluxDBClient from influxdb import InfluxDBClient
from datetime import datetime, timezone, date, timedelta from datetime import datetime, timezone, date, timedelta
from Varken import configuration as config
from Varken.helpers import TVShow, Queue from Varken.helpers import TVShow, Queue
class SonarrAPI(object): class SonarrAPI(object):
def __init__(self, servers): def __init__(self, sonarr_servers, influx_server):
# Set Time of initialization # Set Time of initialization
self.now = datetime.now(timezone.utc).astimezone().isoformat() self.now = datetime.now(timezone.utc).astimezone().isoformat()
self.today = str(date.today()) self.today = str(date.today())
self.influx = InfluxDBClient(config.influxdb_url, config.influxdb_port, config.influxdb_username, self.influx = InfluxDBClient(influx_server.url, influx_server.port, influx_server.username,
config.influxdb_password, config.sonarr_influxdb_db_name) influx_server.password, 'plex')
self.influx_payload = [] self.influx_payload = []
self.servers = servers self.servers = sonarr_servers
# Create session to reduce server web thread load, and globally define pageSize for all requests # Create session to reduce server web thread load, and globally define pageSize for all requests
self.session = requests.Session() self.session = requests.Session()
self.session.params = {'pageSize': 1000} self.session.params = {'pageSize': 1000}

2
Varken/varken.py
View file

@ -25,7 +25,7 @@ if __name__ == "__main__":
CONFIG = INIParser() CONFIG = INIParser()
if CONFIG.sonarr_enabled: if CONFIG.sonarr_enabled:
SONARR = SonarrAPI(CONFIG.sonarr_servers) SONARR = SonarrAPI(CONFIG.sonarr_servers, CONFIG.influx_server)
for server in CONFIG.sonarr_servers: for server in CONFIG.sonarr_servers:
if server.queue: if server.queue:
schedule.every().minute.do(threaded, SONARR.get_queue) schedule.every().minute.do(threaded, SONARR.get_queue)