john
a400130c67
Add coach review APIs, pose calibration thresholds, Gitea CI, Kubernetes skeleton, H5 practice page, and mini program tab bar. Co-authored-by: Cursor <cursoragent@cursor.com>
107 lines
3.3 KiB
Python
107 lines
3.3 KiB
Python
"""Seed demo tenant, parent user and child for local development."""
|
|
|
|
from datetime import date, datetime, timezone
|
|
|
|
from sqlalchemy import select
|
|
|
|
from app.core.security import hash_phone
|
|
from app.db.models import Child, Tenant, TrainingPlan, User
|
|
from app.db.session import SessionLocal
|
|
from app.services.training import build_plan_detail, ensure_default_exercises
|
|
from app.schemas.models import TrainingPlanCreateRequest
|
|
|
|
|
|
def seed() -> None:
|
|
db = SessionLocal()
|
|
try:
|
|
tenant = db.scalar(select(Tenant).where(Tenant.name == "Demo Organization"))
|
|
if not tenant:
|
|
tenant = Tenant(name="Demo Organization", type="organization", status="active")
|
|
db.add(tenant)
|
|
db.flush()
|
|
|
|
phone = "18600000000"
|
|
phone_hash = hash_phone(phone)
|
|
user = db.scalar(select(User).where(User.phone_hash == phone_hash))
|
|
if not user:
|
|
user = User(
|
|
tenant_id=tenant.id,
|
|
phone=phone,
|
|
phone_hash=phone_hash,
|
|
role="parent",
|
|
status="active",
|
|
consent_signed=True,
|
|
)
|
|
db.add(user)
|
|
db.flush()
|
|
|
|
coach_hash = hash_phone("18600000001")
|
|
coach = db.scalar(select(User).where(User.phone_hash == coach_hash))
|
|
if not coach:
|
|
coach = User(
|
|
tenant_id=tenant.id,
|
|
phone="18600000001",
|
|
phone_hash=coach_hash,
|
|
role="coach",
|
|
status="active",
|
|
consent_signed=True,
|
|
)
|
|
db.add(coach)
|
|
db.flush()
|
|
|
|
child = db.scalar(
|
|
select(Child).where(Child.parent_user_id == user.id, Child.name == "小明")
|
|
)
|
|
if not child:
|
|
child = Child(
|
|
parent_user_id=user.id,
|
|
tenant_id=tenant.id,
|
|
name="小明",
|
|
gender="male",
|
|
birthday=date(2016, 3, 15),
|
|
height=142.5,
|
|
weight=36.0,
|
|
status="active",
|
|
)
|
|
db.add(child)
|
|
db.flush()
|
|
|
|
ensure_default_exercises(db)
|
|
|
|
plan = db.scalar(
|
|
select(TrainingPlan).where(
|
|
TrainingPlan.child_id == child.id, TrainingPlan.status == "active"
|
|
)
|
|
)
|
|
if not plan:
|
|
body = TrainingPlanCreateRequest.model_validate(
|
|
{"childId": child.id, "goal": "改善头前伸与高低肩", "cycleDays": 28}
|
|
)
|
|
detail = build_plan_detail(body)
|
|
detail["completedDays"] = 2
|
|
detail["currentDay"] = 3
|
|
plan = TrainingPlan(
|
|
child_id=child.id,
|
|
goal=body.goal,
|
|
cycle_days=28,
|
|
status="active",
|
|
plan_detail=detail,
|
|
started_at=datetime.now(timezone.utc),
|
|
)
|
|
db.add(plan)
|
|
|
|
db.commit()
|
|
child_row = db.scalar(
|
|
select(Child).where(Child.parent_user_id == user.id, Child.name == "小明")
|
|
)
|
|
print(
|
|
f"Seed OK · tenant={tenant.id} user={user.id} coach={coach.id if coach else '-'} "
|
|
f"child={child_row.id if child_row else '-'}"
|
|
)
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
seed()
|