BuddyMath / create_mock_user.py
dotandru's picture
fix: Auto-provision dev-bypass-user with Admin roles on startup
898b10c
Raw
History Blame Contribute Delete
1.29 kB
import firebase_admin
from firebase_admin import credentials, firestore
import json
import os
# Manual init for script
FIREBASE_CREDENTIALS_PATH = "serviceAccountKey.json"
STORAGE_BUCKET = "bussymath.firebasestorage.app"
if not firebase_admin._apps:
with open(FIREBASE_CREDENTIALS_PATH, "r", encoding="utf-8") as f:
cred_dict = json.load(f)
cred = credentials.Certificate(cred_dict)
firebase_admin.initialize_app(cred, {
'storageBucket': STORAGE_BUCKET
})
db = firestore.client()
def create_mock_user():
uid = "dev-bypass-user"
user_ref = db.collection('users').document(uid)
today = "2026-03-12" # Static for bypass
month = "2026-03"
user_data = {
'uid': uid,
'name': 'Dev Bypass User',
'status': 'approved',
'tier': 'student_premium',
'used_today': 0,
'used_tokens_this_month': 0,
'monthly_token_budget': 9999999,
'last_usage_month': month,
'quota': {
'limit': 9999,
'used_today': 0,
'last_reset_date': today,
},
'isAdmin': True,
'role': 'admin'
}
user_ref.set(user_data, merge=True)
print(f"✅ Created mock user: {uid}")
if __name__ == "__main__":
create_mock_user()