Hermes CI Fix
3dddd36866
包含前后端完整代码： - 前端：Vue3+Vite+ElementPlus - 后端：FastAPI+SQLAlchemy - 模块：驾驶舱/KPI/战略地图/预警/预算/成本/预测/改善行动 - 当前版本：v1.0.0
123 lines
4.2 KiB
Python
123 lines
4.2 KiB
Python
"""数据库配置"""
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker, declarative_base
|
|
from sqlalchemy import inspect
|
|
import os
|
|
import logging
|
|
|
|
logger = logging.getLogger("cma")
|
|
|
|
DB_USER = os.getenv("CMA_DB_USER", "cma_user")
|
|
DB_PASS = os.getenv("CMA_DB_PASS", "cma_pass_2026")
|
|
DB_HOST = os.getenv("CMA_DB_HOST", "127.0.0.1")
|
|
DB_PORT = os.getenv("CMA_DB_PORT", "3306")
|
|
DB_NAME = os.getenv("CMA_DB_NAME", "cma")
|
|
|
|
DATABASE_URL = "mysql+pymysql://%(user)s:%(password)s@%(host)s:%(port)s/%(name)s?charset=utf8mb4" % {
|
|
"user": DB_USER,
|
|
"password": DB_PASS,
|
|
"host": DB_HOST,
|
|
"port": DB_PORT,
|
|
"name": DB_NAME,
|
|
}
|
|
|
|
_engine = None
|
|
_SessionLocal = None
|
|
Base = declarative_base()
|
|
|
|
|
|
def get_engine():
|
|
global _engine
|
|
if _engine is None:
|
|
_engine = create_engine(DATABASE_URL, echo=False, pool_size=5, max_overflow=10, pool_pre_ping=True)
|
|
return _engine
|
|
|
|
|
|
def get_session_local():
|
|
global _SessionLocal
|
|
if _SessionLocal is None:
|
|
_SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=get_engine())
|
|
return _SessionLocal
|
|
|
|
|
|
def get_db():
|
|
db = get_session_local()()
|
|
try:
|
|
yield db
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
def init_db():
|
|
import app.models
|
|
Base.metadata.create_all(bind=get_engine())
|
|
logger.info("CMA数据库已初始化")
|
|
|
|
# ── 初始化组织层级示例数据 ──
|
|
try:
|
|
inspector = inspect(get_engine())
|
|
if "org_nodes" in inspector.get_table_names():
|
|
Session = get_session_local()
|
|
session = Session()
|
|
try:
|
|
cnt = session.query(app.models.OrgNode).count()
|
|
if cnt == 0:
|
|
_seed_org_data(session)
|
|
finally:
|
|
session.close()
|
|
except Exception as e:
|
|
logger.warning(f"组织数据初始化跳过: {e}")
|
|
|
|
|
|
def _seed_org_data(db_session):
|
|
"""插入5层级组织示例数据"""
|
|
from app.models import OrgNode
|
|
|
|
# 1. 集团
|
|
g = OrgNode(id=1, parent_id=None, name="博海网络科技", code="BH", level=1, sort_order=1, enabled=1)
|
|
db_session.add(g)
|
|
db_session.flush()
|
|
|
|
# 2. 事业部
|
|
depts = [
|
|
OrgNode(parent_id=1, name="技术事业部", code="TECH", level=2, sort_order=1, enabled=1),
|
|
OrgNode(parent_id=1, name="销售事业部", code="SALES", level=2, sort_order=2, enabled=1),
|
|
OrgNode(parent_id=1, name="财务事业部", code="FIN", level=2, sort_order=3, enabled=1),
|
|
]
|
|
db_session.add_all(depts)
|
|
db_session.flush()
|
|
|
|
# 3. 区域/部门级
|
|
regions = [
|
|
OrgNode(parent_id=2, name="华南区域", code="SC", level=3, sort_order=1, enabled=1),
|
|
OrgNode(parent_id=2, name="华东区域", code="EC", level=3, sort_order=2, enabled=1),
|
|
OrgNode(parent_id=3, name="销售一部", code="S1", level=3, sort_order=1, enabled=1),
|
|
OrgNode(parent_id=3, name="销售二部", code="S2", level=3, sort_order=2, enabled=1),
|
|
]
|
|
db_session.add_all(regions)
|
|
db_session.flush()
|
|
|
|
# 4. 部门
|
|
departs = [
|
|
OrgNode(parent_id=5, name="研发部", code="RD", level=4, sort_order=1, enabled=1),
|
|
OrgNode(parent_id=5, name="实施部", code="IMP", level=4, sort_order=2, enabled=1),
|
|
OrgNode(parent_id=5, name="运维部", code="OPS", level=4, sort_order=3, enabled=1),
|
|
OrgNode(parent_id=6, name="前端研发", code="FE", level=4, sort_order=1, enabled=1),
|
|
OrgNode(parent_id=6, name="后端研发", code="BE", level=4, sort_order=2, enabled=1),
|
|
OrgNode(parent_id=8, name="KA客户部", code="KA", level=4, sort_order=1, enabled=1),
|
|
]
|
|
db_session.add_all(departs)
|
|
db_session.flush()
|
|
|
|
# 5. 班组
|
|
teams = [
|
|
OrgNode(parent_id=12, name="前端组", code="FE-TEAM", level=5, sort_order=1, enabled=1),
|
|
OrgNode(parent_id=12, name="后端组", code="BE-TEAM", level=5, sort_order=2, enabled=1),
|
|
OrgNode(parent_id=12, name="测试组", code="QA-TEAM", level=5, sort_order=3, enabled=1),
|
|
OrgNode(parent_id=13, name="实施一组", code="IMP1", level=5, sort_order=1, enabled=1),
|
|
OrgNode(parent_id=13, name="实施二组", code="IMP2", level=5, sort_order=2, enabled=1),
|
|
]
|
|
db_session.add_all(teams)
|
|
db_session.commit()
|
|
logger.info("组织层级示例数据已初始化")
|