65 lines
2.5 KiB
Python
65 lines
2.5 KiB
Python
|
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)
|