nebula/pages/head_page.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
'''
@File    :   head_page.py
@Time    :   2024/11/07 15:19:33
@Author  :   dulip3ng 
@Version :   1.0
@Desc    :   None
'''

import re

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

class HeadPage(BasePage):

    def __init__(self, page:Page, locator):
        super().__init__(page)
        self.locator = locator

    def click_button(self, button_name:str, sub_button_name:str=None):
        """
        点击单据头按钮
        :param button_name: 按钮名称
        :param sub_button_name: 子按钮名称
        :return: None
        """
        if not sub_button_name:
            self.page.locator(self.locator.BUTTON_ARGS_LOC % button_name).click()
        else:
            self.page.locator(self.locator.BUTTON_ARROW_ARGS_LOC % button_name).hover()
            self.page.locator(self.locator.SUB_BUTTON_ARGS_LOC % sub_button_name).click()

    def change_sheet(self, sheet_name:str):
        """
        点击单据头页签
        :param sheet_name: 页签名称
        :return: None
        """
        self.page.locator(self.locator.SHEET_ARGS_LOC % sheet_name).click()

    def _set_text(self, field_name:str, value:str):
        """
        单据头文本字段赋值
        :param field_name: 字段名
        :param value: 字段值
        :return: None
        """
        locator = self.page.locator(self.locator.TEXT_FIELD_ARGS_LOC % field_name)
        locator.clear()
        locator.fill(value)

    def _set_base(self, field_name:str, value:str):
        """
        单据头基础资料字段赋值
        :param field_name: 字段名
        :param value: 字段值
        :return: None
        """
        # 直接点击BASE_FIELD_ARGS_LOC input元素可操作性检查不通过，被BASE_FIELD_SPAN_ARGS_LOC span劫持，这里直接对span进行点击
        self.page.locator(self.locator.BASE_FIELD_SPAN_ARGS_LOC % field_name).click()
        locator = self.page.locator(self.locator.BASE_FIELD_ARGS_LOC % field_name)
        locator.clear()
        locator.fill(value)
        # 输入空格弹出基础资料选项，fill方法会覆盖当前输入内容，使用press_sequentially方法模拟输入
        locator.press_sequentially(" ")
        locator.press("Backspace")
        self.page.locator(self.locator.BASE_ITEM_ARGS_LOC % value).nth(0).click()

    def _set_select(self, field_name:str, item_value:str):
        """
        单据头下拉列表字段赋值
        :param field_name: 字段名
        :param item_value: 字段值
        :return: None
        """
        # Locator.click()普通点击无法弹出下拉选项值，使用force强制点击
        # 注意：force=True强制点击不执行非必要的元素可操作性检查
        self.page.locator(self.locator.SELECT_FIELD_ARGS_LOC % field_name).click()
        self.page.locator(self.locator.SELECT_ITEM_ARGS_LOC % (item_value,item_value)).click()

    def _set_textarea(self, field_name:str, value:str):
        """
        单据头大文本框字段赋值
        :param field_name: 字段名
        :param value: 字段值
        :return: None
        """
        locator = self.page.locator(self.locator.TEXTAREA_FIELD_ARGS_LOC % field_name)
        locator.clear()
        locator.fill(value)

    def _set_multiselect(self, field_name:str, values:str):
        """
        单据头多选下拉列表字段赋值
        :param field_name: 字段名
        :param values: 字段值
        :return: None
        """
        self.page.locator(self.locator.MULTISELECT_FIELD_ARGS_LOC % field_name).click()
        for item_value in values:
            self.page.locator(self.locator.MULTISELECT_ITEM_ARGS_LOC % item_value).click()
        self.page.locator(self.locator.MULTISELECT_OK_BTN_LOC).click()

    def set_value(self, field_name:str, *values:str, field_type: FieldType = None):
        """
        单据头字段赋值统一方法，根据给定或自动识别的字段类型调用不同方法完成赋值
        :param field_name: 字段名
        :param value: 字段值
        :param field_type: 可选参数，字段类型，不填时自动识别字段类型
        :return: None
        """
        if not field_type:
            field_type = self._parse_field_type(field_name)

        match field_type:
            case FieldType.TEXT:
                self._set_text(field_name, values[0])
            case FieldType.SELECT:
                self._set_select(field_name, values[0])
            case FieldType.BASE:
                self._set_base(field_name, values[0])
            case FieldType.TEXTAREA:
                self._set_textarea(field_name, values[0])
            case FieldType.MULTISELECT:
                self._set_multiselect(field_name, *values)

    def set_checked(self, field_name:str):
        """
        单据头勾选复选框
        :param field_name: 复选框名
        :return: None
        """
        self.page.locator(self.locator.CHECKBOX_FIELD_ARGS_LOC % field_name).click()

    def set_text_in_head_body(self, row:int, field_name:str, value:str):
        """
        单据头中的单据体文本字段赋值
        如样品通知单-接收人页签
        :param field_name: 字段名
        :param value: 字段值
        :param row: 行号，从1开始
        :return: None
        """
        self.page.locator(self.locator.TRIGGER_IN_HEAD_BODY_FIELD_ARGS_LOC % (field_name, row-1)).click()
        locator = self.page.locator(self.locator.TEXT_FIELD_IN_HEAD_BODY_ARGS_LOC % (field_name, row-1))
        locator.clear()
        locator.fill(value)

    def set_base_in_head_body(self, row:int, field_name:str, value:str):
        """
        单据头中的单据体基础资料字段赋值
        如样品通知单-接收人页签
        :param field_name: 字段名
        :param value: 字段值
        :param row: 行号，从1开始
        :return: None
        """
        self.page.locator(self.locator.TRIGGER_IN_HEAD_BODY_FIELD_ARGS_LOC % (field_name, row - 1)).click()
        locator = self.page.locator(self.locator.BASE_FIELD_IN_HEAD_BODY_ARGS_LOC % (field_name, row - 1))
        locator.click()
        locator.fill(value)
        # 输入空格弹出基础资料选项，fill方法会覆盖当前输入内容，使用press_sequentially方法模拟输入
        locator.press_sequentially(" ")
        locator.press("Backspace")
        self.page.locator(self.locator.BASE_ITEM_ARGS_LOC % value).nth(0).click()


    def _get_text_value(self, field_name:str) -> str:
        """
        单据头文本字段取值
        文本字段有两种形式，一种是可编辑状态，值存放在input元素中，另一种是锁定状态值存放在span元素中，
        默认取input值，500ms超时未取到时取span中的值
        :param field_name: 字段名
        :return: 字段值
        """
        try:
            locator_input = self.page.locator(self.locator.TEXT_VALUE_INPUT_ARGS_LOC % field_name)
            expect(locator_input).to_be_visible(timeout=500)
            return locator_input.input_value()
        except AssertionError:
            locator_span = self.page.locator(self.locator.TEXT_VALUE_SPAN_ARGS_LOC % field_name)
            expect(locator_span).to_be_visible(timeout=500)
            return locator_span.inner_text()

    def _get_select_value(self, field_name:str) -> str:
        """
        单据头下拉列表字段取值
        下拉列表字段有两种形式，一种是可编辑状态，值存放在input元素中，另一种是锁定状态值存放在span元素中，
        默认取input值，未找到时立即去span中的值
        与取text不同的是text由两个xpath语句获取值，select由一个通用的xpath取值，所以判断方式不同
        :param field_name: 字段名
        :return: 字段值
        """
        try:
            locator = self.page.locator(self.locator.SELECT_VALUE_ARGS_LOC % field_name)
            return locator.input_value()
        except Error:
            return locator.inner_text()
    def _get_base_value(self, field_name:str) -> str:
        """
        获取单据头基础资料字段值
        :param field_name: 字段名
        :return: 字段值
        """
        return self.page.locator(self.locator.BASE_VALUE_ARGS_LOC % field_name).inner_text()

    def _get_textarea_value(self, field_name:str):
        """
        单据头大文本字段取值
        :param field_name: 字段名
        :return: 字段值
        """
        return self.page.locator(self.locator.TEXTAREA_FIELD_ARGS_LOC % field_name).input_value()

    def _get_multiselect_value(self, field_name:str):
        """
        单据头多选下拉列表字段取值
        :param field_name: 字段名
        :return: 字段值
        """
        return self.page.locator(self.locator.MULTISELECT_VALUE_ARGS_LOC % field_name).inner_text()

    def get_value(self, field_name:str, field_type: FieldType=None) -> str:
        """
        单据头字段取值统一方法，根据给定或自动识别的字段类型调用不同取值方法取值
        :param field_name: 字段名
        :param field_type: 可选参数，字段类型
        :return: 字段值
        """
        if not field_type:
            field_type = self._parse_field_type(field_name)

        match field_type:
            case FieldType.TEXT:
                return self._get_text_value(field_name)
            case FieldType.SELECT:
                return self._get_select_value(field_name)
            case FieldType.BASE:
                return self._get_base_value(field_name)
            case FieldType.MULTISELECT:
                return self._get_multiselect_value(field_name)
            case FieldType.TEXTAREA:
                return self._get_textarea_value(field_name)
            case _:
                return ""


    def _parse_field_type(self, field_name:str) -> FieldType:
        """
        分析字段类型，set_value(), get_value()中调用的核心方法
        注意：金蝶版本升级后可能不同字段类型判断依据有变化，字段类型是根据元素"data-attdata"属性值判断，
        定位器可用性检查通过不代表分析字段类型可正常使用
        :param field_name: 字段名
        :return: 字段类型 FieldType
        """
        data_role = self.page.locator(self.locator.PARSE_FIELD_TYPE_ARGS_LOC % field_name).get_attribute("data-attdata")
        pattern = "xtype\":\"([a-zA-Z0-9]+)"
        match_strs = re.findall(pattern, data_role)
        if match_strs:
            match match_strs[0]:
                case "bosf7field":
                    return FieldType.BASE
                case "kdeditcombo" | "kdcombo":
                    return FieldType.SELECT
                case "kdmulticombo":
                    return FieldType.MULTISELECT
                case "kdtextarea" | "kdlangtextarea" | "kdtabpage":
                    return FieldType.TEXTAREA
                case _:
                    return FieldType.TEXT
        return FieldType.UNKNOWN

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

        **用法：**
        HeadPage.click_button_in_head_body("新增行")

        HeadPage.click_button_in_head_body("新增行", "复制行")

        :param button_name: 按钮名称
        :param sub_button_name: 可选参数，子按钮名称
        :return: None
        """
        if not sub_button_name:
            self.page.locator(self.locator.BUTTON_IN_HEAD_BODY_ARGS_LOC % button_name).click()
        else:
            self.page.locator(self.locator.BUTTON_ARROW_IN_HEAD_BODY_ARGS_LOC % button_name).hover()
            self.page.locator(self.locator.SUB_BUTTON_IN_HEAD_BODY_ARGS_LOC % sub_button_name).click()
        self._click_blank()