sys_stuffinfo_conf_mapper.py 4.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160
  1. # -*- coding: utf-8 -*-
  2. # @Author : YY
  3. # @FileName: sys_stuff_info_conf_mapper.py
  4. # @Time : 2025-12-22 04:49:51
  5. from typing import List
  6. from datetime import datetime
  7. from flask import g
  8. from sqlalchemy import select, update, delete
  9. from ruoyi_admin.ext import db
  10. from ruoyi_common.domain.entity import SysStuffInfoConf
  11. from ruoyi_system.domain.po import SysStuffInfoConfPo
  12. class SysStuffInfoConfMapper:
  13. """员工配置表Mapper"""
  14. @staticmethod
  15. def select_stuffinfo_conf_list(sys: SysStuffInfoConf) -> List[SysStuffInfoConf]:
  16. """
  17. 查询员工配置表列表
  18. Args:
  19. sys (sys_stuff_info_conf): 员工配置表对象
  20. Returns:
  21. List[sys_stuff_info_conf]: 员工配置表列表
  22. """
  23. try:
  24. # 构建查询条件
  25. stmt = select(SysStuffInfoConfPo)
  26. if sys.conf_id is not None:
  27. stmt = stmt.where(SysStuffInfoConfPo.conf_id == sys.conf_id)
  28. if sys.module_index is not None:
  29. stmt = stmt.where(SysStuffInfoConfPo.module_index == sys.module_index)
  30. if sys.category is not None:
  31. stmt = stmt.where(SysStuffInfoConfPo.category == sys.category)
  32. if sys.attributes is not None:
  33. stmt = stmt.where(SysStuffInfoConfPo.attributes == sys.attributes)
  34. if "criterian_meta" in g and g.criterian_meta.page:
  35. g.criterian_meta.page.stmt = stmt
  36. result = db.session.execute(stmt).scalars().all()
  37. return [SysStuffInfoConf.model_validate(item) for item in result] if result else []
  38. except Exception as e:
  39. print(f"查询员工配置表列表出错: {e}")
  40. return []
  41. @staticmethod
  42. def select_sys_stuff_info_conf_by_id(conf_id: int) -> SysStuffInfoConf:
  43. """
  44. 根据ID查询员工配置表
  45. Args:
  46. conf_id (int): conf_id
  47. Returns:
  48. sys_stuff_info_conf: 员工配置表对象
  49. """
  50. try:
  51. result = db.session.get(SysStuffInfoConfPo, conf_id)
  52. return SysStuffInfoConf.model_validate(result) if result else None
  53. except Exception as e:
  54. print(f"根据ID查询员工配置表出错: {e}")
  55. return None
  56. @staticmethod
  57. def insert_sys_stuff_info_conf(sys: SysStuffInfoConf) -> int:
  58. """
  59. 新增员工配置表
  60. Args:
  61. sys (sys_stuff_info_conf): 员工配置表对象
  62. Returns:
  63. int: 插入的记录数
  64. """
  65. try:
  66. now = datetime.now()
  67. new_po = SysStuffInfoConfPo()
  68. new_po.conf_id = sys.conf_id
  69. new_po.module_index = sys.module_index
  70. new_po.category = sys.category
  71. new_po.attributes = sys.attributes
  72. db.session.add(new_po)
  73. db.session.commit()
  74. sys.conf_id = new_po.conf_id
  75. return 1
  76. except Exception as e:
  77. db.session.rollback()
  78. print(f"新增员工配置表出错: {e}")
  79. return 0
  80. @staticmethod
  81. def update_sys_stuff_info_conf(sys: SysStuffInfoConf) -> int:
  82. """
  83. 修改员工配置表
  84. Args:
  85. sys (sys_stuff_info_conf): 员工配置表对象
  86. Returns:
  87. int: 更新的记录数
  88. """
  89. try:
  90. existing = db.session.get(SysStuffInfoConfPo, sys.conf_id)
  91. if not existing:
  92. return 0
  93. now = datetime.now()
  94. # 主键不参与更新
  95. existing.module_index = sys.module_index
  96. existing.category = sys.category
  97. existing.attributes = sys.attributes
  98. db.session.commit()
  99. return 1
  100. except Exception as e:
  101. db.session.rollback()
  102. print(f"修改员工配置表出错: {e}")
  103. return 0
  104. @staticmethod
  105. def delete_sys_stuff_info_conf_by_ids(ids: List[int]) -> int:
  106. """
  107. 批量删除员工配置表
  108. Args:
  109. ids (List[int]): ID列表
  110. Returns:
  111. int: 删除的记录数
  112. """
  113. try:
  114. stmt = delete(SysStuffInfoConfPo).where(SysStuffInfoConfPo.conf_id.in_(ids))
  115. result = db.session.execute(stmt)
  116. db.session.commit()
  117. return result.rowcount
  118. except Exception as e:
  119. db.session.rollback()
  120. print(f"批量删除员工配置表出错: {e}")
  121. return 0