diff --git a/src/series/views/add_series.py b/src/series/views/add_series.py index 65f585e..a47308c 100644 --- a/src/series/views/add_series.py +++ b/src/series/views/add_series.py @@ -29,8 +29,8 @@ class AddSeriesView(LoginRequiredMixin, CreateView): except StatusCodeError: messages.info(self.request, 'TV Series can not added. Please try again.') return redirect('add-series') - except MaximumUsageError: - messages.info(self.request, f"IMDB API: {data['errorMessage']}") + except MaximumUsageError as e: + messages.info(self.request, str(e)) return redirect('add-series') except APIError: form.add_error('imdb_id', 'ID is not correct.') @@ -51,8 +51,8 @@ class AddSeriesView(LoginRequiredMixin, CreateView): except StatusCodeError: messages.info(self.request, 'TV Series can not added. Please try again.') return redirect('add-series') - except MaximumUsageError: - messages.info(self.request, f"IMDB API: {data['errorMessage']}") + except MaximumUsageError as e: + messages.info(self.request, str(e)) return redirect('add-series') except APIError: form.add_error('imdb_id', 'ID is not correct.') diff --git a/src/series/views/update_series.py b/src/series/views/update_series.py index f06bc52..e777477 100644 --- a/src/series/views/update_series.py +++ b/src/series/views/update_series.py @@ -37,8 +37,8 @@ class UpdateSeriesView(LoginRequiredMixin, UpdateView): except StatusCodeError: messages.info(self.request, 'TV Series can not added. Please try again.') return redirect('update-series', self.kwargs.get('slug')) - except MaximumUsageError: - messages.info(self.request, f"IMDB API: {data['errorMessage']}") + except MaximumUsageError 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.') @@ -58,8 +58,8 @@ class UpdateSeriesView(LoginRequiredMixin, UpdateView): except StatusCodeError: messages.info(self.request, 'TV Series can not added. Please try again.') return redirect('update-series', self.kwargs.get('slug')) - except MaximumUsageError: - messages.info(self.request, f"IMDB API: {data['errorMessage']}") + except MaximumUsageError 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.')