From 58e9ac17b1c38a5f0d629b3361079a7530af8770 Mon Sep 17 00:00:00 2001 From: Dave Gallant Date: Tue, 8 Oct 2024 22:57:50 -0400 Subject: [PATCH] Add custom User-Agent --- rfd_notify/constants.py | 2 ++ rfd_notify/rfd.py | 4 +++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/rfd_notify/constants.py b/rfd_notify/constants.py index 2186548..1cd29d2 100644 --- a/rfd_notify/constants.py +++ b/rfd_notify/constants.py @@ -1 +1,3 @@ API_BASE_URL = "https://forums.redflagdeals.com" + +HEADERS = {"User-Agent": "rfd-notify"} diff --git a/rfd_notify/rfd.py b/rfd_notify/rfd.py index c3c27db..309bd1d 100644 --- a/rfd_notify/rfd.py +++ b/rfd_notify/rfd.py @@ -4,7 +4,7 @@ import re import shelve import requests from loguru import logger -from constants import API_BASE_URL +from constants import API_BASE_URL, HEADERS from config import Config from notifications import send_notification @@ -22,6 +22,7 @@ def get_topic(topic_id: int) -> List[Post]: try: response = requests.get( f"{API_BASE_URL}/api/topics/{topic_id}/posts?per_page=1&page=1", + headers=HEADERS, timeout=30, ) if response.status_code != 200: @@ -41,6 +42,7 @@ def get_topics(forum_id: int, pages: int) -> List[Topic]: for page in range(1, pages + 1): response = requests.get( f"{API_BASE_URL}/api/topics?forum_id={forum_id}&per_page=40&page={page}", + headers=HEADERS, timeout=30, ) if response.status_code != 200: