添加系统统计信息实体类
This commit is contained in:
parent
fe61902e5e
commit
9fcc92a0fc
@ -0,0 +1,105 @@
|
||||
from datetime import datetime
|
||||
from fastapi import APIRouter, Depends, Form, Request
|
||||
from pydantic_validation_decorator import ValidateFields
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
from config.enums import BusinessType
|
||||
from config.get_db import get_db
|
||||
from module_admin.annotation.log_annotation import Log
|
||||
from module_admin.aspect.interface_auth import CheckUserInterfaceAuth
|
||||
from module_admin.entity.vo.user_vo import CurrentUserModel
|
||||
from module_admin.service.login_service import LoginService
|
||||
from module_admin.service.sys_statistics_service import Sys_statisticsService
|
||||
from module_admin.entity.vo.sys_statistics_vo import DeleteSys_statisticsModel, Sys_statisticsModel, Sys_statisticsPageQueryModel
|
||||
from utils.common_util import bytes2file_response
|
||||
from utils.log_util import logger
|
||||
from utils.page_util import PageResponseModel
|
||||
from utils.response_util import ResponseUtil
|
||||
|
||||
|
||||
sys_statisticsController = APIRouter(prefix='/system/sys_statistics', dependencies=[Depends(LoginService.get_current_user)])
|
||||
|
||||
|
||||
@sys_statisticsController.get(
|
||||
'/list', response_model=PageResponseModel, dependencies=[Depends(CheckUserInterfaceAuth('system:sys_statistics:list'))]
|
||||
)
|
||||
async def get_system_sys_statistics_list(
|
||||
request: Request,
|
||||
sys_statistics_page_query: Sys_statisticsPageQueryModel = Depends(Sys_statisticsPageQueryModel.as_query),
|
||||
query_db: AsyncSession = Depends(get_db),
|
||||
):
|
||||
# 获取分页数据
|
||||
sys_statistics_page_query_result = await Sys_statisticsService.get_sys_statistics_list_services(query_db, sys_statistics_page_query, is_page=True)
|
||||
logger.info('获取成功')
|
||||
|
||||
return ResponseUtil.success(model_content=sys_statistics_page_query_result)
|
||||
|
||||
|
||||
@sys_statisticsController.post('', dependencies=[Depends(CheckUserInterfaceAuth('system:sys_statistics:add'))])
|
||||
@ValidateFields(validate_model='add_sys_statistics')
|
||||
@Log(title='系统统计数据', business_type=BusinessType.INSERT)
|
||||
async def add_system_sys_statistics(
|
||||
request: Request,
|
||||
add_sys_statistics: Sys_statisticsModel,
|
||||
query_db: AsyncSession = Depends(get_db),
|
||||
current_user: CurrentUserModel = Depends(LoginService.get_current_user),
|
||||
):
|
||||
add_sys_statistics.create_time = datetime.now()
|
||||
add_sys_statistics.create_by = current_user.user.user_name
|
||||
add_sys_statistics.update_time = datetime.now()
|
||||
add_sys_statistics.update_by = current_user.user.user_name
|
||||
add_sys_statistics_result = await Sys_statisticsService.add_sys_statistics_services(query_db, add_sys_statistics)
|
||||
logger.info(add_sys_statistics_result.message)
|
||||
|
||||
return ResponseUtil.success(msg=add_sys_statistics_result.message)
|
||||
|
||||
|
||||
@sys_statisticsController.put('', dependencies=[Depends(CheckUserInterfaceAuth('system:sys_statistics:edit'))])
|
||||
@ValidateFields(validate_model='edit_sys_statistics')
|
||||
@Log(title='系统统计数据', business_type=BusinessType.UPDATE)
|
||||
async def edit_system_sys_statistics(
|
||||
request: Request,
|
||||
edit_sys_statistics: Sys_statisticsModel,
|
||||
query_db: AsyncSession = Depends(get_db),
|
||||
current_user: CurrentUserModel = Depends(LoginService.get_current_user),
|
||||
):
|
||||
edit_sys_statistics.update_by = current_user.user.user_name
|
||||
edit_sys_statistics.update_time = datetime.now()
|
||||
edit_sys_statistics_result = await Sys_statisticsService.edit_sys_statistics_services(query_db, edit_sys_statistics)
|
||||
logger.info(edit_sys_statistics_result.message)
|
||||
|
||||
return ResponseUtil.success(msg=edit_sys_statistics_result.message)
|
||||
|
||||
|
||||
# @sys_statisticsController.delete('/{statistic_ids}', dependencies=[Depends(CheckUserInterfaceAuth('system:sys_statistics:remove'))])
|
||||
# @Log(title='系统统计数据', business_type=BusinessType.DELETE)
|
||||
# async def delete_system_sys_statistics(request: Request, statistic_ids: str, query_db: AsyncSession = Depends(get_db)):
|
||||
# delete_sys_statistics = DeleteSys_statisticsModel(statisticIds=statistic_ids)
|
||||
# delete_sys_statistics_result = await Sys_statisticsService.delete_sys_statistics_services(query_db, delete_sys_statistics)
|
||||
# logger.info(delete_sys_statistics_result.message)
|
||||
#
|
||||
# return ResponseUtil.success(msg=delete_sys_statistics_result.message)
|
||||
#
|
||||
#
|
||||
# @sys_statisticsController.get(
|
||||
# '/{statistic_id}', response_model=Sys_statisticsModel, dependencies=[Depends(CheckUserInterfaceAuth('system:sys_statistics:query'))]
|
||||
# )
|
||||
# async def query_detail_system_sys_statistics(request: Request, statistic_id: int, query_db: AsyncSession = Depends(get_db)):
|
||||
# sys_statistics_detail_result = await Sys_statisticsService.sys_statistics_detail_services(query_db, statistic_id)
|
||||
# logger.info(f'获取statistic_id为{statistic_id}的信息成功')
|
||||
#
|
||||
# return ResponseUtil.success(data=sys_statistics_detail_result)
|
||||
#
|
||||
#
|
||||
# @sys_statisticsController.post('/export', dependencies=[Depends(CheckUserInterfaceAuth('system:sys_statistics:export'))])
|
||||
# @Log(title='系统统计数据', business_type=BusinessType.EXPORT)
|
||||
# async def export_system_sys_statistics_list(
|
||||
# request: Request,
|
||||
# sys_statistics_page_query: Sys_statisticsPageQueryModel = Form(),
|
||||
# query_db: AsyncSession = Depends(get_db),
|
||||
# ):
|
||||
# # 获取全量数据
|
||||
# sys_statistics_query_result = await Sys_statisticsService.get_sys_statistics_list_services(query_db, sys_statistics_page_query, is_page=False)
|
||||
# sys_statistics_export_result = await Sys_statisticsService.export_sys_statistics_list_services(sys_statistics_query_result)
|
||||
# logger.info('导出成功')
|
||||
#
|
||||
# return ResponseUtil.streaming(data=bytes2file_response(sys_statistics_export_result))
|
||||
117
ruoyi-fastapi-backend/module_admin/dao/sys_statistics_dao.py
Normal file
117
ruoyi-fastapi-backend/module_admin/dao/sys_statistics_dao.py
Normal file
@ -0,0 +1,117 @@
|
||||
from sqlalchemy import delete, select, update
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
from module_admin.entity.do.sys_statistics_do import SysStatistics
|
||||
from module_admin.entity.vo.sys_statistics_vo import Sys_statisticsModel, Sys_statisticsPageQueryModel
|
||||
from utils.page_util import PageUtil
|
||||
|
||||
|
||||
class Sys_statisticsDao:
|
||||
"""
|
||||
系统统计数据模块数据库操作层
|
||||
"""
|
||||
|
||||
@classmethod
|
||||
async def get_sys_statistics_detail_by_id(cls, db: AsyncSession, statistic_id: int):
|
||||
"""
|
||||
根据统计项目ID获取系统统计数据详细信息
|
||||
|
||||
:param db: orm对象
|
||||
:param statistic_id: 统计项目ID
|
||||
:return: 系统统计数据信息对象
|
||||
"""
|
||||
sys_statistics_info = (
|
||||
(
|
||||
await db.execute(
|
||||
select(SysStatistics)
|
||||
.where(
|
||||
SysStatistics.statistic_id == statistic_id
|
||||
)
|
||||
)
|
||||
)
|
||||
.scalars()
|
||||
.first()
|
||||
)
|
||||
|
||||
return sys_statistics_info
|
||||
|
||||
@classmethod
|
||||
async def get_sys_statistics_detail_by_info(cls, db: AsyncSession, sys_statistics: Sys_statisticsModel):
|
||||
"""
|
||||
根据系统统计数据参数获取系统统计数据信息
|
||||
|
||||
:param db: orm对象
|
||||
:param sys_statistics: 系统统计数据参数对象
|
||||
:return: 系统统计数据信息对象
|
||||
"""
|
||||
sys_statistics_info = (
|
||||
(
|
||||
await db.execute(
|
||||
select(SysStatistics).where(
|
||||
)
|
||||
)
|
||||
)
|
||||
.scalars()
|
||||
.first()
|
||||
)
|
||||
|
||||
return sys_statistics_info
|
||||
|
||||
@classmethod
|
||||
async def get_sys_statistics_list(cls, db: AsyncSession, query_object: Sys_statisticsPageQueryModel, is_page: bool = False):
|
||||
"""
|
||||
根据查询参数获取系统统计数据列表信息
|
||||
|
||||
:param db: orm对象
|
||||
:param query_object: 查询参数对象
|
||||
:param is_page: 是否开启分页
|
||||
:return: 系统统计数据列表信息对象
|
||||
"""
|
||||
query = (
|
||||
select(SysStatistics)
|
||||
.where(
|
||||
SysStatistics.stat_date == query_object.stat_date if query_object.stat_date else True,
|
||||
)
|
||||
.order_by(SysStatistics.statistic_id)
|
||||
.distinct()
|
||||
)
|
||||
sys_statistics_list = await PageUtil.paginate(db, query, query_object.page_num, query_object.page_size, is_page)
|
||||
|
||||
return sys_statistics_list
|
||||
|
||||
@classmethod
|
||||
async def add_sys_statistics_dao(cls, db: AsyncSession, sys_statistics: Sys_statisticsModel):
|
||||
"""
|
||||
新增系统统计数据数据库操作
|
||||
|
||||
:param db: orm对象
|
||||
:param sys_statistics: 系统统计数据对象
|
||||
:return:
|
||||
"""
|
||||
db_sys_statistics = SysStatistics(**sys_statistics.model_dump(exclude={}))
|
||||
db.add(db_sys_statistics)
|
||||
await db.flush()
|
||||
|
||||
return db_sys_statistics
|
||||
|
||||
@classmethod
|
||||
async def edit_sys_statistics_dao(cls, db: AsyncSession, sys_statistics: dict):
|
||||
"""
|
||||
编辑系统统计数据数据库操作
|
||||
|
||||
:param db: orm对象
|
||||
:param sys_statistics: 需要更新的系统统计数据字典
|
||||
:return:
|
||||
"""
|
||||
await db.execute(update(SysStatistics), [sys_statistics])
|
||||
|
||||
@classmethod
|
||||
async def delete_sys_statistics_dao(cls, db: AsyncSession, sys_statistics: Sys_statisticsModel):
|
||||
"""
|
||||
删除系统统计数据数据库操作
|
||||
|
||||
:param db: orm对象
|
||||
:param sys_statistics: 系统统计数据对象
|
||||
:return:
|
||||
"""
|
||||
await db.execute(delete(SysStatistics).where(SysStatistics.statistic_id.in_([sys_statistics.statistic_id])))
|
||||
|
||||
@ -0,0 +1,25 @@
|
||||
from sqlalchemy import Date, DateTime, String, BigInteger, Column, Integer
|
||||
from config.database import Base
|
||||
|
||||
|
||||
class SysStatistics(Base):
|
||||
"""
|
||||
系统统计数据表
|
||||
"""
|
||||
|
||||
__tablename__ = 'sys_statistics'
|
||||
|
||||
statistic_id = Column(BigInteger, primary_key=True, autoincrement=True, nullable=False, comment='统计项目ID')
|
||||
stat_date = Column(Date, nullable=False, comment='统计日期')
|
||||
llm_call = Column(BigInteger, nullable=True, comment='大模型调用量')
|
||||
access_control_count = Column(BigInteger, nullable=True, comment='门禁通行量')
|
||||
visitor_guide = Column(BigInteger, nullable=True, comment='访客引导次数')
|
||||
exhibition_explanation = Column(BigInteger, nullable=True, comment='展厅讲解次数')
|
||||
access_control_success_rate = Column(Integer, nullable=True, comment='门禁识别成功率(%)')
|
||||
create_time = Column(DateTime, nullable=True, comment='创建时间')
|
||||
create_by = Column(String(64), nullable=True, comment='创建者')
|
||||
update_time = Column(DateTime, nullable=False, comment='修改时间')
|
||||
update_by = Column(String(64), nullable=True, comment='修改者')
|
||||
|
||||
|
||||
|
||||
@ -0,0 +1,69 @@
|
||||
from datetime import datetime, date
|
||||
from pydantic import BaseModel, ConfigDict, Field
|
||||
from pydantic.alias_generators import to_camel
|
||||
from pydantic_validation_decorator import NotBlank
|
||||
from typing import Optional
|
||||
from module_admin.annotation.pydantic_annotation import as_query
|
||||
|
||||
|
||||
|
||||
|
||||
class Sys_statisticsModel(BaseModel):
|
||||
"""
|
||||
系统统计数据表对应pydantic模型
|
||||
"""
|
||||
model_config = ConfigDict(alias_generator=to_camel, from_attributes=True)
|
||||
|
||||
statistic_id: Optional[int] = Field(default=None, description='统计项目ID')
|
||||
stat_date: Optional[date] = Field(default=None, description='统计日期')
|
||||
llm_call: Optional[int] = Field(default=None, description='大模型调用量')
|
||||
access_control_count: Optional[int] = Field(default=None, description='门禁通行量')
|
||||
visitor_guide: Optional[int] = Field(default=None, description='访客引导次数')
|
||||
exhibition_explanation: Optional[int] = Field(default=None, description='展厅讲解次数')
|
||||
access_control_success_rate: Optional[int] = Field(default=None, description='门禁识别成功率(%)')
|
||||
create_time: Optional[datetime] = Field(default=None, description='创建时间')
|
||||
create_by: Optional[str] = Field(default=None, description='创建者')
|
||||
update_time: Optional[datetime] = Field(default=None, description='修改时间')
|
||||
update_by: Optional[str] = Field(default=None, description='修改者')
|
||||
|
||||
@NotBlank(field_name='stat_date', message='统计日期不能为空')
|
||||
def get_stat_date(self):
|
||||
return self.stat_date
|
||||
|
||||
@NotBlank(field_name='update_time', message='修改时间不能为空')
|
||||
def get_update_time(self):
|
||||
return self.update_time
|
||||
|
||||
|
||||
def validate_fields(self):
|
||||
self.get_stat_date()
|
||||
self.get_update_time()
|
||||
|
||||
|
||||
|
||||
|
||||
class Sys_statisticsQueryModel(Sys_statisticsModel):
|
||||
"""
|
||||
系统统计数据不分页查询模型
|
||||
"""
|
||||
pass
|
||||
|
||||
|
||||
@as_query
|
||||
class Sys_statisticsPageQueryModel(Sys_statisticsQueryModel):
|
||||
"""
|
||||
系统统计数据分页查询模型
|
||||
"""
|
||||
|
||||
page_num: int = Field(default=1, description='当前页码')
|
||||
page_size: int = Field(default=10, description='每页记录数')
|
||||
|
||||
|
||||
class DeleteSys_statisticsModel(BaseModel):
|
||||
"""
|
||||
删除系统统计数据模型
|
||||
"""
|
||||
|
||||
model_config = ConfigDict(alias_generator=to_camel)
|
||||
|
||||
statistic_ids: str = Field(description='需要删除的统计项目ID')
|
||||
@ -0,0 +1,136 @@
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
from typing import List
|
||||
from config.constant import CommonConstant
|
||||
from exceptions.exception import ServiceException
|
||||
from module_admin.entity.vo.common_vo import CrudResponseModel
|
||||
from module_admin.dao.sys_statistics_dao import Sys_statisticsDao
|
||||
from module_admin.entity.vo.sys_statistics_vo import DeleteSys_statisticsModel, Sys_statisticsModel, Sys_statisticsPageQueryModel
|
||||
from utils.common_util import CamelCaseUtil
|
||||
from utils.excel_util import ExcelUtil
|
||||
|
||||
|
||||
class Sys_statisticsService:
|
||||
"""
|
||||
系统统计数据模块服务层
|
||||
"""
|
||||
|
||||
@classmethod
|
||||
async def get_sys_statistics_list_services(
|
||||
cls, query_db: AsyncSession, query_object: Sys_statisticsPageQueryModel, is_page: bool = False
|
||||
):
|
||||
"""
|
||||
获取系统统计数据列表信息service
|
||||
|
||||
:param query_db: orm对象
|
||||
:param query_object: 查询参数对象
|
||||
:param is_page: 是否开启分页
|
||||
:return: 系统统计数据列表信息对象
|
||||
"""
|
||||
sys_statistics_list_result = await Sys_statisticsDao.get_sys_statistics_list(query_db, query_object, is_page)
|
||||
|
||||
return sys_statistics_list_result
|
||||
|
||||
|
||||
@classmethod
|
||||
async def add_sys_statistics_services(cls, query_db: AsyncSession, page_object: Sys_statisticsModel):
|
||||
"""
|
||||
新增系统统计数据信息service
|
||||
|
||||
:param query_db: orm对象
|
||||
:param page_object: 新增系统统计数据对象
|
||||
:return: 新增系统统计数据校验结果
|
||||
"""
|
||||
try:
|
||||
await Sys_statisticsDao.add_sys_statistics_dao(query_db, page_object)
|
||||
await query_db.commit()
|
||||
return CrudResponseModel(is_success=True, message='新增成功')
|
||||
except Exception as e:
|
||||
await query_db.rollback()
|
||||
raise e
|
||||
|
||||
@classmethod
|
||||
async def edit_sys_statistics_services(cls, query_db: AsyncSession, page_object: Sys_statisticsModel):
|
||||
"""
|
||||
编辑系统统计数据信息service
|
||||
|
||||
:param query_db: orm对象
|
||||
:param page_object: 编辑系统统计数据对象
|
||||
:return: 编辑系统统计数据校验结果
|
||||
"""
|
||||
edit_sys_statistics = page_object.model_dump(exclude_unset=True, exclude={'create_time', 'create_by', })
|
||||
sys_statistics_info = await cls.sys_statistics_detail_services(query_db, page_object.statistic_id)
|
||||
if sys_statistics_info.statistic_id:
|
||||
try:
|
||||
await Sys_statisticsDao.edit_sys_statistics_dao(query_db, edit_sys_statistics)
|
||||
await query_db.commit()
|
||||
return CrudResponseModel(is_success=True, message='更新成功')
|
||||
except Exception as e:
|
||||
await query_db.rollback()
|
||||
raise e
|
||||
else:
|
||||
raise ServiceException(message='系统统计数据不存在')
|
||||
|
||||
@classmethod
|
||||
async def delete_sys_statistics_services(cls, query_db: AsyncSession, page_object: DeleteSys_statisticsModel):
|
||||
"""
|
||||
删除系统统计数据信息service
|
||||
|
||||
:param query_db: orm对象
|
||||
:param page_object: 删除系统统计数据对象
|
||||
:return: 删除系统统计数据校验结果
|
||||
"""
|
||||
if page_object.statistic_ids:
|
||||
statistic_id_list = page_object.statistic_ids.split(',')
|
||||
try:
|
||||
for statistic_id in statistic_id_list:
|
||||
await Sys_statisticsDao.delete_sys_statistics_dao(query_db, Sys_statisticsModel(statisticId=statistic_id))
|
||||
await query_db.commit()
|
||||
return CrudResponseModel(is_success=True, message='删除成功')
|
||||
except Exception as e:
|
||||
await query_db.rollback()
|
||||
raise e
|
||||
else:
|
||||
raise ServiceException(message='传入统计项目ID为空')
|
||||
|
||||
@classmethod
|
||||
async def sys_statistics_detail_services(cls, query_db: AsyncSession, statistic_id: int):
|
||||
"""
|
||||
获取系统统计数据详细信息service
|
||||
|
||||
:param query_db: orm对象
|
||||
:param statistic_id: 统计项目ID
|
||||
:return: 统计项目ID对应的信息
|
||||
"""
|
||||
sys_statistics = await Sys_statisticsDao.get_sys_statistics_detail_by_id(query_db, statistic_id=statistic_id)
|
||||
if sys_statistics:
|
||||
result = Sys_statisticsModel(**CamelCaseUtil.transform_result(sys_statistics))
|
||||
else:
|
||||
result = Sys_statisticsModel(**dict())
|
||||
|
||||
return result
|
||||
|
||||
@staticmethod
|
||||
async def export_sys_statistics_list_services(sys_statistics_list: List):
|
||||
"""
|
||||
导出系统统计数据信息service
|
||||
|
||||
:param sys_statistics_list: 系统统计数据信息列表
|
||||
:return: 系统统计数据信息对应excel的二进制数据
|
||||
"""
|
||||
# 创建一个映射字典，将英文键映射到中文键
|
||||
mapping_dict = {
|
||||
'statisticId': '统计项目ID',
|
||||
'statDate': '统计日期',
|
||||
'llmCall': '大模型调用量',
|
||||
'accessControlCount': '门禁通行量',
|
||||
'visitorGuide': '访客引导次数',
|
||||
'exhibitionExplanation': '展厅讲解次数',
|
||||
'accessControlSuccessRate': '门禁识别成功率(%)',
|
||||
'createTime': '创建时间',
|
||||
'createBy': '创建者',
|
||||
'updateTime': '修改时间',
|
||||
'updateBy': '修改者',
|
||||
}
|
||||
binary_data = ExcelUtil.export_list2excel(sys_statistics_list, mapping_dict)
|
||||
|
||||
return binary_data
|
||||
Loading…
Reference in New Issue
Block a user