240 lines
9.2 KiB
Python
240 lines
9.2 KiB
Python
"""添加测试数据"""
|
|
|
|
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()
|