117 lines
3.5 KiB
Python
117 lines
3.5 KiB
Python
|
import time
|
||
|
from urllib.parse import urljoin
|
||
|
from datetime import datetime, timedelta
|
||
|
import requests as RQ
|
||
|
|
||
|
from utils import timed_cache
|
||
|
|
||
|
|
||
|
class Sonarr(object):
|
||
|
def __init__(self, url, api_key):
|
||
|
self.url = url
|
||
|
self.api_key = api_key
|
||
|
self.root_folder = self.get("api/v3/rootFolder")[0]["path"]
|
||
|
self.quality_profile = self.get("api/v3/qualityprofile")[0]
|
||
|
self.language_profile = self.get("api/v3/languageprofile")[0]
|
||
|
|
||
|
def get(self, url, **kwargs):
|
||
|
kwargs["apikey"] = self.api_key
|
||
|
kwargs["_"] = str(int(time.time()))
|
||
|
res = RQ.get(urljoin(self.url, url), params=kwargs)
|
||
|
res.raise_for_status()
|
||
|
try:
|
||
|
return res.json()
|
||
|
except Exception:
|
||
|
return res.text
|
||
|
|
||
|
def search(self, query):
|
||
|
return self.get("api/v3/series/lookup", term=query)
|
||
|
|
||
|
def status(self):
|
||
|
return self.get("api/v3/system/status")
|
||
|
|
||
|
@timed_cache()
|
||
|
def history(self, pageSize=500):
|
||
|
return self.get(
|
||
|
"api/v3/history",
|
||
|
page=1,
|
||
|
pageSize=500,
|
||
|
sortDirection="descending",
|
||
|
sortKey="date",
|
||
|
)
|
||
|
|
||
|
@timed_cache()
|
||
|
def calendar(self, days=30):
|
||
|
today = datetime.today()
|
||
|
start = today - timedelta(days=days)
|
||
|
end = today + timedelta(days=days)
|
||
|
return self.get(
|
||
|
"api/v3/calendar",
|
||
|
unmonitored=False,
|
||
|
start=start.isoformat(),
|
||
|
end=end.isoformat(),
|
||
|
)
|
||
|
|
||
|
@timed_cache()
|
||
|
def series(self, series_id=None, keys=None):
|
||
|
if series_id is None:
|
||
|
return self.get("api/v3/series")
|
||
|
ret = {}
|
||
|
ret["series"] = self.get("api/v3/series/{}".format(series_id))
|
||
|
ret["episodes"] = self.get("api/v3/episode", seriesId=series_id)
|
||
|
ret["episodeFile"] = self.get("api/v3/episodeFile", seriesId=series_id)
|
||
|
ret["queue"] = self.get("api/v3/queue/details", seriesId=series_id)
|
||
|
return ret
|
||
|
|
||
|
@timed_cache(seconds=60)
|
||
|
def queue(self, **kwargs):
|
||
|
data = []
|
||
|
page = 1
|
||
|
while True:
|
||
|
res = self.get("api/v3/queue", page=page, pageSize=100, **kwargs)
|
||
|
data = res.get("records", [])
|
||
|
page += 1
|
||
|
if len(data) >= res.get("totalRecords", 0):
|
||
|
break
|
||
|
return data
|
||
|
|
||
|
@timed_cache()
|
||
|
def details(self, episode_id):
|
||
|
return self.get("api/v3/queue/details", episodeIds=episode_id)
|
||
|
|
||
|
@timed_cache()
|
||
|
def episodes(self, series_id):
|
||
|
return self.get("api/v3/episode", seriesId=series_id)
|
||
|
|
||
|
def add(self, data):
|
||
|
data["qualityProfileId"] = self.quality_profile["id"]
|
||
|
data["languageProfileId"] = self.language_profile["id"]
|
||
|
data["rootFolderPath"] = self.root_folder
|
||
|
data["addOptions"] = {
|
||
|
"ignoreEpisodesWithoutFiles": False,
|
||
|
"ignoreEpisodesWithFiles": True,
|
||
|
"searchForMissingEpisodes": True,
|
||
|
}
|
||
|
data["seasonFolder"] = True
|
||
|
params = dict(apikey=self.api_key)
|
||
|
res = RQ.post(
|
||
|
urljoin(
|
||
|
self.url,
|
||
|
"api/v3/series"),
|
||
|
json=data,
|
||
|
params=params)
|
||
|
return res.json()
|
||
|
|
||
|
def log(self, limit=0):
|
||
|
return self.get(
|
||
|
"api/v3/log",
|
||
|
page=1,
|
||
|
pageSize=(limit or 1024),
|
||
|
sortDirection="descending",
|
||
|
sortKey="time",
|
||
|
)
|
||
|
|
||
|
def test(self):
|
||
|
self.status()
|
||
|
return {}
|