refactor: Include logger.catch with reraise

This commit is contained in:
TrisNol 2023-11-11 14:02:00 +01:00
parent 5dcf8ecf55
commit 05ea0fbb33
2 changed files with 4 additions and 0 deletions

View File

@ -4,6 +4,7 @@ from datetime import datetime
import requests import requests
import xmltodict import xmltodict
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
from loguru import logger
from selenium import webdriver from selenium import webdriver
from aki_prj23_transparenzregister.models.news import News from aki_prj23_transparenzregister.models.news import News
@ -19,6 +20,7 @@ class HandelsblattRSS(BaseNewsExtractor):
"""Constructor.""" """Constructor."""
super().__init__("https://www.handelsblatt.com/contentexport/feed") super().__init__("https://www.handelsblatt.com/contentexport/feed")
@logger.catch(reraise=True)
def get_news_for_category(self, category: str = "unternehmen") -> list[News] | None: def get_news_for_category(self, category: str = "unternehmen") -> list[News] | None:
"""Retrieve news for the given category from the Handelsblatt RSS feed. """Retrieve news for the given category from the Handelsblatt RSS feed.

View File

@ -1,6 +1,7 @@
"""Tageschau API news extractor.""" """Tageschau API news extractor."""
import requests import requests
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
from loguru import logger
from aki_prj23_transparenzregister.models.news import News from aki_prj23_transparenzregister.models.news import News
from aki_prj23_transparenzregister.utils.data_extraction.news.base import ( from aki_prj23_transparenzregister.utils.data_extraction.news.base import (
@ -15,6 +16,7 @@ class TagesschauAPI(BaseNewsExtractor):
"""Constructor.""" """Constructor."""
super().__init__("https://www.tagesschau.de/api2") super().__init__("https://www.tagesschau.de/api2")
@logger.catch(reraise=True)
def get_news_for_category(self, category: str = "wirtschaft") -> list[News] | None: def get_news_for_category(self, category: str = "wirtschaft") -> list[News] | None:
"""Retrieve news for the given category from the Tageschau API. """Retrieve news for the given category from the Tageschau API.