From 5137d75e9f31cd861b8fe50bdd079816c8e0a8e1 Mon Sep 17 00:00:00 2001 From: joel Date: Sat, 1 Nov 2025 21:51:57 -0700 Subject: [PATCH 1/2] get login from env every time --- watttime/api.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/watttime/api.py b/watttime/api.py index a772f197..4e250b4e 100644 --- a/watttime/api.py +++ b/watttime/api.py @@ -37,8 +37,12 @@ def __init__( worker_count (int): The number of worker threads to use for multithreading. Default is min(10, (os.cpu_count() or 1) * 2). """ - self.username = username or os.getenv("WATTTIME_USER") - self.password = password or os.getenv("WATTTIME_PASSWORD") + + if username: + os.environ["WATTTIME_USER"] = username + if password: + os.environ["WATTTIME_PASSWORD"] = password + self.token = None self.headers = None self.token_valid_until = None @@ -67,7 +71,7 @@ def _login(self): url = f"{self.url_base}/login" rsp = self.session.get( url, - auth=requests.auth.HTTPBasicAuth(self.username, self.password), + auth=requests.auth.HTTPBasicAuth(os.getenv("WATTTIME_USER"), os.getenv("WATTTIME_PASSWORD")), timeout=(10, 60), ) rsp.raise_for_status() @@ -150,8 +154,8 @@ def register(self, email: str, organization: Optional[str] = None) -> None: url = f"{self.url_base}/register" params = { - "username": self.username, - "password": self.password, + "username": os.getenv("WATTTIME_USER"), + "password": os.getenv("WATTTIME_PASSWORD"), "email": email, "org": organization, } @@ -159,7 +163,7 @@ def register(self, email: str, organization: Optional[str] = None) -> None: rsp = self.session.post(url, json=params, timeout=(10, 60)) rsp.raise_for_status() print( - f"Successfully registered {self.username}, please check {email} for a verification email" + f"Successfully registered {os.getenv("WATTTIME_USER")}, please check {email} for a verification email" ) @cache From e193b9d19821f1732531e542cb4dacdd9fd72de7 Mon Sep 17 00:00:00 2001 From: joel Date: Sat, 1 Nov 2025 21:59:37 -0700 Subject: [PATCH 2/2] fix nested quotes in register log --- watttime/api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/watttime/api.py b/watttime/api.py index 4e250b4e..8c78919b 100644 --- a/watttime/api.py +++ b/watttime/api.py @@ -163,7 +163,7 @@ def register(self, email: str, organization: Optional[str] = None) -> None: rsp = self.session.post(url, json=params, timeout=(10, 60)) rsp.raise_for_status() print( - f"Successfully registered {os.getenv("WATTTIME_USER")}, please check {email} for a verification email" + f"Successfully registered {os.getenv('WATTTIME_USER')}, please check {email} for a verification email" ) @cache