mirror of
https://github.com/meisnate12/Plex-Meta-Manager
synced 2024-11-26 14:30:25 +00:00
63 lines
2.3 KiB
Python
63 lines
2.3 KiB
Python
import logging, math, re, requests
|
|
from lxml import html
|
|
from modules import util
|
|
from modules.util import Failed
|
|
from retrying import retry
|
|
|
|
logger = logging.getLogger("Plex Meta Manager")
|
|
|
|
class OMDbObj:
|
|
def __init__(self, data):
|
|
self._data = data
|
|
self.title = data["Title"]
|
|
try:
|
|
self.year = int(data["Year"])
|
|
except (ValueError, TypeError):
|
|
self.year = None
|
|
self.content_rating = data["Rated"]
|
|
self.genres = util.get_list(data["Genre"])
|
|
self.genres_str = data["Genre"]
|
|
try:
|
|
self.imdb_rating = float(data["imdbRating"])
|
|
except (ValueError, TypeError):
|
|
self.imdb_rating = None
|
|
try:
|
|
self.imdb_votes = int(str(data["imdbVotes"]).replace(',', ''))
|
|
except (ValueError, TypeError):
|
|
self.imdb_votes = None
|
|
try:
|
|
self.metacritic_rating = int(data["Metascore"])
|
|
except (ValueError, TypeError):
|
|
self.metacritic_rating = None
|
|
self.imdb_id = data["imdbID"]
|
|
self.type = data["Type"]
|
|
|
|
class OMDbAPI:
|
|
def __init__(self, params, Cache=None):
|
|
self.url = "http://www.omdbapi.com/"
|
|
self.apikey = params["apikey"]
|
|
self.cache = params["omdb_cache"]
|
|
self.cache_expiration = params["omdb_cache_expiration"]
|
|
self.limit = False
|
|
Cache.omdb_expiration = self.cache_expiration
|
|
self.Cache = Cache
|
|
self.get_omdb("tt0080684")
|
|
|
|
#@retry(stop_max_attempt_number=6, wait_fixed=10000, retry_on_exception=util.retry_if_not_failed)
|
|
def get_omdb(self, imdb_id):
|
|
expired = None
|
|
if self.cache and self.Cache:
|
|
omdb_dict, expired = self.Cache.query_omdb(imdb_id)
|
|
if omdb_dict and expired is False:
|
|
return OMDbObj(omdb_dict)
|
|
response = requests.get(self.url, params={"i": imdb_id, "apikey": self.apikey})
|
|
if response.status_code < 400:
|
|
omdb = OMDbObj(response.json())
|
|
if self.cache and self.Cache:
|
|
self.Cache.update_omdb(expired, omdb)
|
|
return omdb
|
|
else:
|
|
error = response.json()['Error']
|
|
if error == "Request limit reached!":
|
|
self.limit = True
|
|
raise Failed(f"OMDb Error: {error}")
|