refactor: set cookies after status code check

antwxne · Jun 28, 2022 · 7504968 · 7504968
1 parent a5830aa
commit 7504968
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="CVAT wrapper",
-    version="0.0.18",
+    version="1.0.0",
     author="antwxne",
     author_email="antoine.desruet@epitech.eu",
     description="Python wrapper for CVAT API",

diff --git a/src/CVAT/__init__.py b/src/CVAT/__init__.py
@@ -34,8 +34,6 @@ def __init__(self, username: str = "admin", password: str = "admin", url: str =
         }
         response: Response = self.session.post(url=f'{self.url}/api/auth/login',
                                                json=body)
-        self.session.headers = {"Authorization": f'Token {response.json()["key"]}'}
         if response.status_code != 200:
             raise ValueError("Bad credentials")
-
-
+        self.session.headers = {"Authorization": f'Token {response.json()["key"]}'}