#!/usr/bin/env python
# -*- coding: utf-8 -*-
'''
    @Time : 2020/8/18 14:13
    @Author : Shiji
    @Site : 
    @File : common.py
    @Software: PyCharm

'''
# -*- coding:UTF-8 -*-
from selenium import webdriver
import time,os
from PIL import Image
import tesserocr
from selenium.common.exceptions import NoSuchElementException

class common(object):
    #新建对象就自动创建浏览器并且最大化窗口
    def __init__(self):
        # 加启动配置
        option = webdriver.ChromeOptions()
        option.add_experimental_option("excludeSwitches", ['enable-automation'])# 屏蔽'CHROME正受到组件控制'的提示
        prefs = {"": ""}
        prefs["credentials_enable_service"] = False
        prefs["profile.password_manager_enabled"] = False
        option.add_experimental_option("prefs", prefs)                          # 屏蔽'保存密码'提示框
        self.driver = webdriver.Chrome(chrome_options=option,executable_path="./webdriver/chromedriver.exe")
        self.driver.implicitly_wait(10)
        self.driver.maximize_window()
    #打开网页功能
    def open(self,url):
        self.driver.get(url)
        self.driver.implicitly_wait(5)

    # 关闭网页功能
    def close(self):
        time.sleep(10)
        self.driver.close()
    # 定位元素功能
    def locateElement(self,type,value):
        if type=="id":
            el=self.driver.find_element_by_id(value)
        elif type=="name":
            el=self.driver.find_element_by_name(value)
        elif type=="class_name":
            el=self.driver.find_element_by_class_name(value)
        elif type=="tag_name":
            el=self.driver.find_element_by_tag_name(value)
        elif type=="link_text":
            el=self.driver.find_element_by_link_text(value)
        elif type=="partial_link_text":
            el=self.driver.find_element_by_partial_link_text(value)
        elif type=="xpath":
            el=self.driver.find_element_by_xpath(value)
        elif type=="css_selector":
            el=self.driver.find_element_by_css_selector(value)

        return el

    #点击元素功能
    def click(self,type,value):
        # 调用locateElement定位元素
        el=self.locateElement(type,value)
        #调用click()进行点击操作
        el.click()

    # 对定位到元素进行输入
    def input_data(self, type, value,data):
        # 调用locateElement定位元素
        el = self.locateElement(type, value)
        # 调用send_keys进行输入
        el.clear()
        el.send_keys(data)

    # 获取定位到的元素中的文本内容<a>text</a>
    def getText(self, type, value):
        # 调用locateElement定位元素
        el = self.locateElement(type, value)
        # 返回文本内容
        return el.text

    # 判断元素是否存在，返回布尔值
    def isElementPresent(self, type, value):
        try:
            element = self.locateElement(type, value)
        except NoSuchElementException as e:
            return False
        return True


    # 获取定位到的元素中的标签值
    def getAttribute(self, type, value,name):
        # 调用locateElement定位元素
        el = self.locateElement(type, value)
        # 返回文本内容
        return el.get_attribute(name)

    # 获取元素截图内容
    def get_element_code_img(self,type,locator, img_path):
        self.driver.save_screenshot(img_path)
        element = self.locateElement(type,locator)
        left = element.location['x']
        top = element.location['y']
        right = element.location['x'] + element.size['width']
        bottom = element.location['y'] + element.size['height']
        im = Image.open(img_path)
        im = im.crop((left, top, right, bottom))
        im = im.convert('L')
        threshold = 150
        table = []
        for i in range(256):
            if i < threshold:
                table.append(0)
            else:
                table.append(1)
        im = im.point(table, '1')
        result_code = tesserocr.image_to_text(im)
        if os.path.exists(img_path):  # 如果文件存在
            # 删除文件，可使用以下两种方法。
            os.remove(img_path)
            # os.unlink(path)
        else:
            print('no such file:%s' % img_path)  # 则返回文件不存在
        return result_code

    # 删除对象时自动执行的方法
    def __del__(self):
        time.sleep(5)
        self.driver.close()