fix parse env. vars. in python client - rk

rjojjr · Sep 28, 2024 · c11727a · c11727a
1 parent 9d5a6cd
commit c11727a
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/client-modules/python/src/taskslock/client/utils/host_utils.py b/client-modules/python/src/taskslock/client/utils/host_utils.py
@@ -2,17 +2,13 @@
 
 
 def get_host() -> str:
-    try:
-        return os.environ.get('TASKS_LOCK_API_HOST')
-    except Exception as e:
-        return 'localhost'
+    host = os.environ.get('TASKS_LOCK_API_HOST')
+    return host if host is not None else 'localhost'
 
 
 def get_protocol() -> str:
-    try:
-        return os.environ.get('TASKS_LOCK_API_PROTOCOL')
-    except Exception as e:
-        return 'http'
+    proto = os.environ.get('TASKS_LOCK_API_PROTOCOL')
+    return proto if proto is not None else 'http'
 
 
 def get_port() -> int: