hankin
815cbf9d8c
- 更新 .gitignore：全面覆盖环境变量、数据库、日志、缓存、上传文件 - 移除误跟踪的 server/venv/、crm_data.db、.env 文件 - 新增 server/.env.example 模板 - 新增合同管理、利润核算、AI教练等功能模块 - 新增 Playwright e2e 测试套件 - 前后端多项功能升级和 bug 修复
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
"""add xinyu lubricant company
|
|
|
|
Revision ID: b2c3d4e5f6a7
|
|
Revises: a1b2c3d4e5f6
|
|
Create Date: 2026-03-19
|
|
"""
|
|
from alembic import op
|
|
|
|
revision = "b2c3d4e5f6a7"
|
|
down_revision = "a1b2c3d4e5f6"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
XINYU_COMPANY_ID = "aaaaaaaa-bbbb-cccc-dddd-eeeeeeee0002"
|
|
|
|
|
|
def upgrade() -> None:
|
|
# 1. 插入第二个公司：新宇润滑油
|
|
op.execute(f"""
|
|
INSERT INTO sys_companies (id, name, code, is_active)
|
|
VALUES ('{XINYU_COMPANY_ID}', '新宇润滑油', 'XY-LUB', true)
|
|
ON CONFLICT (id) DO NOTHING
|
|
""")
|
|
|
|
# 2. 将所有现有用户关联到新宇润滑油（非默认）
|
|
op.execute(f"""
|
|
INSERT INTO sys_user_companies (id, user_id, company_id, is_default)
|
|
SELECT gen_random_uuid(), id, '{XINYU_COMPANY_ID}'::uuid, false
|
|
FROM sys_users
|
|
WHERE id NOT IN (
|
|
SELECT user_id FROM sys_user_companies
|
|
WHERE company_id = '{XINYU_COMPANY_ID}'::uuid
|
|
)
|
|
""")
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.execute(f"""
|
|
DELETE FROM sys_user_companies WHERE company_id = '{XINYU_COMPANY_ID}'::uuid
|
|
""")
|
|
op.execute(f"""
|
|
DELETE FROM sys_companies WHERE id = '{XINYU_COMPANY_ID}'::uuid
|
|
""")
|