aykhans/series-robot-web
1
0
Fork 0
mirror of https://github.com/aykhans/series-robot-web.git synced 2025-04-08 10:24:01 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added InvalidAPIKey exception to imdb_api_access

Browse Source
This commit is contained in:
Aykhan 2023-07-22 18:39:29 +04:00
parent 995da2fcaa
commit b87da3b032
9 changed files with 86 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/account/views/profile_editing.py
View File

@ -16,21 +16,26 @@ def profile_editing_view(request):
if 'imdb_api_key' in form.changed_data:
try:
get_request(f"https://imdb-api.com/en/API/Title/{request.POST['imdb_api_key']}/tt0110413")
except StatusCodeError:
messages.info(request, 'Account not created. Please try again later')
return redirect('profile-editing')
except MaximumUsageError as e:
messages.info(request, str(e))
return redirect('profile-editing')
except APIError as e:
except (APIError, InvalidAPIKey) as e:
if e.message == 'Invalid API Key':
form.add_error('imdb_api_key', 'Invalid API Key')
return render(request, 'profile_editing.html', context={"form": form})
messages.info(request, str(e))
return redirect('profile-editing')
if 'email' in form.changed_data:
to_email = form.cleaned_data.get('email')
if to_email is not None:
if form.cleaned_data.get('send_email'):
form.add_error('send_email', 'You must verify the email before enabling the send email feature')

6
src/account/views/register.py
View File

@ -16,16 +16,20 @@ def register_view(request): # sourcery skip: extract-method
if form.is_valid():
try:
get_request(f"https://imdb-api.com/en/API/Title/{request.POST['imdb_api_key']}/tt0110413")
except StatusCodeError:
messages.info(request, 'Account not created. Please try again later')
return redirect('register')
except MaximumUsageError as e:
messages.info(request, str(e))
return redirect('register')
except APIError as e:
except (APIError, InvalidAPIKey) as e:
if e.message == 'Invalid API Key':
form.add_error('imdb_api_key', 'Invalid API Key')
return render(request, 'register.html', context={"form": form})
messages.info(request, str(e))
return redirect('register')

14
src/imdb_api_access/_requests.py
View File

@ -4,7 +4,14 @@ from typing import (Union,
from .exceptions import *
def get_request(url: str) -> Union[StatusCodeError, APIError, MaximumUsageError, Dict]:
def get_request(url: str) -> Union[
StatusCodeError,
APIError,
MaximumUsageError,
InvalidAPIKey,
Dict
]:
raw_data = get(url, timeout=7)
if raw_data.status_code != 200:
@ -14,5 +21,10 @@ def get_request(url: str) -> Union[StatusCodeError, APIError, MaximumUsageError,
if data['errorMessage']:
if 'Maximum usage' in data['errorMessage']:
raise MaximumUsageError(data['errorMessage'])
elif 'Upgrade your account to use the service' in data['errorMessage']:
raise InvalidAPIKey()
raise APIError(data['errorMessage'])
return data

7
src/imdb_api_access/exceptions.py
View File

@ -19,5 +19,12 @@ class MaximumUsageError(Exception):
def __init__(self, message: str):
self.message = message
def __str__(self) -> str:
return f"IMDB API: {self.message}"
class InvalidAPIKey(Exception):
def __init__(self, message: str = 'Upgrade your imdb-api account to use the service'):
self.message = message
def __str__(self) -> str:
return f"IMDB API: {self.message}"

30
src/imdb_api_access/series_counter.py
View File

@ -14,11 +14,13 @@ class SeriesCounter:
self.new_series_list: List[NewSeries] = []
self.error_series: List[SeriesModel] = []
def get_episode_count(self, series: SeriesModel, data_seasons: Dict) -> Union[StatusCodeError,
APIError,
MaximumUsageError,
Dict]:
# sourcery skip: aware-datetime-for-utc
def get_episode_count(self, series: SeriesModel, data_seasons: Dict) -> Union[
StatusCodeError,
APIError,
MaximumUsageError,
Dict
]:
now_date = datetime.strptime(datetime.strftime(datetime.utcnow(),'%d %b %Y'), '%d %b %Y')
data_return = {
'new_episodes_count': 0,
@ -46,10 +48,15 @@ class SeriesCounter:
except ValueError: return data_return
return data_return
def find_new_series(self, series: List[SeriesModel]) -> Union[MaximumUsageError, None]:
def find_new_series(self, series: List[SeriesModel]) -> Union[
MaximumUsageError,
InvalidAPIKey,
None
]:
for s in series:
try:
data = get_request(f"https://imdb-api.com/en/API/Title/{self.api_key}/{s.imdb_id}")
except (StatusCodeError, APIError):
self.error_series.append(s)
@ -63,10 +70,13 @@ class SeriesCounter:
except (StatusCodeError, APIError):
self.error_series.append(s)
def find_last_episode(self, series: SeriesModel) -> Union[MaximumUsageError,
StatusCodeError,
APIError,
NewSeries]:
def find_last_episode(self, series: SeriesModel) -> Union[
MaximumUsageError,
StatusCodeError,
InvalidAPIKey,
APIError,
NewSeries
]:
data = get_request(f"https://imdb-api.com/en/API/Title/{self.api_key}/{series.imdb_id}")
data = self.get_episode_count(series, data)

9
src/series/tasks.py
View File

@ -2,7 +2,10 @@ from celery import shared_task
from account.models import User
from django.template.loader import render_to_string
from imdb_api_access import SeriesCounter
from imdb_api_access import MaximumUsageError
from imdb_api_access import (
MaximumUsageError,
InvalidAPIKey
)
from django.core.mail import (EmailMessage,
get_connection)
# from celery.utils.log import get_task_logger
@ -22,6 +25,7 @@ def compose_email(user):
try:
series_counter.find_new_series(series)
series_len = len(series_counter.new_series_list)
except MaximumUsageError as e:
if series_len := len(series_counter.new_series_list):
for updated_series in series_counter.new_series_list:
@ -37,6 +41,9 @@ def compose_email(user):
)
return
except InvalidAPIKey:
return
if series_len:
for updated_series in series_counter.new_series_list:
update_series(updated_series)

12
src/series/views/add_series.py
View File

@ -26,12 +26,15 @@ class AddSeriesView(LoginRequiredMixin, CreateView):
try:
data = get_request(f"https://imdb-api.com/en/API/Title/{series.user.imdb_api_key}/{series.imdb_id}")
except StatusCodeError:
messages.info(self.request, 'TV Series can not added. Please try again.')
return redirect('add-series')
except MaximumUsageError as e:
except (MaximumUsageError, InvalidAPIKey) as e:
messages.info(self.request, str(e))
return redirect('add-series')
except APIError:
form.add_error('imdb_id', 'ID is not correct.')
return self.form_invalid(form)
@ -48,12 +51,15 @@ class AddSeriesView(LoginRequiredMixin, CreateView):
try:
data = get_request(f"https://imdb-api.com/en/API/SeasonEpisodes/{series.user.imdb_api_key}/{series.imdb_id}/{series.watched_season}")
except StatusCodeError:
messages.info(self.request, 'TV Series can not added. Please try again.')
return redirect('add-series')
except MaximumUsageError as e:
except (MaximumUsageError, InvalidAPIKey) as e:
messages.info(self.request, str(e))
return redirect('add-series')
except APIError:
form.add_error('imdb_id', 'ID is not correct.')
return self.form_invalid(form)
@ -87,7 +93,7 @@ class AddSeriesView(LoginRequiredMixin, CreateView):
messages.info(self.request, 'TV Series can not added. Please try again later.')
return redirect('add-series')
except MaximumUsageError as e:
except (MaximumUsageError, InvalidAPIKey) as e:
messages.info(self.request, str(e))
return redirect('homepage')

14
src/series/views/new_episodes.py
View File

@ -3,7 +3,10 @@ from django.contrib.auth.decorators import login_required
from django.contrib import messages
from account.models import User
from imdb_api_access import SeriesCounter
from imdb_api_access import MaximumUsageError
from imdb_api_access import (
MaximumUsageError,
InvalidAPIKey
)
@login_required(login_url='/account/login')
@ -11,10 +14,12 @@ def new_episodes_view(request):
series = User.objects.get(id=request.user.id).series.filter(show=True).order_by('-id')
series_counter = SeriesCounter(request.user.imdb_api_key)
try:
series_counter.find_new_series(series)
series_len = len(series_counter.new_series_list)
error_series_len = len(series_counter.error_series)
except MaximumUsageError as e:
if series_len := len(series_counter.new_series_list):
for updated_series in series_counter.new_series_list:
@ -30,16 +35,23 @@ def new_episodes_view(request):
f"Series could not be updated. {str(e)}")
return redirect('homepage')
except InvalidAPIKey as e:
messages.info(request,
f"Series could not be updated: {str(e.message)}")
return redirect('homepage')
if series_len:
for updated_series in series_counter.new_series_list:
updated_series.series.last_season = updated_series.last_season
updated_series.series.last_episode = updated_series.last_episode
updated_series.series.new_episodes_count = updated_series.new_episodes_count
updated_series.series.save()
if error_series_len:
messages.warning(request,
f"{series_len} series updated ({error_series_len} series could not be updated).")
return redirect('homepage')
messages.warning(request,
f"{series_len} series updated.")
return redirect('homepage')

7
src/series/views/update_series.py
View File

@ -35,12 +35,15 @@ class UpdateSeriesView(LoginRequiredMixin, UpdateView):
try:
data = get_request(f"https://imdb-api.com/en/API/Title/{series.user.imdb_api_key}/{series.imdb_id}")
except StatusCodeError:
messages.info(self.request, 'TV Series can not added. Please try again.')
return redirect('update-series', self.kwargs.get('slug'))
except MaximumUsageError as e:
except (MaximumUsageError, InvalidAPIKey) as e:
messages.info(self.request, str(e))
return redirect('update-series', self.kwargs.get('slug'))
except APIError:
form.add_error('imdb_id', 'ID is not correct.')
return self.form_invalid(form)
@ -95,7 +98,7 @@ class UpdateSeriesView(LoginRequiredMixin, UpdateView):
messages.info(self.request, 'TV Series can not updated. Please try again later.')
return redirect('update-series', self.kwargs.get('slug'))
except MaximumUsageError as e:
except (MaximumUsageError, InvalidAPIKey) as e:
messages.info(self.request, str(e))
return redirect('homepage')