feat: initialize aivideo project
This commit is contained in:
282
backend/app/core/bootstrap.py
Normal file
282
backend/app/core/bootstrap.py
Normal file
@@ -0,0 +1,282 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from datetime import datetime, timedelta
|
||||
from decimal import Decimal
|
||||
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
from app.common.db.session import SessionLocal, engine
|
||||
from app.common.security.password import hash_password
|
||||
from app.common.utils.id_gen import new_invite_code, new_public_id
|
||||
from app.models import Base
|
||||
from app.models.entities import (
|
||||
AdminUser,
|
||||
GrowthRewardRule,
|
||||
PaymentChannel,
|
||||
PricingRule,
|
||||
ProviderAccount,
|
||||
ProviderModel,
|
||||
RechargePlan,
|
||||
RedeemCode,
|
||||
SystemConfig,
|
||||
VideoModel,
|
||||
VideoModelSupplierBinding,
|
||||
)
|
||||
|
||||
|
||||
def init_database() -> None:
|
||||
Base.metadata.create_all(bind=engine)
|
||||
with SessionLocal() as db:
|
||||
seed_defaults(db)
|
||||
|
||||
|
||||
def seed_defaults(db: Session) -> None:
|
||||
from app.common.config.settings import get_settings
|
||||
|
||||
settings = get_settings()
|
||||
|
||||
admin = db.scalar(select(AdminUser).where(AdminUser.username == settings.admin_username))
|
||||
if not admin:
|
||||
admin = AdminUser(
|
||||
username=settings.admin_username,
|
||||
password_hash=hash_password(settings.admin_password),
|
||||
nickname=settings.admin_nickname,
|
||||
is_super_admin=True,
|
||||
status=1,
|
||||
)
|
||||
db.add(admin)
|
||||
|
||||
for rule_type, trigger, points in [
|
||||
("signup_reward", "on_register", 300),
|
||||
("invite_reward", "on_first_consume", 500),
|
||||
]:
|
||||
rule = db.scalar(select(GrowthRewardRule).where(GrowthRewardRule.rule_type == rule_type))
|
||||
if not rule:
|
||||
db.add(
|
||||
GrowthRewardRule(
|
||||
rule_type=rule_type,
|
||||
enabled=True,
|
||||
reward_points=points,
|
||||
trigger_condition=trigger,
|
||||
min_consume_points=settings.invite_reward_min_consume_points,
|
||||
remark=rule_type,
|
||||
)
|
||||
)
|
||||
|
||||
for channel_code, channel_name, provider_type, sort_order in [
|
||||
("alipay", "支付宝", "manual", 10),
|
||||
("wechat_pay", "微信支付", "manual", 20),
|
||||
]:
|
||||
channel = db.scalar(
|
||||
select(PaymentChannel).where(PaymentChannel.channel_code == channel_code)
|
||||
)
|
||||
if not channel:
|
||||
db.add(
|
||||
PaymentChannel(
|
||||
channel_code=channel_code,
|
||||
channel_name=channel_name,
|
||||
provider_type=provider_type,
|
||||
status=1,
|
||||
sort_order=sort_order,
|
||||
)
|
||||
)
|
||||
|
||||
if not db.scalar(select(RechargePlan.id)):
|
||||
db.add_all(
|
||||
[
|
||||
RechargePlan(
|
||||
name="体验包",
|
||||
pay_amount=Decimal("29.90"),
|
||||
point_ratio=100,
|
||||
give_points=2990,
|
||||
bonus_points=200,
|
||||
sort_order=10,
|
||||
status=1,
|
||||
),
|
||||
RechargePlan(
|
||||
name="标准包",
|
||||
pay_amount=Decimal("99.00"),
|
||||
point_ratio=100,
|
||||
give_points=9900,
|
||||
bonus_points=1200,
|
||||
sort_order=20,
|
||||
status=1,
|
||||
),
|
||||
RechargePlan(
|
||||
name="专业包",
|
||||
pay_amount=Decimal("299.00"),
|
||||
point_ratio=100,
|
||||
give_points=29900,
|
||||
bonus_points=4500,
|
||||
sort_order=30,
|
||||
status=1,
|
||||
),
|
||||
]
|
||||
)
|
||||
|
||||
if not db.scalar(select(ProviderAccount.id)):
|
||||
openai_account = ProviderAccount(
|
||||
provider_code="openai-mock",
|
||||
provider_name="OpenAI Mock",
|
||||
api_format="openai_official_video",
|
||||
base_url="mock://openai",
|
||||
api_key_encrypted="mock",
|
||||
timeout_seconds=60,
|
||||
max_retries=3,
|
||||
status=1,
|
||||
)
|
||||
seedance_account = ProviderAccount(
|
||||
provider_code="seedance-mock",
|
||||
provider_name="Seedance Mock",
|
||||
api_format="seedance_video_generation",
|
||||
base_url="mock://seedance",
|
||||
api_key_encrypted="mock",
|
||||
timeout_seconds=60,
|
||||
max_retries=3,
|
||||
status=1,
|
||||
)
|
||||
db.add_all([openai_account, seedance_account])
|
||||
db.flush()
|
||||
|
||||
openai_model = ProviderModel(
|
||||
provider_account_id=openai_account.id,
|
||||
model_code="sora-2",
|
||||
model_name="Sora 2",
|
||||
request_content_type="multipart/form-data",
|
||||
supports_text_to_video=True,
|
||||
supports_image_to_video=True,
|
||||
supports_generate_audio=True,
|
||||
supports_webhook=True,
|
||||
min_duration=4,
|
||||
max_duration=12,
|
||||
default_ratio="16:9",
|
||||
default_resolution="1280x720",
|
||||
status=1,
|
||||
)
|
||||
seedance_model = ProviderModel(
|
||||
provider_account_id=seedance_account.id,
|
||||
model_code="seedance",
|
||||
model_name="Seedance",
|
||||
request_content_type="application/json",
|
||||
supports_text_to_video=True,
|
||||
supports_image_to_video=True,
|
||||
supports_generate_audio=True,
|
||||
supports_webhook=False,
|
||||
min_duration=4,
|
||||
max_duration=12,
|
||||
default_ratio="16:9",
|
||||
default_resolution="1280x720",
|
||||
status=1,
|
||||
)
|
||||
db.add_all([openai_model, seedance_model])
|
||||
db.flush()
|
||||
|
||||
standard_model = VideoModel(
|
||||
model_key="standard-video",
|
||||
model_name="标准视频",
|
||||
frontend_title="标准视频",
|
||||
frontend_description="平衡质量与速度，适合大多数日常创作。",
|
||||
default_duration_seconds=8,
|
||||
default_ratio="16:9",
|
||||
default_resolution="1280x720",
|
||||
sort_order=10,
|
||||
status=1,
|
||||
)
|
||||
fast_model = VideoModel(
|
||||
model_key="fast-video",
|
||||
model_name="高速视频",
|
||||
frontend_title="高速视频",
|
||||
frontend_description="更快返回结果，适合灵感验证与批量尝试。",
|
||||
default_duration_seconds=6,
|
||||
default_ratio="16:9",
|
||||
default_resolution="1280x720",
|
||||
sort_order=20,
|
||||
status=1,
|
||||
)
|
||||
db.add_all([standard_model, fast_model])
|
||||
db.flush()
|
||||
|
||||
db.add_all(
|
||||
[
|
||||
VideoModelSupplierBinding(
|
||||
video_model_id=standard_model.id,
|
||||
provider_model_id=openai_model.id,
|
||||
routing_priority=10,
|
||||
is_primary=True,
|
||||
status=1,
|
||||
),
|
||||
VideoModelSupplierBinding(
|
||||
video_model_id=fast_model.id,
|
||||
provider_model_id=seedance_model.id,
|
||||
routing_priority=10,
|
||||
is_primary=True,
|
||||
status=1,
|
||||
),
|
||||
]
|
||||
)
|
||||
db.add_all(
|
||||
[
|
||||
PricingRule(
|
||||
rule_name="标准视频默认价格",
|
||||
video_model_id=standard_model.id,
|
||||
points_per_second=120,
|
||||
minimum_points=500,
|
||||
effective_at=datetime.utcnow() - timedelta(days=1),
|
||||
version_no=1,
|
||||
status=1,
|
||||
),
|
||||
PricingRule(
|
||||
rule_name="高速视频默认价格",
|
||||
video_model_id=fast_model.id,
|
||||
points_per_second=90,
|
||||
minimum_points=400,
|
||||
effective_at=datetime.utcnow() - timedelta(days=1),
|
||||
version_no=1,
|
||||
status=1,
|
||||
),
|
||||
]
|
||||
)
|
||||
|
||||
default_configs = {
|
||||
"site.title": ("AIVideo", "site"),
|
||||
"site.notice": ("欢迎体验 AIVideo 本地开发版。", "site"),
|
||||
"reward.signup.enabled": ("1", "reward"),
|
||||
"reward.signup.points": ("300", "reward"),
|
||||
"reward.invite.enabled": ("1", "reward"),
|
||||
"reward.invite.points": ("500", "reward"),
|
||||
"invite.code.enabled": ("1", "invite"),
|
||||
"task.default_poll_interval_seconds": ("5", "task"),
|
||||
}
|
||||
for key, (value, group_name) in default_configs.items():
|
||||
if not db.scalar(select(SystemConfig).where(SystemConfig.config_key == key)):
|
||||
db.add(
|
||||
SystemConfig(
|
||||
config_key=key,
|
||||
config_value=value,
|
||||
value_type="string",
|
||||
group_name=group_name,
|
||||
is_public=1 if key.startswith("site.") else 0,
|
||||
)
|
||||
)
|
||||
|
||||
if not db.scalar(select(RedeemCode.id)):
|
||||
db.add_all(
|
||||
[
|
||||
RedeemCode(
|
||||
batch_no="WELCOME",
|
||||
redeem_code="SPRING-2026-ABCD-1234",
|
||||
points=1000,
|
||||
status="unused",
|
||||
),
|
||||
RedeemCode(
|
||||
batch_no="WELCOME",
|
||||
redeem_code="SPRING-2026-EFGH-5678",
|
||||
points=1500,
|
||||
status="unused",
|
||||
),
|
||||
]
|
||||
)
|
||||
|
||||
db.commit()
|
||||
|
||||
Reference in New Issue
Block a user