This commit is contained in:
huangzhenpc
2025-02-26 10:21:48 +08:00
parent a9e29c9cf5
commit 5cc138b69a
3 changed files with 38 additions and 25 deletions

View File

@@ -91,7 +91,7 @@ def register_account(client, domain, username=None):
'message': '未收到验证邮件'
}
def batch_register(domain, count, concurrent=2):
def batch_register(domain, count, concurrent=2, api_url=None):
"""
批量注册账号
@@ -99,6 +99,7 @@ def batch_register(domain, count, concurrent=2):
domain: 邮箱域名
count: 注册数量
concurrent: 并发数
api_url: API基础URL默认为None使用客户端默认值
返回:
注册结果列表
@@ -106,7 +107,7 @@ def batch_register(domain, count, concurrent=2):
logger.info(f"开始批量注册 {count} 个账号,域名: {domain},并发数: {concurrent}")
# 创建API客户端
client = EmailApiClient()
client = EmailApiClient(api_base_url=api_url) if api_url else EmailApiClient()
# 检查系统状态
status = client.check_system_status()
@@ -178,11 +179,12 @@ def main():
parser.add_argument('--count', type=int, default=5, help='注册数量')
parser.add_argument('--concurrent', type=int, default=2, help='并发数')
parser.add_argument('--output', type=str, default='registration_results.json', help='结果输出文件')
parser.add_argument('--api-url', type=str, help='邮件系统API基础URL默认为http://74.48.75.19:5000/api')
args = parser.parse_args()
# 执行批量注册
results = batch_register(args.domain, args.count, args.concurrent)
results = batch_register(args.domain, args.count, args.concurrent, args.api_url)
# 保存结果
if results: