diff --git a/ruoyi-fastapi-backend/module_admin/controller/sys_statistics_controller.py b/ruoyi-fastapi-backend/module_admin/controller/sys_statistics_controller.py new file mode 100644 index 0000000..1485df7 --- /dev/null +++ b/ruoyi-fastapi-backend/module_admin/controller/sys_statistics_controller.py @@ -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)) diff --git a/ruoyi-fastapi-backend/module_admin/dao/sys_statistics_dao.py b/ruoyi-fastapi-backend/module_admin/dao/sys_statistics_dao.py new file mode 100644 index 0000000..610ed44 --- /dev/null +++ b/ruoyi-fastapi-backend/module_admin/dao/sys_statistics_dao.py @@ -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]))) + diff --git a/ruoyi-fastapi-backend/module_admin/entity/do/sys_statistics_do.py b/ruoyi-fastapi-backend/module_admin/entity/do/sys_statistics_do.py new file mode 100644 index 0000000..38d0918 --- /dev/null +++ b/ruoyi-fastapi-backend/module_admin/entity/do/sys_statistics_do.py @@ -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='修改者') + + + diff --git a/ruoyi-fastapi-backend/module_admin/entity/vo/sys_statistics_vo.py b/ruoyi-fastapi-backend/module_admin/entity/vo/sys_statistics_vo.py new file mode 100644 index 0000000..92a263c --- /dev/null +++ b/ruoyi-fastapi-backend/module_admin/entity/vo/sys_statistics_vo.py @@ -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') diff --git a/ruoyi-fastapi-backend/module_admin/service/sys_statistics_service.py b/ruoyi-fastapi-backend/module_admin/service/sys_statistics_service.py new file mode 100644 index 0000000..f20ada1 --- /dev/null +++ b/ruoyi-fastapi-backend/module_admin/service/sys_statistics_service.py @@ -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