path: root/common/ui_common/objects/geolocation.py

import os
import random
import time

from selenium import webdriver
# from selenium.webdriver import Keys
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By

from cases.conftest import my_driver
from common.driver_common.mywebdriver import MyWebDriver
from common.driver_common.random_name import RandomName
from common.driver_common.screenshot import screenshot_on_failure
from common.read_data.read_data import ReadData
from common.ui_common.login_logout.loginout import LogInOut
from common.ui_common.objects.objects_public_operations import ObjectsPublicOperations, get_txt_filenames, \
    download_files
from common.ui_common.profiles.profiles_public_operations import ProfilesPublicOperations
from config.workpath import workdir
from page_element.objects_element_position import *

class geolocation:

    def __init__(self,demo_fixture: MyWebDriver):
        self.workdir = workdir
        self.my_random = RandomName()
        self.driver= demo_fixture
        #调用profiles公共方法操作、删除、查询
        self.objects_po = ObjectsPublicOperations(self.driver)
        self.profile_pub = ProfilesPublicOperations(self.driver)

    @screenshot_on_failure
    def geolocation_case(self,data:{}):

        self.driver.implicitly_wait(10)
        self._create(data)
        self._modify(data)
        self._delete(data)

    def _create(self,data:{}):
        #跳转到Geolocation页面
        self._goto_subObjectPage()
        self.loading_parameter(data)
        #点击Create
        self.driver.find_element(By.ID,listPage_object_geolocation_createButton_posId).click()
        if data["Create"]["model"] == "0":
            # 添加ip item
            self._add_item(data)
            #输入Geolocation选项，ip
            self._input_geolocation(data)
            self.Confirm_creation(data)

    def _goto_subObjectPage(self):
        self.objects_po.licenese_Check()
        # 菜单操作 跳转到mobile_identities页面
        self.driver.find_element(By.ID, mainPage_firstLevelMenu_Objects_posId).click()
        self.driver.find_element(By.ID, mainPage_secondLevelMenu_Geolocation_posId).click()

    def loading_parameter(self,data):
        # self.objects_po.Modify_object(data)
        self.modify_status = "0" #查看修改条件 0 不修改，1修改
        self.create_nameid = self.my_random.random_geo_name_id(count=random.randint(1,9))
        self.random_item_ipv4 = self.my_random.random_ipv4()
        self.random_item_ipv6 = self.my_random.random_ipv6()

    def _input_geolocation(self,data):
        if data["Createtype"] == "Country_and_Region":
            #添加新建Continent
            if data["Continent"] != "null":
                self.driver.find_element(By.ID,geo_ObjectCreate_Continent_select_button_posID).click()
                self.driver.find_element(By.XPATH,geo_Geographic_add_continent_button_posXpath).click()
                self.driver.find_element(By.XPATH,geo_Geographic_add_continent_text_input_posXpath).send_keys(data["Continent"])
                self.driver.find_element(By.ID,geo_Geographic_add_continent_save_button_posID).click()
                self.driver.find_element(By.ID,geo_Geographic_add_continent_cancel_button_posID).click()
            else:
                data['Create']['model'] = "2"
                return
            #添加Geo Name ID
            self._input_geolocation_value(data, "Geo Name ID", data["Geo Name ID"],geo_ObjectCreate_geo_name_id_posID)
            #添加Country Abbreviation
            self._input_geolocation_value(data, "Country Abbreviation", data["Country Abbreviation"],geo_ObjectCreate_Country_Abbreviation_posID)
            #添加Country and Region
            if data["Country and Region"] != "null":
                self.driver.find_element(By.ID,geo_ObjectCreate_Country_Region_posID).send_keys(data["Country and Region"])
            else: return
            # 添加Super Administrative Area
            if data["Super Administrative Area"] != "null":
                self.driver.find_element(By.XPATH,geo_ObjectCreate_add_super_administrative_area_button_posXpath).click()
                self.driver.find_element(By.ID,geo_ObjectCreate_super_administrative_area_posID).click()
                self.driver.find_element(By.XPATH,geo_Geographic_add_custom_button_posXpath).click()
                self.driver.find_element(By.XPATH,geo_ObjectCreate_add_super_administrative_area_text_posXpath).send_keys(data["Super Administrative Area"])
                self.driver.find_element(By.ID,geo_Geographic_add_super_administrative_area_save_button_posID).click()
                self.driver.find_element(By.XPATH,"(//button[@id='interceptionadd_allcancelobject9'])[2]").click()
            else: return
            #添加Administrative Area
            if data["Administrative Area"] != "null":
                self.driver.find_element(By.XPATH, geo_ObjectCreate_add_super_administrative_area_button_posXpath).click()
                self.driver.find_element(By.ID, geo_ObjectCreate_super_administrative_area_posID).click()
                self.driver.find_element(By.XPATH, geo_Geographic_add_custom_button_posXpath).click()
                self.driver.find_element(By.XPATH, geo_ObjectCreate_add_super_administrative_area_text_posXpath).send_keys(data["Administrative Area"])
                self.driver.find_element(By.ID, geo_Geographic_add_super_administrative_area_save_button_posID).click()
                self.driver.find_element(By.XPATH, "(//button[@id='interceptionadd_allcancelobject9'])[2]").click()
            else: return

        elif data["Createtype"] == "City":
            #切换到City类型
            self.driver.find_element(By.XPATH,geo_ObjectCreate_type_city_posXpath).click()
            #添加Country
            self.driver.find_element(By.ID,geo_ObjectCreate_super_administrative_area_posID).click()
            self.driver.find_element(By.XPATH,f"//div[@id='IP-library-country-tree']//p[normalize-space(text())='{data['Continent']}']").click()
            self.driver.find_element(By.XPATH,f"//div[@id='IP-library-country-tree']//div[normalize-space(text())='{data['Country']}']").click()
            self.driver.find_element(By.ID, geo_Geographic_add_continent_cancel_button_posID).click()
            #添加Name ID
            self._input_geolocation_value(data, "Geo Name ID", data["Geo Name ID"],geo_ObjectCreate_geo_name_id_posID)
            #添加Super Administrative Area
            self.driver.find_element(By.XPATH, geo_ObjectCreate_add_super_administrative_area_button_posXpath).click()
            self.driver.find_element(By.ID, geo_ObjectCreate_super_administrative_area_posID).click()
            self.driver.find_element(By.XPATH,f"//div[@class='rm-content-tree rm-c-list']//span[text()='{data['Super Administrative Area']}']").click()
            self.driver.find_element(By.XPATH, "(//button[@id='interceptionadd_allcancelobject9'])[2]").click()
            #添加Administrative Area
            self.driver.find_element(By.XPATH, geo_ObjectCreate_add_super_administrative_area_button_posXpath).click()
            self.driver.find_element(By.ID, geo_ObjectCreate_super_administrative_area_posID).click()
            self.driver.find_element(By.XPATH,f"//div[@class='rm-content-tree rm-c-list']//span[text()='{data['Administrative Area']}']").click()
            self.driver.find_element(By.XPATH, "(//button[@id='interceptionadd_allcancelobject9'])[2]").click()
        self.driver.find_element(By.XPATH, geo_ObjectCreate_Longitude_posXPATH).clear()
        self.driver.find_element(By.XPATH, geo_ObjectCreate_Latitude_posXPATH).clear()
        self.driver.find_element(By.XPATH, geo_ObjectCreate_Longitude_posXPATH).send_keys(data['Longitude'])
        self.driver.find_element(By.XPATH, geo_ObjectCreate_Latitude_posXPATH).send_keys(data['Latitude'])

    def _input_geolocation_value(self,data,data_type,data_value,element_text):
        if data_type == 'Geo Name ID' and data_value == 'random':
            self.driver.find_element(By.ID, element_text).send_keys(self.create_nameid)
        elif data_value != 'null':
            self.driver.find_element(By.ID, element_text).send_keys(data_value)
            if data["Subtype"] == "error_object":
                data['Create']['model'] = "2"
        elif data_value == 'null':
            data['Create']['model'] = "2"
            pass


    def _add_item(self,data):
        if data['Create']['item'] == "null":
            data['Create']['model'] = "2"
        if data['Create']['model'] == "0":
            if self.modify_status == "0":
                #新建item
                self.driver.find_element(By.ID, geo_ObjectCreate_ip_add_button_posID).click()
                if data['Create']['type'] == "new_object":
                    for item in data['Create']['item'].keys():
                        if item == "IPv4":
                            for ipv4_address in data['Create']['item'][item]:
                                self.driver.find_element(By.ID, geo_ObjectCreate_ip_item_add_button_posID).click()
                                if ipv4_address =='random':
                                    self.driver.find_element(By.XPATH,geoObjectPage_button_editItem_text_posXpath).send_keys(self.random_item_ipv4)
                                else:
                                    self.driver.find_element(By.XPATH, geoObjectPage_button_editItem_text_posXpath).send_keys(ipv4_address)
                                self.driver.find_element(By.XPATH,geo_ObjectCreate_ip_item_text_save_button_posXpath).click()
                        elif item == "IPv6":
                            for ipv6_address in data['Create']['item'][item]:
                                self.driver.find_element(By.ID, geo_ObjectCreate_ip_item_add_button_posID).click()
                                self.driver.find_element(By.ID, geo_ObjectCreate_ip_item_type_posID).click()
                                self.driver.find_element(By.XPATH, geo_ObjectCreate_ip_item_type_ipv6_posXpath).click()
                                if ipv6_address =='random':
                                    self.driver.find_element(By.XPATH,geoObjectPage_button_editItem_text_posXpath).send_keys(self.random_item_ipv6)
                                else:
                                    self.driver.find_element(By.XPATH, geoObjectPage_button_editItem_text_posXpath).send_keys(ipv6_address)
                                self.driver.find_element(By.XPATH,geo_ObjectCreate_ip_item_text_save_button_posXpath).click()
                        if data['Subtype'] == "error_item":
                            assert self.driver.find_element(By.XPATH,geoObjectPage_item_error_posXpath).text == "IP format error", "断言错误ip类型"
                            data['Create']['model'] ="2"
            elif self.modify_status == "1":  # add item
                for item in data['Create']['item_modify']['add_data'].keys():
                    if item == "IPv4":
                        self.driver.find_element(By.ID, geo_ObjectCreate_ip_item_add_button_posID).click()
                        self.driver.find_element(By.XPATH, geoObjectPage_button_editItem_text_posXpath).send_keys(data['Create']['item_modify']['add_data'][item])
                        self.driver.find_element(By.XPATH,geo_ObjectCreate_ip_item_text_save_button_posXpath).click()
                    elif item == "IPv6":
                        self.driver.find_element(By.ID, geo_ObjectCreate_ip_item_add_button_posID).click()
                        self.driver.find_element(By.ID, geo_ObjectCreate_ip_item_type_posID).click()
                        self.driver.find_element(By.XPATH, geo_ObjectCreate_ip_item_type_ipv6_posXpath).click()
                        self.driver.find_element(By.XPATH, geoObjectPage_button_editItem_text_posXpath).send_keys(data['Create']['item_modify']['add_data'][item])
                        self.driver.find_element(By.XPATH,geo_ObjectCreate_ip_item_text_save_button_posXpath).click()
            elif self.modify_status == "2":  # add item
                    self.driver.find_element(By.XPATH, geoObjectPage_button_editItem_posXpath).click()
                    description_elem = self.driver.find_element(By.XPATH, geoObjectPage_button_editItem_text_posXpath)
                    description_elem.clear()
                    description_elem.send_keys(data['Create']['item_modify']['modify_data'])
                    self.driver.find_element(By.XPATH, geoObjectPage_button_editItem_text_Save_posXpath).click()
                    if data['Create']['item_type'] == "error_item":
                        assert self.driver.find_element(By.XPATH,geoObjectPage_item_error_posXpath).text == "IP format error","断言错误ip类型"
                        self.driver.find_element(By.XPATH,geo_ObjectCreate_button_Cancel_posXpath).click()
                        data["Create"]["model"] = "1"
            if isinstance(data["Longitude"], str) or isinstance(data["Latitude"], str):
                data['Create']['model'] = "2"
            elif data["Longitude"] < -180 or data["Latitude"] > 180 or data["Latitude"] < -90 or data["Longitude"] > 90:
                data['Create']['model'] = "2"
        else:
            pass

    def Confirm_creation(self,data):
        if data["Create"]["model"] == "0":
            self.driver.find_element(By.XPATH, geo_ObjectCreate_button_OK_posXpath).click()
            self.driver.find_element(By.XPATH, geo_ObjectCreate_button_warningSaveYes_posXpath).click()
            self.extract_ele()
        #对不规范Object的验证
        elif data["Create"]["model"] == "2":
            if data["Geo Name ID"] == "null":
                self._check_null(geo_ObjectCreate_geo_name_error_posXpath)
                # self.driver.find_element(By.XPATH, geo_ObjectCreate_button_OK_posXpath).click()
                # assert self.driver.find_element(By.XPATH,geo_ObjectCreate_geo_name_error_posXpath).text == "Cannot be empty","NameID为空校验"
            elif data["Geo Name ID"] != "random":   #判断正常输入数据但是不符合格式，即输入字符
                self.driver.find_element(By.XPATH, geo_ObjectCreate_button_OK_posXpath).click()
                assert self.driver.find_element(By.XPATH,geo_ObjectCreate_geo_name_error_posXpath).text == "Please enter the correct format", "NameID为空校验"
            elif data['Create']['item'] == "null":
                self._check_null(geo_ObjectCreate_city_ip_range_error_posXpath)
                # self.driver.find_element(By.XPATH, geo_ObjectCreate_button_OK_posXpath).click()
                # assert self.driver.find_element(By.XPATH,geo_ObjectCreate_city_ip_range_error_posXpath).text == "Cannot be empty","IP Range为空校验"
            elif data['Continent'] =="null":
                self._check_null(geo_ObjectCreate_contry_error_posXpath)
            if data["Createtype"] == "Country_and_Region":
                if data["Country Abbreviation"] == "null":
                    self._check_null(geo_ObjectCreate_country_abbr_error_posXpath)
                    # self.driver.find_element(By.XPATH, geo_ObjectCreate_button_OK_posXpath).click()
                    # assert self.driver.find_element(By.XPATH,geo_ObjectCreate_country_abbr_error_posXpath).text == "Cannot be empty","Country Abbre为空"
                elif data["Country and Region"] == "null":
                    self._check_null(geo_ObjectCreate_country_region_error_posXpath)
                    # self.driver.find_element(By.XPATH, geo_ObjectCreate_button_OK_posXpath).click()
                    # assert self.driver.find_element(By.XPATH,geo_ObjectCreate_country_region_error_posXpath).text == "Cannot be empty", "Country Abbre为空"
            elif data['Super Administrative Area'] == "null" or data['Administrative Area'] == "null":
                self.driver.find_element(By.XPATH, geo_ObjectCreate_button_OK_posXpath).click()
                self.driver.find_element(By.XPATH,ip_addressObjectPage_button_warningSaveYes_posXpath).click()
                assert self.driver.find_element(By.XPATH,main_listPage_object_error_alter_postXpath).text == "Super Administrative Area is empty" or "Administrative Area is empty"
            if isinstance(data["Longitude"],str) or isinstance(data["Latitude"],str):
                self.driver.find_element(By.XPATH, geo_ObjectCreate_button_OK_posXpath).click()
                assert self.driver.find_element(By.XPATH,geo_ObjectCreate_Longitude_error_posXpath).text == "Please enter the correct format"
                assert self.driver.find_element(By.XPATH,geo_ObjectCreate_Latitude_error_posXpath).text == "Please enter the correct format"
            elif data["Longitude"] < -180 or data["Latitude"] > 180 or data["Latitude"] < -90 or data["Longitude"] > 90:
                self.driver.find_element(By.XPATH, geo_ObjectCreate_button_OK_posXpath).click()
                assert self.driver.find_element(By.XPATH,geo_ObjectCreate_Longitude_error_posXpath).text == "Please enter the correct format"
                assert self.driver.find_element(By.XPATH,geo_ObjectCreate_Latitude_error_posXpath).text == "Please enter the correct format"
        else:
            pass

    def _check_null(self,element_error_warn):
        self.driver.find_element(By.XPATH, geo_ObjectCreate_button_OK_posXpath).click()
        assert self.driver.find_element(By.XPATH,element_error_warn).text == "Cannot be empty", "Country Abbre为空"


    def _search(self, data):
        search_type = data["Search_type"].split('&')
        for search_type in search_type:
            if search_type == "IP":
                self.driver.find_element(By.ID, listPage_objectSearch_ip_address_selectLabel_posId).click()
                self.driver.find_element(By.ID, listPage_objectSearch_ip_address_selectLabel_posId).send_keys(data["Create"]["item"]["IPv4"])
                self.driver.find_element(By.XPATH, listPage_objectSearch_geo_select_IP_posXpath).click()
            elif search_type == "Geo Name ID":
                self.driver.find_element(By.ID, listPage_objectSearch_ip_address_selectLabel_posId).click()
                self.driver.find_element(By.ID, listPage_objectSearch_ip_address_selectLabel_posId).send_keys(self.table_dict['Geo Name ID'])
                self.driver.find_element(By.XPATH, listPage_objectSearch_geo_select_GeoNameID_posXpath).click()
            elif search_type == "Country Abbreviation":
                self.driver.find_element(By.ID, listPage_objectSearch_ip_address_selectLabel_posId).click()
                self.driver.find_element(By.ID, listPage_objectSearch_ip_address_selectLabel_posId).send_keys(self.table_dict['Country Abbreviation'])
                self.driver.find_element(By.XPATH,listPage_objectSearch_geo_select_CountryAbbreviation_posXpath).click()
            elif search_type == "Country and Region":
                self.driver.find_element(By.ID, listPage_objectSearch_ip_address_selectLabel_posId).click()
                self.driver.find_element(By.ID, listPage_objectSearch_ip_address_selectLabel_posId).send_keys(self.table_dict['Country and Region'])
                self.driver.find_element(By.XPATH,listPage_objectSearch_geo_select_CountryandRegion_posXpath).click()
            elif search_type == "Super Administrative Area":
                self.driver.find_element(By.ID, listPage_objectSearch_ip_address_selectLabel_posId).click()
                self.driver.find_element(By.ID, listPage_objectSearch_ip_address_selectLabel_posId).send_keys(self.table_dict['Super Administrative Area'])
                self.driver.find_element(By.XPATH,listPage_objectSearch_geo_select_SuperAdministrativeArea_posXpath).click()
            elif search_type == "Administrative Area":
                self.driver.find_element(By.ID, listPage_objectSearch_ip_address_selectLabel_posId).click()
                self.driver.find_element(By.ID, listPage_objectSearch_ip_address_selectLabel_posId).send_keys(self.table_dict['Administrative Area'])
                self.driver.find_element(By.XPATH,listPage_objectSearch_geo_select_AdministrativeArea_posXpath).click()
            elif search_type == 'Sub Administrative Area':
                self.driver.find_element(By.ID, listPage_objectSearch_ip_address_selectLabel_posId).click()
                self.driver.find_element(By.ID, listPage_objectSearch_ip_address_selectLabel_posId).send_keys(self.table_dict['Sub Administrative Area'])
                self.driver.find_element(By.XPATH, listPage_objectSearch_geo_select_SubAdministrativeArea_posXpath).click()
            else:  # 全局查询
                self.driver.find_element(By.ID, listPage_objectSearch_ip_address_selectLabel_posId).send_keys()
                self.driver.find_element(By.ID, mainPage_ObjectSearch_buttonSearch_posId).click()
        self.driver.find_element(By.ID, mainPage_ObjectSearch_buttonSearch_posId).click()

    def extract_ele(self):
        # self.driver.isElementExist(Element='//div/div[3]/table/tbody/tr[1]')  # 查看第一行是否存在数据
        self.driver.isElementExist(
            Element='((//div[@class!="itemDetails cursor"]//table[@class="el-table__body"])[1]//tr)[1]')  # 查看第一行是否存在数据
        if self.driver.Exist:
            # 展示所有列
            self.driver.find_element(By.XPATH, mainPage_ObjectSearch_Column_settings_posXpath).click()  # 点击列设置
            all_checked_xpath = '//*[@class="all-btn"]//*[@class="el-checkbox__input is-checked"]'
            if self.driver.element_isExist(By.XPATH, all_checked_xpath):  # 如果全选被选中
                self.driver.find_element(By.XPATH,
                                         mainPage_ObjectSearch_Column_settings_click_Cancel_Button_posXpath).click()  # 点击Cancel按钮
            else:
                self.driver.find_element(By.XPATH,
                                         mainPage_ObjectSearch_Column_settings_select_allCheckBox_posXpath).click()  # 全选
                self.driver.find_element(By.XPATH,
                                         mainPage_ObjectSearch_Column_settings_click_OK_Button_posXpath).click()  # 点击OK按钮

            # 公共方法 Detail特殊，需要使用时特殊提取
            list_headers_template = '//div[@class="el-table__header-wrapper"]//span[text()="{}"]'
            list_firstRow_values_template = '//div[contains(@class,"el-table__body-wrapper")]//tr[1]//td[count({a}/../../../preceding-sibling::th)+1]{b}'

            #获取标题元素
            list_headers_name_id_posXpath = list_headers_template.format("Geo Name ID")
            list_headers_vsys_id_posXpath = list_headers_template.format("Vsys ID")
            list_headers_countryabbr_posXpath = list_headers_template.format("Country Abbreviation")
            list_headers_country_region_posXpath = list_headers_template.format("Country and Region")
            list_headers_super_admin_area_posXpath = list_headers_template.format("Super Administrative Area")
            list_headers_admin_area_posXpath = list_headers_template.format("Administrative Area")
            list_headers_sub_admin_area_posXpath = list_headers_template.format("Sub Administrative Area")

            #获取第一个Object的neirong
            list_firstRow_values_name_id_posXpath = list_firstRow_values_template.format(a=list_headers_name_id_posXpath,b='//div[@class="table-status-item-id"]//span')
            list_firstRow_values_vsys_id_posXpath = list_firstRow_values_template.format(a=list_headers_vsys_id_posXpath, b='//div[@class="table-hide"]')
            list_firstRow_values_countyrabbr_posXpath = list_firstRow_values_template.format(a=list_headers_countryabbr_posXpath, b='//div[@class="table-hide"]')
            list_firstRow_values_country_region_posXpath = list_firstRow_values_template.format(a=list_headers_country_region_posXpath,b='//div[@class="table-hide"]')
            list_firstRow_values_super_admin_area_posXpath = list_firstRow_values_template.format(a=list_headers_super_admin_area_posXpath,b='//div[@class="table-hide"]')
            list_firstRow_values_admin_area_posXpath = list_firstRow_values_template.format(a=list_headers_admin_area_posXpath, b='//div[@class="table-hide"]')
            list_firstRow_values_sub_admin_area_posXpath = list_firstRow_values_template.format(a=list_headers_sub_admin_area_posXpath, b='//div[@class="table-hide"]')

            self.key_id = self.driver.find_element(By.XPATH, list_headers_name_id_posXpath).text
            self.value_id = self.driver.find_element(By.XPATH, list_firstRow_values_name_id_posXpath).text
            self.key_vsys_id = self.driver.find_element(By.XPATH, list_headers_vsys_id_posXpath).text
            self.value_vsys_id = self.driver.find_element(By.XPATH, list_firstRow_values_vsys_id_posXpath).text
            self.key_countryabbr = self.driver.find_element(By.XPATH, list_headers_countryabbr_posXpath).text
            self.value_countryabbr = self.driver.find_element(By.XPATH, list_firstRow_values_countyrabbr_posXpath).text
            self.key_country_region = self.driver.find_element(By.XPATH, list_headers_country_region_posXpath).text
            self.value_country_region = self.driver.find_element(By.XPATH, list_firstRow_values_country_region_posXpath).text
            self.key_super_admin_area = self.driver.find_element(By.XPATH, list_headers_super_admin_area_posXpath).text
            self.value_super_admin_area = self.driver.find_element(By.XPATH, list_firstRow_values_super_admin_area_posXpath).text
            self.key_admin_area = self.driver.find_element(By.XPATH, list_headers_admin_area_posXpath).text
            self.value_admin_area = self.driver.find_element(By.XPATH, list_firstRow_values_admin_area_posXpath).text
            self.key_sub_admin_area = self.driver.find_element(By.XPATH, list_headers_sub_admin_area_posXpath).text
            self.value_sub_admin_area = self.driver.find_element(By.XPATH, list_firstRow_values_sub_admin_area_posXpath).text

            self.table_dict = {self.key_id: self.value_id, self.key_vsys_id: self.value_vsys_id,
                               self.key_countryabbr: self.value_countryabbr, self.key_country_region: self.value_country_region,
                               self.key_super_admin_area: self.value_super_admin_area,
                               self.key_admin_area: self.value_admin_area,
                               self.key_sub_admin_area: self.value_sub_admin_area}
            print("extract_ele TEST:{}".format(self.table_dict))

    def _modify(self, data: {}):
        if "item_modify" in data['Create']:
            self._search(data)
            self.driver.find_element(By.XPATH, listPage_object_ip_address_select_First_object_posXpath).click()
            self.driver.find_element(By.ID, listPage_object_geolocation_editButton_posId).click()
            if data['Create']['item_modify']['type'] == "add":
                self.modify_status = "1"
                self._add_item(data)
                self.Confirm_creation(data)
            if data['Create']['item_modify']['type'] == "check":
                self.driver.find_element(By.ID, mainPage_GeolocationSearch_buttonSearch_Item_posId).send_keys(data['Create']['item_modify']['check_data'] + Keys.ENTER)
                self.item_check_content = self.driver.find_element(By.XPATH,geoObjectPage_button_addItem_first_item_posXpath).text
                assert self.item_check_content == data['Create']['item_modify']['check_data'], "断言检查item文本正确"
            if data['Create']['item_modify']['type'] == "modify":
                self.modify_status = "2"
                self._add_item(data)
                self.Confirm_creation(data)
        else:
            pass

    def _delete(self, data: {}, search_value = None,dele_flag=None):
        if self.modify_status == "1" or self.modify_status == "2":
            self.driver.find_element(By.XPATH,listPage_object_ip_address_select_First_object_posXpath).click()  # 选择第一个对象
            self.driver.find_element(By.ID, listPage_object_geolocation_delButton_posId).click()
            self.driver.find_element(By.XPATH, listPage_object_geolocation_delYesButton_posXpath).click()
            # self.Page_Tips = self.driver.find_element(By.XPATH, listPage_object_ip_address_del_after_ele_posXpath).text
            # assert self.Page_Tips == "No Data", "校验修改Name是否生效"
        elif data['Subtype'] =="not_delete":
            pass
        elif data['Create']['model'] == "2":
            pass
        else:
            self._goto_subObjectPage()
            self._search(data)
            self.driver.find_element(By.XPATH,listPage_object_ip_address_select_First_object_posXpath).click()  # 选择第一个对象
            self.driver.find_element(By.ID, listPage_object_geolocation_delButton_posId).click()
            self.driver.find_element(By.XPATH, listPage_object_geolocation_delYesButton_posXpath).click()
            time.sleep(3)

    def _page_turn_case(self,data:{}):
        self._goto_subObjectPage()
        self.objects_po.pages_turning(data)

    @screenshot_on_failure
    def _verify_audit_logs(self, data):
        # 新建
        self._create(data)  # 新建
        self._search(data)  # 创建后查询
        # 选中并进入编辑页面
        self.driver.find_element(By.XPATH, listPage_object_ip_address_select_First_object_posXpath).click()
        self.driver.find_element(By.ID, listPage_object_geolocation_editButton_posId).click()
        # 点击Audit Logs
        self.driver.find_element(By.XPATH, acc_ObjectDetailPage_auditLogs_posXpath).click()
        self.driver.find_element(By.XPATH,acc_ObjectDetailPage_auditLogsDrawer_firstRowLog_checkBox_posXpath).click()  # 勾选第一行log CheckBox
        self.driver.find_element(By.XPATH,acc_ObjectDetailPage_auditLogsDrawer_compareButton_posXpath).click()  # 点击compare 按钮
        try:
            operation = self.driver.find_element(By.XPATH, acc_ObjectDetailPage_auditLogsDrawer_compareDrawer_operationText_posXpath).text
            assert operation == "Create", "最新一条audit log 非 Create log！！！！"
            print("校验成功，获取到 Create log")
        except Exception as e:
            print(e)
            raise
        self._delete(data)

    def _itemDuplicateCheck(self, data):
        #新建Object
        self._create(data)
        #重复创建Object
        self._goto_subObjectPage()
        self.driver.find_element(By.ID, listPage_object_geolocation_createButton_posId).click()
        # 输入Geolocation选项，ip
        self.create_nameid = self.my_random.random_geo_name_id(count=random.randint(1, 9))
        self._input_geolocation(data)
        self._add_item(data)
        #重复校验
        is_dup_alert_elem_exist = self.driver.element_isExist(By.XPATH, acc_ObjectDetailPage_dupAlert_posXpath)
        assert is_dup_alert_elem_exist, "校验失败！重复提示标志存在为：{}".format(is_dup_alert_elem_exist)
        # 点击重复标志
        self.driver.find_element(By.XPATH, acc_ObjectDetailPage_dupAlert_posXpath).click()
        total_numer_text = self.driver.find_element(By.XPATH,
                                                    acc_ObjectDetailPage_dupAlert_duplicateListTotal_posXpath).text
        total_number = [s for s in total_numer_text.split(":")][-1]
        assert int(total_number) >= 1, "校验失败，此时total_number = {}".format(total_number)
        self.Confirm_creation(data)
        for i in range(2):
            self._search(data)
            self.driver.find_element(By.XPATH,listPage_object_ip_address_select_First_object_posXpath).click()  # 选择第一个对象
            self.driver.find_element(By.ID, listPage_object_geolocation_delButton_posId).click()
            self.driver.find_element(By.XPATH, listPage_object_geolocation_delYesButton_posXpath).click()


    def _add_continent_error(self,data):
        self._goto_subObjectPage()
        self.driver.find_element(By.ID, listPage_object_geolocation_createButton_posId).click()
        self.driver.find_element(By.ID, geo_ObjectCreate_Continent_select_button_posID).click()
        self.driver.find_element(By.XPATH, geo_Geographic_add_continent_button_posXpath).click()
        self.driver.find_element(By.XPATH, geo_Geographic_add_continent_text_input_posXpath).send_keys(data["Continent"])
        self.driver.find_element(By.ID, geo_Geographic_add_continent_save_button_posID).click()
        if data["Type"] == "empty":
            assert self.driver.find_element(By.XPATH,"//*[@id='right_locationList']//div[@class='font-color']").text == "Cannot be empty","新建continent为空判断失败"
        elif data["Type"] == "format":
            assert self.driver.find_element(By.XPATH,"//*[@id='right_locationList']//div[@class='font-color']").text == "Format error", "新建continent格式判断失败"
        elif data["Type"] == "search":
            self.driver.find_element(By.ID,geo_Geographic_add_continent_search_button_posID).send_keys(data["Continent"] + Keys.ENTER)
            if data["Search_type"] == "0":
                assert self.driver.find_element(By.XPATH,"//*[@class='el-collapse']//div[@role='tab']/div/div/div[2]/p[1]").text == data["Continent"]
            else:
                assert self.driver.find_element(By.XPATH,"//*[@id='right_locationList']/div[2]//div[normalize-space(text())='No Data']").text == "No Data"


    def _sideslip_check(self,data):
        # 切换到geo页面
        self._goto_subObjectPage()
        self.loading_parameter(data)
        # 点击Create
        self.driver.find_element(By.ID, listPage_object_geolocation_createButton_posId).click()
        if data["Checktype"] == "Super Administrative Area" or data["Checktype"] == "Administrative Area" or data["Checktype"] == "Sub Administrative Area":
            # 切换到City类型
            self.driver.find_element(By.XPATH, geo_ObjectCreate_type_city_posXpath).click()
            # 添加Country
            self.driver.find_element(By.ID, geo_ObjectCreate_super_administrative_area_posID).click()
            self.driver.find_element(By.XPATH,f"//div[@id='IP-library-country-tree']//p[normalize-space(text())='{data['Continent']}']").click()
            self.driver.find_element(By.XPATH,f"//div[@id='IP-library-country-tree']//div[normalize-space(text())='{data['Country']}']").click()
            self.driver.find_element(By.ID, geo_Geographic_add_continent_cancel_button_posID).click()
        if data["Checktype"] == "Super Administrative Area":
            #校验Super Administrative Area
            self.check_error(data["Type"], data["Search_type"], data["Super Administrative Area"])
        elif data["Checktype"] == "Administrative Area":
            self.driver.find_element(By.XPATH, geo_ObjectCreate_add_super_administrative_area_button_posXpath).click()
            self.driver.find_element(By.ID, geo_ObjectCreate_super_administrative_area_posID).click()
            self.driver.find_element(By.XPATH,f"//div[@class='rm-content-tree rm-c-list']//span[text()='{data['Super Administrative Area']}']").click()
            self.driver.find_element(By.XPATH, "(//button[@id='interceptionadd_allcancelobject9'])[2]").click()
            # 校验Administrative Area
            self.check_error(data["Type"], data["Search_type"], data["Administrative Area"])
        elif data["Checktype"] == "Sub Administrative Area":
            # 添加Super Administrative Area
            self.driver.find_element(By.XPATH, geo_ObjectCreate_add_super_administrative_area_button_posXpath).click()
            self.driver.find_element(By.ID, geo_ObjectCreate_super_administrative_area_posID).click()
            self.driver.find_element(By.XPATH,f"//div[@class='rm-content-tree rm-c-list']//span[text()='{data['Super Administrative Area']}']").click()
            self.driver.find_element(By.XPATH, "(//button[@id='interceptionadd_allcancelobject9'])[2]").click()
            # 添加Administrative Area
            self.driver.find_element(By.XPATH, geo_ObjectCreate_add_super_administrative_area_button_posXpath).click()
            self.driver.find_element(By.ID, geo_ObjectCreate_super_administrative_area_posID).click()
            self.driver.find_element(By.XPATH,f"//div[@class='rm-content-tree rm-c-list']//span[text()='{data['Administrative Area']}']").click()
            self.driver.find_element(By.XPATH, "(//button[@id='interceptionadd_allcancelobject9'])[2]").click()
            #校验Sub Administrative Area
            self.check_error(data["Type"], data["Search_type"], data["Sub Administrative Area"])
        elif data["Checktype"] == "Locality":
            self._input_geolocation(data)
            #添加Sub Administrative Area
            self._input_sideslip_value(data["Sub Administrative Area"])
            #校验Locality
            self.check_error(data["Type"], data["Search_type"], data["Locality"])
        elif data["Checktype"] == "Dependent Locality":
            self._input_geolocation(data)
            # 添加Sub Administrative Area
            self._input_sideslip_value(data["Sub Administrative Area"])
            #添加Locality
            self._input_sideslip_value(data["Locality"])
            #校验Dependent Locality
            self.check_error(data["Type"], data["Search_type"], data["Dependent Locality"])
        elif data["Checktype"] == "Double Dependent Locality":
            self._input_geolocation(data)
            # 添加Sub Administrative Area
            self._input_sideslip_value(data["Sub Administrative Area"])
            # 添加Locality
            self._input_sideslip_value(data["Locality"])
            #添加Dependent Locality
            self._input_sideslip_value(data["Dependent Locality"])
            #校验Double Dependent Locality
            self.check_error(data["Type"], data["Search_type"], data["Double Dependent Locality"])

    def _input_sideslip_value(self,data_value):
        self.driver.find_element(By.XPATH, geo_ObjectCreate_add_super_administrative_area_button_posXpath).click()
        self.driver.find_element(By.ID, geo_ObjectCreate_super_administrative_area_posID).click()
        self.driver.find_element(By.XPATH, geo_Geographic_add_custom_button_posXpath).click()
        self.driver.find_element(By.XPATH, geo_ObjectCreate_add_super_administrative_area_text_posXpath).send_keys(data_value)
        self.driver.find_element(By.ID, geo_Geographic_add_super_administrative_area_save_button_posID).click()

    def check_error(self,data_type,search_type,data_value):
        self.driver.find_element(By.XPATH, geo_ObjectCreate_add_super_administrative_area_button_posXpath).click()
        self.driver.find_element(By.ID, geo_ObjectCreate_super_administrative_area_posID).click()
        if data_type == "empty" or data_type == "format":
            self.driver.find_element(By.XPATH, geo_Geographic_add_custom_button_posXpath).click()
            self.driver.find_element(By.XPATH, geo_ObjectCreate_add_super_administrative_area_text_posXpath).send_keys(data_value)
            self.driver.find_element(By.ID, geo_Geographic_add_super_administrative_area_save_button_posID).click()
            assert self.driver.find_element(By.XPATH,main_listPage_object_error_alter_postXpath).text == 'Cannot be empty and content cannot contain "."'
        elif data_type == "search":
            self.driver.find_element(By.ID, geo_Geographic_super_admin_search_button_posID).send_keys(data_value + Keys.ENTER)
            if search_type == "0":
                assert self.driver.find_element(By.XPATH, geo_ObjectCreate_sideslip_fist_data_posXpath).text == data_value
            else:
                assert self.driver.find_element(By.XPATH,geo_ObjectCreate_sideslip_no_data_posXpath).text == "No Data"


    def _View_statistics_case(self, data):
        self._goto_subObjectPage()
        self._create(data)
        self._search(data)
        self.driver.find_element(By.XPATH, main_listPage_object_statistics_view_postXpath).click()
        self.driver.isElementExist(Element="(//div[@class='ObjectStatisticSlider'])[1]")
        assert self.driver.Exist, "Statistics界面打开"
        self.driver.find_element(By.XPATH, "(//div[@class='ObjectStatisticSlider']//span[text()='Cancel'])[2]").click()
        # 详情界面Statistics验证
        self.driver.find_element(By.XPATH, listPage_object_ip_address_select_First_object_posXpath).click()
        self.driver.find_element(By.ID, listPage_object_geolocation_editButton_posId).click()
        self.driver.find_element(By.XPATH, "//div[@class='right-detail']//span[text()='Statistics']").click()
        self.driver.isElementExist(Element="(//div[@class='ObjectStatisticSlider'])[1]")
        assert self.driver.Exist, "Statistics界面打开"
        self._delete(data)

    @screenshot_on_failure
    def check_link_case(self, data, link_list_dict):
        # link简单link校验测试
        """
        :param data: link_list_dict=[{"link_dst_cluster": "42.49-User4Link", "link_dst_vsys":"Vsys2test", "url": "http://192.168.42.49/#/login","username": "admin","passwd": "admin"}])
        :param link_list_dict:
        :return:
        """
        self._goto_subObjectPage()
        self._create(data)
        time.sleep(3)
        self._search(data)
        time.sleep(2)
        self.first_row_checkBox_element = self.driver.find_element(By.XPATH,
                                                                   "//div[@class='el-checkbox-group']//span[@class='el-checkbox__input']").click()  # 勾选第一行非本vsys元素
        # 调用公共的link创建逻辑
        self.objects_po.linkdata(link_list_dict, geo_listpage_linkButton_posId,
                                 geo_listpage_linkSave_posXpath,
                                 geo_listpage_linkAdd_posXpath,
                                 geo_listpage_linkOk_posXpath)
        profile_public = ProfilesPublicOperations(self.driver)
        profile_public.change_vsys(vsys_name="UIAutoTestVsys")
        # 删除数据
        self._goto_subObjectPage()
        self._delete(data)  # 删除数据
        time.sleep(3)
        # 删除其它目标link数据逻辑
        self._del_linkDstData(data, link_list_dict)

    @screenshot_on_failure
    def _del_linkDstData(self, data, link_list_dict):
        loginout = LogInOut(self.driver)
        loginout.logout()
        time.sleep(1)
        for link_index in range(len(link_list_dict)):
            loginout.login_other(url=link_list_dict[link_index]["url"], username=link_list_dict[link_index]["username"],
                                 passwd=link_list_dict[link_index]["passwd"])
            profile_public = ProfilesPublicOperations(self.driver)
            profile_public.change_vsys(vsys_name=link_list_dict[link_index]["link_dst_vsys"])
            self._goto_subObjectPage()
            # self._query_link(data)
            self._delete(data)  # 删除数据
            time.sleep(3)
            # 退出账号
            loginout.logout()
            time.sleep(1)
        # 再次登录源tsg，测试退出使用
        loginout.login()

    def verify_search(self, data):
        self._create(data)
        self._search(data)
        total = self.driver.find_element(By.XPATH, "//div[@class='page-box-containcheck']//span[contains(text(),'Total')]").text
        count = total.split("Total:")
        if int(count[1]) != 1:
            raise AssertionError("搜索结果错误，请查看")
        self._delete(data)

    def _edit_built_in_objects(self):
        ip_type = random.choice(["IPv4","IPv6"])
        if ip_type == "IPv4":
            geo_ip = self.my_random.random_ipv4()
        else:
            geo_ip = self.my_random.random_ipv6()
        # 切换Vsys
        self.profile_pub.change_vsys(vsys_name="vsys0")
        self._goto_subObjectPage()
        # 点击Built-in
        self.driver.find_element(By.ID, 'IPLibraryAll1').click()
        # 勾选第一个
        self.driver.find_element(By.XPATH, asn_listPage_first_row_checkBox_posXpath).click()
        # 进入其编辑页面
        self.driver.find_element(By.ID, listPage_object_geolocation_editButton_posId).click()
        #判断不可编辑按钮存在
        self.driver.isElementExist("(//div[@class='IconBtn iconfont icon-Edit fontsize18 IPIcon disabled-btn'])[1]")
        # 新增Item
        # self._items_asn_add(geo_ipv4)
        self.driver.find_element(By.ID, geo_ObjectCreate_ip_item_add_button_posID).click()
        if ip_type == "IPv4":
            self.driver.find_element(By.XPATH, geoObjectPage_button_editItem_text_posXpath).send_keys(geo_ip)
            self.driver.find_element(By.XPATH, geo_ObjectCreate_ip_item_text_save_button_posXpath).click()
        elif ip_type == "IPv6":
            self.driver.find_element(By.ID, geo_ObjectCreate_ip_item_type_posID).click()
            self.driver.find_element(By.XPATH, geo_ObjectCreate_ip_item_type_ipv6_posXpath).click()
            self.driver.find_element(By.XPATH, geoObjectPage_button_editItem_text_posXpath).send_keys(geo_ip)
            self.driver.find_element(By.XPATH, geo_ObjectCreate_ip_item_text_save_button_posXpath).click()
        # 保存Object
        # self._object_detail_page_save()
        self.driver.find_element(By.XPATH, geo_ObjectCreate_button_OK_posXpath).click()
        self.driver.find_element(By.XPATH, geo_ObjectCreate_button_warningSaveYes_posXpath).click()
        # 校验是否添加成功
        ## 勾选第一个
        self.driver.find_element(By.XPATH, asn_listPage_first_row_checkBox_posXpath).click()
        # #进入其编辑页面
        self.driver.find_element(By.ID, listPage_object_geolocation_editButton_posId).click()
        ##搜索Item
        # self._items_asn_search(geo_ip)
        self.driver.find_element(By.ID, mainPage_GeolocationSearch_buttonSearch_Item_posId).send_keys(geo_ip + Keys.ENTER)
        self.item_check_content = self.driver.find_element(By.XPATH,geoObjectPage_button_addItem_first_item_posXpath).text
        assert self.item_check_content == geo_ip, "断言检查item文本正确"
        # 清除测试数据
        # self._items_asn_del(geo_ip)
        self.driver.find_element(By.XPATH,"//div[@class='IconBtn iconfont icon-Delete_X fontsize18 IPIcon']").click()
        # 保存Object
        # self._object_detail_page_save()
        self.driver.find_element(By.XPATH, geo_ObjectCreate_button_OK_posXpath).click()
        self.driver.find_element(By.XPATH, geo_ObjectCreate_button_warningSaveYes_posXpath).click()

    def modify_area(self,data):
        # 跳转到Geolocation页面
        self._goto_subObjectPage()
        self.loading_parameter(data)
        # 点击Create
        self.driver.find_element(By.ID, listPage_object_geolocation_createButton_posId).click()
        # 添加ip item
        self._add_item(data)
        # 输入Geolocation选项，ip
        self._input_geolocation(data)
        # 添加Sub Administrative Area
        self._input_sideslip_value(data["Sub Administrative Area"])
        # 添加Locality
        self._input_sideslip_value(data["Locality"])
        self.Confirm_creation(data)
        self._search(data)
        self.driver.find_element(By.XPATH, listPage_object_ip_address_select_First_object_posXpath).click()
        self.driver.find_element(By.ID, listPage_object_geolocation_editButton_posId).click()
        if data["Create"]["modify"]["type"] == "0":
            self.driver.find_element(By.XPATH,"//div[text()='Super Administrative Area']//parent::*//parent::*//span[@class='width80percentage long-hide float-left']").click()
        elif data["Create"]["modify"]["type"] == "1":
            self.driver.find_element(By.XPATH,"//div[text()='Administrative Area']//parent::*//parent::*//span[@class='width80percentage long-hide float-left']").click()
        elif data["Create"]["modify"]["type"] == "2":
            self.driver.find_element(By.XPATH,"//div[text()='Sub Administrative Area']//parent::*//parent::*//span[@class='width80percentage long-hide float-left']").click()
        self.driver.find_element(By.XPATH, geo_Geographic_add_custom_button_posXpath).click()
        self.driver.find_element(By.XPATH, geo_ObjectCreate_add_super_administrative_area_text_posXpath).send_keys(data["Create"]["modify"]["modify_data"])
        self.driver.find_element(By.ID, geo_Geographic_add_super_administrative_area_save_button_posID).click()
        self.Confirm_creation(data)
        self._delete(data)

    create=_create
    query = _search
    delete = _delete
if __name__ == '__main__':
    chrome_option = webdriver.ChromeOptions()
    a = MyWebDriver(
        command_executor="http://192.168.39.77:4444",
        options=chrome_option
    )
    a.get("http://192.168.44.72")
    a.maximize_window()
    a.find_element(By.XPATH, "//*[@id='app']/div/div[1]/div[2]/div[2]/div[1]/div/input").send_keys("youzhijiang")
    a.find_element(By.XPATH, "//*[@id='app']/div/div[1]/div[2]/div[2]/div[2]/div/input").send_keys("youzhijiang1")
    a.find_element(By.ID, "login").click()
    time.sleep(2)
    r = geolocation(a)
    # model=0/1 ====0:单个   1group
    # type =IMSI/Phone_Number/IMEI
    data={
        "ids": "用例1、Geolocation_Country_and_Region的增删改查",
        "Geo Name ID": "random",
        "Continent": "Autoui",
        "Country Abbreviation": "Autoui",
        "Country and Region": "Autoui",
        "Super Administrative Area": "Autoui",
        "Administrative Area": "Autoui",
        "Longitude": 0,
        "Latitude": 0,
        "Create": {
            "model": "0",
            "type": "new_object",
            "item": {
                "IPv4": ["1.1.1.1", "3.3.3.3"],
                "IPv6": ["1:1:1::1/24"]
            }
        },
        "Createtype": "Country_and_Region",
        "Subtype": "",
        "Search_type": "Geo Name ID"
    }
    r.geolocation_case(data)
    # r._sideslip_check(data)