From d42a2e59d61574eb3a51f851a52642ae03611e84 Mon Sep 17 00:00:00 2001 From: quihi Date: Fri, 30 Dec 2022 00:52:08 -0700 Subject: [PATCH] set custom user agent --- AO3/requester.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/AO3/requester.py b/AO3/requester.py index e3d1c4e..67dd814 100644 --- a/AO3/requester.py +++ b/AO3/requester.py @@ -4,6 +4,9 @@ import requests +CUSTOM_USERAGENT = "ao3-api-bot" + + class Requester: """Requester object""" @@ -60,6 +63,10 @@ def request(self, *args, **kwargs): self._requests.append(time.time()) self.total += 1 + if "headers" not in kwargs: + kwargs["headers"] = {"User-Agent": CUSTOM_USERAGENT} + else: + kwargs["headers"]["User-Agent"] = CUSTOM_USERAGENT if "session" in kwargs: sess = kwargs["session"] del kwargs["session"]