group_manage/ruoyi_generator/service/column_service.py

# -*- coding: utf-8 -*-
# @Author  : YY
# @FileName: column_service.py

from typing import List, Tuple
from ruoyi_generator.domain.entity import GenTableColumn
from ruoyi_generator.mapper import gen_table_column_mapper


class GenTableColumnService:
    """代码生成表列服务类"""
    
    def select_gen_table_column_list_by_table_id(self, table_id: int) -> Tuple[List[GenTableColumn], int]:
        """
        根据表ID查询代码生成表列列表
        
        Args:
            table_id (int): 表ID
            
        Returns:
            Tuple[List[GenTableColumn], int]: 代码生成表列列表和总数
        """
        columns = gen_table_column_mapper.select_list_by_table_id(table_id)
        # 实现分页逻辑
        page_num = 1
        page_size = 10
        # 从flask的g对象或request中获取分页参数
        try:
            from flask import g, request
            if hasattr(g, 'criterian_meta') and g.criterian_meta.page:
                page_num = g.criterian_meta.page.page_num
                page_size = g.criterian_meta.page.page_size
            elif request and request.args.get('pageNum') and request.args.get('pageSize'):
                page_num = int(request.args.get('pageNum', 1))
                page_size = int(request.args.get('pageSize', 10))
        except Exception:
            pass
            
        start = (page_num - 1) * page_size
        end = start + page_size
        paginated_columns = columns[start:end]
        return paginated_columns, len(columns)

    def insert_gen_table_column(self, gen_table_column: GenTableColumn) -> bool:
        """
        插入代码生成表列
        
        Args:
            gen_table_column (GenTableColumn): 代码生成表列对象
            
        Returns:
            bool: 是否成功
        """
        try:
            gen_table_column.create_by = "admin"
            result = gen_table_column_mapper.insert(gen_table_column)
            return result > 0
        except Exception as e:
            print(f"插入代码生成表列失败: {e}")
            return False

    def update_gen_table_column(self, gen_table_column: GenTableColumn) -> bool:
        """
        更新代码生成表列
        
        Args:
            gen_table_column (GenTableColumn): 代码生成表列对象
            
        Returns:
            bool: 是否成功
        """
        try:
            gen_table_column.update_by = "admin"
            gen_table_column_mapper.update(gen_table_column)
            return True
        except Exception as e:
            print(f"更新代码生成表列失败: {e}")
            return False

    def delete_gen_table_column_by_id(self, column_id: int) -> bool:
        """
        根据ID删除代码生成表列
        
        Args:
            column_id (int): 列ID
            
        Returns:
            bool: 是否成功
        """
        try:
            gen_table_column_mapper.delete_by_id(column_id)
            return True
        except Exception as e:
            print(f"删除代码生成表列失败: {e}")
            return False