增强：添加批量注册功能和支持文件

2025-02-26 10:16:12 +08:00
parent 71c9653d54
commit a9e29c9cf5
11 changed files with 1392 additions and 192 deletions
--- a/test_email_by_address.py
+++ b/test_email_by_address.py
@@ -0,0 +1,116 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+import requests
+import json
+import sys
+import time
+
+def get_emails_by_address(email_address, limit=10, unread_only=False, since=None):
+    """
+    通过邮箱地址获取最新邮件
+    
+    参数:
+        email_address: 完整邮箱地址 (例如: user@example.com)
+        limit: 返回的邮件数量 (默认: 10)
+        unread_only: 是否只返回未读邮件 (默认: False)
+        since: 从指定时间戳后获取邮件 (可选)
+    
+    返回:
+        API响应的JSON数据
+    """
+    base_url = "http://localhost:5000/api/emails/by-address"
+    
+    # 构建参数
+    params = {
+        'email_address': email_address,
+        'limit': limit
+    }
+    
+    if unread_only:
+        params['unread_only'] = 'true'
+        
+    if since:
+        params['since'] = since
+    
+    # 发送请求
+    try:
+        print(f"请求 URL: {base_url}?{'&'.join([f'{k}={v}' for k, v in params.items()])}")
+        response = requests.get(base_url, params=params)
+        
+        # 打印响应信息
+        print(f"状态码: {response.status_code}")
+        
+        if response.status_code == 200:
+            data = response.json()
+            return data
+        else:
+            print(f"请求失败: {response.text}")
+            return None
+    except Exception as e:
+        print(f"请求出错: {str(e)}")
+        return None
+
+def format_email_info(email):
+    """格式化邮件信息显示"""
+    result = []
+    result.append(f"ID: {email.get('id')}")
+    result.append(f"主题: {email.get('subject')}")
+    result.append(f"发件人: {email.get('sender')}")
+    result.append(f"收件人: {email.get('recipients')}")
+    result.append(f"接收时间: {email.get('received_at')}")
+    
+    if 'verification_code' in email and email['verification_code']:
+        result.append(f"验证码: {email.get('verification_code')}")
+        
+    if 'verification_link' in email and email['verification_link']:
+        result.append(f"验证链接: {email.get('verification_link')}")
+    
+    return "\n".join(result)
+
+def main():
+    # 检查命令行参数
+    if len(sys.argv) < 2:
+        print("用法: python test_email_by_address.py <邮箱地址> [limit] [unread_only] [since]")
+        print("例如: python test_email_by_address.py test@example.com 5 true")
+        return
+    
+    # 解析参数
+    email_address = sys.argv[1]
+    limit = int(sys.argv[2]) if len(sys.argv) > 2 else 10
+    unread_only = sys.argv[3].lower() == 'true' if len(sys.argv) > 3 else False
+    since = float(sys.argv[4]) if len(sys.argv) > 4 else None
+    
+    print(f"获取邮箱 {email_address} 的最新邮件")
+    print(f"参数: limit={limit}, unread_only={unread_only}, since={since}")
+    
+    # 获取邮件
+    result = get_emails_by_address(email_address, limit, unread_only, since)
+    
+    # 显示结果
+    if result and result.get('success'):
+        print("\n===== 查询结果 =====")
+        print(f"邮箱地址: {result.get('email_address')}")
+        print(f"邮箱ID: {result.get('mailbox_id')}")
+        print(f"总邮件数: {result.get('total')}")
+        print(f"返回邮件数: {result.get('count')}")
+        
+        # 显示邮件详情
+        emails = result.get('emails', [])
+        if emails:
+            print("\n----- 邮件列表 -----")
+            for i, email in enumerate(emails, 1):
+                print(f"\n邮件 {i}:")
+                print(format_email_info(email))
+                print("-" * 40)
+        else:
+            print("\n没有找到邮件")
+            
+        # 返回时间戳，可用于下次查询
+        print(f"\n当前时间戳: {result.get('timestamp')}")
+        print("在下次查询时可以使用此时间戳作为since参数，仅获取新邮件")
+    else:
+        print("查询失败或没有结果")
+
+if __name__ == "__main__":
+    main()