postgres change

Author: Johannes Hötter

onetask/__init__.py

@@ -13,17 +13,13 @@ def __init__(
         self, user_name: str, password: str, project_id: str, stage: str = "prod"
     ):
         settings.set_stage(stage)
-        if stage in ["prod", "test", "dev"]:
-            self.session_token = api_calls.create_session_token(
-                user_name=user_name, password=password
-            )
-            if self.session_token is not None:
-                msg.good("Logged in to system.")
-            else:
-                msg.fail("Could not log in. Please check your username and password.")
+        self.session_token = api_calls.create_session_token(
+            user_name=user_name, password=password
+        )
+        if self.session_token is not None:
+            msg.good("Logged in to system.")
         else:
-            self.session_token = None
-            msg.info("Sending requests to localhost")
+            msg.fail("Could not log in. Please check your username and password.")
         self.project_id = project_id
 
     def register_custom_lf(self, lf: Callable) -> None:

onetask/api_calls.py

@@ -47,9 +47,8 @@ def __init__(
         headers = {
             "Content-Type": "application/json",
             "User-Agent": f"python-sdk-{version}",
+            "Authorization": f"Bearer {session_token}",
         }
-        if session_token:
-            headers["Authorization"] = f"Bearer {session_token}"
 
         if data is None:
             self.response = requests.request(self.method, url, headers=headers)

onetask/settings.py

@@ -16,7 +16,7 @@ def get_base_url():
     elif STAGE == "dev":
         return "https://app.dev.onetask.ai"
     else:
-        return "http://localhost:8000"
+        return STAGE
 
 
 def get_authentication_url():

setup.py

@@ -10,7 +10,7 @@
 
 setup(
     name="onetask",
-    version="0.1.2",
+    version="0.1.6",
     author="onetask",
     author_email="[email protected]",
     description="Official Python SDK for the onetask API",