from sqlalchemy import delete, select, update, and_
from sqlalchemy.ext.asyncio import AsyncSession
from module_admin.entity.do.metadata_config_do import BatchBusiLabelConfig, BatchDataopLabelConfig, \
    BatchDatatypeLabelConfig, BatchTableRelaLabelConfig

from utils.page_util import PageUtil


class BatchLabelConfigDAO:

    # region === BatchBusiLabelConfig ===

    @classmethod
    async def get_busi_detail_by_id(cls, db: AsyncSession, onum: int):
        result = await db.execute(select(BatchBusiLabelConfig).where(BatchBusiLabelConfig.onum == onum))
        return result.scalars().first()

    @classmethod
    async def get_busi_list(cls, db: AsyncSession, query_object, is_page: bool = False):
        query = select(BatchBusiLabelConfig).where(
            BatchBusiLabelConfig.mdl_name == query_object.mdl_name if query_object.mdl_name else True,
            BatchBusiLabelConfig.ssys_id == query_object.ssys_id if query_object.ssys_id else True,
        ).order_by(BatchBusiLabelConfig.upd_time.desc())

        return await PageUtil.paginate(db, query, query_object.page_num, query_object.page_size, is_page)

    @classmethod
    async def get_col_optype_list(cls, db: AsyncSession, query_object, is_page: bool = False):
        query = select(BatchDataopLabelConfig).where(
            BatchDataopLabelConfig.mdl_name.like(f'%{query_object.mdl_name}%') if query_object.mdl_name else True,
            BatchDataopLabelConfig.ssys_id == query_object.ssys_id if query_object.ssys_id else True,
        ).order_by(BatchDataopLabelConfig.upd_time.desc())

        return await PageUtil.paginate(db, query, query_object.page_num, query_object.page_size, is_page)

    @classmethod
    async def add_busi(cls, db: AsyncSession, obj):
        db_obj = BatchBusiLabelConfig(**obj.model_dump())
        db.add(db_obj)
        await db.flush()
        return db_obj

    @classmethod
    async def edit_busi(cls, db: AsyncSession, onum: int, update_data: dict):
        await db.execute(update(BatchBusiLabelConfig).where(BatchBusiLabelConfig.onum == onum).values(**update_data))

    @classmethod
    async def delete_busi(cls, db: AsyncSession, onum_list: list[int]):
        await db.execute(delete(BatchBusiLabelConfig).where(BatchBusiLabelConfig.onum.in_(onum_list)))

    # endregion

    # region === BatchDataopLabelConfig ===

    @classmethod
    async def get_dataop_detail_by_id(cls, db: AsyncSession, onum: int):
        result = await db.execute(select(BatchDataopLabelConfig).where(BatchDataopLabelConfig.onum == onum))
        return result.scalars().first()

    @classmethod
    async def get_dataop_list(cls, db: AsyncSession, query_object, is_page: bool = False):
        query = select(BatchDataopLabelConfig).where(
            BatchDataopLabelConfig.mdl_name == query_object.mdl_name if query_object.mdl_name else True,
            BatchDataopLabelConfig.ssys_id == query_object.ssys_id if query_object.ssys_id else True,
        )

        return await PageUtil.paginate(db, query, query_object.page_num, query_object.page_size, is_page)

    @classmethod
    async def add_dataop(cls, db: AsyncSession, obj):
        db.add(obj)
        await db.flush()
        return obj

    @classmethod
    async def edit_dataop(cls, db: AsyncSession, onum: int, update_data: dict):
        await db.execute(
            update(BatchDataopLabelConfig).where(BatchDataopLabelConfig.onum == onum).values(**update_data))

    @classmethod
    async def delete_dataop(cls, db: AsyncSession, onum_list: list[int]):
        await db.execute(delete(BatchDataopLabelConfig).where(BatchDataopLabelConfig.onum.in_(onum_list)))

    # endregion

    # region === BatchDatatypeLabelConfig ===

    @classmethod
    async def get_datatype_detail_by_id(cls, db: AsyncSession, onum: int):
        result = await db.execute(select(BatchDatatypeLabelConfig).where(BatchDatatypeLabelConfig.onum == onum))
        return result.scalars().first()

    @classmethod
    async def get_table_rela_detail_by_id(cls, db: AsyncSession, onum: int):
        result = await db.execute(select(BatchTableRelaLabelConfig).where(BatchTableRelaLabelConfig.onum == onum))
        return result.scalars().first()

    @classmethod
    async def get_datatype_list(cls, db: AsyncSession, query_object, is_page: bool = False):
        query = select(BatchDatatypeLabelConfig).where(
            BatchDatatypeLabelConfig.mdl_name == query_object.mdl_name if query_object.mdl_name else True,
            BatchDatatypeLabelConfig.ssys_id == query_object.ssys_id if query_object.ssys_id else True,
        ).order_by(BatchDatatypeLabelConfig.upd_time.desc())

        return await PageUtil.paginate(db, query, query_object.page_num, query_object.page_size, is_page)

    @classmethod
    async def get_table_rel_list(cls, db: AsyncSession, query_object, is_page: bool = False):
        query = select(BatchTableRelaLabelConfig).where(
         BatchTableRelaLabelConfig.a_mdl_name.like(f'%{query_object.a_mdl_name}%') if query_object.a_mdl_name else True,
         BatchTableRelaLabelConfig.b_mdl_name.like(f'%{query_object.b_mdl_name}%') if query_object.b_mdl_name else True,
         BatchTableRelaLabelConfig.a_ssys_id == query_object.a_ssys_id if query_object.a_ssys_id else True,
         BatchTableRelaLabelConfig.a_ssys_id == query_object.a_ssys_id if query_object.a_ssys_id else True,
        ).order_by(BatchTableRelaLabelConfig.upd_time.desc())
        return await PageUtil.paginate(db, query, query_object.page_num, query_object.page_size, is_page)

    @classmethod
    async def add_datatype(cls, db: AsyncSession, obj):
        db_obj = BatchDatatypeLabelConfig(**obj.model_dump())
        db.add(db_obj)
        await db.flush()
        return db_obj

    @classmethod
    async def add_table_rela(cls, db: AsyncSession, obj):
        db_obj = BatchTableRelaLabelConfig(**obj.model_dump())
        db.add(db_obj)
        await db.flush()
        return db_obj

    @classmethod
    async def edit_datatype(cls, db: AsyncSession, onum: int, update_data: dict):
        await db.execute(
            update(BatchDatatypeLabelConfig).where(BatchDatatypeLabelConfig.onum == onum).values(**update_data))

    @classmethod
    async def edit_table_rela(cls, db: AsyncSession, onum: int, update_data: dict):
        await db.execute(
            update(BatchTableRelaLabelConfig).where(BatchTableRelaLabelConfig.onum == onum).values(**update_data))

    @classmethod
    async def delete_datatype(cls, db: AsyncSession, onum_list: list[int]):
        await db.execute(delete(BatchDatatypeLabelConfig).where(BatchDatatypeLabelConfig.onum.in_(onum_list)))

    @classmethod
    async def delete_table_rela(cls, db: AsyncSession, onum_list: list[int]):
        await db.execute(delete(BatchTableRelaLabelConfig).where(BatchTableRelaLabelConfig.onum.in_(onum_list)))

    # endregion