200 lines
8.5 KiB
Python
200 lines
8.5 KiB
Python
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}"
|
||
)) |