202 lines
8.3 KiB
Python
202 lines
8.3 KiB
Python
"""种子数据 —— 用于测试和演示"""
|
|
from datetime import date, timedelta
|
|
from database import SessionLocal, engine, Base
|
|
from models import *
|
|
from auth import hash_password
|
|
|
|
Base.metadata.create_all(bind=engine)
|
|
db = SessionLocal()
|
|
|
|
|
|
def seed():
|
|
# 清空数据
|
|
for table in reversed(Base.metadata.sorted_tables):
|
|
db.execute(table.delete())
|
|
db.commit()
|
|
|
|
# ── 初始化内置角色 ──
|
|
roles = {}
|
|
for role_name, role_def in BUILTIN_ROLES.items():
|
|
role = Role(
|
|
name=role_name,
|
|
description=role_def["description"],
|
|
permissions=role_def["permissions"],
|
|
is_system=1,
|
|
)
|
|
db.add(role)
|
|
roles[role_name] = role
|
|
db.flush()
|
|
|
|
# ── 用户 ──
|
|
users = [
|
|
User(username="admin", password_hash=hash_password("admin123"),
|
|
name="老板", phase_group=PhaseGroup.PRODUCTION, role_id=roles["超级管理员"].id, monthly_salary=0),
|
|
User(username="zhangsan", password_hash=hash_password("123456"),
|
|
name="张三", phase_group=PhaseGroup.PRE, role_id=roles["组长"].id, monthly_salary=15000),
|
|
User(username="lisi", password_hash=hash_password("123456"),
|
|
name="李四", phase_group=PhaseGroup.PRODUCTION, role_id=roles["组长"].id, monthly_salary=18000),
|
|
User(username="wangwu", password_hash=hash_password("123456"),
|
|
name="王五", phase_group=PhaseGroup.PRODUCTION, role_id=roles["成员"].id, monthly_salary=12000),
|
|
User(username="zhaoliu", password_hash=hash_password("123456"),
|
|
name="赵六", phase_group=PhaseGroup.PRODUCTION, role_id=roles["成员"].id, monthly_salary=12000),
|
|
User(username="sunqi", password_hash=hash_password("123456"),
|
|
name="孙七", phase_group=PhaseGroup.POST, role_id=roles["成员"].id, monthly_salary=13000),
|
|
User(username="producer", password_hash=hash_password("123456"),
|
|
name="陈制片", phase_group=PhaseGroup.PRODUCTION, role_id=roles["主管"].id, monthly_salary=20000),
|
|
]
|
|
db.add_all(users)
|
|
db.flush()
|
|
|
|
admin, zhangsan, lisi, wangwu, zhaoliu, sunqi, producer = users
|
|
|
|
# ── 项目 ──
|
|
proj_a = Project(
|
|
name="星际漫游 第一季", project_type=ProjectType.CLIENT_FORMAL,
|
|
leader_id=lisi.id, current_phase=PhaseGroup.PRODUCTION,
|
|
episode_duration_minutes=5, episode_count=13,
|
|
estimated_completion_date=date.today() + timedelta(days=60),
|
|
contract_amount=100000,
|
|
)
|
|
proj_b = Project(
|
|
name="品牌方 TVC", project_type=ProjectType.CLIENT_FORMAL,
|
|
leader_id=lisi.id, current_phase=PhaseGroup.PRODUCTION,
|
|
episode_duration_minutes=1, episode_count=3,
|
|
estimated_completion_date=date.today() + timedelta(days=20),
|
|
contract_amount=50000,
|
|
)
|
|
proj_c = Project(
|
|
name="甲方风格测试", project_type=ProjectType.CLIENT_TEST,
|
|
leader_id=zhangsan.id, current_phase=PhaseGroup.PRE,
|
|
episode_duration_minutes=1, episode_count=1,
|
|
)
|
|
proj_d = Project(
|
|
name="AI 短剧原创", project_type=ProjectType.INTERNAL_ORIGINAL,
|
|
leader_id=lisi.id, current_phase=PhaseGroup.PRE,
|
|
episode_duration_minutes=8, episode_count=6,
|
|
estimated_completion_date=date.today() + timedelta(days=90),
|
|
)
|
|
db.add_all([proj_a, proj_b, proj_c, proj_d])
|
|
db.flush()
|
|
|
|
# ── 内容提交（模拟近两周的数据） ──
|
|
base_date = date.today() - timedelta(days=14)
|
|
submissions = []
|
|
|
|
for i in range(5):
|
|
d = base_date + timedelta(days=i)
|
|
submissions.append(Submission(
|
|
user_id=zhangsan.id, project_id=proj_a.id,
|
|
project_phase=PhaseGroup.PRE, work_type=WorkType.PLAN,
|
|
content_type=ContentType.DESIGN, total_seconds=0,
|
|
submit_to=SubmitTo.INTERNAL, description=f"角色设定第{i+1}版",
|
|
submit_date=d,
|
|
))
|
|
for i in range(3):
|
|
d = base_date + timedelta(days=i + 5)
|
|
submissions.append(Submission(
|
|
user_id=zhangsan.id, project_id=proj_d.id,
|
|
project_phase=PhaseGroup.PRE, work_type=WorkType.PLAN,
|
|
content_type=ContentType.DESIGN, total_seconds=0,
|
|
submit_to=SubmitTo.INTERNAL, description=f"剧本大纲第{i+1}稿",
|
|
submit_date=d,
|
|
))
|
|
|
|
for i in range(10):
|
|
d = base_date + timedelta(days=i)
|
|
secs = 45 + (i % 3) * 15
|
|
wt = WorkType.TEST if i < 2 else WorkType.PRODUCTION
|
|
submissions.append(Submission(
|
|
user_id=lisi.id, project_id=proj_a.id,
|
|
project_phase=PhaseGroup.PRODUCTION, work_type=wt,
|
|
content_type=ContentType.ANIMATION, total_seconds=secs,
|
|
duration_minutes=secs // 60, duration_seconds=secs % 60,
|
|
submit_to=SubmitTo.INTERNAL, description=f"第1集片段 - 场景{i+1}",
|
|
submit_date=d,
|
|
))
|
|
|
|
for i in range(8):
|
|
d = base_date + timedelta(days=i)
|
|
secs = 30 + (i % 4) * 20
|
|
submissions.append(Submission(
|
|
user_id=wangwu.id, project_id=proj_a.id,
|
|
project_phase=PhaseGroup.PRODUCTION, work_type=WorkType.PRODUCTION,
|
|
content_type=ContentType.ANIMATION, total_seconds=secs,
|
|
duration_minutes=secs // 60, duration_seconds=secs % 60,
|
|
submit_to=SubmitTo.LEADER, description=f"第2集动画片段{i+1}",
|
|
submit_date=d,
|
|
))
|
|
for i in range(4):
|
|
d = base_date + timedelta(days=i + 8)
|
|
secs = 20 + i * 10
|
|
submissions.append(Submission(
|
|
user_id=wangwu.id, project_id=proj_b.id,
|
|
project_phase=PhaseGroup.PRODUCTION, work_type=WorkType.PRODUCTION,
|
|
content_type=ContentType.ANIMATION, total_seconds=secs,
|
|
duration_minutes=secs // 60, duration_seconds=secs % 60,
|
|
submit_to=SubmitTo.LEADER, description=f"TVC 片段{i+1}",
|
|
submit_date=d,
|
|
))
|
|
|
|
for i in range(10):
|
|
d = base_date + timedelta(days=i)
|
|
secs = 50 + (i % 2) * 30
|
|
wt = WorkType.TEST if i < 1 else WorkType.PRODUCTION
|
|
submissions.append(Submission(
|
|
user_id=zhaoliu.id, project_id=proj_a.id,
|
|
project_phase=PhaseGroup.PRODUCTION, work_type=wt,
|
|
content_type=ContentType.ANIMATION, total_seconds=secs,
|
|
duration_minutes=secs // 60, duration_seconds=secs % 60,
|
|
submit_to=SubmitTo.LEADER, description=f"第3集场景动画{i+1}",
|
|
submit_date=d,
|
|
))
|
|
|
|
for i in range(3):
|
|
d = base_date + timedelta(days=i + 10)
|
|
submissions.append(Submission(
|
|
user_id=sunqi.id, project_id=proj_a.id,
|
|
project_phase=PhaseGroup.POST, work_type=WorkType.PRODUCTION,
|
|
content_type=ContentType.EDITING, total_seconds=0,
|
|
submit_to=SubmitTo.PRODUCER, description=f"第{i+1}集粗剪完成",
|
|
submit_date=d,
|
|
))
|
|
submissions.append(Submission(
|
|
user_id=sunqi.id, project_id=proj_a.id,
|
|
project_phase=PhaseGroup.POST, work_type=WorkType.PRODUCTION,
|
|
content_type=ContentType.ANIMATION, total_seconds=15,
|
|
duration_seconds=15,
|
|
submit_to=SubmitTo.PRODUCER, description="第1集补拍修改镜头",
|
|
submit_date=base_date + timedelta(days=12),
|
|
))
|
|
|
|
db.add_all(submissions)
|
|
|
|
db.add(AIToolCost(
|
|
tool_name="Midjourney", subscription_period=SubscriptionPeriod.MONTHLY,
|
|
amount=200, allocation_type=CostAllocationType.TEAM,
|
|
recorded_by=producer.id, record_date=date.today().replace(day=1),
|
|
))
|
|
db.add(AIToolCost(
|
|
tool_name="Runway", subscription_period=SubscriptionPeriod.MONTHLY,
|
|
amount=600, allocation_type=CostAllocationType.PROJECT,
|
|
project_id=proj_a.id,
|
|
recorded_by=producer.id, record_date=date.today().replace(day=1),
|
|
))
|
|
|
|
db.add(OutsourceCost(
|
|
project_id=proj_a.id, outsource_type=OutsourceType.ANIMATION,
|
|
episode_start=10, episode_end=13, amount=20000,
|
|
recorded_by=producer.id, record_date=date.today() - timedelta(days=5),
|
|
))
|
|
|
|
db.commit()
|
|
print("[OK] seed data generated")
|
|
print(f" - roles: {len(roles)}")
|
|
print(f" - users: {len(users)}")
|
|
print(f" - projects: 4")
|
|
print(f" - submissions: {len(submissions)}")
|
|
print(f" - default account: admin / admin123")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
seed()
|