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

added sickchill. fixes #48

Browse source
This commit is contained in:
Nicholas St. Germain 2018-12-16 00:55:59 -06:00
parent 056d211d99
commit 78b7765553
5 changed files with 208 additions and 92 deletions
Download patch file Download diff file Expand all files Collapse all files

9
Varken.py
View file

@ -21,6 +21,7 @@ from varken.sonarr import SonarrAPI
from varken.tautulli import TautulliAPI from varken.tautulli import TautulliAPI
from varken.radarr import RadarrAPI from varken.radarr import RadarrAPI
from varken.ombi import OmbiAPI from varken.ombi import OmbiAPI
from varken.sickchill import SickChillAPI
from varken.cisco import CiscoAPI from varken.cisco import CiscoAPI
from varken.dbmanager import DBManager from varken.dbmanager import DBManager
from varken.varkenlogger import VarkenLogger from varken.varkenlogger import VarkenLogger
@ -110,6 +111,12 @@ if __name__ == "__main__":
if server.request_total_counts: if server.request_total_counts:
schedule.every(server.request_total_run_seconds).seconds.do(threaded, OMBI.get_all_requests) schedule.every(server.request_total_run_seconds).seconds.do(threaded, OMBI.get_all_requests)
if CONFIG.sickchill_enabled:
for server in CONFIG.sickchill_servers:
SICKCHILL = SickChillAPI(server, DBMANAGER)
if server.get_missing:
schedule.every(server.get_missing_run_seconds).seconds.do(threaded, SICKCHILL.get_missing)
if CONFIG.ciscoasa_enabled: if CONFIG.ciscoasa_enabled:
for firewall in CONFIG.ciscoasa_firewalls: for firewall in CONFIG.ciscoasa_firewalls:
ASA = CiscoAPI(firewall, DBMANAGER) ASA = CiscoAPI(firewall, DBMANAGER)
@ -117,7 +124,7 @@ if __name__ == "__main__":
# Run all on startup # Run all on startup
SERVICES_ENABLED = [CONFIG.ombi_enabled, CONFIG.radarr_enabled, CONFIG.tautulli_enabled, SERVICES_ENABLED = [CONFIG.ombi_enabled, CONFIG.radarr_enabled, CONFIG.tautulli_enabled,
CONFIG.sonarr_enabled, CONFIG.ciscoasa_enabled] CONFIG.sonarr_enabled, CONFIG.ciscoasa_enabled, CONFIG.sickchill_enabled]
if not [enabled for enabled in SERVICES_ENABLED if enabled]: if not [enabled for enabled in SERVICES_ENABLED if enabled]:
exit("All services disabled. Exiting") exit("All services disabled. Exiting")
schedule.run_all() schedule.run_all()

25
data/varken.example.ini
View file

@ -11,6 +11,7 @@ radarr_server_ids = 1,2
tautulli_server_ids = 1 tautulli_server_ids = 1
ombi_server_ids = 1 ombi_server_ids = 1
ciscoasa_firewall_ids = false ciscoasa_firewall_ids = false
sickchill_server_ids = false
[influxdb] [influxdb]
url = influxdb.domain.tld url = influxdb.domain.tld
@ -23,7 +24,7 @@ url = tautulli.domain.tld:8181
fallback_ip = 0.0.0.0 fallback_ip = 0.0.0.0
apikey = xxxxxxxxxxxxxxxx apikey = xxxxxxxxxxxxxxxx
ssl = false ssl = false
verify_ssl = true verify_ssl = false
get_activity = true get_activity = true
get_activity_run_seconds = 30 get_activity_run_seconds = 30
@ -31,7 +32,7 @@ get_activity_run_seconds = 30
url = sonarr1.domain.tld:8989 url = sonarr1.domain.tld:8989
apikey = xxxxxxxxxxxxxxxx apikey = xxxxxxxxxxxxxxxx
ssl = false ssl = false
verify_ssl = true verify_ssl = false
missing_days = 7 missing_days = 7
missing_days_run_seconds = 300 missing_days_run_seconds = 300
future_days = 1 future_days = 1
@ -43,7 +44,7 @@ queue_run_seconds = 300
url = sonarr2.domain.tld:8989 url = sonarr2.domain.tld:8989
apikey = yyyyyyyyyyyyyyyy apikey = yyyyyyyyyyyyyyyy
ssl = false ssl = false
verify_ssl = true verify_ssl = false
missing_days = 7 missing_days = 7
missing_days_run_seconds = 300 missing_days_run_seconds = 300
future_days = 1 future_days = 1
@ -55,7 +56,7 @@ queue_run_seconds = 300
url = radarr1.domain.tld url = radarr1.domain.tld
apikey = xxxxxxxxxxxxxxxx apikey = xxxxxxxxxxxxxxxx
ssl = false ssl = false
verify_ssl = true verify_ssl = false
queue = true queue = true
queue_run_seconds = 300 queue_run_seconds = 300
get_missing = true get_missing = true
@ -65,7 +66,7 @@ get_missing_run_seconds = 300
url = radarr2.domain.tld url = radarr2.domain.tld
apikey = yyyyyyyyyyyyyyyy apikey = yyyyyyyyyyyyyyyy
ssl = false ssl = false
verify_ssl = true verify_ssl = false
queue = true queue = true
queue_run_seconds = 300 queue_run_seconds = 300
get_missing = true get_missing = true
@ -75,17 +76,27 @@ get_missing_run_seconds = 300
url = ombi.domain.tld url = ombi.domain.tld
apikey = xxxxxxxxxxxxxxxx apikey = xxxxxxxxxxxxxxxx
ssl = false ssl = false
verify_ssl = true verify_ssl = false
get_request_type_counts = true get_request_type_counts = true
request_type_run_seconds = 300 request_type_run_seconds = 300
get_request_total_counts = true get_request_total_counts = true
request_total_run_seconds = 300 request_total_run_seconds = 300
[sickchill-1]
url = sickchill.domain.tld:8081
apikey = xxxxxxxxxxxxxxxx
ssl = false
verify_ssl = false
get_missing = true
get_missing_run_seconds = 300
[ciscoasa-1] [ciscoasa-1]
url = firewall.domain.tld url = firewall.domain.tld
username = cisco username = cisco
password = cisco password = cisco
outside_interface = WAN outside_interface = WAN
ssl = false ssl = false
verify_ssl = true verify_ssl = false
get_bandwidth_run_seconds = 300 get_bandwidth_run_seconds = 300

172
varken/iniparser.py
View file

@ -8,6 +8,7 @@ from os.path import join, exists
from varken.helpers import clean_sid_check from varken.helpers import clean_sid_check
from varken.varkenlogger import BlacklistFilter from varken.varkenlogger import BlacklistFilter
from varken.structures import SonarrServer, RadarrServer, OmbiServer, TautulliServer, InfluxServer, CiscoASAFirewall from varken.structures import SonarrServer, RadarrServer, OmbiServer, TautulliServer, InfluxServer, CiscoASAFirewall
from varken.structures import SickChillServer
class INIParser(object): class INIParser(object):
@ -31,6 +32,9 @@ class INIParser(object):
self.tautulli_enabled = False self.tautulli_enabled = False
self.tautulli_servers = [] self.tautulli_servers = []
self.sickchill_enabled = False
self.sickchill_servers = []
self.ciscoasa_enabled = False self.ciscoasa_enabled = False
self.ciscoasa_firewalls = [] self.ciscoasa_firewalls = []
@ -72,8 +76,8 @@ class INIParser(object):
url_check = url url_check = url
inc_port = include_port inc_port = include_port
search = (r'(?:(?:[A-Z0-9](?:[A-Z0-9-]{0,61}[A-Z0-9])?\.)+(?:[A-Z]{2,6}\.?|[A-Z0-9-]{2,}\.?)|' #domain... search = (r'(?:(?:[A-Z0-9](?:[A-Z0-9-]{0,61}[A-Z0-9])?\.)+(?:[A-Z]{2,6}\.?|[A-Z0-9-]{2,}\.?)|' # domain...
r'localhost|' #localhost... r'localhost|' # localhost...
r'\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})' # ...or ip r'\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})' # ...or ip
) )
@ -89,13 +93,14 @@ class INIParser(object):
valid = re.match(regex, url_check) is not None valid = re.match(regex, url_check) is not None
if not valid: if not valid:
if inc_port: if inc_port:
self.logger.error('%s is invalid! URL must host/IP and port if not 80 or 443. ie. localhost:8080', url_check) self.logger.error('%s is invalid! URL must host/IP and port if not 80 or 443. ie. localhost:8080',
url_check)
exit(1) exit(1)
else: else:
self.logger.error('%s is invalid! URL must host/IP. ie. localhost', url_check) self.logger.error('%s is invalid! URL must host/IP. ie. localhost', url_check)
exit(1) exit(1)
else: else:
self.logger.debug('%s is a vlaid URL in the config.', url_check) self.logger.debug('%s is a valid URL in the config.', url_check)
return url_check return url_check
def parse_opts(self): def parse_opts(self):
@ -116,37 +121,30 @@ class INIParser(object):
if self.sonarr_enabled: if self.sonarr_enabled:
for server_id in self.sonarr_enabled: for server_id in self.sonarr_enabled:
sonarr_section = 'sonarr-' + str(server_id) section = 'sonarr-' + str(server_id)
try: try:
url = self.url_check(self.config.get(sonarr_section, 'url')) url = self.url_check(self.config.get(section, 'url'))
apikey = self.config.get(sonarr_section, 'apikey') apikey = self.config.get(section, 'apikey')
scheme = 'https://' if self.config.getboolean( scheme = 'https://' if self.config.getboolean(section, 'ssl') else 'http://'
sonarr_section, 'ssl') else 'http://'
verify_ssl = self.config.getboolean( verify_ssl = self.config.getboolean(section, 'verify_ssl')
sonarr_section, 'verify_ssl')
if scheme != 'https://': if scheme != 'https://':
verify_ssl = False verify_ssl = False
queue = self.config.getboolean(sonarr_section, 'queue') queue = self.config.getboolean(section, 'queue')
missing_days = self.config.getint( missing_days = self.config.getint(section, 'missing_days')
sonarr_section, 'missing_days')
future_days = self.config.getint( future_days = self.config.getint(section, 'future_days')
sonarr_section, 'future_days')
missing_days_run_seconds = self.config.getint( missing_days_run_seconds = self.config.getint(section, 'missing_days_run_seconds')
sonarr_section, 'missing_days_run_seconds')
future_days_run_seconds = self.config.getint( future_days_run_seconds = self.config.getint(section, 'future_days_run_seconds')
sonarr_section, 'future_days_run_seconds')
queue_run_seconds = self.config.getint( queue_run_seconds = self.config.getint(section, 'queue_run_seconds')
sonarr_section, 'queue_run_seconds')
server = SonarrServer(server_id, scheme + url, apikey, verify_ssl, missing_days, server = SonarrServer(server_id, scheme + url, apikey, verify_ssl, missing_days,
missing_days_run_seconds, future_days, future_days_run_seconds, missing_days_run_seconds, future_days, future_days_run_seconds,
@ -154,40 +152,35 @@ class INIParser(object):
self.sonarr_servers.append(server) self.sonarr_servers.append(server)
except configparser.NoOptionError as e: except configparser.NoOptionError as e:
self.radarr_enabled = False self.sonarr_enabled = False
self.logger.error( self.logger.error(
'%s disabled. Error: %s', sonarr_section, e) '%s disabled. Error: %s', section, e)
# Parse Radarr options # Parse Radarr options
self.radarr_enabled = self.enable_check('radarr_server_ids') self.radarr_enabled = self.enable_check('radarr_server_ids')
if self.radarr_enabled: if self.radarr_enabled:
for server_id in self.radarr_enabled: for server_id in self.radarr_enabled:
radarr_section = 'radarr-' + str(server_id) section = 'radarr-' + str(server_id)
try: try:
url = self.url_check(self.config.get(radarr_section, 'url')) url = self.url_check(self.config.get(section, 'url'))
apikey = self.config.get(radarr_section, 'apikey') apikey = self.config.get(section, 'apikey')
scheme = 'https://' if self.config.getboolean( scheme = 'https://' if self.config.getboolean(section, 'ssl') else 'http://'
radarr_section, 'ssl') else 'http://'
verify_ssl = self.config.getboolean( verify_ssl = self.config.getboolean(section, 'verify_ssl')
radarr_section, 'verify_ssl')
if scheme != 'https://': if scheme != 'https://':
verify_ssl = False verify_ssl = False
queue = self.config.getboolean(radarr_section, 'queue') queue = self.config.getboolean(section, 'queue')
queue_run_seconds = self.config.getint( queue_run_seconds = self.config.getint(section, 'queue_run_seconds')
radarr_section, 'queue_run_seconds')
get_missing = self.config.getboolean( get_missing = self.config.getboolean(section, 'get_missing')
radarr_section, 'get_missing')
get_missing_run_seconds = self.config.getint( get_missing_run_seconds = self.config.getint(section, 'get_missing_run_seconds')
radarr_section, 'get_missing_run_seconds')
server = RadarrServer(server_id, scheme + url, apikey, verify_ssl, queue, queue_run_seconds, server = RadarrServer(server_id, scheme + url, apikey, verify_ssl, queue, queue_run_seconds,
get_missing, get_missing_run_seconds) get_missing, get_missing_run_seconds)
@ -195,36 +188,31 @@ class INIParser(object):
except configparser.NoOptionError as e: except configparser.NoOptionError as e:
self.radarr_enabled = False self.radarr_enabled = False
self.logger.error( self.logger.error(
'%s disabled. Error: %s', radarr_section, e) '%s disabled. Error: %s', section, e)
# Parse Tautulli options # Parse Tautulli options
self.tautulli_enabled = self.enable_check('tautulli_server_ids') self.tautulli_enabled = self.enable_check('tautulli_server_ids')
if self.tautulli_enabled: if self.tautulli_enabled:
for server_id in self.tautulli_enabled: for server_id in self.tautulli_enabled:
tautulli_section = 'tautulli-' + str(server_id) section = 'tautulli-' + str(server_id)
try: try:
url = self.url_check(self.config.get(tautulli_section, 'url')) url = self.url_check(self.config.get(section, 'url'))
fallback_ip = self.config.get( fallback_ip = self.config.get(section, 'fallback_ip')
tautulli_section, 'fallback_ip')
apikey = self.config.get(tautulli_section, 'apikey') apikey = self.config.get(section, 'apikey')
scheme = 'https://' if self.config.getboolean( scheme = 'https://' if self.config.getboolean(section, 'ssl') else 'http://'
tautulli_section, 'ssl') else 'http://'
verify_ssl = self.config.getboolean( verify_ssl = self.config.getboolean(section, 'verify_ssl')
tautulli_section, 'verify_ssl')
if scheme != 'https://': if scheme != 'https://':
verify_ssl = False verify_ssl = False
get_activity = self.config.getboolean( get_activity = self.config.getboolean(section, 'get_activity')
tautulli_section, 'get_activity')
get_activity_run_seconds = self.config.getint( get_activity_run_seconds = self.config.getint(section, 'get_activity_run_seconds')
tautulli_section, 'get_activity_run_seconds')
server = TautulliServer(server_id, scheme + url, fallback_ip, apikey, verify_ssl, get_activity, server = TautulliServer(server_id, scheme + url, fallback_ip, apikey, verify_ssl, get_activity,
get_activity_run_seconds) get_activity_run_seconds)
@ -232,39 +220,33 @@ class INIParser(object):
except configparser.NoOptionError as e: except configparser.NoOptionError as e:
self.tautulli_enabled = False self.tautulli_enabled = False
self.logger.error( self.logger.error(
'%s disabled. Error: %s', tautulli_section, e) '%s disabled. Error: %s', section, e)
# Parse Ombi options # Parse Ombi options
self.ombi_enabled = self.enable_check('ombi_server_ids') self.ombi_enabled = self.enable_check('ombi_server_ids')
if self.ombi_enabled: if self.ombi_enabled:
for server_id in self.ombi_enabled: for server_id in self.ombi_enabled:
ombi_section = 'ombi-' + str(server_id) section = 'ombi-' + str(server_id)
try: try:
url = self.url_check(self.config.get(ombi_section, 'url')) url = self.url_check(self.config.get(section, 'url'))
apikey = self.config.get(ombi_section, 'apikey') apikey = self.config.get(section, 'apikey')
scheme = 'https://' if self.config.getboolean( scheme = 'https://' if self.config.getboolean(section, 'ssl') else 'http://'
ombi_section, 'ssl') else 'http://'
verify_ssl = self.config.getboolean( verify_ssl = self.config.getboolean(section, 'verify_ssl')
ombi_section, 'verify_ssl')
if scheme != 'https://': if scheme != 'https://':
verify_ssl = False verify_ssl = False
request_type_counts = self.config.getboolean( request_type_counts = self.config.getboolean(section, 'get_request_type_counts')
ombi_section, 'get_request_type_counts')
request_type_run_seconds = self.config.getint( request_type_run_seconds = self.config.getint(section, 'request_type_run_seconds')
ombi_section, 'request_type_run_seconds')
request_total_counts = self.config.getboolean( request_total_counts = self.config.getboolean(section, 'get_request_total_counts')
ombi_section, 'get_request_total_counts')
request_total_run_seconds = self.config.getint( request_total_run_seconds = self.config.getint(section, 'request_total_run_seconds')
ombi_section, 'request_total_run_seconds')
server = OmbiServer(server_id, scheme + url, apikey, verify_ssl, request_type_counts, server = OmbiServer(server_id, scheme + url, apikey, verify_ssl, request_type_counts,
request_type_run_seconds, request_total_counts, request_total_run_seconds) request_type_run_seconds, request_total_counts, request_total_run_seconds)
@ -272,35 +254,61 @@ class INIParser(object):
except configparser.NoOptionError as e: except configparser.NoOptionError as e:
self.ombi_enabled = False self.ombi_enabled = False
self.logger.error( self.logger.error(
'%s disabled. Error: %s', ombi_section, e) '%s disabled. Error: %s', section, e)
# Parse SickChill options
self.sickchill_enabled = self.enable_check('sickchill_server_ids')
if self.sickchill_enabled:
for server_id in self.sickchill_enabled:
section = 'sickchill-' + str(server_id)
try:
url = self.url_check(self.config.get(section, 'url'))
apikey = self.config.get(section, 'apikey')
scheme = 'https://' if self.config.getboolean(section, 'ssl') else 'http://'
verify_ssl = self.config.getboolean(section, 'verify_ssl')
if scheme != 'https://':
verify_ssl = False
get_missing = self.config.getboolean(section, 'get_missing')
get_missing_run_seconds = self.config.getint(section, 'get_missing_run_seconds')
server = SickChillServer(server_id, scheme + url, apikey, verify_ssl,
get_missing, get_missing_run_seconds)
self.sickchill_servers.append(server)
except configparser.NoOptionError as e:
self.sickchill_enabled = False
self.logger.error(
'%s disabled. Error: %s', section, e)
# Parse ASA opts # Parse ASA opts
self.ciscoasa_enabled = self.enable_check('ciscoasa_firewall_ids') self.ciscoasa_enabled = self.enable_check('ciscoasa_firewall_ids')
if self.ciscoasa_enabled: if self.ciscoasa_enabled:
for firewall_id in self.ciscoasa_enabled: for firewall_id in self.ciscoasa_enabled:
ciscoasa_section = 'ciscoasa-' + str(firewall_id) section = 'ciscoasa-' + str(firewall_id)
try: try:
url = self.url_check(self.config.get(ciscoasa_section, 'url')) url = self.url_check(self.config.get(section, 'url'))
username = self.config.get(ciscoasa_section, 'username') username = self.config.get(section, 'username')
password = self.config.get(ciscoasa_section, 'password') password = self.config.get(section, 'password')
scheme = 'https://' if self.config.getboolean( scheme = 'https://' if self.config.getboolean(section, 'ssl') else 'http://'
ciscoasa_section, 'ssl') else 'http://'
verify_ssl = self.config.getboolean( verify_ssl = self.config.getboolean(section, 'verify_ssl')
ciscoasa_section, 'verify_ssl')
if scheme != 'https://': if scheme != 'https://':
verify_ssl = False verify_ssl = False
outside_interface = self.config.get( outside_interface = self.config.get(section, 'outside_interface')
ciscoasa_section, 'outside_interface')
get_bandwidth_run_seconds = self.config.getint( get_bandwidth_run_seconds = self.config.getint(section, 'get_bandwidth_run_seconds')
ciscoasa_section, 'get_bandwidth_run_seconds')
firewall = CiscoASAFirewall(firewall_id, scheme + url, username, password, outside_interface, firewall = CiscoASAFirewall(firewall_id, scheme + url, username, password, outside_interface,
verify_ssl, get_bandwidth_run_seconds) verify_ssl, get_bandwidth_run_seconds)
@ -308,4 +316,4 @@ class INIParser(object):
except configparser.NoOptionError as e: except configparser.NoOptionError as e:
self.ciscoasa_enabled = False self.ciscoasa_enabled = False
self.logger.error( self.logger.error(
'%s disabled. Error: %s', ciscoasa_section, e) '%s disabled. Error: %s', section, e)

64
varken/sickchill.py Normal file
View file

@ -0,0 +1,64 @@
import logging
from requests import Session, Request
from datetime import datetime, timezone, date, timedelta
from varken.helpers import hashit, connection_handler
from varken.structures import SickChillTVShow
class SickChillAPI(object):
def __init__(self, server, dbmanager):
self.dbmanager = dbmanager
self.server = server
# Create session to reduce server web thread load, and globally define pageSize for all requests
self.session = Session()
self.session.params = {'limit': 1000}
self.endpoint = f"/api/{self.server.api_key}"
self.logger = logging.getLogger()
def __repr__(self):
return "<sickchill-{}>".format(self.server.id)
def get_missing(self):
now = datetime.now(timezone.utc).astimezone().isoformat()
influx_payload = []
params = {'cmd': 'future', 'paused': 1, 'type': 'missed|today|soon|later|snatched'}
req = self.session.prepare_request(Request('GET', self.server.url + self.endpoint, params=params))
get = connection_handler(self.session, req, self.server.verify_ssl)
if not get:
return
try:
for key, section in get['data'].items():
get['data'][key] = [SickChillTVShow(**show) for show in section]
except TypeError as e:
self.logger.error('TypeError has occurred : %s while creating SickChillTVShow structure', e)
return
for key, section in get['data'].items():
for show in section:
sxe = 'S{:0>2}E{:0>2}'.format(show.season, show.episode)
hash_id = hashit('{}{}{}'.format(self.server.id, show.show_name, sxe))
missing_types = [(0, 'future'), (1, 'later'), (2, 'soon'), (3, 'today'), (4, 'missed')]
influx_payload.append(
{
"measurement": "SickChill",
"tags": {
"type": [item[0] for item in missing_types if key in item][0],
"indexerid": show.indexerid,
"server": self.server.id,
"name": show.show_name,
"epname": show.ep_name,
"sxe": sxe,
"airdate": show.airdate,
},
"time": now,
"fields": {
"hash": hash_id
}
}
)
self.dbmanager.write_points(influx_payload)

26
varken/structures.py
View file

@ -70,6 +70,16 @@ class InfluxServer(NamedTuple):
username: str = 'root' username: str = 'root'
password: str = 'root' password: str = 'root'
class SickChillServer(NamedTuple):
id: int = None
url: str = None
api_key: str = None
verify_ssl: bool = False
get_missing: bool = False
get_missing_run_seconds: int = 30
class CiscoASAFirewall(NamedTuple): class CiscoASAFirewall(NamedTuple):
id: int = None id: int = None
url: str = '192.168.1.1' url: str = '192.168.1.1'
@ -384,3 +394,19 @@ class OmbiTVRequest(NamedTuple):
totalSeasons: int = None totalSeasons: int = None
childRequests: list = None childRequests: list = None
id: int = None id: int = None
class SickChillTVShow(NamedTuple):
airdate: str = None
airs: str = None
ep_name: str = None
ep_plot: str = None
episode: int = None
indexerid: int = None
network: str = None
paused: int = None
quality: str = None
season: int = None
show_name: str = None
show_status: str = None
tvdbid: int = None
weekday: int = None