50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
"""
|
||
博主账号初始化脚本。
|
||
|
||
用法：
|
||
1. 在 .env 中配置 BLOGGER_USERNAME / BLOGGER_EMAIL / BLOGGER_PASSWORD
|
||
2. 运行：python -m backend.seed
|
||
|
||
脚本会创建全局唯一博主账号（bcrypt 哈希存储密码）；已存在则跳过。
|
||
"""
|
||
|
||
import os
|
||
|
||
from .auth import hash_password
|
||
from .database import PROJECT_ROOT, SessionLocal, init_db
|
||
from .models import ROLE_BLOGGER, User
|
||
|
||
|
||
def seed_blogger() -> None:
|
||
"""创建唯一博主账号（已存在则跳过）。"""
|
||
username = os.getenv("BLOGGER_USERNAME", "").strip()
|
||
email = os.getenv("BLOGGER_EMAIL", "").strip()
|
||
password = os.getenv("BLOGGER_PASSWORD", "")
|
||
|
||
if not (username and email and password):
|
||
raise SystemExit("请在 .env 中配置 BLOGGER_USERNAME / BLOGGER_EMAIL / BLOGGER_PASSWORD")
|
||
if "@" not in email:
|
||
raise SystemExit("BLOGGER_EMAIL 格式不正确")
|
||
|
||
init_db()
|
||
db = SessionLocal()
|
||
try:
|
||
exists = db.query(User).filter((User.email == email) | (User.username == username)).first()
|
||
if exists is not None:
|
||
print(f"博主账号已存在（id={exists.id}, username={exists.username}），跳过创建")
|
||
return
|
||
blogger = User(
|
||
email=email,
|
||
username=username,
|
||
password_hash=hash_password(password),
|
||
role=ROLE_BLOGGER,
|
||
)
|
||
db.add(blogger)
|
||
db.commit()
|
||
print(f"博主账号创建成功（id={blogger.id}, username={blogger.username}）")
|
||
finally:
|
||
db.close()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
seed_blogger() |