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

var separation and args added to sonarr

Browse source
This commit is contained in:
Nicholas St. Germain 2018-11-28 22:20:09 -06:00
parent 8f47246be1
commit ad376fcfa8
1 changed files with 14 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

19
Varken/iniparser.py
View file

@ -1,25 +1,29 @@
import sys import sys
import configparser import configparser
from Varken.helpers import Server, TautulliServer, InfluxServer from Varken.helpers import Server, TautulliServer, SonarrServer, InfluxServer
class INIParser(object): class INIParser(object):
def __init__(self): def __init__(self):
self.config = configparser.ConfigParser() self.config = configparser.ConfigParser()
self.influx_server = InfluxServer() self.influx_server = InfluxServer()
self.sonarr_enabled = False self.sonarr_enabled = False
self.sonarr_servers = [] self.sonarr_servers = []
self.radarr_enabled = False self.radarr_enabled = False
self.radarr_servers = [] self.radarr_servers = []
self.ombi_enabled = False self.ombi_enabled = False
self.ombi_server = None self.ombi_server = None
self.tautulli_enabled = False self.tautulli_enabled = False
self.tautulli_server = None self.tautulli_server = None
self.asa_enabled = False self.asa_enabled = False
self.asa = None self.asa = None
self.read_file()
self.parse_opts() self.parse_opts()
def read_file(self): def read_file(self):
@ -27,6 +31,7 @@ class INIParser(object):
self.config.read_file(config_ini) self.config.read_file(config_ini)
def parse_opts(self): def parse_opts(self):
self.read_file()
# Parse InfluxDB options # Parse InfluxDB options
url = self.config.get('influxdb', 'url') url = self.config.get('influxdb', 'url')
port = self.config.getint('influxdb', 'port') port = self.config.getint('influxdb', 'port')
@ -49,8 +54,12 @@ class INIParser(object):
apikey = self.config.get(sonarr_section, 'apikey') apikey = self.config.get(sonarr_section, 'apikey')
scheme = 'https://' if self.config.getboolean(sonarr_section, 'ssl') else 'http://' scheme = 'https://' if self.config.getboolean(sonarr_section, 'ssl') else 'http://'
verify_ssl = self.config.getboolean(sonarr_section, 'verify_ssl') verify_ssl = self.config.getboolean(sonarr_section, 'verify_ssl')
queue = self.config.getboolean(sonarr_section, 'queue')
missing_days = self.config.getint(sonarr_section, 'missing_days')
future_days = self.config.getint(sonarr_section, 'future_days')
self.sonarr_servers.append(Server(server_id, scheme + url, apikey, verify_ssl)) self.sonarr_servers.append(SonarrServer(server_id, scheme + url, apikey, verify_ssl,
missing_days, future_days, queue))
# Parse Radarr options # Parse Radarr options
try: try:
@ -60,7 +69,7 @@ class INIParser(object):
self.radarr_enabled = True self.radarr_enabled = True
sids = self.config.get('global', 'radarr_server_ids').strip(' ').split(',') sids = self.config.get('global', 'radarr_server_ids').strip(' ').split(',')
for server_id in sids: for server_id in sids:
radarr_section = 'sonarr-' + server_id radarr_section = 'radarr-' + server_id
url = self.config.get(radarr_section, 'url') url = self.config.get(radarr_section, 'url')
apikey = self.config.get(radarr_section, 'apikey') apikey = self.config.get(radarr_section, 'apikey')
scheme = 'https://' if self.config.getboolean(radarr_section, 'ssl') else 'http://' scheme = 'https://' if self.config.getboolean(radarr_section, 'ssl') else 'http://'