93 lines
3.7 KiB
Python
93 lines
3.7 KiB
Python
|
from django.core.management.base import BaseCommand
|
|||
|
|
from django.contrib.auth import get_user_model
|
|||
|
|
from django.utils import timezone
|
|||
|
|
|
|||
|
|
from plans.models import Plan
|
|||
|
|
from domains.models import Domain
|
|||
|
|
|
|||
|
|
|
|||
|
|
User = get_user_model()
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Command(BaseCommand):
|
|||
|
|
help = "导入演示套餐并可选创建演示域名。使用 --user-id 为指定用户创建域名。"
|
|||
|
|
|
|||
|
|
def add_arguments(self, parser):
|
|||
|
|
parser.add_argument('--user-id', type=int, help='为该用户创建示例域名（可选）')
|
|||
|
|
|
|||
|
|
def handle(self, *args, **options):
|
|||
|
|
free, _ = Plan.objects.get_or_create(
|
|||
|
|
name='Free',
|
|||
|
|
defaults={
|
|||
|
|
'description': '基础免费套餐，每域名每月包含 15GB。',
|
|||
|
|
'billing_mode': 'per_domain_monthly',
|
|||
|
|
'base_price_per_domain': 0,
|
|||
|
|
'included_traffic_gb_per_domain': 15,
|
|||
|
|
'overage_price_per_gb': 0.50,
|
|||
|
|
'allow_overage': True,
|
|||
|
|
'is_active': True,
|
|||
|
|
'is_public': True,
|
|||
|
|
'features': {
|
|||
|
|
'waf_enabled': False,
|
|||
|
|
'http3_enabled': False,
|
|||
|
|
'logs_enabled': False,
|
|||
|
|
'websocket_enabled': True,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
pro, _ = Plan.objects.get_or_create(
|
|||
|
|
name='Pro',
|
|||
|
|
defaults={
|
|||
|
|
'description': '专业版，每域名每月包含 200GB，支持更多功能。',
|
|||
|
|
'billing_mode': 'per_domain_monthly',
|
|||
|
|
'base_price_per_domain': 49.00,
|
|||
|
|
'included_traffic_gb_per_domain': 200,
|
|||
|
|
'overage_price_per_gb': 0.30,
|
|||
|
|
'allow_overage': True,
|
|||
|
|
'is_active': True,
|
|||
|
|
'is_public': True,
|
|||
|
|
'features': {
|
|||
|
|
'waf_enabled': True,
|
|||
|
|
'http3_enabled': True,
|
|||
|
|
'logs_enabled': True,
|
|||
|
|
'websocket_enabled': True,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
self.stdout.write(self.style.SUCCESS(f"已准备套餐：Free({free.id}), Pro({pro.id})"))
|
|||
|
|
|
|||
|
|
user_id = options.get('user_id')
|
|||
|
|
if not user_id:
|
|||
|
|
self.stdout.write(self.style.NOTICE("未提供 --user-id，跳过示例域名创建。"))
|
|||
|
|
return
|
|||
|
|
user = User.objects.filter(id=user_id).first()
|
|||
|
|
if not user:
|
|||
|
|
self.stdout.write(self.style.ERROR(f"用户 {user_id} 不存在，跳过域名创建。"))
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
today = timezone.now().date()
|
|||
|
|
month_start = today.replace(day=1)
|
|||
|
|
# 下月第一天
|
|||
|
|
if month_start.month == 12:
|
|||
|
|
next_month_start = month_start.replace(year=month_start.year + 1, month=1, day=1)
|
|||
|
|
else:
|
|||
|
|
next_month_start = month_start.replace(month=month_start.month + 1, day=1)
|
|||
|
|
current_cycle_end = next_month_start - timezone.timedelta(days=1)
|
|||
|
|
|
|||
|
|
d, created = Domain.objects.get_or_create(
|
|||
|
|
user=user,
|
|||
|
|
name='example.com',
|
|||
|
|
defaults={
|
|||
|
|
'status': Domain.STATUS_PENDING,
|
|||
|
|
'current_plan': free,
|
|||
|
|
'current_cycle_start': month_start,
|
|||
|
|
'current_cycle_end': current_cycle_end,
|
|||
|
|
'cname_targets': {'www.example.com': 'www.cdn.example.com'},
|
|||
|
|
'origin_config': {'host': 'origin.example.com', 'protocol': 'http', 'port': 80},
|
|||
|
|
'edge_server_id': None,
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
if created:
|
|||
|
|
self.stdout.write(self.style.SUCCESS(f"已创建演示域名：{d.name}（用户ID {user.id}）"))
|
|||
|
|
else:
|
|||
|
|
self.stdout.write(self.style.WARNING(f"演示域名已存在：{d.name}（用户ID {user.id}）"))
|