Skip to content

Commit

Permalink
add api key to omdb and clean tvdb
Browse files Browse the repository at this point in the history
  • Loading branch information
getzze committed May 9, 2024
1 parent c46cab3 commit 409f2df
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 15 deletions.
2 changes: 1 addition & 1 deletion pytest.ini
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[pytest]
norecursedirs = build dist env .tox .eggs
addopts = --pep8 --flakes --doctest-glob='*.rst'
#addopts = --pep8 --flakes --doctest-glob='*.rst'
pep8maxlinelength = 120
pep8ignore =
docs/conf.py ALL
Expand Down
15 changes: 7 additions & 8 deletions subliminal/refiners/omdb.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,11 +10,16 @@

logger = logging.getLogger(__name__)

#: OMDB subliminal API key
OMDB_API_KEY = '44d5b275'


class OMDBClient(object):
base_url = 'http://www.omdbapi.com'

def __init__(self, version=1, session=None, headers=None, timeout=10):
def __init__(self, apikey=None, version=1, session=None, headers=None, timeout=10):
self.apikey = apikey or OMDB_API_KEY

#: Session for the requests
self.session = session or requests.Session()
self.session.timeout = timeout
Expand All @@ -24,7 +29,7 @@ def __init__(self, version=1, session=None, headers=None, timeout=10):

def get(self, id=None, title=None, type=None, year=None, plot='short', tomatoes=False):
# build the params
params = {}
params = {"apikey": self.apikey}
if id:
params['i'] = id
if title:
Expand Down Expand Up @@ -105,12 +110,6 @@ def refine(video, apikey=None, **kwargs):
* :attr:`~subliminal.video.Video.imdb_id`
"""
if not apikey:
logger.warning('No apikey. Skipping omdb refiner.')
return

omdb_client.session.params['apikey'] = apikey

if isinstance(video, Episode):
# exit if the information is complete
if video.series_imdb_id:
Expand Down
12 changes: 6 additions & 6 deletions subliminal/refiners/tvdb.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
# -*- coding: utf-8 -*-
from datetime import datetime, timedelta
from datetime import datetime, timedelta, timezone
from functools import wraps
import logging
import re
Expand Down Expand Up @@ -64,7 +64,7 @@ def __init__(self, apikey=None, username=None, password=None, language='en', ses
self.password = password

#: Last token acquisition date
self.token_date = datetime.utcnow() - self.token_lifespan
self.token_date = datetime.now(timezone.utc) - self.token_lifespan

#: Session for the requests
self.session = session or requests.Session()
Expand All @@ -89,11 +89,11 @@ def token(self):

@property
def token_expired(self):
return datetime.utcnow() - self.token_date > self.token_lifespan
return datetime.now(timezone.utc) - self.token_date > self.token_lifespan

@property
def token_needs_refresh(self):
return datetime.utcnow() - self.token_date > self.refresh_token_every
return datetime.now(timezone.utc) - self.token_date > self.refresh_token_every

def login(self):
"""Login"""
Expand All @@ -106,7 +106,7 @@ def login(self):
self.session.headers['Authorization'] = 'Bearer ' + r.json()['token']

# update token_date
self.token_date = datetime.utcnow()
self.token_date = datetime.now(timezone.utc)

def refresh_token(self):
"""Refresh token"""
Expand All @@ -118,7 +118,7 @@ def refresh_token(self):
self.session.headers['Authorization'] = 'Bearer ' + r.json()['token']

# update token_date
self.token_date = datetime.utcnow()
self.token_date = datetime.now(timezone.utc)

@requires_auth
def search_series(self, name=None, imdb_id=None, zap2it_id=None):
Expand Down

0 comments on commit 409f2df

Please sign in to comment.