80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import requests
|
|
import json
|
|
|
|
# 基础URL
|
|
BASE_URL = "http://localhost:5000/api"
|
|
|
|
def create_domain(name, description):
|
|
"""创建一个新域名"""
|
|
print(f"创建域名: {name}")
|
|
url = f"{BASE_URL}/domains"
|
|
data = {
|
|
"name": name,
|
|
"description": description
|
|
}
|
|
|
|
try:
|
|
response = requests.post(url, json=data)
|
|
print(f"状态码: {response.status_code}")
|
|
print(f"响应内容: {response.text}")
|
|
response.raise_for_status()
|
|
result = response.json()
|
|
print(f"域名创建成功: {result}")
|
|
return result.get("domain", {})
|
|
except Exception as e:
|
|
print(f"域名创建失败: {str(e)}")
|
|
return None
|
|
|
|
def create_mailbox(domain_id, address, description=""):
|
|
"""在指定域名下创建邮箱"""
|
|
print(f"在域 {domain_id} 下创建邮箱: {address}")
|
|
url = f"{BASE_URL}/mailboxes"
|
|
data = {
|
|
"domain_id": domain_id,
|
|
"address": address,
|
|
"description": description
|
|
}
|
|
|
|
try:
|
|
response = requests.post(url, json=data)
|
|
print(f"状态码: {response.status_code}")
|
|
print(f"响应内容: {response.text}")
|
|
response.raise_for_status()
|
|
result = response.json()
|
|
print(f"邮箱创建成功: {result}")
|
|
return result.get("mailbox", {})
|
|
except Exception as e:
|
|
print(f"邮箱创建失败: {str(e)}")
|
|
return None
|
|
|
|
def main():
|
|
# 检查API状态
|
|
try:
|
|
status_response = requests.get(f"{BASE_URL}/status")
|
|
print(f"API状态: {status_response.status_code}")
|
|
print(f"API响应: {status_response.text[:200]}...")
|
|
except Exception as e:
|
|
print(f"API状态检查失败: {str(e)}")
|
|
|
|
# 创建测试域名 nosqli.com
|
|
domain = create_domain("nosqli.com", "测试域名")
|
|
if not domain:
|
|
print("无法继续，域名创建失败")
|
|
return
|
|
|
|
# 获取域名ID
|
|
domain_id = domain.get("id")
|
|
|
|
# 创建测试邮箱 testaa@nosqli.com
|
|
mailbox = create_mailbox(domain_id, "testaa", "测试用户")
|
|
|
|
print("\n测试环境准备完成:")
|
|
print(f"- 域名: nosqli.com (ID: {domain_id})")
|
|
if mailbox:
|
|
print(f"- 邮箱: testaa@nosqli.com (ID: {mailbox.get('id')})")
|
|
|
|
if __name__ == "__main__":
|
|
main() |