1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

# coding:utf-8

import os

import subprocess

from PIL import Image

def get_captcha(driver, captcha_id, full_screen_img_path, captcha_img_path, captcha_final_path, txt_path, ocr_path):

    # 浏览器界面截图

    driver.save_screenshot(full_screen_img_path)

    # 找到验证码图片，得到它的坐标

    element = driver.find_element_by_id(captcha_id)

    left = element.location['x']

    top = element.location['y']

    right = element.location['x'] + element.size['width']

    bottom = element.location['y'] + element.size['height']

    left, top, right, bottom = int(left), int(top), int(right), int(bottom)

    img = Image.open(full_screen_img_path)

    img = img.crop((left, top, right, bottom))

    # 得到验证码图片

    img.save(captcha_img_path)

    # 打开验证码图片

    img = Image.open(captcha_img_path)

    # 颜色直方图，255种颜色，255为白色

    # 新建一张图片(大小和原图大小相同，背景颜色为255白色)

    img_new = Image.new('P', img.size, 255)

    for x in range(img.size[1]):

        for y in range(img.size[0]):

            # 遍历图片的xy坐标像素点颜色

            pix = img.getpixel((y, x))

            # print(pix)

            # 自己调色，r=0，g=0，b>0为蓝色

            if pix[0] < 20 and pix[1] < 20 and pix[2] > 50:

                # 把遍历的结果放到新图片上，0为透明度，不透明

                img_new.putpixel((y, x), 0)

    img_new.save(captcha_final_path, format='png')

    # 通过tesseract工具解析验证码图片，生成文本

    os.system(ocr_path)

    # 读取txt文件里面的验证码

    with open(txt_path, 'r') as f:

        if f.read():

            t = f.read().strip()

            # 去掉中间空格

            if ' ' in t:

                t = t.replace(' ', '')

            if t.isdigit() and len(t) == 4:

                return t

            else:

                return 'fail'

def check_resp(result, msg):

    if msg in result:

        return 'pass'

    else:

        return 'failed'

# 接口 - 识别验证码

def get_captcha(captcha_img_path, captcha_final_path, txt_path, ocr_path):

    # 打开验证码图片

    img = Image.open(captcha_img_path)

    # 新建一张图片(大小和原图大小相同，背景颜色为255白色)

    img_new = Image.new('P', img.size, 55)

    for x in range(img.size[1]):

        for y in range(img.size[0]):

            # 遍历图片的xy坐标像素点颜色

            pix = img.getpixel((y, x))

            # print(pix)

            # 自己调色，r=0，g=0，b>0为蓝色

            if pix[0] < 20 and pix[1] < 20 and pix[2] > 50:

                # 把遍历的结果放到新图片上，0为透明度，不透明

                img_new.putpixel((y, x), 0)

    img_new.save(captcha_final_path, format='png')

    # 通过tesseract工具解析验证码图片，生成文本，【Tesseract-OCR必须和jpg的根目录必须相同，如C盘、D盘！！！】

    os.system(ocr_path)

    # 读取txt文件里面的验证码

    with open(txt_path, 'r') as f:

        if r.read():

            t = f.read().strip()

            # 去掉中间空格

            if ' ' in t:

                t = t.replace(' ', '')

            # 如果是数字且长度为4，就返回数字，如果不是就返回 fail

            if t.isdigit() and len(t) == 4:

                return t

            else:

                return fail