From 05ea0fbb3321e2270088b4568a692b6795be1652 Mon Sep 17 00:00:00 2001
From: TrisNol <tristan.nolde@yahoo.de>
Date: Sat, 11 Nov 2023 14:02:00 +0100
Subject: [PATCH] refactor: Include logger.catch with reraise

---
 .../utils/data_extraction/news/handelsblatt.py                  | 2 ++
 .../utils/data_extraction/news/tagesschau.py                    | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/src/aki_prj23_transparenzregister/utils/data_extraction/news/handelsblatt.py b/src/aki_prj23_transparenzregister/utils/data_extraction/news/handelsblatt.py
index 2cadd92..a1af5d5 100644
--- a/src/aki_prj23_transparenzregister/utils/data_extraction/news/handelsblatt.py
+++ b/src/aki_prj23_transparenzregister/utils/data_extraction/news/handelsblatt.py
@@ -4,6 +4,7 @@ from datetime import datetime
 import requests
 import xmltodict
 from bs4 import BeautifulSoup
+from loguru import logger
 from selenium import webdriver
 
 from aki_prj23_transparenzregister.models.news import News
@@ -19,6 +20,7 @@ class HandelsblattRSS(BaseNewsExtractor):
         """Constructor."""
         super().__init__("https://www.handelsblatt.com/contentexport/feed")
 
+    @logger.catch(reraise=True)
     def get_news_for_category(self, category: str = "unternehmen") -> list[News] | None:
         """Retrieve news for the given category from the Handelsblatt RSS feed.
 
diff --git a/src/aki_prj23_transparenzregister/utils/data_extraction/news/tagesschau.py b/src/aki_prj23_transparenzregister/utils/data_extraction/news/tagesschau.py
index 43cb322..cabeaee 100644
--- a/src/aki_prj23_transparenzregister/utils/data_extraction/news/tagesschau.py
+++ b/src/aki_prj23_transparenzregister/utils/data_extraction/news/tagesschau.py
@@ -1,6 +1,7 @@
 """Tageschau API news extractor."""
 import requests
 from bs4 import BeautifulSoup
+from loguru import logger
 
 from aki_prj23_transparenzregister.models.news import News
 from aki_prj23_transparenzregister.utils.data_extraction.news.base import (
@@ -15,6 +16,7 @@ class TagesschauAPI(BaseNewsExtractor):
         """Constructor."""
         super().__init__("https://www.tagesschau.de/api2")
 
+    @logger.catch(reraise=True)
     def get_news_for_category(self, category: str = "wirtschaft") -> list[News] | None:
         """Retrieve news for the given category from the Tageschau API.