From ae70c95df37e5ae66e5a6be1c86030b92d04e324 Mon Sep 17 00:00:00 2001
From: keonly <keon.lee@proton.me>
Date: Sat, 25 Nov 2023 17:01:49 +0900
Subject: [PATCH] feat: send webhooks to slack app when scraping

---
 configurations/secrets.py |  8 ++++++++
 scrap/utils/runner.py     | 25 +++++++++++++++++++------
 2 files changed, 27 insertions(+), 6 deletions(-)

diff --git a/configurations/secrets.py b/configurations/secrets.py
index 5a72cd4..cd4ef62 100644
--- a/configurations/secrets.py
+++ b/configurations/secrets.py
@@ -37,3 +37,11 @@ class EmailSecrets:
     sender_email = str(os.getenv("SCRAP_SENDER_EMAIL") or "")
     receiver_email = str(os.getenv("SCRAP_RECEIVER_EMAIL") or "")
     password = str(os.getenv("SCRAP_EMAIL_PASSWORD") or "")
+
+
+class WebhookSecrets:
+    """
+    스크랩 결과 웹훅 전송에 필요한 키를 정의합니다.
+    """
+
+    webhook_url = str(os.getenv("WEBHOOK_URL") or "")
diff --git a/scrap/utils/runner.py b/scrap/utils/runner.py
index dd121b2..d40a162 100644
--- a/scrap/utils/runner.py
+++ b/scrap/utils/runner.py
@@ -10,6 +10,8 @@
 from tqdm import tqdm
 from abc import *
 
+from configurations.secrets import WebhookSecrets
+
 from scrap.utils.export import export_results_to_json, export_results_to_txt
 from scrap.utils.database import save_to_database
 from scrap.utils.types import ScrapResult, ScrapBasicArgument
@@ -30,6 +32,7 @@
 from scrap.local_councils import *
 from scrap.metropolitan_council import *
 from scrap.national_council import *
+from requests import post
 from requests.exceptions import Timeout
 
 
@@ -64,6 +67,16 @@ def handle_errors(self, cid: int | str, error):
             self.parseerror_count += 1
         logging.error(f"| {cid} | 오류: {error}")
 
+    def send_webhook(self, message: str) -> None:
+        webhook_url = WebhookSecrets.webhook_url
+        payload = {"text": message}
+
+        response = requests.post(webhook_url, json=payload)
+        if response.status_code != 200:
+            raise ValueError(
+                f"Request to slack returned an error {response.status_code}, the response is:\n{response.text}"
+            )
+
     @abstractmethod
     def run(self) -> Dict[str, ScrapResult]:
         pass
@@ -136,9 +149,9 @@ def run(self, cids: Iterable[int]) -> Dict[int, ScrapResult]:
             except Exception as e:
                 self.handle_errors(cid, e)
 
-        logging.info(
-            f"| 총 실행 횟수: {len(cids)} | 에러: {list(self.error_log.keys())}, 총 {len(self.error_log)}회 | 그 중 정보 없음 횟수: {self.parseerror_count} | 타임아웃 횟수: {self.timeout_count} |"
-        )
+        result_summary = f"| 총 실행 횟수: {len(cids)} | 에러: {list(self.error_log.keys())}, 총 {len(self.error_log)}회 | 그 중 정보 없음 횟수: {self.parseerror_count} | 타임아웃 횟수: {self.timeout_count} |"
+        logging.info(result_summary)
+        self.send_webhook("지방의회 스크랩 결과\n" + result_summary)
 
         return scrape_results
 
@@ -168,9 +181,9 @@ def run(self, cids: Iterable[int]) -> Dict[int, ScrapResult]:
             except Exception as e:
                 self.handle_errors(cid, e)
 
-        logging.info(
-            f"| 총 실행 횟수: {len(cids)} | 에러: {list(self.error_log.keys())}, 총 {len(self.error_log)}회 | 그 중 정보 없음 횟수: {self.parseerror_count} | 타임아웃 횟수: {self.timeout_count} |"
-        )
+        result_summary = f"| 총 실행 횟수: {len(cids)} | 에러: {list(self.error_log.keys())}, 총 {len(self.error_log)}회 | 그 중 정보 없음 횟수: {self.parseerror_count} | 타임아웃 횟수: {self.timeout_count} |"
+        logging.info(result_summary)
+        self.send_webhook("광역의회 스크랩 결과\n" + result_summary)
 
         return scrape_results