pixelarraylib 1.0.0__py3-none-any.whl

arraylib/__init__.py

@@ -0,0 +1,36 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+PixelArray Python开发工具库
+
+这个库包含了常用的开发工具和服务集成：
+- aliyun: 阿里云服务集成 (STS, SMS, OSS, FC, Email, ECS, 内容扫描等)
+- db_utils: 数据库工具 (Redis, MySQL)  
+- decorators: 装饰器工具
+- gitlab: GitLab工具 (PyPI包管理)
+- monitor: 监控工具 (飞书通知)
+- net: 网络请求工具
+- system: 系统工具
+
+使用示例:
+    from arraylib.aliyun import oss
+    from arraylib.db_utils import mysql
+    from arraylib.decorators import decorators
+    from arraylib.gitlab import pypi_package_manager
+"""
+
+__version__ = "1.0.0"
+__author__ = "PixelArray"
+__email__ = "qi.lu@pixelarrayai.com"
+
+# 导出主要模块
+__all__ = [
+    'aliyun',
+    'db_utils', 
+    'decorators',
+    'gitlab',
+    'monitor',
+    'net',
+    'system', 
+]

arraylib/__main__.py

@@ -0,0 +1,126 @@
+#!/usr/bin/env python3
+"""
+ArrayLib 命令行工具入口点
+支持的命令：
+- arraylib create_test_case_files
+- arraylib summary_code_count [options]
+- arraylib collect_code_to_txt [options]
+- arraylib nginx_proxy_to_ecs [options]
+- arraylib remove_empty_lines <input_file> [output_file]
+"""
+
+import sys
+import argparse
+
+
+def main():
+    # 检查是否有子命令
+    if len(sys.argv) < 2:
+        parser = argparse.ArgumentParser(
+            description="ArrayLib 命令行工具",
+            formatter_class=argparse.RawDescriptionHelpFormatter,
+            epilog="""
+示例用法:
+  arraylib create_test_case_files --help                    # 创建测试用例文件
+  arraylib summary_code_count --help                        # 查看统计工具帮助
+  arraylib collect_code_to_txt --help                       # 查看收集工具帮助
+  arraylib nginx_proxy_to_ecs --help                        # 查看Nginx反向代理到ECS工具帮助
+  arraylib remove_empty_lines --help                        # 查看去除空行工具帮助
+            """,
+        )
+        parser.print_help()
+        sys.exit(1)
+
+    command = sys.argv[1]
+
+    if command == "create_test_case_files":
+        # 导入并执行创建测试用例文件功能
+        try:
+            from arraylib.scripts.create_test_case_files import main as create_test_main
+
+            create_test_main()
+        except ImportError as e:
+            print(f"错误：无法导入 create_test_case_files 模块: {e}")
+            sys.exit(1)
+
+    elif command == "summary_code_count":
+        # 导入并执行代码统计功能
+        try:
+            from arraylib.scripts.summary_code_count import main as summary_main
+
+            # 修改sys.argv，移除第一个参数（arraylib），让summary_code_count正确处理参数
+            original_argv = sys.argv
+            sys.argv = [original_argv[0]] + original_argv[2:]
+            summary_main()
+            sys.argv = original_argv
+        except ImportError as e:
+            print(f"错误：无法导入 summary_code_count 模块: {e}")
+            sys.exit(1)
+
+    elif command == "collect_code_to_txt":
+        # 导入并执行代码收集功能
+        try:
+            from arraylib.scripts.collect_code_to_txt import main as collect_main
+
+            # 修改sys.argv，移除第一个参数（arraylib），让collect_code_to_txt正确处理参数
+            original_argv = sys.argv
+            sys.argv = [original_argv[0]] + original_argv[2:]
+            collect_main()
+            sys.argv = original_argv
+        except ImportError as e:
+            print(f"错误：无法导入 collect_code_to_txt 模块: {e}")
+            sys.exit(1)
+
+    elif command == "nginx_proxy_to_ecs":
+        # 导入并执行Nginx反向代理到ECS功能
+        try:
+            from arraylib.scripts.nginx_proxy_to_ecs import main as nginx_proxy_to_ecs
+
+            # 修改sys.argv，移除第一个参数（arraylib），让nginx_proxy_to_ecs正确处理参数
+            original_argv = sys.argv
+            sys.argv = [original_argv[0]] + original_argv[2:]
+            nginx_proxy_to_ecs()
+            sys.argv = original_argv
+        except ImportError as e:
+            print(f"错误：无法导入 nginx_proxy_to_ecs 模块: {e}")
+            sys.exit(1)
+
+    elif command == "remove_empty_lines":
+        # 导入并执行去除空行功能
+        try:
+            from arraylib.scripts.remove_empty_lines import main as remove_empty_lines_main
+
+            # 修改sys.argv，移除第一个参数（arraylib），让remove_empty_lines正确处理参数
+            original_argv = sys.argv
+            sys.argv = [original_argv[0]] + original_argv[2:]
+            remove_empty_lines_main()
+            sys.argv = original_argv
+        except ImportError as e:
+            print(f"错误：无法导入 remove_empty_lines 模块: {e}")
+            sys.exit(1)
+
+    elif command in ["-h", "--help"]:
+        parser = argparse.ArgumentParser(
+            description="ArrayLib 命令行工具",
+            formatter_class=argparse.RawDescriptionHelpFormatter,
+            epilog="""
+示例用法:
+  arraylib create_test_case_files --help                    # 创建测试用例文件
+  arraylib summary_code_count --help                        # 查看统计工具帮助
+  arraylib collect_code_to_txt --help                       # 查看收集工具帮助
+  arraylib nginx_proxy_to_ecs --help                        # 查看Nginx反向代理到ECS工具帮助
+  arraylib remove_empty_lines --help                        # 查看去除空行工具帮助
+            """,
+        )
+        parser.print_help()
+    else:
+        print(f"错误：未知命令 '{command}'")
+        print(
+            "可用命令：create_test_case_files, summary_code_count, collect_code_to_txt, nginx_proxy_to_ecs, remove_empty_lines"
+        )
+        print("使用 'arraylib --help' 查看帮助")
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()

arraylib/aliyun/aliyun_email.py

@@ -0,0 +1,130 @@
+import os
+import time
+import random
+import traceback
+from alibabacloud_dm20151123.client import Client as Dm20151123Client
+from alibabacloud_tea_openapi import models as open_api_models
+from alibabacloud_dm20151123 import models as dm_20151123_models
+from alibabacloud_tea_util import models as util_models
+from arraylib.monitor.feishu import Feishu
+
+feishu_alert = Feishu("devtoolkit服务报警")
+
+
+class AliyunEmailSender:
+    def __init__(self, access_key_id, access_key_secret):
+        self.client = Dm20151123Client(
+            open_api_models.Config(
+                access_key_id=access_key_id,
+                access_key_secret=access_key_secret,
+                endpoint="dm.aliyuncs.com",
+            )
+        )
+
+    def generate_verification_code(self, length=6):
+        """
+        description:
+            生成数字验证码
+        parameters:
+            length(int): 验证码长度
+        return:
+            str: 验证码
+        """
+        return "".join(str(random.randint(0, 9)) for _ in range(length))
+
+    def create_verification_email_content(
+        self, username, verification_code, email_type, validity=15
+    ):
+        """
+        description:
+            创建邮件HTML内容
+        parameters:
+            username(str): 用户名
+            validity(int): 验证码有效期
+        return:
+            str: 邮件HTML内容
+        """
+        email_type_str = "注册" if email_type == "register" else "找回密码"
+        return f"""
+        <html>
+        <head>
+            <style>
+                body {{ font-family: Arial, sans-serif; line-height: 1.6; }}
+                .container {{ max-width: 600px; margin: 0 auto; padding: 20px; border: 1px solid #e1e8f7; border-radius: 10px; }}
+                .header {{ background: linear-gradient(to right, #1a2980, #26d0ce); color: white; padding: 15px; text-align: center; border-radius: 8px; }}
+                .code {{ font-size: 28px; font-weight: bold; text-align: center; margin: 25px 0; letter-spacing: 5px; color: #e74c3c; }}
+                .footer {{ margin-top: 30px; text-align: center; color: #777; font-size: 12px; }}
+            </style>
+        </head>
+        <body>
+            <div class="container">
+                <div class="header">
+                    <h2>北京矩阵像素科技有限公司</h2>
+                </div>
+                
+                <p>尊敬的 <strong>{username}</strong>：</p>
+                <p>您正在进行邮箱{email_type_str}验证，您的验证码为：</p>
+                <div class="code">{verification_code}</div>
+                <p>该验证码 <strong>{validity}分钟</strong> 内有效，请尽快完成验证。</p>
+                <p>如非本人操作，请忽略此邮件。</p>
+                
+                <div class="footer">    
+                    <p>阿里云邮件推送服务 | 安全验证</p>
+                    <p>© {time.strftime('%Y')} 企业注册系统 版权所有</p>
+                </div>
+            </div>
+        </body>
+        </html>
+        """
+
+    def get_email_subject(self, email_type):
+        if email_type == "register":
+            return "您的注册验证码"
+        elif email_type == "forgot":
+            return "您的找回密码验证码"
+        else:
+            return "您的验证码"
+
+    def send_verification_email(
+        self, to_address, verification_code, email_type, username="用户"
+    ):
+        """
+        description:
+            发送验证邮件
+        parameters:
+            to_address(str): 收件人邮箱地址
+            email_type(str): 邮件类型
+            username(str): 用户名
+        return:
+            dict: 发送结果
+        """
+        assert email_type in [
+            "register",
+            "forgot",
+        ], "邮件类型错误，当前只有register和forgot两种类型"
+        try:
+            response = self.client.single_send_mail_with_options(
+                dm_20151123_models.SingleSendMailRequest(
+                    account_name="captcha_new@captcha.pixelarrayai.com",
+                    address_type=1,
+                    to_address=to_address,
+                    subject=self.get_email_subject(email_type),
+                    html_body=self.create_verification_email_content(
+                        username=username,
+                        verification_code=verification_code,
+                        email_type=email_type,
+                    ),
+                    reply_to_address=False,
+                    from_alias="系统验证",  # 发信人昵称
+                ),
+                util_models.RuntimeOptions(),
+            )
+            if response.status_code == 200:
+                return True
+            else:
+                feishu_alert.send(f"发送验证邮件失败: {response}")
+                return False
+
+        except Exception as e:
+            feishu_alert.send(traceback.format_exc())
+            return False