switch between stages

Author: Johannes Hötter

onetask/__init__.py

@@ -10,9 +10,10 @@
 
 class Client:
     def __init__(
-        self, user_name: str, password: str, project_id: str, debug: bool = False
+        self, user_name: str, password: str, project_id: str, stage: str = "prod"
     ):
-        if not debug:
+        settings.set_stage(stage)
+        if stage in ["prod", "test", "dev"]:
             self.session_token = api_calls.create_session_token(
                 user_name=user_name, password=password
             )

onetask/settings.py

@@ -1,14 +1,22 @@
 # -*- coding: utf-8 -*-
-LOCALHOST_SWITCH = False
+STAGE: str
 
 
-def set_to_localhost():
-    global LOCALHOST_SWITCH
-    LOCALHOST_SWITCH = True
+def set_stage(stage):
+    global STAGE
+    STAGE = stage
 
 
 def get_base_url():
-    return "http://localhost:8000" if LOCALHOST_SWITCH else "https://app.dev.onetask.ai"
+    global STAGE
+    if STAGE == "prod":
+        return "https://app.onetask.ai"
+    elif STAGE == "test":
+        return "https://app.test.onetask.ai"
+    elif STAGE == "dev":
+        return "https://app.dev.onetask.ai"
+    else:
+        return "http://localhost:8000"
 
 
 def get_authentication_url():

setup.py

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