33
44
55class User :
6- def __init__ (self , name , key , slack_id = None ):
6+ def __init__ (self , name , key , email = None ):
77 self .name = name
88 self .key = key
9- self .slack_id = slack_id
9+ self .email = email
1010
1111 @classmethod
1212 def get_by_key (cls , key ):
@@ -34,14 +34,14 @@ def save(self):
3434 if existing_user :
3535 # Update existing user data
3636 cursor .execute (
37- "UPDATE users SET name = ?, slack_id = ? WHERE key = ?" ,
38- (self .name , self .slack_id , self .key ),
37+ "UPDATE users SET name = ?, key = ? WHERE email = ?" ,
38+ (self .name , self .key , self .email if self . email else None ),
3939 )
4040 else :
4141 # Create new user
4242 cursor .execute (
43- "INSERT INTO users (name, key, slack_id ) VALUES (?, ?, ?)" ,
44- (self .name , self .key , self .slack_id ),
43+ "INSERT INTO users (name, key, email ) VALUES (?, ?, ?)" ,
44+ (self .name , self .key , self .email if self . email else None ),
4545 )
4646 number_of_new_user = 1
4747 conn .commit ()
@@ -79,7 +79,7 @@ def get_permissions(cls, user_key=None):
7979 if user_key :
8080 cursor .execute (
8181 """
82- SELECT users.name, users.key,
82+ SELECT users.name, users.key, users.email,
8383 (SELECT operation_time
8484 FROM event_logs
8585 WHERE user_key = users.key
@@ -93,7 +93,7 @@ def get_permissions(cls, user_key=None):
9393 else :
9494 cursor .execute (
9595 """
96- SELECT users.name, users.key,
96+ SELECT users.name, users.key, users.email,
9797 (SELECT operation_time
9898 FROM event_logs
9999 WHERE user_key = users.key
@@ -106,7 +106,9 @@ def get_permissions(cls, user_key=None):
106106 for row in cursor .fetchall ():
107107 user_name = row [0 ]
108108 user_key = row [1 ]
109- latest_activity = row [2 ]
109+ user_email = row [2 ] if row [2 ] else None # Handle email being None
110+ latest_activity = row [3 ]
111+
110112
111113 # Get device permissions for the current user
112114 device_permissions = []
@@ -138,14 +140,16 @@ def get_permissions(cls, user_key=None):
138140 user_record = {
139141 "user_name" : user_name ,
140142 "user_key" : user_key ,
143+ "user_email" : user_email ,
141144 "permissions" : device_permissions ,
142145 "latest_activity" : latest_activity ,
143146 }
144147 user_data .append (user_record )
145148
146149 # Close the database connection
147150 conn .close ()
148-
151+
152+ print (f"User data fetched: { user_data } " ) # Debugging output
149153 return user_data
150154
151155 def has_permission_for_device (self , device_id ):
0 commit comments