223 lines
8.1 KiB
Python
223 lines
8.1 KiB
Python
"""
|
||
字典数据一键初始化脚本
|
||
=============================================================
|
||
|
||
以本文件内置的 DICT_SEED 为数据源，把所有字典（dict + dict_item）幂等写入数据库
|
||
（已存在的字典按 code 跳过）。
|
||
|
||
用法：
|
||
python scripts/seed_dict.py
|
||
|
||
说明：
|
||
- 一次性初始化工具，数据入库后可删除本文件
|
||
- DICT_SEED 是本项目全部数据字典的唯一种子数据源（运行时展示一律走 dict_utils
|
||
从字典表读取，勿在业务代码里再硬编码枚举）
|
||
- 之后请在后台"字典管理"维护字典项（写入会触发 dict_utils 缓存失效，前端即时生效）
|
||
"""
|
||
|
||
import os
|
||
import sys
|
||
|
||
# 将项目根加入 sys.path（脚本以 `python scripts/xxx.py` 运行时 cwd 是 scripts/）
|
||
ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||
if ROOT not in sys.path:
|
||
sys.path.insert(0, ROOT)
|
||
|
||
# ============================================================
|
||
# 系统全部数据字典种子定义（幂等初始化：已存在的 code 跳过）
|
||
# code：字典编码（稳定业务键，如 gender、status、operation_type）
|
||
# name：字典中文名；note：字典备注（可选）
|
||
# items：字典项，value 为存储值（写入时统一转为字符串，dict_item.value 是 String 列），
|
||
# name 为显示文本，顺序即 sort 排序
|
||
# ============================================================
|
||
|
||
DICT_SEED = [
|
||
{
|
||
"code": "gender", "name": "性别", "note": "用户性别",
|
||
"items": [
|
||
{"value": "1", "name": "男"},
|
||
{"value": "2", "name": "女"},
|
||
{"value": "3", "name": "保密"},
|
||
],
|
||
},
|
||
{
|
||
"code": "status", "name": "状态", "note": "通用启用状态",
|
||
"items": [
|
||
{"value": "1", "name": "正常"},
|
||
{"value": "2", "name": "禁用"},
|
||
],
|
||
},
|
||
{
|
||
"code": "param_type", "name": "参数类型",
|
||
"items": [
|
||
{"value": "0", "name": "系统参数"},
|
||
{"value": "1", "name": "业务参数"},
|
||
],
|
||
},
|
||
{
|
||
"code": "link_type", "name": "友链类型",
|
||
"items": [
|
||
{"value": "1", "name": "友情链接"},
|
||
{"value": "2", "name": "合作伙伴"},
|
||
],
|
||
},
|
||
{
|
||
"code": "link_form", "name": "友链形式",
|
||
"items": [
|
||
{"value": "1", "name": "文字链接"},
|
||
{"value": "2", "name": "图片链接"},
|
||
],
|
||
},
|
||
{
|
||
"code": "config_item_type", "name": "配置项类型", "note": "配置项输入控件类型",
|
||
"items": [
|
||
{"value": "text", "name": "单行文本"},
|
||
{"value": "textarea", "name": "多行文本"},
|
||
{"value": "ueditor", "name": "富文本编辑器"},
|
||
{"value": "date", "name": "日期"},
|
||
{"value": "datetime", "name": "时间"},
|
||
{"value": "number", "name": "数字"},
|
||
{"value": "select", "name": "下拉框"},
|
||
{"value": "radio", "name": "单选框"},
|
||
{"value": "checkbox", "name": "复选框"},
|
||
{"value": "image", "name": "单张图片"},
|
||
{"value": "images", "name": "多张图片"},
|
||
{"value": "password", "name": "密码"},
|
||
{"value": "icon", "name": "字体图标"},
|
||
{"value": "file", "name": "单个文件"},
|
||
{"value": "files", "name": "多个文件"},
|
||
{"value": "hidden", "name": "隐藏"},
|
||
{"value": "readonly", "name": "只读文本"},
|
||
],
|
||
},
|
||
{
|
||
"code": "city_level", "name": "城市级别", "note": "行政区划级别",
|
||
"items": [
|
||
{"value": "1", "name": "省份"},
|
||
{"value": "2", "name": "城市"},
|
||
{"value": "3", "name": "县区"},
|
||
{"value": "4", "name": "街道"},
|
||
],
|
||
},
|
||
{
|
||
"code": "notice_type", "name": "通知类型",
|
||
"items": [
|
||
{"value": "1", "name": "通知"},
|
||
{"value": "2", "name": "公告"},
|
||
],
|
||
},
|
||
{
|
||
"code": "operation_source", "name": "操作日志来源",
|
||
"items": [
|
||
{"value": "0", "name": "后台系统"},
|
||
{"value": "1", "name": "手机站"},
|
||
{"value": "2", "name": "客户端"},
|
||
{"value": "3", "name": "其他"},
|
||
],
|
||
},
|
||
{
|
||
"code": "operation_type", "name": "操作日志类型",
|
||
"items": [
|
||
{"value": "1", "name": "登录"},
|
||
{"value": "2", "name": "退出"},
|
||
{"value": "3", "name": "新增"},
|
||
{"value": "4", "name": "修改"},
|
||
{"value": "5", "name": "删除"},
|
||
{"value": "6", "name": "查询"},
|
||
{"value": "7", "name": "导出"},
|
||
{"value": "8", "name": "导入"},
|
||
{"value": "9", "name": "提交"},
|
||
{"value": "10", "name": "撤销"},
|
||
{"value": "11", "name": "审核"},
|
||
{"value": "12", "name": "强退"},
|
||
{"value": "13", "name": "开账号"},
|
||
{"value": "14", "name": "清空"},
|
||
{"value": "15", "name": "授权"},
|
||
{"value": "16", "name": "缓存"},
|
||
{"value": "17", "name": "生成代码"},
|
||
{"value": "18", "name": "其他"},
|
||
],
|
||
},
|
||
{
|
||
"code": "login_type", "name": "登录日志类型",
|
||
"items": [
|
||
{"value": "1", "name": "登录"},
|
||
{"value": "2", "name": "退出"},
|
||
],
|
||
},
|
||
{
|
||
"code": "article_status", "name": "文章状态",
|
||
"items": [
|
||
{"value": "0", "name": "正常"},
|
||
{"value": "1", "name": "下架"},
|
||
],
|
||
},
|
||
{
|
||
"code": "job_log_status", "name": "任务日志状态", "note": "定时任务执行状态",
|
||
"items": [
|
||
{"value": "0", "name": "正常"},
|
||
{"value": "1", "name": "失败"},
|
||
],
|
||
},
|
||
{
|
||
"code": "job_status", "name": "任务状态", "note": "定时任务状态",
|
||
"items": [
|
||
{"value": "0", "name": "未发布"},
|
||
{"value": "1", "name": "运行中"},
|
||
{"value": "2", "name": "暂停"},
|
||
{"value": "3", "name": "删除"},
|
||
],
|
||
},
|
||
{
|
||
"code": "job_execute_policy", "name": "执行策略", "note": "定时任务执行策略",
|
||
"items": [
|
||
{"value": "1", "name": "立即执行"},
|
||
{"value": "2", "name": "执行一次"},
|
||
{"value": "3", "name": "放弃执行"},
|
||
],
|
||
},
|
||
]
|
||
|
||
|
||
def seed_dict(app) -> int:
|
||
"""将 DICT_SEED 幂等写入 app 绑定的数据库，返回新增字典数（可测试）。"""
|
||
from src.core.database import db
|
||
from src.modules.dictionary import dict_utils
|
||
from src.modules.dictionary.dict.repository import DictRepository
|
||
from src.modules.dictionary.dict_item.repository import DictItemRepository
|
||
|
||
created = 0
|
||
with app.app_context():
|
||
for sort, spec in enumerate(DICT_SEED, 1):
|
||
if DictRepository.first_by({"code": spec["code"]}):
|
||
print(f" [skip] {spec['code']}（已存在）")
|
||
continue
|
||
dict_obj = DictRepository.create(
|
||
name=spec["name"], code=spec["code"],
|
||
note=spec.get("note", ""), sort=sort, create_user="system",
|
||
)
|
||
for idx, item in enumerate(spec["items"], 1):
|
||
DictItemRepository.create(
|
||
dict_id=dict_obj.id,
|
||
name=item["name"], value=str(item["value"]),
|
||
sort=item.get("sort", idx), create_user="system",
|
||
)
|
||
db.session.commit()
|
||
created += 1
|
||
print(f" [ok] {spec['code']}（{len(spec['items'])} 项）")
|
||
# 失效取值缓存，保证后续读取走新数据
|
||
dict_utils.clear_cache()
|
||
return created
|
||
|
||
|
||
def main():
|
||
from src.app import create_app
|
||
|
||
print(f"开始初始化字典数据（共 {len(DICT_SEED)} 个字典）...")
|
||
created = seed_dict(create_app())
|
||
print(f"完成：新增 {created} 个字典，跳过 {len(DICT_SEED) - created} 个已存在")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|