db_query.py

# -*- coding:utf-8 -*-
# 作者：齐振鋆
# 日期：2023/3/14
# 描述:
import logging

import allure

from unit.public.DataDic import DataDic

log = logging.getLogger(__name__)
dd = DataDic()


def get_results(index, col):
    def query(func):
        def wrapper(self, *args, **kwargs):
            db = dd.get_value('db')
            sql = func(self, *args, **kwargs)
            log.info(sql)
            result = db.exec(sql)
            try:
                return result if index is None else result[index] if col is None else result[index][col]
            except IndexError:
                return "IndexError"

        return wrapper

    return query


class DBQuery(object):
    @get_results(0, 0)
    def get_stock_cargo(self, warehouse):
        """仓库是否启用货位管理"""
        with allure.step('DB:查询仓库是否启用货位管理'):
            return """select 
                    SPACE_MANAGEMENT 
                from 
                    BO_EU_STOCK_CARGO_USING 
                where 
                    BUSINESS_UNITS='{}';""".format(warehouse)

    @get_results(0, 0)
    def get_pro_num(self, warehouse, good):
        """获取生产批号"""
        with allure.step('DB:获取生产批号'):
            return """select
                    F_PRO_NUM
                from
                    BO_EU_STOCK_CURRENT
                where
                    F_ENTREPOT_CODE = '{warehouse}'
                    and F_COMMODITY_CODE = '{good}'
                    and F_CARGO_CODE in (
                    select
                        F_CARGO_CODE
                    from
                        BO_EU_STOCK_MGMT_CARGO_SPACE
                    where
                        F_ENTREPOT_CODE = '{warehouse}'
                        and F_COMMODITY_CODE = '{good}');""".format(**locals())

    @get_results(0, 0)
    def get_serial_flag(self, warehouse, good):
        """商品序列号标识"""
        with allure.step('DB:查询商品序列号标识'):
            return """select 
                    F_IS_SERIAL_NO 
                from