nebula/pages/body_page.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
'''
@File    :   body_page.py
@Time    :   2024/11/12 08:14:19
@Author  :   dulip3ng 
@Version :   1.0
@Desc    :   None
'''

from playwright.sync_api import Page, Error
from pages.base_page import BasePage
from common.field_type import FieldType

class BodyPage(BasePage):

    def __init__(self, page:Page, locator):
        super().__init__(page)
        self.locator = locator
    
    def change_sheet(self, sheet_name:str, order:int=1):
        """
        切换单据体页签
        :param sheet_name: 页面名称
        :param order: 可选参数，存在同名页签时指明第几个，从1开始
        :return: None
        """
        self.page.locator(self.locator.SHEET_ARGS_LOC % (sheet_name, order)).click()

    def click_button(self, button_name:str, sub_button_name:str=None):
        """
        点击单据体按钮或子按钮
        如：新增行

        **用法：**
        BodyPage.click_button("新增行")

        BodyPage.click_button("新增行", "复制行")

        :param button_name: 按钮名称
        :param sub_button_name: 可选参数，子按钮名称
        :return: None
        """
        self.page.wait_for_load_state("networkidle")
        if not sub_button_name:
            self.page.locator(self.locator.BUTTON_ARGS_LOC % button_name).first.click()
        else:
            self.page.locator(self.locator.BUTTON_ARROW_ARGS_LOC % button_name).first.hover()
            self.page.locator(self.locator.SUB_BUTTON_ARGS_LOC % sub_button_name).click()
        self._click_blank()

    def click_line(self, row:int):
        """
        选中单据体行（通过点击行序号）
        :param row: 行号，从1开始
        :return: None
        """
        self.page.locator(self.locator.ACTIVATE_FIELD_ARGS_LOC % ("序号", row-1, 1)).click()

    def _activate_field(self, row:int, field_name:str, order:int=1):
        """
        激活字段，单据体字段未激活状态时统一为span元素，无法识别字段类型，无法针对不同字段类型操作
        :param field_name: 字段名
        :param row: 行号，从1开始
        :param order: 可选参数，存在同名字段时指明第几个，从1开始
        :return: None
        """
        self.page.wait_for_load_state("networkidle")
        self.page.locator(self.locator.ACTIVATE_FIELD_ARGS_LOC % (field_name, row-1, order)).click()

    def _set_text(self, row:int, field_name:str, value:str, order:int=1):
        """
        单据体文本字段赋值
        :param field_name: 字段名
        :param value: 字段值
        :param row: 行号
        :param order: 可选参数，存在同名字段时指明第几个，从1开始
        :return: None
        """
        self._activate_field(row, field_name, order)
        locator = self.page.locator(self.locator.TEXT_FIELD_ARGS_LOC % (field_name, row-1, order))
        locator.clear()
        locator.fill(value)
        self.click_line(1)

    def _set_select(self, row:int, field_name:str, value:str, order:int=1):
        """
        单据体下拉列表字段赋值
        :param field_name: 字段名
        :param value: 选项值
        :param row: 行号
        :param order: 可选参数，存在同名字段时指明第几个，从1开始
        :return: None
        """
        self.page.locator(self.locator.ACTIVATE_FIELD_ARGS_LOC % (field_name, row-1, order)).click()
        self.page.locator(self.locator.SELECT_ITEM_ARGS_LOC % value).click()
        self.click_line(1)

    def _set_base(self, row:int, field_name:str, value:str, order:int=1):
        """
        单据体基础资料类型字段赋值
        :param field_name: 字段名
        :param value: 字段值
        :param row: 行号
        :param order: 可选参数，存在同名字段时指明第几个，从1开始
        :return: None
        """
        self._activate_field(row, field_name, order)
        locator = self.page.locator(self.locator.BASE_FIELD_ARGS_LOC % (field_name, row-1, order))
        locator.click()
        locator.clear()
        locator.fill(value)
        locator.press_sequentially(" ")
        locator.press("Backspace")
        self.page.locator(self.locator.BASE_ITEM_ARGS_LOC % value).nth(0).click()
        self.click_line(1)

    def set_value(self, row:int, field_name:str, value:str, order:int=1, field_type:FieldType = None):
        """
        单据体字段赋值统一方法，根据给定或自动识别的字段类型调用不同方法完成赋值

        **用法：**

        BodyPage.set_value("衬衣订单号", "26024021", 1)

        :param field_name: 字段名
        :param value: 字段值
        :param row: 行号
        :param order: 可选参数，存在同名字段时指明第几个，从1开始
        :param field_type: 可选参数，字段类型，不填时自动识别
        :return: None
        """
        if not field_type:
            field_type = self._parse_field(row, field_name, order)

        match field_type:
            case FieldType.TEXT:
                self._set_text(row, field_name, value, order)
            case FieldType.SELECT:
                self._set_select(row, field_name, value, order)
            case FieldType.BASE:
                self._set_base(row, field_name, value, order)

    def _parse_field(self, row:int, field_name:str, order:int=1):
        """
        识别单据体字段类型
        :param field_name: 字段名
        :param order: 可选参数，存在同名字段时指明第几个，从1开始
        :return: 字段类型
        """
        self._activate_field(row, field_name, order)
        data_role = self.page.locator(self.locator.PARSE_FILED_ARGS_LOC % (field_name, row-1, order)).get_attribute("data-role")
        self.click_line(1)
        match data_role:
            case "kdbosf7":
                return FieldType.BASE
            case "dropdownlist":
                return FieldType.SELECT
            case "kdtextfield" | "kdnumberfield" | "datepicker" | "kdmultilang" | "datetimepicker":
                return FieldType.TEXT

    def get_row_count(self):
        """
        获取单据体总行数
        :return: 行数
        """
        return self.page.locator(self.locator.ROW_COUNT_LOC).inner_text()

    def get_value(self, row:int, field_name:str, order:int=1):
        """
        获取单据体字段值，不区分字段类型
        :param field_name: 字段名
        :param row: 行号
        :param order: 可选参数，存在同名字段时指明第几个，从1开始
        :return: 字段值
        """
        return self.page.locator(self.locator.FIELD_VALUE_ARGS_LOC % (field_name, row-1, order)).inner_text()

    def set_checked(self, row:int, field_name:str):
        """
        勾选单据体复选框
        :param field_name: 字段名
        :param row: 行号
        :return: None
        """
        self.page.locator(self.locator.CHECKBOX_FIELD_ARGS_LOC % (field_name, row-1)).click()
        self.click_line(1)

    def set_text_in_body_head(self, field_name:str, value:str):
        """
        单据体中单据头形式文本字段赋值
        如：销售订单->物料数据->面料规格
        :param field_name: 字段名
        :param value: 字段值
        :return: None
        """
        locator = self.page.locator(self.locator.TEXT_FIELD_IN_BODY_HEAD_ARGS_LOC % field_name)
        locator.clear()
        locator.fill(value)
        self._click_blank()

    def set_select_in_body_head(self, field_name:str, value:str):
        """
        单据体中单据头形式下拉列表字段赋值
        如：销售订单->物料数据->货主类型
        :param field_name: 字段名
        :param value: 选项值
        :return: None
        """
        self.page.locator(self.locator.SELECT_FIELD_IN_BODY_HEAD_ARGS_LOC % field_name).click()
        self.page.locator(self.locator.SELECT_ITEM_ARGS_LOC % value).click()
        self._click_blank()

    def set_base_in_body_head(self, field_name:str, value:str):
        """
        单据体中单据头形式基础资料字段赋值
        如：销售订单->物料数据->面料来源
        :param field_name: 字段名
        :param value: 选项值
        :return: None
        """
        self.page.locator(self.locator.BASE_FIELD_IN_BODY_HEAD_SPAN_ARGS_LOC % field_name).click()
        locator = self.page.locator(self.locator.BASE_FIELD_IN_BODY_HEAD_ARGS_LOC % field_name)
        locator.clear()
        locator.fill(value)
        locator.press_sequentially(" ")
        locator.press("Backspace")
        self.page.locator(self.locator.BASE_ITEM_ARGS_LOC % value).nth(0).click()
        self._click_blank()

    def get_text_in_body_head(self, field_name:str, order:int=1):
        """
        单据体中单据头形式文本字段取值
        如：销售订单->物料数据->面料规格
        :param field_name: 字段名
        :param order: 可选参数，存在同名字段时指明第几个，从1开始
        :return: 字段值
        """
        # 字段值可能存在1. input中，2.span或其他元素中，目标元素不是input时input_value()方法报错，捕获并换为inner_text()获取
        try:
            return self.page.locator(self.locator.TEXT_VALUE_IN_BODY_HEAD_ARGS_LOC % (field_name, field_name, order)).input_value()
        except Error:
            return self.page.locator(self.locator.TEXT_VALUE_IN_BODY_HEAD_ARGS_LOC % (field_name, field_name, order)).inner_text()

    def get_select_in_body_head(self, field_name:str, order:int=1):
        """
        单据体中单据头形式下拉列表字段取值
        如：销售订单->物料数据->货主类型
        :param field_name: 字段名
        :param order: 可选参数，存在同名字段时指明第几个，从1开始
        :return: 字段值
        """
        return self.page.locator(self.locator.SELECT_VALUE_IN_BODY_HEAD_ARGS_LOC % (field_name, order)).inner_text()

    def get_base_in_body_head(self, field_name:str, order:int=1):
        """
        单据体中单据头形式基础资料字段取值
        如：销售订单->物料数据->面料来源
        :param field_name: 字段名
        :param order: 可选参数，存在同名字段时指明第几个，从1开始
        :return: 字段值
        """
        return self.page.locator(self.locator.BASE_VALUE_IN_BODY_HEAD_ARGS_LOC % (field_name, order)).inner_text()