83 lines
3.6 KiB
Python
83 lines
3.6 KiB
Python
|
from django.core.management.base import BaseCommand
|
|||
|
|
from django.utils import timezone
|
|||
|
|
from django.db import transaction
|
|||
|
|
|
|||
|
|
from billing.models import Invoice
|
|||
|
|
from core.models import SystemSettings, OperationLog
|
|||
|
|
from core.goedge_client import GoEdgeClient
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Command(BaseCommand):
|
|||
|
|
help = '对逾期未支付账单执行未支付策略（停服或限速）。支持 dry-run 与用户/时间过滤。'
|
|||
|
|
|
|||
|
|
def add_arguments(self, parser):
|
|||
|
|
parser.add_argument('--days-overdue', type=int, default=0, help='逾期天数阈值（默认0，表示只要未支付即处理）')
|
|||
|
|
parser.add_argument('--user-id', type=int, default=0, help='仅处理指定用户（可选）')
|
|||
|
|
parser.add_argument('--dry-run', action='store_true', default=False, help='试运行，不落库与不调用接口')
|
|||
|
|
|
|||
|
|
@transaction.atomic
|
|||
|
|
def handle(self, *args, **options):
|
|||
|
|
days_overdue = int(options.get('days_overdue') or 0)
|
|||
|
|
user_id = int(options.get('user_id') or 0)
|
|||
|
|
dry = bool(options.get('dry_run'))
|
|||
|
|
|
|||
|
|
today = timezone.now().date()
|
|||
|
|
qs = Invoice.objects.filter(status=Invoice.STATUS_UNPAID)
|
|||
|
|
if user_id:
|
|||
|
|
qs = qs.filter(user_id=user_id)
|
|||
|
|
if days_overdue > 0:
|
|||
|
|
# 以账单周期结束日为逾期基准
|
|||
|
|
cutoff = today - timezone.timedelta(days=days_overdue)
|
|||
|
|
qs = qs.filter(period_end__lte=cutoff)
|
|||
|
|
|
|||
|
|
invoices = list(qs)
|
|||
|
|
if not invoices:
|
|||
|
|
self.stdout.write(self.style.NOTICE('无符合条件的未支付账单'))
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
sys = SystemSettings.objects.order_by('id').first()
|
|||
|
|
policy = getattr(sys, 'default_overage_policy', {}) or {}
|
|||
|
|
action_type = (policy.get('action') or '').lower()
|
|||
|
|
limit_bps = int(policy.get('limit_bps') or 0)
|
|||
|
|
|
|||
|
|
client = None
|
|||
|
|
if not dry:
|
|||
|
|
client = GoEdgeClient()
|
|||
|
|
|
|||
|
|
total_domains = 0
|
|||
|
|
for inv in invoices:
|
|||
|
|
applied = []
|
|||
|
|
items = inv.items.select_related('domain').all()
|
|||
|
|
for it in items:
|
|||
|
|
d = it.domain
|
|||
|
|
if not d or not d.edge_server_id:
|
|||
|
|
continue
|
|||
|
|
total_domains += 1
|
|||
|
|
if dry:
|
|||
|
|
applied.append({'domain': d.name, 'action': action_type or 'shutdown', 'limit_bps': limit_bps})
|
|||
|
|
continue
|
|||
|
|
web_id = client.find_server_web_id(int(d.edge_server_id))
|
|||
|
|
if not web_id:
|
|||
|
|
continue
|
|||
|
|
if action_type == 'limit' and limit_bps > 0:
|
|||
|
|
client.update_http_web_request_limit(web_id, {'isOn': True, 'rateBytes': int(limit_bps)})
|
|||
|
|
applied.append({'domain': d.name, 'action': 'limit', 'rateBytes': limit_bps})
|
|||
|
|
else:
|
|||
|
|
client.update_http_web_shutdown(web_id, {'isOn': True})
|
|||
|
|
if d.status != 'suspended':
|
|||
|
|
d.status = 'suspended'
|
|||
|
|
d.save(update_fields=['status', 'updated_at'])
|
|||
|
|
applied.append({'domain': d.name, 'action': 'shutdown'})
|
|||
|
|
if applied:
|
|||
|
|
try:
|
|||
|
|
OperationLog.objects.create(
|
|||
|
|
actor=None,
|
|||
|
|
action='apply_overage_policy_cron',
|
|||
|
|
target=f"Invoice#{inv.id}",
|
|||
|
|
detail=str(applied)
|
|||
|
|
)
|
|||
|
|
except Exception:
|
|||
|
|
pass
|
|||
|
|
self.stdout.write(self.style.SUCCESS(f"Invoice#{inv.id} 应用策略：{len(applied)} 个域名"))
|
|||
|
|
|
|||
|
|
self.stdout.write(self.style.SUCCESS(f'完成：处理账单 {len(invoices)} 个，涉及域名 {total_domains} 个'))
|