cv2

from selenium import webdriver
import time
import cv2 as cv
from PIL import ImageGrab
from pymouse import PyMouse

def find_img(source_path):
driver = webdriver.Firefox()
driver.get(‘http://nt2j.kbm.xeai.lanlinker.com:54010/xpp-fg-login’)
driver.find_element_by_id(‘account’).click()
driver.find_element_by_id(‘account’).send_keys(‘661’)
driver.find_element_by_id(‘password’).click()
driver.find_element_by_id(‘password’).send_keys(‘444444’)

driver.find_element_by_id('login-btn').click()  # 登陆


target_path = r'./big.png'  # 对当前屏幕截图的路径
ImageGrab.grab().save(target_path)  # 对当前屏幕截图

target = cv.imread(target_path)  # 打开屏幕截图
source = cv.imread(source_path)  # 打开模板截图   需要点击的图片
result = cv.matchTemplate(target, source, cv.TM_CCOEFF_NORMED)
pipeidu = cv.minMaxLoc(result)[1]  # 匹配度
pos_start = cv.minMaxLoc(result)[3]  # 匹配成功后的起始坐标
# 如果匹配成功，计算出中心点，以列表形式传出
if pipeidu > 0.95:
    x = int(pos_start[0]) + int(source.shape[1] / 2)
    y = int(pos_start[1]) + int(source.shape[0] / 2)
    print(x,y)
    PyMouse().click(x,y)

if name == ‘main’:
find = find_img(r’test/gctp.png’)