freedakgmail
33b4c734aa
后端: - 新增认证(auth)、任务(tasks)、映射(mappings)、对账(reconciliation)、异常(exceptions)、导出(exports) API - 新增核心模块: database, security, permissions, tenant, exceptions, error_handlers - 新增数据模型: user, company, reconciliation_task, field_mapping, uploaded_file 等 - 新增服务层: ai_recognizer, file_parser, file_storage, mapping, reconciliation 等 - 添加数据库迁移脚本 前端: - 新增登录页面和仪表盘页面 - 新增任务列表、任务详情、字段映射页面 - 新增异常处理页面和规则设置页面 - 新增 API 代理路由 /api/[...path] - 新增 UI 组件库 (button, card, dialog, input, table 等) - 新增 auth 组件 (ProtectedRoute, PermissionGate) - 新增 layout 组件 (Header, Sidebar) - 新增 mapping 组件 (FieldMappingTable, AISuggestionPanel) - 新增 API 客户端和 hooks (useAsync, useToast, usePermission 等) - 新增状态管理 (auth-store, company-store, ui-store) - 集成 Tailwind CSS 和 shadcn/ui 组件库 其他: - 添加 Alembic 数据库迁移配置 - 添加初始化示例数据脚本 - 更新项目文档
113 lines
2.7 KiB
Python
113 lines
2.7 KiB
Python
from typing import Optional
|
|
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.models.company import Company, CompanyStatus
|
|
from app.schemas.company import CompanyCreate, CompanyUpdate
|
|
|
|
|
|
class CompanyService:
|
|
"""企业服务"""
|
|
|
|
@staticmethod
|
|
async def create_company(
|
|
db: AsyncSession,
|
|
company_data: CompanyCreate,
|
|
) -> Company:
|
|
"""
|
|
创建企业
|
|
|
|
Args:
|
|
db: 数据库会话
|
|
company_data: 企业数据
|
|
|
|
Returns:
|
|
创建的企业对象
|
|
"""
|
|
company = Company(**company_data.model_dump())
|
|
db.add(company)
|
|
await db.commit()
|
|
await db.refresh(company)
|
|
return company
|
|
|
|
@staticmethod
|
|
async def get_company(
|
|
db: AsyncSession,
|
|
company_id: int,
|
|
) -> Optional[Company]:
|
|
"""
|
|
获取企业信息
|
|
|
|
Args:
|
|
db: 数据库会话
|
|
company_id: 企业 ID
|
|
|
|
Returns:
|
|
企业对象，不存在则返回 None
|
|
"""
|
|
result = await db.execute(
|
|
select(Company).where(
|
|
Company.id == company_id,
|
|
Company.status != CompanyStatus.DELETED.value,
|
|
)
|
|
)
|
|
return result.scalar_one_or_none()
|
|
|
|
@staticmethod
|
|
async def update_company(
|
|
db: AsyncSession,
|
|
company_id: int,
|
|
company_data: CompanyUpdate,
|
|
) -> Optional[Company]:
|
|
"""
|
|
更新企业信息
|
|
|
|
Args:
|
|
db: 数据库会话
|
|
company_id: 企业 ID
|
|
company_data: 更新数据
|
|
|
|
Returns:
|
|
更新后的企业对象，不存在则返回 None
|
|
"""
|
|
company = await CompanyService.get_company(db, company_id)
|
|
if not company:
|
|
return None
|
|
|
|
update_data = company_data.model_dump(exclude_unset=True)
|
|
for field, value in update_data.items():
|
|
setattr(company, field, value)
|
|
|
|
await db.commit()
|
|
await db.refresh(company)
|
|
return company
|
|
|
|
@staticmethod
|
|
async def list_companies(
|
|
db: AsyncSession,
|
|
skip: int = 0,
|
|
limit: int = 100,
|
|
) -> list[Company]:
|
|
"""
|
|
获取企业列表
|
|
|
|
Args:
|
|
db: 数据库会话
|
|
skip: 跳过数量
|
|
limit: 返回数量
|
|
|
|
Returns:
|
|
企业列表
|
|
"""
|
|
result = await db.execute(
|
|
select(Company)
|
|
.where(Company.status != CompanyStatus.DELETED.value)
|
|
.offset(skip)
|
|
.limit(limit)
|
|
)
|
|
return list(result.scalars().all())
|
|
|
|
|
|
# 单例实例
|
|
company_service = CompanyService() |