easy-captcha-python 0.1.0__py3-none-any.whl

easy_captcha/__init__.py

@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+"""
+EasyCaptcha - Python图形验证码生成库
+支持GIF、中文、算术等类型
+"""
+
+__version__ = "1.0.0"
+
+from .captcha.spec_captcha import SpecCaptcha
+from .captcha.gif_captcha import GifCaptcha
+from .captcha.chinese_captcha import ChineseCaptcha
+from .captcha.chinese_gif_captcha import ChineseGifCaptcha
+from .captcha.arithmetic_captcha import ArithmeticCaptcha
+from .constants import (
+    TYPE_DEFAULT,
+    TYPE_ONLY_NUMBER,
+    TYPE_ONLY_CHAR,
+    TYPE_ONLY_UPPER,
+    TYPE_ONLY_LOWER,
+    TYPE_NUM_AND_UPPER,
+    FONT_1, FONT_2, FONT_3, FONT_4, FONT_5,
+    FONT_6, FONT_7, FONT_8, FONT_9, FONT_10
+)
+
+__all__ = [
+    'SpecCaptcha',
+    'GifCaptcha',
+    'ChineseCaptcha',
+    'ChineseGifCaptcha',
+    'ArithmeticCaptcha',
+    'TYPE_DEFAULT',
+    'TYPE_ONLY_NUMBER',
+    'TYPE_ONLY_CHAR',
+    'TYPE_ONLY_UPPER',
+    'TYPE_ONLY_LOWER',
+    'TYPE_NUM_AND_UPPER',
+    'FONT_1', 'FONT_2', 'FONT_3', 'FONT_4', 'FONT_5',
+    'FONT_6', 'FONT_7', 'FONT_8', 'FONT_9', 'FONT_10'
+]
+

easy_captcha/base/__init__.py

@@ -0,0 +1,17 @@
+# -*- coding: utf-8 -*-
+"""
+基础类模块
+"""
+
+from .randoms import Randoms
+from .captcha import Captcha
+from .arithmetic_captcha_abstract import ArithmeticCaptchaAbstract
+from .chinese_captcha_abstract import ChineseCaptchaAbstract
+
+__all__ = [
+    'Randoms',
+    'Captcha',
+    'ArithmeticCaptchaAbstract',
+    'ChineseCaptchaAbstract'
+]
+

easy_captcha/base/arithmetic_captcha_abstract.py

@@ -0,0 +1,79 @@
+# -*- coding: utf-8 -*-
+"""
+算术验证码抽象类
+"""
+
+from .captcha import Captcha
+
+
+class ArithmeticCaptchaAbstract(Captcha):
+    """算术验证码抽象基类"""
+    
+    def __init__(self):
+        super().__init__()
+        self._len = 2  # 算术验证码默认2位数运算
+        self._arithmetic_string = None  # 算术公式字符串
+    
+    def _alphas(self):
+        """
+        生成随机算术验证码
+        
+        Returns:
+            list: 验证码字符列表（这里是计算结果）
+        """
+        formula_parts = []
+        
+        # 生成算术表达式
+        for i in range(self._len):
+            # 生成0-9的随机数字
+            num = self.num(10)
+            formula_parts.append(str(num))
+            
+            # 如果不是最后一个数字，添加运算符
+            if i < self._len - 1:
+                operator_type = self.num(1, 4)  # 1: +, 2: -, 3: x
+                if operator_type == 1:
+                    formula_parts.append('+')
+                elif operator_type == 2:
+                    formula_parts.append('-')
+                else:  # operator_type == 3
+                    formula_parts.append('x')
+        
+        # 构建公式字符串
+        formula = ''.join(formula_parts)
+        
+        # 计算结果
+        try:
+            # 将 'x' 替换为 '*' 进行计算
+            result = eval(formula.replace('x', '*'))
+            # 确保结果为整数
+            result = int(result)
+            self._chars = str(result)
+        except:
+            # 如果计算失败，默认为0
+            self._chars = '0'
+        
+        # 保存公式字符串（显示用）
+        self._arithmetic_string = formula + '=?'
+        
+        return list(self._chars)
+    
+    def get_arithmetic_string(self):
+        """
+        获取算术公式字符串
+        
+        Returns:
+            str: 算术公式，如 "3+2=?"
+        """
+        self.check_alpha()
+        return self._arithmetic_string
+    
+    def set_arithmetic_string(self, arithmetic_string):
+        """
+        设置算术公式字符串
+        
+        Args:
+            arithmetic_string: 算术公式字符串
+        """
+        self._arithmetic_string = arithmetic_string
+

easy_captcha/base/captcha.py

@@ -0,0 +1,337 @@
+# -*- coding: utf-8 -*-
+"""
+验证码抽象基类
+"""
+
+import base64
+import os
+from abc import ABC, abstractmethod
+from io import BytesIO
+from typing import Optional, Tuple
+
+from PIL import Image, ImageDraw, ImageFont
+
+from .randoms import Randoms
+from ..constants import *
+
+
+class Captcha(Randoms, ABC):
+    """验证码抽象基类"""
+    
+    def __init__(self):
+        self._font = None
+        self._len = 5  # 验证码字符长度
+        self._width = 130  # 验证码宽度
+        self._height = 48  # 验证码高度
+        self._char_type = TYPE_DEFAULT  # 验证码字符类型
+        self._chars = None  # 当前验证码文本
+        self._font_size = 32  # 字体大小
+    
+    def _alphas(self):
+        """
+        生成随机验证码字符
+        
+        Returns:
+            list: 验证码字符列表
+        """
+        chars = []
+        for i in range(self._len):
+            if self._char_type == TYPE_ONLY_NUMBER:
+                # 纯数字
+                chars.append(self.alpha(self.NUM_MAX_INDEX))
+            elif self._char_type == TYPE_ONLY_CHAR:
+                # 纯字母
+                chars.append(self.alpha(self.CHAR_MIN_INDEX, self.CHAR_MAX_INDEX))
+            elif self._char_type == TYPE_ONLY_UPPER:
+                # 纯大写字母
+                chars.append(self.alpha(self.UPPER_MIN_INDEX, self.UPPER_MAX_INDEX))
+            elif self._char_type == TYPE_ONLY_LOWER:
+                # 纯小写字母
+                chars.append(self.alpha(self.LOWER_MIN_INDEX, self.LOWER_MAX_INDEX))
+            elif self._char_type == TYPE_NUM_AND_UPPER:
+                # 数字和大写字母
+                chars.append(self.alpha(self.UPPER_MAX_INDEX))
+            else:
+                # 默认：数字和字母混合
+                chars.append(self.alpha())
+        
+        self._chars = ''.join(chars)
+        return chars
+    
+    def _color(self, fc=None, bc=None):
+        """
+        生成随机颜色
+        
+        Args:
+            fc: 前景色最小值 (0-255)
+            bc: 背景色最大值 (0-255)
+            
+        Returns:
+            tuple: RGB颜色元组
+        """
+        if fc is not None and bc is not None:
+            if fc > 255:
+                fc = 255
+            if bc > 255:
+                bc = 255
+            r = fc + self.num(bc - fc)
+            g = fc + self.num(bc - fc)
+            b = fc + self.num(bc - fc)
+            return (r, g, b)
+        else:
+            # 返回预定义的常用颜色
+            return COLORS[self.num(len(COLORS))]
+    
+    def check_alpha(self):
+        """检查验证码是否已生成，如果没有则生成"""
+        if self._chars is None:
+            self._alphas()
+    
+    def text(self):
+        """
+        获取验证码文本
+        
+        Returns:
+            str: 验证码文本
+        """
+        self.check_alpha()
+        return self._chars
+    
+    def text_char(self):
+        """
+        获取验证码字符列表
+        
+        Returns:
+            list: 验证码字符列表
+        """
+        self.check_alpha()
+        return list(self._chars)
+    
+    @abstractmethod
+    def out(self, stream: BytesIO) -> bool:
+        """
+        输出验证码到流
+        
+        Args:
+            stream: 输出流
+            
+        Returns:
+            bool: 是否成功
+        """
+        pass
+    
+    @abstractmethod
+    def to_base64(self, prefix: str = "") -> str:
+        """
+        输出base64编码
+        
+        Args:
+            prefix: base64前缀
+            
+        Returns:
+            str: base64编码字符串
+        """
+        pass
+    
+    def _to_base64_impl(self, prefix: str = "") -> str:
+        """
+        base64编码实现
+
+        Args:
+            prefix: base64前缀
+
+        Returns:
+            str: base64编码字符串
+        """
+        stream = BytesIO()
+        self.out(stream)
+        b64_data = base64.b64encode(stream.getvalue()).decode('utf-8')
+        return prefix + b64_data
+
+    def draw_line(self, num: int, draw: ImageDraw.Draw, color: Optional[Tuple[int, int, int]] = None):
+        """
+        绘制干扰线
+
+        Args:
+            num: 线条数量
+            draw: ImageDraw对象
+            color: 线条颜色，None则随机
+        """
+        for i in range(num):
+            line_color = color if color else self._color()
+            x1 = self.num(-10, self._width - 10)
+            y1 = self.num(5, self._height - 5)
+            x2 = self.num(10, self._width + 10)
+            y2 = self.num(2, self._height - 2)
+            draw.line([(x1, y1), (x2, y2)], fill=line_color, width=1)
+
+    def draw_oval(self, num: int, draw: ImageDraw.Draw, color: Optional[Tuple[int, int, int]] = None):
+        """
+        绘制干扰圆
+
+        Args:
+            num: 圆圈数量
+            draw: ImageDraw对象
+            color: 圆圈颜色，None则随机
+        """
+        for i in range(num):
+            oval_color = color if color else self._color()
+            w = 5 + self.num(10)
+            x = self.num(self._width - 25)
+            y = self.num(self._height - 15)
+            draw.ellipse([x, y, x + w, y + w], outline=oval_color)
+
+    def draw_bezier_curve(self, num: int, draw: ImageDraw.Draw, color: Optional[Tuple[int, int, int]] = None):
+        """
+        绘制贝塞尔曲线
+
+        Args:
+            num: 曲线数量
+            draw: ImageDraw对象
+            color: 曲线颜色，None则随机
+        """
+        for i in range(num):
+            curve_color = color if color else self._color()
+
+            # 起点和终点
+            x1 = 5
+            y1 = self.num(5, self._height // 2)
+            x2 = self._width - 5
+            y2 = self.num(self._height // 2, self._height - 5)
+
+            # 控制点
+            ctrlx = self.num(self._width // 4, self._width * 3 // 4)
+            ctrly = self.num(5, self._height - 5)
+
+            # 随机交换起点和终点的y坐标
+            if self.num(2) == 0:
+                y1, y2 = y2, y1
+
+            # 随机选择二阶或三阶贝塞尔曲线
+            if self.num(2) == 0:
+                # 二阶贝塞尔曲线
+                points = self._quadratic_bezier_points((x1, y1), (ctrlx, ctrly), (x2, y2), 50)
+            else:
+                # 三阶贝塞尔曲线
+                ctrlx1 = self.num(self._width // 4, self._width * 3 // 4)
+                ctrly1 = self.num(5, self._height - 5)
+                points = self._cubic_bezier_points((x1, y1), (ctrlx, ctrly), (ctrlx1, ctrly1), (x2, y2), 50)
+
+            draw.line(points, fill=curve_color, width=2)
+
+    def _quadratic_bezier_points(self, p0, p1, p2, num_points):
+        """
+        计算二阶贝塞尔曲线上的点
+
+        Args:
+            p0: 起点 (x, y)
+            p1: 控制点 (x, y)
+            p2: 终点 (x, y)
+            num_points: 点的数量
+
+        Returns:
+            list: 点的列表
+        """
+        points = []
+        for i in range(num_points + 1):
+            t = i / num_points
+            x = (1 - t) ** 2 * p0[0] + 2 * (1 - t) * t * p1[0] + t ** 2 * p2[0]
+            y = (1 - t) ** 2 * p0[1] + 2 * (1 - t) * t * p1[1] + t ** 2 * p2[1]
+            points.append((x, y))
+        return points
+
+    def _cubic_bezier_points(self, p0, p1, p2, p3, num_points):
+        """
+        计算三阶贝塞尔曲线上的点
+
+        Args:
+            p0: 起点 (x, y)
+            p1: 控制点1 (x, y)
+            p2: 控制点2 (x, y)
+            p3: 终点 (x, y)
+            num_points: 点的数量
+
+        Returns:
+            list: 点的列表
+        """
+        points = []
+        for i in range(num_points + 1):
+            t = i / num_points
+            x = (1 - t) ** 3 * p0[0] + 3 * (1 - t) ** 2 * t * p1[0] + 3 * (1 - t) * t ** 2 * p2[0] + t ** 3 * p3[0]
+            y = (1 - t) ** 3 * p0[1] + 3 * (1 - t) ** 2 * t * p1[1] + 3 * (1 - t) * t ** 2 * p2[1] + t ** 3 * p3[1]
+            points.append((x, y))
+        return points
+
+    def get_font(self):
+        """
+        获取字体
+
+        Returns:
+            ImageFont: 字体对象
+        """
+        if self._font is None:
+            self.set_font(FONT_1)
+        return self._font
+
+    def set_font(self, font, size: int = 32):
+        """
+        设置字体
+
+        Args:
+            font: 字体索引(0-9)或字体文件路径
+            size: 字体大小
+        """
+        self._font_size = size
+
+        if isinstance(font, int):
+            # 使用内置字体
+            font_path = os.path.join(
+                os.path.dirname(os.path.dirname(__file__)),
+                'fonts',
+                FONT_NAMES[font]
+            )
+            try:
+                self._font = ImageFont.truetype(font_path, size)
+            except:
+                # 如果加载失败，使用默认字体
+                self._font = ImageFont.load_default()
+        else:
+            # 使用自定义字体文件
+            try:
+                self._font = ImageFont.truetype(font, size)
+            except:
+                self._font = ImageFont.load_default()
+
+    # Getter和Setter方法
+    @property
+    def len(self):
+        return self._len
+
+    @len.setter
+    def len(self, value):
+        self._len = value
+
+    @property
+    def width(self):
+        return self._width
+
+    @width.setter
+    def width(self, value):
+        self._width = value
+
+    @property
+    def height(self):
+        return self._height
+
+    @height.setter
+    def height(self, value):
+        self._height = value
+
+    @property
+    def char_type(self):
+        return self._char_type
+
+    @char_type.setter
+    def char_type(self, value):
+        self._char_type = value
+

easy_captcha/base/chinese_captcha_abstract.py

@@ -0,0 +1,81 @@
+# -*- coding: utf-8 -*-
+"""
+中文验证码抽象类
+"""
+
+from .captcha import Captcha
+
+
+class ChineseCaptchaAbstract(Captcha):
+    """中文验证码抽象基类"""
+    
+    # 常用汉字字符集
+    CHINESE_CHARS = (
+        "的一是在不了有和人这中大为上个国我以要他时来用们生到作地于出就分对成会可主发年动同工也能下过子说产种面而方后多定行学法所民得经十三之进着等部度家电力里如水化高自二理起小物现实加量都两体制机当使点从业本去把性好应开它合还因由其些然前外天政四日那社义事平形相全表间样与关各重新线内数正心反你明看原又么利比或但质气第向道命此变条只没结解问意建月公无系军很情者最立代想已通并提直题党程展五果料象员革位入常文总次品式活设及管特件长求老头基资边流路级少图山统接知较将组见计别她手角期根论运农指几九区强放决西被干做必战先回则任取据处队南给色光门即保治北造百规热领七海口东导器压志世金增争济阶油思术极交受联什认六共权收证改清己美再采转更单风切打白教速花带安场身车例真务具万每目至达走积示议声报斗完类八离华名确才科张信马节话米整空元况今集温传土许步群广石记需段研界拉林律叫且究观越织装影算低持音众书布复容儿须际商非验连断深难近矿千周委素技备半办青省列习响约支般史感劳便团往酸历市克何除消构府称太准精值号率族维划选标写存候毛亲快效斯院查江型眼王按格养易置派层片始却专状育厂京识适属圆包火住调满县局照参红细引听该铁价严"
+    )
+    
+    def __init__(self):
+        super().__init__()
+        self._len = 4  # 中文验证码默认4个字
+        # 中文验证码使用系统字体
+        self.set_font_for_chinese()
+    
+    def set_font_for_chinese(self):
+        """设置中文字体"""
+        import platform
+        
+        # 根据操作系统选择合适的中文字体
+        system = platform.system()
+        
+        try:
+            if system == 'Windows':
+                # Windows系统使用微软雅黑或宋体
+                try:
+                    from PIL import ImageFont
+                    self._font = ImageFont.truetype("msyh.ttc", 28)  # 微软雅黑
+                except:
+                    try:
+                        self._font = ImageFont.truetype("simsun.ttc", 28)  # 宋体
+                    except:
+                        self._font = ImageFont.truetype("C:/Windows/Fonts/simhei.ttf", 28)  # 黑体
+            elif system == 'Darwin':  # macOS
+                from PIL import ImageFont
+                self._font = ImageFont.truetype("/System/Library/Fonts/PingFang.ttc", 28)
+            else:  # Linux
+                from PIL import ImageFont
+                # 尝试常见的Linux中文字体
+                try:
+                    self._font = ImageFont.truetype("/usr/share/fonts/truetype/wqy/wqy-microhei.ttc", 28)
+                except:
+                    try:
+                        self._font = ImageFont.truetype("/usr/share/fonts/truetype/droid/DroidSansFallbackFull.ttf", 28)
+                    except:
+                        self._font = ImageFont.load_default()
+        except:
+            from PIL import ImageFont
+            self._font = ImageFont.load_default()
+    
+    def _alphas(self):
+        """
+        生成随机中文验证码
+        
+        Returns:
+            list: 验证码字符列表
+        """
+        chars = []
+        for i in range(self._len):
+            chars.append(self._alpha_han())
+        
+        self._chars = ''.join(chars)
+        return chars
+    
+    @classmethod
+    def _alpha_han(cls):
+        """
+        返回随机汉字
+        
+        Returns:
+            str: 随机汉字
+        """
+        return cls.CHINESE_CHARS[cls.num(len(cls.CHINESE_CHARS))]
+

easy_captcha/base/randoms.py

@@ -0,0 +1,77 @@
+# -*- coding: utf-8 -*-
+"""
+随机数工具类
+"""
+
+import secrets
+
+
+class Randoms:
+    """随机数生成工具类"""
+    
+    # 验证码字符集，去除了0、O、I、L等容易混淆的字母
+    ALPHA = [
+        '2', '3', '4', '5', '6', '7', '8', '9',
+        'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'J', 'K', 'M', 'N', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z',
+        'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'j', 'k', 'm', 'n', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z'
+    ]
+    
+    # 数字的最大索引（不包括）
+    NUM_MAX_INDEX = 8
+    # 字符的最小索引（包括）
+    CHAR_MIN_INDEX = NUM_MAX_INDEX
+    # 字符的最大索引（不包括）
+    CHAR_MAX_INDEX = len(ALPHA)
+    # 大写字符最小索引
+    UPPER_MIN_INDEX = CHAR_MIN_INDEX
+    # 大写字符最大索引
+    UPPER_MAX_INDEX = UPPER_MIN_INDEX + 23
+    # 小写字母最小索引
+    LOWER_MIN_INDEX = UPPER_MAX_INDEX
+    # 小写字母最大索引
+    LOWER_MAX_INDEX = CHAR_MAX_INDEX
+    
+    @staticmethod
+    def num(min_val=None, max_val=None):
+        """
+        生成随机数
+        
+        Args:
+            min_val: 最小值（包括），如果为None则从0开始
+            max_val: 最大值（不包括），如果为None则min_val作为最大值
+            
+        Returns:
+            int: 随机数
+        """
+        if min_val is None and max_val is None:
+            raise ValueError("至少需要提供一个参数")
+        
+        if max_val is None:
+            # 只提供一个参数，生成0到min_val之间的随机数
+            return secrets.randbelow(min_val)
+        else:
+            # 提供两个参数，生成min_val到max_val之间的随机数
+            return min_val + secrets.randbelow(max_val - min_val)
+    
+    @classmethod
+    def alpha(cls, min_index=None, max_index=None):
+        """
+        返回ALPHA中的随机字符
+        
+        Args:
+            min_index: 最小索引（包括）
+            max_index: 最大索引（不包括）
+            
+        Returns:
+            str: 随机字符
+        """
+        if min_index is None and max_index is None:
+            # 返回所有字符中的随机一个
+            return cls.ALPHA[cls.num(len(cls.ALPHA))]
+        elif max_index is None:
+            # 返回0到min_index之间的随机字符
+            return cls.ALPHA[cls.num(min_index)]
+        else:
+            # 返回min_index到max_index之间的随机字符
+            return cls.ALPHA[cls.num(min_index, max_index)]
+

easy_captcha/captcha/__init__.py

@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+"""
+验证码实现模块
+"""
+
+from .spec_captcha import SpecCaptcha
+from .gif_captcha import GifCaptcha
+from .chinese_captcha import ChineseCaptcha
+from .chinese_gif_captcha import ChineseGifCaptcha
+from .arithmetic_captcha import ArithmeticCaptcha
+
+__all__ = [
+    'SpecCaptcha',
+    'GifCaptcha',
+    'ChineseCaptcha',
+    'ChineseGifCaptcha',
+    'ArithmeticCaptcha'
+]
+