Christian Krakau-Louis
0de2eb5830
- Added authentication routes for login, registration, password reset, and account deletion in `auth.py`. - Implemented user profile management, including updating user information and changing passwords. - Created dashboard views to display user-specific information and recent activity in `dashboard.py`. - Developed leaderboard views to show team rankings and points in `leaderboard.py`. - Added QR code generation and redemption functionality in `qr.py` and `redeem.py`. - Implemented team management features, allowing users to create and join teams in `teams.py`. - Introduced session debugging utility to assist with session-related issues in `debug_session.py`. - Configured Docker Compose for MySQL database and FastAPI application with environment variables. - Updated requirements.txt to include necessary dependencies for the application.
135 lines
4.6 KiB
Python
135 lines
4.6 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Seed the database with initial testing data.
|
|
"""
|
|
import random
|
|
from datetime import datetime, timedelta
|
|
from sqlalchemy import inspect
|
|
from sqlalchemy.orm import Session
|
|
|
|
from .models import User, Team, TeamMembership, QRTicket
|
|
from .db import SessionLocal
|
|
|
|
def table_has_column(engine, table_name, column_name):
|
|
"""Check if a table has a specific column."""
|
|
inspector = inspect(engine)
|
|
if table_name not in inspector.get_table_names():
|
|
return False
|
|
columns = [col['name'] for col in inspector.get_columns(table_name)]
|
|
return column_name in columns
|
|
|
|
def seed_db():
|
|
"""Seed the database with test data."""
|
|
db = SessionLocal()
|
|
|
|
try:
|
|
# Only seed if tables are empty
|
|
if db.query(User).count() > 0:
|
|
print("Database already has data. Skipping seeding.")
|
|
return
|
|
|
|
# Create users
|
|
users = [
|
|
User(username="john_quizmaster", email="john@example.com", hashed_password="password123"),
|
|
User(username="sarah_johnson", email="sarah@example.com", hashed_password="password123"),
|
|
User(username="mike_peters", email="mike@example.com", hashed_password="password123"),
|
|
User(username="emma_wilson", email="emma@example.com", hashed_password="password123"),
|
|
User(username="robert_brown", email="robert@example.com", hashed_password="password123"),
|
|
]
|
|
db.add_all(users)
|
|
db.commit()
|
|
|
|
# Create teams
|
|
has_is_public = table_has_column(db.bind, 'teams', 'is_public')
|
|
has_created_at = table_has_column(db.bind, 'teams', 'created_at')
|
|
has_description = table_has_column(db.bind, 'teams', 'description')
|
|
|
|
teams = []
|
|
for i, name in enumerate(["Quiz Wizards", "Trivia Titans", "Beer Brainiacs", "Knowledge Knights"]):
|
|
team_attrs = {"name": name}
|
|
if has_is_public:
|
|
team_attrs["is_public"] = i % 2 == 1 # Alternate public/private
|
|
if has_description:
|
|
team_attrs["description"] = f"A team of quiz enthusiasts called {name}"
|
|
teams.append(Team(**team_attrs))
|
|
|
|
db.add_all(teams)
|
|
db.commit()
|
|
|
|
# Create team memberships
|
|
has_joined_at = table_has_column(db.bind, 'team_membership', 'joined_at')
|
|
|
|
memberships = []
|
|
membership_data = [
|
|
# Quiz Wizards
|
|
(1, 1, True, 160),
|
|
(2, 1, False, 155),
|
|
(3, 1, False, 130),
|
|
(4, 1, False, 90),
|
|
(5, 1, False, 45),
|
|
# Trivia Titans
|
|
(2, 2, True, 150),
|
|
(1, 2, False, 145),
|
|
# Beer Brainiacs
|
|
(3, 3, True, 120),
|
|
]
|
|
|
|
for user_id, team_id, is_admin, days_ago in membership_data:
|
|
membership_attrs = {
|
|
"user_id": user_id,
|
|
"team_id": team_id,
|
|
"is_admin": is_admin
|
|
}
|
|
if has_joined_at:
|
|
membership_attrs["joined_at"] = datetime.now() - timedelta(days=days_ago)
|
|
memberships.append(TeamMembership(**membership_attrs))
|
|
|
|
db.add_all(memberships)
|
|
db.commit()
|
|
|
|
# Create QR tickets
|
|
has_created_at = table_has_column(db.bind, 'qr_tickets', 'created_at')
|
|
has_redeemed_at = table_has_column(db.bind, 'qr_tickets', 'redeemed_at')
|
|
has_event_name = table_has_column(db.bind, 'qr_tickets', 'event_name')
|
|
|
|
event_names = [
|
|
"Music Trivia Night",
|
|
"History Night",
|
|
"Movie Trivia Night",
|
|
"Sports Quiz",
|
|
"General Knowledge"
|
|
]
|
|
|
|
# Create some basic tickets
|
|
tickets = []
|
|
for i in range(15):
|
|
points = random.choice([5, 10, 15, 20, 25])
|
|
team_id = random.randint(1, len(teams))
|
|
user_id = random.randint(1, len(users))
|
|
|
|
ticket_attrs = {
|
|
"code": f"TICKET{i:03d}",
|
|
"points": points,
|
|
"redeemed_by": user_id,
|
|
"redeemed_at_team": team_id,
|
|
"used": True
|
|
}
|
|
|
|
if has_event_name:
|
|
ticket_attrs["event_name"] = random.choice(event_names)
|
|
|
|
tickets.append(QRTicket(**ticket_attrs))
|
|
|
|
db.add_all(tickets)
|
|
db.commit()
|
|
|
|
print("Database seeded successfully!")
|
|
|
|
except Exception as e:
|
|
print(f"Error seeding database: {e}")
|
|
finally:
|
|
db.close()
|
|
|
|
if __name__ == "__main__":
|
|
seed_db()
|