Update main.py

main.py

@@ -43,6 +43,61 @@ except Exception as e:
     exit(1)
 
 
+### ======================================================================= ###
+###                  ONE-TIME ADMIN ACCOUNT CREATION SCRIPT                 ###
+###                                                                         ###
+###   This block will run automatically when the server starts.             ###
+###   After confirming the accounts are created, DELETE this entire block.  ###
+### ======================================================================= ###
+try:
+    logging.info("--- [RUN-ONCE SCRIPT] Checking/Creating Admin Accounts ---")
+    
+    admins_to_create = [
+        {'email': 'rairorr@gmail.com', 'password': 'sqx2025!', 'displayName': 'Rai'},
+        {'email': 'helperzhou@gmail.com', 'password': 'sqx2025!', 'displayName': 'Helper Zhou'}
+    ]
+
+    for admin_data in admins_to_create:
+        email = admin_data['email']
+        try:
+            # 1. Check if the user already exists in Firebase Auth
+            user = auth.get_user_by_email(email)
+            logging.info(f"Admin user '{email}' already exists. Ensuring admin status in Firestore.")
+            # 2. If they exist, just ensure their Firestore profile is set to admin
+            db.collection('users').document(user.uid).set({'isAdmin': True}, merge=True)
+            
+        except auth.UserNotFoundError:
+            # 3. If the user does not exist, create them
+            logging.info(f"Admin user '{email}' not found. Creating new account.")
+            new_user = auth.create_user(
+                email=admin_data['email'],
+                password=admin_data['password'],
+                display_name=admin_data['displayName']
+            )
+            
+            # 4. Create their profile in Firestore and set them as an admin
+            user_profile = {
+                'uid': new_user.uid,
+                'email': new_user.email,
+                'displayName': new_user.display_name,
+                'isAdmin': True, # <<< CRITICAL: Set as admin
+                'phone': None,
+                'phoneStatus': 'unsubmitted',
+                'createdAt': firestore.SERVER_TIMESTAMP
+            }
+            db.collection('users').document(new_user.uid).set(user_profile)
+            logging.info(f"Successfully created admin account for '{email}'.")
+
+    logging.info("--- [RUN-ONCE SCRIPT] Admin setup complete. You can now delete this script block. ---")
+
+except Exception as e:
+    logging.error(f"--- [RUN-ONCE SCRIPT] An error occurred during admin setup: {e} ---")
+### ======================================================================= ###
+###                          END OF ONE-TIME SCRIPT                         ###
+### ======================================================================= ###
+
+
+
 # -----------------------------------------------------------------------------
 # 2. AUTHORIZATION MIDDLEWARE (HELPER FUNCTIONS)
 # -----------------------------------------------------------------------------