Add timezone in User object

skit-ai · Nov 30, 2023 · 2a857ce · 2a857ce
1 parent 6c63a72
commit 2a857ce
Show file tree

Hide file tree

Showing 4 changed files with 531 additions and 295 deletions.
diff --git a/ccb/core.py b/ccb/core.py
@@ -20,7 +20,7 @@ def load_users_from_user_group(client: slack.WebClient, user_group: str) -> List
     users = []
     for i in tqdm(user_ids, desc=f"Collecting user info for group {user_group}"):
         u = client.users_info(user=i)["user"]
-        users.append(User(u["id"], u["real_name"]))
+        users.append(User(u["id"], u["real_name"], u["tz"]))
 
     return users
 
@@ -41,7 +41,7 @@ def load_users(client: slack.WebClient) -> List[User]:
             continue
         if member["id"] == "USLACKBOT":
             continue
-        users.append(User(member["id"], member["real_name"]))
+        users.append(User(member["id"], member["real_name"], u["tz"]))
     return users
 
 

diff --git a/ccb/types.py b/ccb/types.py
@@ -5,3 +5,4 @@
 class User:
     id: str
     name: str
+    tz: str
-Original file line number
+Diff line change
@@ Expand Up / @@ -5,3 +5,4 @@ @@
     class User:
         id: str
         name: str
+        tz: str