chore: initialize tuhui repository
This commit is contained in:
Codex
239
backend/seed_data.py
Normal file
239
backend/seed_data.py
Normal file
@@ -0,0 +1,239 @@
|
||||
"""添加测试数据"""
|
||||
|
||||
from sqlalchemy.orm import Session
|
||||
from app.core.database import SessionLocal
|
||||
from app.core.security import get_password_hash
|
||||
from app.models.user import User
|
||||
from app.models.work import Work
|
||||
|
||||
def seed_data():
|
||||
"""添加测试数据"""
|
||||
db = SessionLocal()
|
||||
|
||||
try:
|
||||
# 创建测试用户（使用手机号）
|
||||
test_user = User(
|
||||
phone="13800138000",
|
||||
password_hash=get_password_hash("123456"),
|
||||
nickname="测试用户",
|
||||
balance=100.0 # 给测试用户100元余额
|
||||
)
|
||||
db.add(test_user)
|
||||
|
||||
# 创建测试作品
|
||||
test_works = [
|
||||
Work(
|
||||
title="活动海报设计",
|
||||
category="活动",
|
||||
designer="呵呵",
|
||||
level=3,
|
||||
level_text="设计师",
|
||||
price=9.9,
|
||||
thumbnail_image="/uploads/thumbnail/1.jpg",
|
||||
watermarked_image="/uploads/watermarked/1.jpg",
|
||||
original_image="/uploads/original/1.jpg",
|
||||
description="高品质活动海报设计，适合各类活动宣传",
|
||||
tags='["海报", "活动", "设计"]'
|
||||
),
|
||||
Work(
|
||||
title="马年招聘海报",
|
||||
category="活动",
|
||||
designer="六十六号屯",
|
||||
level=4,
|
||||
level_text="资深设计师",
|
||||
price=15.9,
|
||||
thumbnail_image="/uploads/thumbnail/2.jpg",
|
||||
watermarked_image="/uploads/watermarked/2.jpg",
|
||||
original_image="/uploads/original/2.jpg",
|
||||
description="马年主题招聘海报，创意十足",
|
||||
tags='["招聘", "马年", "海报"]'
|
||||
),
|
||||
Work(
|
||||
title="新中式主视觉",
|
||||
category="中式",
|
||||
designer="小鱼是设计",
|
||||
level=4,
|
||||
level_text="资深设计师",
|
||||
price=19.9,
|
||||
thumbnail_image="/uploads/thumbnail/3.jpg",
|
||||
watermarked_image="/uploads/watermarked/3.jpg",
|
||||
original_image="/uploads/original/3.jpg",
|
||||
description="新中式风格主视觉设计，高端大气",
|
||||
tags='["中式", "主视觉", "设计"]'
|
||||
),
|
||||
Work(
|
||||
title="品牌VI设计",
|
||||
category="直播",
|
||||
designer="设计狮",
|
||||
level=5,
|
||||
level_text="首席设计师",
|
||||
price=29.9,
|
||||
thumbnail_image="/uploads/thumbnail/4.jpg",
|
||||
watermarked_image="/uploads/watermarked/4.jpg",
|
||||
original_image="/uploads/original/4.jpg",
|
||||
description="完整品牌VI视觉识别系统设计",
|
||||
tags='["品牌", "VI", "Logo"]'
|
||||
),
|
||||
Work(
|
||||
title="电商促销Banner",
|
||||
category="活动",
|
||||
designer="小美设计",
|
||||
level=3,
|
||||
level_text="设计师",
|
||||
price=12.9,
|
||||
thumbnail_image="/uploads/thumbnail/5.jpg",
|
||||
watermarked_image="/uploads/watermarked/5.jpg",
|
||||
original_image="/uploads/original/5.jpg",
|
||||
description="电商平台促销活动Banner设计",
|
||||
tags='["电商", "Banner", "促销"]'
|
||||
),
|
||||
Work(
|
||||
title="春节主题海报",
|
||||
category="周年庆",
|
||||
designer="创意工作室",
|
||||
level=4,
|
||||
level_text="资深设计师",
|
||||
price=18.9,
|
||||
thumbnail_image="/uploads/thumbnail/6.jpg",
|
||||
watermarked_image="/uploads/watermarked/6.jpg",
|
||||
original_image="/uploads/original/6.jpg",
|
||||
description="春节节日氛围海报设计",
|
||||
tags='["春节", "节日", "海报"]'
|
||||
),
|
||||
Work(
|
||||
title="移动端UI界面",
|
||||
category="直播",
|
||||
designer="UI设计师",
|
||||
level=4,
|
||||
level_text="资深设计师",
|
||||
price=25.9,
|
||||
thumbnail_image="/uploads/thumbnail/7.jpg",
|
||||
watermarked_image="/uploads/watermarked/7.jpg",
|
||||
original_image="/uploads/original/7.jpg",
|
||||
description="简洁大气的移动端应用界面设计",
|
||||
tags='["UI", "移动端", "界面"]'
|
||||
),
|
||||
Work(
|
||||
title="扁平风格插画",
|
||||
category="活动",
|
||||
designer="插画师小陈",
|
||||
level=5,
|
||||
level_text="首席设计师",
|
||||
price=22.9,
|
||||
thumbnail_image="/uploads/thumbnail/8.jpg",
|
||||
watermarked_image="/uploads/watermarked/8.jpg",
|
||||
original_image="/uploads/original/8.jpg",
|
||||
description="扁平风格商业插画设计",
|
||||
tags='["插画", "扁平", "商业"]'
|
||||
),
|
||||
Work(
|
||||
title="风光摄影作品",
|
||||
category="中式",
|
||||
designer="摄影师老王",
|
||||
level=6,
|
||||
level_text="大师",
|
||||
price=39.9,
|
||||
thumbnail_image="/uploads/thumbnail/9.jpg",
|
||||
watermarked_image="/uploads/watermarked/9.jpg",
|
||||
original_image="/uploads/original/9.jpg",
|
||||
description="自然风光摄影精品",
|
||||
tags='["摄影", "风光", "自然"]'
|
||||
),
|
||||
Work(
|
||||
title="简约Logo设计",
|
||||
category="直播",
|
||||
designer="品牌设计师",
|
||||
level=5,
|
||||
level_text="首席设计师",
|
||||
price=35.9,
|
||||
thumbnail_image="/uploads/thumbnail/10.jpg",
|
||||
watermarked_image="/uploads/watermarked/10.jpg",
|
||||
original_image="/uploads/original/10.jpg",
|
||||
description="简约现代风格Logo设计",
|
||||
tags='["Logo", "简约", "品牌"]'
|
||||
),
|
||||
Work(
|
||||
title="产品包装设计",
|
||||
category="活动",
|
||||
designer="包装设计工作室",
|
||||
level=4,
|
||||
level_text="资深设计师",
|
||||
price=28.9,
|
||||
thumbnail_image="/uploads/thumbnail/11.jpg",
|
||||
watermarked_image="/uploads/watermarked/11.jpg",
|
||||
original_image="/uploads/original/11.jpg",
|
||||
description="创意产品包装盒设计",
|
||||
tags='["包装", "产品", "创意"]'
|
||||
),
|
||||
Work(
|
||||
title="商务名片设计",
|
||||
category="直播",
|
||||
designer="名片专家",
|
||||
level=3,
|
||||
level_text="设计师",
|
||||
price=8.9,
|
||||
thumbnail_image="/uploads/thumbnail/12.jpg",
|
||||
watermarked_image="/uploads/watermarked/12.jpg",
|
||||
original_image="/uploads/original/12.jpg",
|
||||
description="高端商务名片设计",
|
||||
tags='["名片", "商务", "印刷"]'
|
||||
),
|
||||
Work(
|
||||
title="宣传单页设计",
|
||||
category="活动",
|
||||
designer="平面设计师",
|
||||
level=3,
|
||||
level_text="设计师",
|
||||
price=11.9,
|
||||
thumbnail_image="/uploads/thumbnail/13.jpg",
|
||||
watermarked_image="/uploads/watermarked/13.jpg",
|
||||
original_image="/uploads/original/13.jpg",
|
||||
description="活动宣传单页设计",
|
||||
tags='["宣传单", "活动", "印刷"]'
|
||||
),
|
||||
Work(
|
||||
title="企业画册设计",
|
||||
category="中式",
|
||||
designer="画册设计团队",
|
||||
level=5,
|
||||
level_text="首席设计师",
|
||||
price=45.9,
|
||||
thumbnail_image="/uploads/thumbnail/14.jpg",
|
||||
watermarked_image="/uploads/watermarked/14.jpg",
|
||||
original_image="/uploads/original/14.jpg",
|
||||
description="企业宣传画册整体设计",
|
||||
tags='["画册", "企业", "宣传"]'
|
||||
),
|
||||
Work(
|
||||
title="展板海报设计",
|
||||
category="周年庆",
|
||||
designer="展览设计师",
|
||||
level=4,
|
||||
level_text="资深设计师",
|
||||
price=24.9,
|
||||
thumbnail_image="/uploads/thumbnail/15.jpg",
|
||||
watermarked_image="/uploads/watermarked/15.jpg",
|
||||
original_image="/uploads/original/15.jpg",
|
||||
description="展览展示展板设计",
|
||||
tags='["展板", "展览", "海报"]'
|
||||
),
|
||||
]
|
||||
|
||||
for work in test_works:
|
||||
db.add(work)
|
||||
|
||||
db.commit()
|
||||
print("✅ 测试数据添加成功！")
|
||||
print("\n测试账号：")
|
||||
print("手机号: 13800138000")
|
||||
print("密码: 123456")
|
||||
print("余额: 100.0 元")
|
||||
|
||||
except Exception as e:
|
||||
print(f"❌ 错误: {e}")
|
||||
db.rollback()
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
if __name__ == "__main__":
|
||||
seed_data()
|
||||
Reference in New Issue
Block a user