pikachu.py

import base64
import os
import subprocess
import time
from difflib import SequenceMatcher

import ddddocr
import pandas as pd
import psutil
from loguru import logger as LOG
from retrying import retry
from selenium import webdriver
from selenium.common.exceptions import TimeoutException
from selenium.webdriver import ActionChains
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.wait import WebDriverWait

import uiautomation as auto


def page_has_loaded(driver):
    return driver.execute_script("return document.readyState") == "complete"


def back_code(data):
    ocr = ddddocr.DdddOcr()
    print(data)
    img_imf = data
    img_imf = img_imf.replace('data:image/png;base64,', '')

    # 将data:image/jpg;base64格式的数据转化为图片
    page_content = base64.b64decode(img_imf)
    file_path = './code.png'
    with open(file_path, 'wb') as f:
        f.write(page_content)
    with open("code.png", 'rb') as f:
        image = f.read()

    res = ocr.classification(image)
    return res


class wait_page_load:

    def __init__(self, browser, timeout=5):

        self.browser = browser

        self.timeout = timeout

    def __enter__(self):
        WebDriverWait(self.browser, 20, 0.5).until(
            EC.visibility_of_element_located((By.XPATH,
                                              '//*[@id="app"]/div/div/div/div/div/div/div[1]/div[1]/div[2]/div/div[4]/div[2]/table/tbody/tr/td[3]/div')))

        self.old_page = self.browser.find_element(By.XPATH,
                                                  '//*[@id="app"]/div/div/div/div/div/div/div[1]/div[1]/div[2]/div/div[4]/div[2]/table/tbody/tr/td[3]/div')

        LOG.info(f'Old page id is {self.old_page.id}')

    def __exit__(self, *_):

        start = time.time()

        while time.time() - start < self.timeout:

            new_page = self.browser.find_element(By.XPATH,
                                                 '//*[@id="app"]/div/div/div/div/div/div/div[1]/div[1]/div[2]/div/div[4]/div[2]/table/tbody/tr/td[3]/div')

            LOG.info(f'New page id is {new_page.id}')

            if new_page.id != self.old_page.id:

                return True

            else:

                time.sleep(0.1)
        return False


# def download_ChromeDriver():
#     autodriver.chromedriver()


def format_cookies(cookies):
    """Convert Selenium cookies to Requests compatible format"""
    req_cookies = {}
    for cookie in cookies:
        req_cookies[cookie["name"]] = cookie["value"]
    return req_cookies


def similarity(a, b):
    return SequenceMatcher(None, a, b).ratio()


def find_closest_match(d, value):
    """Return the key in a dictionary with the closest match to a certain value"""
    closest_key = None
    closest_diff = 0
    for k, v in d.items():
        diff = similarity(value, k)
        if diff > closest_diff:
            closest_diff = diff
            closest_key = k
    return closest_key


class Pikachu:
    """
        皮卡丘:selenium 自动化fsp
    """
    ChromeDriver = None
    iframe_list = []
    panel_address = {
        '普通食品': 'https://fsp.samr.gov.cn/pgctest_ui/#/jyjc/ptspjy/sjtb?isIframeOut=true&state=authorize_code&token=',
        '保健食品': 'https://fsp.samr.gov.cn/pgctest_ui/#/jyjc/bjspjy/sjtb1?isIframeOut=true&state=authorize_code&token=',
        '农产品': 'https://fsp.samr.gov.cn/pgctest_ui/#/jyjc/sxjncpjy/sjtb2?isIframeOut=true&state=authorize_code&token='
    }

    pop_address = {
        '普通食品': 'https://fsp.samr.gov.cn/pgctest_ui/#/window-fill-pusp?sampleCode={sampleNo}&collapseNo=&sdId={sdid}&title=%E6%99%AE%E9%80%9A%E9%A3%9F%E5%93%81%E6%A3%80%E9%AA%8C%E6%95%B0%E6%8D%AE%E5%A1%AB%E6%8A%A5&isEdit=true&isIframeOut=true&state=authorize_code&token={token}&username={username}',
        '保健食品': 'https://fsp.samr.gov.cn/pgctest_ui/#/window-fill-bjsp?sampleCode={sampleNo}&collapseNo=&sdId={sdid}&title=%E4%BF%9D%E5%81%A5%E9%A3%9F%E5%93%81%E6%A3%80%E9%AA%8C%E6%95%B0%E6%8D%AE%E5%A1%AB%E6%8A%A5&isEdit=true&isIframeOut=true&state=authorize_code&token={token}&username={username}',
        '农产品': 'https://fsp.samr.gov.cn/pgctest_ui/#/window-fill?sampleCode={sampleNo}&collapseNo=&sdId={sdid}&title=%E5%86%9C%E4%BA%A7%E5%93%81%E6%A3%80%E9%AA%8C%E6%95%B0%E6%8D%AE%E5%A1%AB%E6%8A%A5&isEdit=true&isIframeOut=true&state=authorize_code&token={token}&username={username}'
    }

    def __init__(self, username='3333420602', password='Aa123456!', chrome_location=None, chromedriver_path=None):
        auto.SetGlobalSearchTimeout(2)
        self.now_panelname = None
        self.rq_cookies = None
        self.username = username
        self.password = password
        print('a')
        self.readFormXlsx()
        self.opt = Options()
        self.opt.add_argument('--incognito')
        self.opt.add_argument('start-maximized')
        self.opt.add_argument('disable-gpu')

        self.opt.add_argument('-ignore-certificate-errors')  # 主要是该条
        self.opt.add_argument('-ignore-ssl-errors')
        self.opt.add_experimental_option('excludeSwitches', ['enable-automation'])
        self.chrome_clean()
        if chrome_location:
            self.opt.binary_location = chrome_location
            self.ChromeDriver = webdriver.Chrome(options=self.opt, service=Service(chromedriver_path))
        else:
            print('aaa')
        self.login()
        # self._switch_panel('普通食品')

        self.rq_cookies = format_cookies(self.ChromeDriver.get_cookies())
        print(self.rq_cookies)

    @retry(stop_max_attempt_number=5)
    def login(self, username=None, password=None):
        """
        登录
        :param username:
        :param password:
        :return: ToF
        """
        if not username:
            username = self.username
        else:
            self.username = username

        if not password:
            password = self.password
        else:
            self.password = password
        LOG.info("开始登录")
        self.ChromeDriver.get("https://fsp.samr.gov.cn/")
        self.Notice_click()
        self.wait_xpath('//*[@id="pane-0"]/div[1]/div/div/input').send_keys(username)
        self.wait_xpath('//*[@id="pane-0"]/div[2]/div/div/input').send_keys(password)
        # print(self.wait_xpath('//img[@class="login-code-img"]').get_attribute('src'))
        captcha_code = back_code(self.wait_xpath('//img[@class="login-code-img"]').get_attribute('src'))
        print(captcha_code)
        self.wait_xpath('//*[@id="pane-0"]/div[3]/div/div/div[1]/div/input').send_keys(captcha_code)

        self.wait_xpath('//*[@id="pane-0"]/div[4]/div/button').click()

        WebDriverWait(self.ChromeDriver, 20, 0.5).until(
            EC.text_to_be_present_in_element((By.XPATH, '//*[@id="tab-/dashboard"]/span'), '首页'))

        LOG.success('登录成功')

        self._to_init_pane()
        return True

    def _to_index(self):
        """
        Jump https://fsp.samr.gov.cn/#/dashboard

        :return: ToF
        """
        self.ChromeDriver.get("https://fsp.samr.gov.cn/#/dashboard")
        # self.ChromeDriver.refresh()

    def _to_init_pane(self):
        """
        初始化页面
        :return:
        """
        # 普通食品
        time.sleep(1)
        self.wait_xpath('//span[text()="国家食品安全抽样检验信息系统"]').click()
        self.wait_xpath('//span[text()="接样管理"]').click()

        # self.ChromeDriver.switch_to.frame(1)
        # self.wait_xpath('//span[text()="切换业务类型"]').click()
        #
        # self.wait_xpath('/html/body/div[2]/div/div[2]/form/div/div/div/div/input').click()
        # self.wait_xpath('//span[text()="广东省及省以下任务"]').click()
        # self.wait_xpath('//span[text()="确认"]').click()
        # self.wait_xpath('//span[text()="普通食品检验"]').click()
        # self.wait_xpath('//span[text()="普通食品数据填报"]').click()
        # self._wait_xpath('//span[text()="保健食品检验"]').click()
        # self._wait_xpath('//span[text()="保健食品数据填报"]').click()
        # self._wait_xpath('//span[text()="市县级农产品检验"]').click()
        # self._wait_xpath('//span[text()="农产品数据填报"]').click()
        # time.sleep(20)
        WebDriverWait(self.ChromeDriver, 20).until(page_has_loaded)
        time.sleep(10)
        self.get_cookies()
        LOG.success('初始化界面成功')

    def pop_panel(self, sampleNo, sdid, panelname):
        """pop_address = {
        '普通食品': 'https://fsp.samr.gov.cn/pgctest_ui/#/window-fill-pusp?sampleCode={sampleNo}&collapseNo=&sdId=7439362090928835499&title=%E6%99%AE%E9%80%9A%E9%A3%9F%E5%93%81%E6%A3%80%E9%AA%8C%E6%95%B0%E6%8D%AE%E5%A1%AB%E6%8A%A5&isEdit=true&isIframeOut=true&state=authorize_code&token={token}&username={username}',
        '保健食品': 'https://fsp.samr.gov.cn/pgctest_ui/#/jyjc/bjspjy/sjtb1?isIframeOut=true&state=authorize_code&token=',
        '农产品': 'https://fsp.samr.gov.cn/pgctest_ui/#/jyjc/sxjncpjy/sjtb2?isIframeOut=true&state=authorize_code&token='
    }"""
        url = self.pop_address[panelname].format(sampleNo=sampleNo, token=self.get_token(), sdid=sdid,
                                                 username=self.username)

        time.sleep(3)
        self.ChromeDriver.get(url)
        time.sleep(5)
        # print(url)
        # self.ChromeDriver.navigate().to(url)

    def switch_panel(self, panelname):
        """
        panel_address = {
        '普通食品': 'https://fsp.samr.gov.cn/pgctest_ui/#/jyjc/ptspjy/sjtb?isIframeOut=true&state=authorize_code&token=',
        '保健食品': 'https://fsp.samr.gov.cn/pgctest_ui/#/jyjc/bjspjy/sjtb1?isIframeOut=true&state=authorize_code&token=',
        '农产品': 'https://fsp.samr.gov.cn/pgctest_ui/#/jyjc/sxjncpjy/sjtb2?isIframeOut=true&state=authorize_code&token='
        }
        :param panelname:
        :return:
        """
        self.now_panelname = panelname

        self.ChromeDriver.get(self.panel_address[panelname] + '{}&username={}'.format(self.get_token(), self.username
                                                                                      ))
        self.wait_xpath('//span[text()=" 全部 "]', 60)
        self.rq_cookies = format_cookies(self.ChromeDriver.get_cookies())

    def _switch_panel_food(self):
        """
        切换 food pane
        :return:
        """
        self.ChromeDriver.switch_to.default_content()
        self.wait_xpath('//*[@id="tab-/gjspaqcyjyxxxt/jyjcqyd/ptspjy/sjtb1"]/span[1]').click()
        self.ChromeDriver.switch_to.frame(0)

    def _switch_panel_health(self):
        """
        切换保健品
        :return:
        """
        self.ChromeDriver.switch_to.default_content()
        self.wait_xpath('//*[@id="tab-/gjspaqcyjyxxxt/jyjcqyd/bjspjy/sjtb2"]/span[1]').click()
        self.ChromeDriver.switch_to.frame(1)

    def _switch_panel_agriculture(self):
        """
        切换到农产品
        :return:
        """
        self.ChromeDriver.switch_to.default_content()
        self.wait_xpath('//*[@id="tab-/gjspaqcyjyxxxt/jyjcqyd/sxjncpjy/sjtb3"]/span[1]').click()
        self.ChromeDriver.switch_to.frame(2)

    def _switch_panel_food_double_click(self):
        """
        切换 food pane
        :return:
        """
        self.ChromeDriver.switch_to.default_content()
        ele = self.wait_xpath('//*[@id="tab-/gjspaqcyjyxxxt/jyjcqyd/ptspjy/sjtb1"]/span[1]')
        ActionChains(self.ChromeDriver).double_click(ele).perform()

        self.ChromeDriver.switch_to.frame(0)

    def _switch_panel_health_double_click(self):
        """
        切换保健品
        :return:
        """
        self.ChromeDriver.switch_to.default_content()
        ele = self.wait_xpath('//*[@id="tab-/gjspaqcyjyxxxt/jyjcqyd/bjspjy/sjtb2"]/span[1]')
        ActionChains(self.ChromeDriver).double_click(ele).perform()

        self.ChromeDriver.switch_to.frame(1)

    def _switch_panel_agriculture_double_click(self):
        """
        切换到农产品
        :return:
        """
        self.ChromeDriver.switch_to.default_content()
        ele = self.wait_xpath('//*[@id="tab-/gjspaqcyjyxxxt/jyjcqyd/sxjncpjy/sjtb3"]/span[1]')
        ActionChains(self.ChromeDriver).double_click(ele).perform()

        self.ChromeDriver.switch_to.frame(2)

    def sendkey_sampling_number(self, key):
        ele = self.wait_xpath('//input[@placeholder="请输入抽样编号"]')
        ele.clear()
        ele.send_keys(key)

    def _sendkey_sampling_time_start(self, key):
        ele = self.wait_xpath(
            '//*[@id="app"]/div/div/div/div/div/div/div[1]/div[1]/div[1]/div/div[2]/form/div[2]/div/div/input')
        ele.clear()
        ele.send_keys(key)

    def _sendkey_sampling_time_end(self, key):
        ele = self.wait_xpath(
            '//*[@id="app"]/div/div/div/div/div/div/div[1]/div[1]/div[1]/div/div[2]/form/div[3]/div/div/input')
        ele.clear()
        ele.send_keys(key)

    def _sendkey_task_source(self, key):
        ele = self.wait_xpath('//input[@placeholder="请输入任务来源"]')
        ele.clear()
        ele.send_keys(key)

    def switch_role(self, rolename):
        self.wait_xpath('//span[text()="切换角色"]').click()
        # 主检人
        self.wait_xpath('//span[text()="{}"]'.format(rolename)).click()

    def button_click(self, btname):
        ele = self.wait_xpath('//span[text()="{}"]'.format(btname))
        # ele = self.wait_xpath("//*[contains(text(),'{}')]".format(btname))
        ActionChains(self.ChromeDriver).move_to_element(ele).click(ele).perform()

    def wait_table_refresh(self):
        while len(self.wait_xpath(
                '//*[@id="app"]/div/div/div/div/div/div/div[1]/div[1]/div[2]/div/div[4]/div[2]/table').find_elements(
            By.TAG_NAME, 'tr')) != 1:
            time.sleep(0.5)
            print('-----------------------------------------')
        with wait_page_load(self.ChromeDriver):
            LOG.info('Button will click!')
        ele = self.wait_xpath(
            '//*[@id="app"]/div/div/div/div/div/div/div[1]/div[1]/div[2]/div/div[4]/div[2]/table/tbody/tr/td[1]/div/label/span')
        ActionChains(self.ChromeDriver).move_to_element(ele).click(ele).perform()
        LOG.info('表格刷新成功')

    def readFormXlsx(self):
        pass

    def select_ukey(self, ukeyname):

        pass

    def select_ukey_single_hnca(self, ukeyname):

        ele = self.wait_xpath('//span[text()="刷新证书信息 "]', 5)
        ActionChains(self.ChromeDriver).move_to_element(ele).click(ele).perform()
        t = 0
        while True:
            if t > 10:
                raise TimeoutException
            try:
                try:
                    ele = self.wait_xpath('/html/body/div[3]/div/div[2]/div[3]/div/input', 3)
                except Exception as e:
                    ele = self.wait_xpath('/html/body/div[3]/div/div[2]/div[3]/div[1]/input', 3)
                ActionChains(self.ChromeDriver).move_to_element(ele).click(ele).perform()
                # print('//li[starts‐with(text()," CN=520112199603300023,OU=绿城农科检测技术有限公司,O={}")'.format(ukeyname))
                # print('//li[substring(text(), string-length(text()) - string-length('{}') +1) = '{}']'.format(ukeyname))
                # str_ukey = 'OU=绿城农科检测技术有限公司,O={}'.format(ukeyname)
                # str_ukey2 = 'OU={},O=绿城农科检测技术有限公司'.format(ukeyname)
                # print(str_ukey)

                ele = self.wait_xpath(
                    "/html/body/div[5]/div[1]/div[1]/ul/li",
                    3)
            except Exception as e:
                # self.RestartHuaCA()
                if t == 7:
                    self.ChromeDriver.refresh()
                    time.sleep(2)
                    self.button_click('数据提交并签章 ')
                # time.sleep(2)
                LOG.info(e)
                t += 1
                continue
            k = ele.text
            LOG.info(k)
            print(ukeyname, k)
            if ukeyname == '张棋':
                ukeyname = '张琪'
            if str(ukeyname) not in str(k):
                LOG.error('ukeyname not in')
                t += 1
                continue

            break
        # ActionChains(self.ChromeDriver).move_to_element(ele).click(ele).perform()
        # print(ele)
        ele = self.wait_xpath('//span[text()="刷新证书信息 "]')
        print('刷新证书信息', ele)
        ActionChains(self.ChromeDriver).move_to_element(ele).click(ele).perform()

        self.wait_xpath('//*[@id="UserPwd"]').send_keys('111111')

    def select_ukey_unit_hnca(self, ukeyname):

        ele = self.wait_xpath('//span[text()="刷新证书信息 "]', 5)
        ActionChains(self.ChromeDriver).move_to_element(ele).click(ele).perform()
        ele = self.wait_xpath('//input[@placeholder="请选择"]', 5)

        ActionChains(self.ChromeDriver).move_to_element(ele).click(ele).perform()
        # print('//li[starts‐with(text()," CN=520112199603300023,OU=绿城农科检测技术有限公司,O={}")'.format(ukeyname))
        # print('//li[substring(text(), string-length(text()) - string-length('{}') +1) = '{}']'.format(ukeyname))
        str_ukey = 'OU=绿城农科检测技术有限公司,O={}'.format(ukeyname)
        str_ukey2 = 'OU={},O=绿城农科检测技术有限公司'.format(ukeyname)
        print(str_ukey)

        ele = self.wait_xpath(
            "//li[contains(text(), '{}')]".format(str_ukey),
            5)

        ActionChains(self.ChromeDriver).move_to_element(ele).click(ele).perform()
        print(ele)
        ele = self.wait_xpath('//span[text()="刷新证书信息 "]')
        ActionChains(self.ChromeDriver).move_to_element(ele).click(ele).perform()

        self.wait_xpath('//*[@id="UserPwd"]').send_keys('111111')

    def select_ukey_single_bjca(self, ukeyname):
        time.sleep(2)
        ele = self.wait_xpath('//span[text()="刷新证书信息 "]')
        ActionChains(self.ChromeDriver).move_to_element(ele).click(ele).perform()
        time.sleep(1)
        ele = self.wait_xpath('/html/body/div[3]/div/div[2]/div[3]/div/input')

        ActionChains(self.ChromeDriver).move_to_element(ele).click(ele).perform()

        ele = self.wait_xpath('//li[text()=" {} "]'.format(ukeyname))
        ActionChains(self.ChromeDriver).move_to_element(ele).click(ele).perform()
        # ele = self._wait_xpath('//span[text()="刷新证书信息 "]')
        # ActionChains(self.ChromeDriver).move_to_element(ele).click(ele).perform()
        time.sleep(1)
        self.wait_xpath('//input[@id="UserPwd"]').send_keys(
            self.UKEY_CONFIG[ukeyname]['password'])

    def _error_flash_panle(self):
        self.ChromeDriver.switch_to.default_content()
        tabs = self.ChromeDriver.find_elements(By.XPATH, '//*[@role="tab"]')
        print(tabs)
        for tab in tabs:
            if tab.get_attribute('aria-selected'):
                tab_name = tab.find_element(By.TAG_NAME, 'span').text
                if tab_name == '普通食品数据填报':
                    self._switch_panel_food_double_click()
                elif tab_name == '保健食品数据填报':
                    self._switch_panel_health_double_click()
                elif tab_name == '农产品数据填报':
                    self._switch_panel_agriculture_double_click()
                print(tab_name)
                break
        time.sleep(10)

    def Notice_click(self, btname='确 定'):
        try:
            js_bottom = "window.scrollTo(0,document.body.scrollHeight)"
            self.ChromeDriver.execute_script(js_bottom)
            self.wait_xpath('//span[text()="{}"]'.format(btname), wait_time=5).click()
        except Exception as e:
            print(e)
        try:
            self.wait_xpath('//span[text()="{}"]'.format('已 读'), wait_time=5).click()
        except Exception as e:
            print(e)

    def wait_xpath(self, xpath_c, wait_time=20):
        WebDriverWait(self.ChromeDriver, wait_time, 0.5).until(EC.presence_of_element_located((By.XPATH, xpath_c)))
        WebDriverWait(self.ChromeDriver, wait_time, 0.5).until(EC.visibility_of_element_located((By.XPATH, xpath_c)))
        WebDriverWait(self.ChromeDriver, wait_time, 0.5).until(EC.element_to_be_clickable((By.XPATH, xpath_c)))
        ele = self.ChromeDriver.find_element(By.XPATH, xpath_c)
        print(ele)
        return ele

    def chrome_clean(self):

        if self.ChromeDriver:
            self.ChromeDriver.close()
            self.ChromeDriver = None
        # self.ChromeDriver.quit()
        LOG.info("开始清理chrome")
        while self.chrome_check_alive:
            os.system('taskkill /F /im chrome.exe')
            LOG.info('关闭chrome.exe')
        LOG.info("开始清理ChromeDriver")
        while self.chromedrver_check_alive:
            os.system('taskkill /F /im chromedriver.exe')
            LOG.info('关闭chromedriver.exe')

    @property
    @retry(stop_max_attempt_number=5)
    def chrome_check_alive(self):
        pl = psutil.pids()
        for pid in pl:
            if psutil.Process(pid).name() == 'chrome.exe':
                return True
        return False

    @property
    @retry(stop_max_attempt_number=5)
    def chromedrver_check_alive(self):
        pl = psutil.pids()
        for pid in pl:
            if psutil.Process(pid).name() == 'chromedriver.exe':
                return True
        return False

    def test_sendkey(self, key):
        self.sendkey_sampling_number(key)
        self._sendkey_sampling_time_start('2022-12-01')
        self._sendkey_sampling_time_end('2022-12-31')
        self._sendkey_task_source(key)

    def get_cookies(self):
        self.rq_cookies = format_cookies(self.ChromeDriver.get_cookies())

    def get_k(self, name='portal-fpp247--1.01.016-fpp_token'):
        k = find_closest_match(format_cookies(self.ChromeDriver.get_cookies()), name)
        return k

    def get_token(self):
        return self.ChromeDriver.get_cookie(self.get_k('portal-fpp247--1.01.016-fpp_token'))['value']

    def ukey_check(self, ukeyName, xpath_se):
        pass

    def RestartHuaCA(self):
        while self.huaCACT_check_alive:
            os.system('taskkill /F /im CTICAServiceClient.exe')
            LOG.info('关闭CTICAServiceClient.exe')
        time.sleep(1)
        while self.huaCA_check_alive:
            os.system('taskkill /F /im HCT.exe')
            LOG.info('关闭HCT.exe')

        time.sleep(2)

        # os.system('C:\\Users\\iron\\Desktop\\hct.bat')
        os.system('hct.bat')

    def RestartUsbrsa(self):

        while self.usbrsa_check_alive:
            os.system('taskkill /F /im usbrpa.exe')
            LOG.info('关闭usbrpa.exe')

        time.sleep(0.5)
        a = None
        # os.system('C:\\Users\\iron\\Desktop\\hct.bat')
        os.system('start1.bat')

    def stop_huaCA(self):
        while self.huaCACT_check_alive:
            os.system('taskkill /F /im CTICAServiceClient.exe')
            LOG.info('关闭CTICAServiceClient.exe')
        while self.huaCA_check_alive:
            os.system('taskkill /F /im HCT.exe')
            LOG.info('关闭HCT.exe')

    @property
    @retry(stop_max_attempt_number=5)
    def usbrsa_check_alive(self):
        pl = psutil.pids()
        for pid in pl:
            if psutil.Process(pid).name() == 'usbrpa.exe':
                return True
        return False

    @property
    @retry(stop_max_attempt_number=5)
    def huaCA_check_alive(self):
        pl = psutil.pids()
        for pid in pl:
            if psutil.Process(pid).name() == 'HCT.exe':
                return True
        return False

    @property
    @retry(stop_max_attempt_number=5)
    def huaCACT_check_alive(self):
        pl = psutil.pids()
        for pid in pl:
            if psutil.Process(pid).name() == 'CTICAServiceClient.exe':
                return True
        return False

    def clean_Ico(self):
        auto.GetRootControl()

        Pane = auto.PaneControl(Depth=1, Name="任务栏")
        v_Button = Pane.ButtonControl(Depth=2, name="通知 V 形")
        v_Button.Click()
        Pane1 = auto.PaneControl(Depth=1, Name="通知溢出", timeout=1)
        for a in Pane1.GetFirstChildControl().GetChildren():
            a.MoveCursorToMyCenter()
            # time.sleep(1)

    def switch_p(self, taskname):
        self.ChromeDriver.refresh()
        time.sleep(2)
        WebDriverWait(self.ChromeDriver, 10).until(page_has_loaded)
        self.wait_xpath('//span[text()="国家食品安全抽样检验信息系统"]').click()
        self.wait_xpath('//span[text()="接样管理"]').click()
        WebDriverWait(self.ChromeDriver, 10).until(page_has_loaded)
        time.sleep(10)
        self.ChromeDriver.switch_to.frame(self.ChromeDriver.find_element(By.TAG_NAME, 'iframe'))
        self.wait_xpath('//span[text()="切换业务类型"]').click()

        self.wait_xpath('/html/body/div[2]/div/div[2]/form/div/div/div/div/input').click()
        # ul = self.ChromeDriver.find_element(By.TAG_NAME, 'ul')
        # self.ChromeDriver.execute_script("arguments[0].scrollIntoView(true);", ul)
        # lis = ul.find_elements(By.TAG_NAME, 'li')
        # __n = len(lis)
        # __i = 0
        # while True:
        #     if __i >= __n:
        #         __i = 0
        #     try:
        #
        #         if lis[__i].is_displayed():
        #             element = lis[__i]
        #             self.ChromeDriver.execute_script("arguments[0].scrollIntoView(true);", element)
        #             self.ChromeDriver.execute_script("arguments[0].click();", element)
        #         element = self.wait_xpath('//span[text()="{}"]'.format(taskname))
        #         self.ChromeDriver.execute_script("arguments[0].scrollIntoView(true);", element)
        #         self.ChromeDriver.execute_script("arguments[0].click();", element)
        #         break
        #     except Exception as e:
        #         __i += 1
        #         print(e)
        n = 0
        while True:

            try:
                scrollTop_js = "var q = document.getElementsByClassName('el-scrollbar')[0].scrollTop=0"
                n += 10
                self.ChromeDriver.execute_script(scrollTop_js)

                element = self.wait_xpath('//span[text()="{}"]'.format(taskname))
                self.ChromeDriver.execute_script("arguments[0].scrollIntoView(true);", element)
                self.ChromeDriver.execute_script("arguments[0].click();", element)
                break
            except Exception as e:
                print(e)
                continue

        time.sleep(2)
        # self.wait_xpath('//span[text()="{}"]'.format(taskname)).click()
        self.wait_xpath('//span[text()="确认"]').click()
        time.sleep(30)


if __name__ == '__main__':
    current_directory = os.getcwd()
    LOG.info("当前运行目录:" + current_directory)
    chrome_location = current_directory + '\\chrome-win64\\chrome.exe'
    chromedriver_path = current_directory + '\\driver\\chromedriver.exe'
    p = Pikachu(username='iron520', password='Jc.x520520', chrome_location=chrome_location,
                chromedriver_path=chromedriver_path)
    # p.login()
    p._to_index()
    a = input()
    p.switch_p(a)
    # p.
    input()
    #