修复
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user