Initial commit

billing/management/commands/generate_invoices.py

@@ -0,0 +1,200 @@
+from datetime import date, timedelta
+from decimal import Decimal, ROUND_HALF_UP
+
+from django.core.management.base import BaseCommand, CommandError
+from django.db import transaction
+from django.utils import timezone
+from django.contrib.auth import get_user_model
+
+from billing.models import Invoice, InvoiceItem
+from domains.models import Domain, DomainTrafficDaily
+from core.models import SystemSettings
+
+
+User = get_user_model()
+
+
+def month_range_from_str(month_str: str):
+    try:
+        year, month = map(int, month_str.split('-'))
+        start = date(year, month, 1)
+    except Exception as e:
+        raise CommandError(f"无效的 --month 参数，格式应为 YYYY-MM：{e}")
+    # 下月第一天
+    if month == 12:
+        next_month_start = date(year + 1, 1, 1)
+    else:
+        next_month_start = date(year, month + 1, 1)
+    end = next_month_start - timedelta(days=1)
+    return start, end
+
+
+def previous_month_range():
+    today = timezone.now().date()
+    this_month_start = today.replace(day=1)
+    last_day_prev = this_month_start - timedelta(days=1)
+    start_prev = last_day_prev.replace(day=1)
+    return start_prev, last_day_prev
+
+
+class Command(BaseCommand):
+    help = "按自然月生成用户账单（套餐费 + 超量流量）。默认生成上一自然月，可用 --month 指定。"
+
+    def add_arguments(self, parser):
+        parser.add_argument('--month', type=str, help='计费月份，格式 YYYY-MM，如 2025-10')
+        parser.add_argument('--user-id', type=int, help='仅为指定用户生成账单（可选）')
+        parser.add_argument('--overwrite', action='store_true', help='若已存在账单则覆盖重建（删除后重建）')
+        parser.add_argument('--dry-run', action='store_true', help='试运行，仅输出计算结果，不写入数据库')
+
+    @transaction.atomic
+    def handle(self, *args, **options):
+        # 计费周期
+        if options.get('month'):
+            period_start, period_end = month_range_from_str(options['month'])
+        else:
+            period_start, period_end = previous_month_range()
+
+        user_id = options.get('user_id')
+        overwrite = options.get('overwrite', False)
+        dry_run = options.get('dry_run', False)
+
+        self.stdout.write(self.style.NOTICE(
+            f"生成账单周期：{period_start} → {period_end}"
+        ))
+
+        users_qs = User.objects.all()
+        if user_id:
+            users_qs = users_qs.filter(id=user_id)
+            if not users_qs.exists():
+                raise CommandError(f"用户 {user_id} 不存在")
+
+        sys = SystemSettings.objects.order_by('id').first()
+
+        total_invoices = 0
+        total_amount = Decimal('0.00')
+
+        for user in users_qs:
+            # 处理已存在账单
+            existing = Invoice.objects.filter(user=user, period_start=period_start, period_end=period_end).first()
+            if existing and not overwrite:
+                self.stdout.write(self.style.WARNING(
+                    f"跳过用户 {user.id}（{user.username}），该周期账单已存在：Invoice#{existing.id}"
+                ))
+                continue
+
+            if existing and overwrite and not dry_run:
+                existing.delete()
+                self.stdout.write(self.style.WARNING(
+                    f"已删除旧账单：用户 {user.id}（{user.username}） Invoice#{existing.id}"
+                ))
+
+            # 汇总用户域名的费用
+            domains = Domain.objects.filter(user=user, status=Domain.STATUS_ACTIVE)
+            amount_plan_total = Decimal('0.00')
+            amount_overage_total = Decimal('0.00')
+            items = []
+
+            for d in domains:
+                plan = d.current_plan
+                base_price = Decimal(str(plan.base_price_per_domain)) if plan and plan.base_price_per_domain is not None else Decimal('0.00')
+                included_gb = Decimal(str(plan.included_traffic_gb_per_domain)) if plan and plan.included_traffic_gb_per_domain is not None else None
+                overage_price = Decimal(str(plan.overage_price_per_gb)) if plan and plan.overage_price_per_gb is not None else Decimal('0.00')
+
+                # 若域名自定义超量单价优先
+                if d.custom_overage_price_per_gb is not None:
+                    overage_price = Decimal(str(d.custom_overage_price_per_gb))
+
+                # 用户级默认免费额度覆盖
+                user_default_override = None
+                if hasattr(user, 'profile') and user.profile.default_free_traffic_gb_per_domain_override is not None:
+                    user_default_override = Decimal(str(user.profile.default_free_traffic_gb_per_domain_override))
+
+                # 系统全局默认
+                sys_default_gb = Decimal(str(sys.default_free_traffic_gb_per_domain)) if sys and sys.default_free_traffic_gb_per_domain is not None else Decimal('0')
+
+                # 最终基础配额：优先 Plan，其次用户覆盖，再次系统默认
+                base_quota_gb = Decimal('0')
+                if included_gb is not None:
+                    base_quota_gb = Decimal(str(included_gb))
+                elif user_default_override is not None:
+                    base_quota_gb = user_default_override
+                else:
+                    base_quota_gb = sys_default_gb
+
+                # 本周期域名额外赠送
+                extra_gb = Decimal(str(d.extra_free_traffic_gb_current_cycle or 0))
+
+                # 统计周期用量
+                traffic_qs = DomainTrafficDaily.objects.filter(domain=d, day__gte=period_start, day__lte=period_end)
+                total_bytes = sum(t.bytes for t in traffic_qs)
+                used_gb = Decimal(str(total_bytes)) / Decimal('1073741824')  # 1024^3
+                used_gb = used_gb.quantize(Decimal('0.001'), rounding=ROUND_HALF_UP)
+
+                # 超量（不为负）
+                over_gb = used_gb - (base_quota_gb + extra_gb)
+                if over_gb < Decimal('0'):
+                    over_gb = Decimal('0')
+                over_gb = over_gb.quantize(Decimal('0.001'), rounding=ROUND_HALF_UP)
+
+                # 费用累计
+                if base_price > 0:
+                    amount_plan_total += base_price
+                    items.append({
+                        'domain': d,
+                        'description': '基础套餐费用',
+                        'quantity': Decimal('1'),
+                        'unit_price': base_price,
+                        'amount': base_price,
+                    })
+
+                if over_gb > 0 and overage_price > 0:
+                    over_amount = (over_gb * overage_price).quantize(Decimal('0.01'), rounding=ROUND_HALF_UP)
+                    amount_overage_total += over_amount
+                    items.append({
+                        'domain': d,
+                        'description': '超量流量费用',
+                        'quantity': over_gb,
+                        'unit_price': overage_price,
+                        'amount': over_amount,
+                    })
+
+            amount_adjustment = Decimal('0.00')
+            amount_total = (amount_plan_total + amount_overage_total + amount_adjustment).quantize(Decimal('0.01'), rounding=ROUND_HALF_UP)
+
+            if dry_run:
+                self.stdout.write(self.style.SUCCESS(
+                    f"[DRY] 用户 {user.id}（{user.username}） 套餐费：￥{amount_plan_total} 超量费：￥{amount_overage_total} 总计：￥{amount_total}"
+                ))
+                continue
+
+            # 创建账单与明细
+            invoice = Invoice.objects.create(
+                user=user,
+                period_start=period_start,
+                period_end=period_end,
+                amount_plan_total=amount_plan_total,
+                amount_overage_total=amount_overage_total,
+                amount_adjustment=amount_adjustment,
+                amount_total=amount_total,
+                status='unpaid',
+            )
+
+            for it in items:
+                InvoiceItem.objects.create(
+                    invoice=invoice,
+                    domain=it['domain'],
+                    description=it['description'],
+                    quantity=it['quantity'],
+                    unit_price=it['unit_price'],
+                    amount=it['amount'],
+                )
+
+            total_invoices += 1
+            total_amount += amount_total
+            self.stdout.write(self.style.SUCCESS(
+                f"已生成账单：Invoice#{invoice.id} 用户 {user.id}（{user.username}） 总额：￥{amount_total}"
+            ))
+
+        self.stdout.write(self.style.SUCCESS(
+            f"生成完成：共 {total_invoices} 个账单，合计金额 ￥{total_amount}"
+        ))