Kushal commited on
Commit
4f772f1
·
1 Parent(s): 448c402

Fix: automate admin rights during signup process

Browse files
Files changed (1) hide show
  1. app/services/auth_service.py +18 -2
app/services/auth_service.py CHANGED
@@ -37,17 +37,25 @@ class AuthService:
37
  password_hash = hash_password(password)
38
 
39
  # Create user
 
 
 
 
40
  user = User(
41
  id=generate_id(),
42
  email=email,
43
  password_hash=password_hash,
44
- name=name
 
45
  )
46
 
47
  db.add(user)
48
  db.commit()
49
  db.refresh(user)
50
 
 
 
 
51
  return user
52
 
53
  @staticmethod
@@ -149,17 +157,25 @@ class AuthService:
149
  return user
150
 
151
  # Create new user
 
 
 
 
152
  user = User(
153
  id=generate_id(),
154
  email=email,
155
  google_id=google_id,
156
- name=name
 
157
  )
158
 
159
  db.add(user)
160
  db.commit()
161
  db.refresh(user)
162
 
 
 
 
163
  return user
164
 
165
  @staticmethod
 
37
  password_hash = hash_password(password)
38
 
39
  # Create user
40
+ import os
41
+ admin_email = os.environ.get('ADMIN_EMAIL')
42
+ is_admin_user = 1 if admin_email and email == admin_email else 0
43
+
44
  user = User(
45
  id=generate_id(),
46
  email=email,
47
  password_hash=password_hash,
48
+ name=name,
49
+ is_admin=is_admin_user
50
  )
51
 
52
  db.add(user)
53
  db.commit()
54
  db.refresh(user)
55
 
56
+ if is_admin_user:
57
+ print(f"✅ User {email} registered and granted ADMIN privileges!")
58
+
59
  return user
60
 
61
  @staticmethod
 
157
  return user
158
 
159
  # Create new user
160
+ import os
161
+ admin_email = os.environ.get('ADMIN_EMAIL')
162
+ is_admin_user = 1 if admin_email and email == admin_email else 0
163
+
164
  user = User(
165
  id=generate_id(),
166
  email=email,
167
  google_id=google_id,
168
+ name=name,
169
+ is_admin=is_admin_user
170
  )
171
 
172
  db.add(user)
173
  db.commit()
174
  db.refresh(user)
175
 
176
+ if is_admin_user:
177
+ print(f"✅ User {email} registered via Google and granted ADMIN privileges!")
178
+
179
  return user
180
 
181
  @staticmethod