74 lines
3.0 KiB
Python
74 lines
3.0 KiB
Python
|
import datetime
|
|||
|
|
from typing import Dict
|
|||
|
|
|
|||
|
|
from django.core.management.base import BaseCommand, CommandError
|
|||
|
|
from django.db import transaction
|
|||
|
|
from django.utils import timezone
|
|||
|
|
|
|||
|
|
from domains.models import Domain, DomainTrafficDaily
|
|||
|
|
from core.goedge_client import GoEdgeClient
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Command(BaseCommand):
|
|||
|
|
help = '从 GoEdge 读取各域名（serverId）的最近每日流量并写入 DomainTrafficDaily'
|
|||
|
|
|
|||
|
|
def add_arguments(self, parser):
|
|||
|
|
parser.add_argument('--days', type=int, default=35, help='拉取最近N天数据（默认35）')
|
|||
|
|
parser.add_argument('--only-active', action='store_true', dest='only_active', default=False, help='仅拉取状态为active的域名')
|
|||
|
|
# 兼容文档中的别名参数
|
|||
|
|
parser.add_argument('--active-only', action='store_true', dest='only_active', help='参数别名，等同 --only-active')
|
|||
|
|
|
|||
|
|
@transaction.atomic
|
|||
|
|
def handle(self, *args, **options):
|
|||
|
|
days: int = options['days']
|
|||
|
|
only_active: bool = options['only_active']
|
|||
|
|
|
|||
|
|
qs = Domain.objects.exclude(edge_server_id__isnull=True)
|
|||
|
|
if only_active:
|
|||
|
|
qs = qs.filter(status=Domain.STATUS_ACTIVE)
|
|||
|
|
|
|||
|
|
domains = list(qs)
|
|||
|
|
if not domains:
|
|||
|
|
self.stdout.write(self.style.WARNING('没有可拉取的域名（缺少 edge_server_id）。'))
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
client = GoEdgeClient()
|
|||
|
|
total_written = 0
|
|||
|
|
|
|||
|
|
for d in domains:
|
|||
|
|
server_id = int(d.edge_server_id)
|
|||
|
|
try:
|
|||
|
|
stats = client.find_latest_server_daily_stats(server_id=server_id, days=days)
|
|||
|
|
except Exception as e:
|
|||
|
|
self.stderr.write(self.style.ERROR(f'[domain={d.name}] 拉取失败：{e}'))
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
# 解析并入库
|
|||
|
|
for s in stats:
|
|||
|
|
day_str = s.get('day')
|
|||
|
|
bytes_val = int(s.get('bytes') or 0)
|
|||
|
|
peak_mbps = float(s.get('peakBandwidthMbps') or 0.0)
|
|||
|
|
# 将 day 格式 YYYYMMDD 转为 date
|
|||
|
|
try:
|
|||
|
|
day_date = datetime.datetime.strptime(day_str, '%Y%m%d').date()
|
|||
|
|
except Exception:
|
|||
|
|
# 某些版本可能返回 YYYY-MM-DD
|
|||
|
|
try:
|
|||
|
|
day_date = datetime.datetime.strptime(day_str, '%Y-%m-%d').date()
|
|||
|
|
except Exception:
|
|||
|
|
self.stderr.write(self.style.WARNING(f'无法解析日期：{day_str}'))
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
obj, created = DomainTrafficDaily.objects.update_or_create(
|
|||
|
|
domain=d,
|
|||
|
|
day=day_date,
|
|||
|
|
defaults={
|
|||
|
|
'bytes': bytes_val,
|
|||
|
|
'peak_bandwidth_mbps': peak_mbps,
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
total_written += 1
|
|||
|
|
|
|||
|
|
self.stdout.write(self.style.SUCCESS(f'[domain={d.name}] 写入 {len(stats)} 条'))
|
|||
|
|
|
|||
|
|
self.stdout.write(self.style.SUCCESS(f'完成，总写入记录：{total_written}'))
|