feat: make mock first to check the api

sinkaroid · May 12, 2022 · 91b1772 · 91b1772
1 parent 08be5b4
commit 91b1772
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/booru/client/danbooru.py b/booru/client/danbooru.py
@@ -63,6 +63,13 @@ def __init__(self, api_key: str = "", login: str = ""):
 
         self.specs = {"api_key": self.api_key, "login": self.login}
 
+        if self.api_key and self.login:
+            self.data = requests.get(f"https://danbooru.donmai.us/profile.json?api_key={self.api_key}&login={self.login}")
+
+            if self.data.status_code != 200:
+                raise ValueError(Booru.error_handling_invalid_auth)
+
+
     async def search(
         self,
         query: str,