222 lines
8.1 KiB
Python
222 lines
8.1 KiB
Python
from django.contrib import messages
|
|
from django.contrib.auth.mixins import LoginRequiredMixin
|
|
from django.core.paginator import Paginator
|
|
from django.db.models import Sum
|
|
from django.shortcuts import get_object_or_404, redirect
|
|
from django.urls import reverse, reverse_lazy
|
|
from django.views import View
|
|
from django.views.generic import FormView, TemplateView
|
|
|
|
from apps.ai.catalog import get_public_model_catalog
|
|
from apps.billing.payment_gateways import PaymentOrderCreateError, payment_callback_mode
|
|
from apps.billing.pricing import NoExchangeRateError
|
|
from apps.billing.models import PointsLedger, RechargeOrder
|
|
from apps.billing.services import (
|
|
RechargeOrderCreateError,
|
|
create_recharge_order,
|
|
get_balance_snapshot,
|
|
)
|
|
from apps.users.models import ApiKey
|
|
|
|
from .forms import ApiKeyCreateForm, RechargeCreateForm
|
|
|
|
|
|
NEW_API_KEY_SESSION_KEY = "portal_new_api_key"
|
|
RECORDS_PAGE_SIZE = 50
|
|
|
|
|
|
def _sum_or_zero(queryset, field_name: str):
|
|
return queryset.aggregate(total=Sum(field_name))["total"] or 0
|
|
|
|
|
|
def get_portal_account_summary(user) -> dict:
|
|
paid_orders = RechargeOrder.objects.filter(
|
|
user=user,
|
|
status=RechargeOrder.Status.PAID,
|
|
)
|
|
consume_entries = PointsLedger.objects.filter(
|
|
user=user,
|
|
change_type=PointsLedger.ChangeType.CONSUME,
|
|
)
|
|
refund_entries = PointsLedger.objects.filter(
|
|
user=user,
|
|
change_type=PointsLedger.ChangeType.REFUND,
|
|
)
|
|
recharge_entries = PointsLedger.objects.filter(
|
|
user=user,
|
|
change_type=PointsLedger.ChangeType.RECHARGE,
|
|
)
|
|
consumed_points = abs(int(_sum_or_zero(consume_entries, "points_delta")))
|
|
refunded_points = int(_sum_or_zero(refund_entries, "points_delta"))
|
|
return {
|
|
"recharge_total_amount": _sum_or_zero(paid_orders, "amount_money"),
|
|
"recharge_points_total": int(_sum_or_zero(recharge_entries, "points_delta")),
|
|
"consumed_points_total": consumed_points,
|
|
"refunded_points_total": refunded_points,
|
|
"net_used_points": consumed_points - refunded_points,
|
|
"paid_recharge_count": paid_orders.count(),
|
|
}
|
|
|
|
|
|
def get_recharge_orders_for_user(user):
|
|
return RechargeOrder.objects.filter(user=user).order_by("-created_at", "-id")
|
|
|
|
|
|
def get_usage_ledger_entries_for_user(user):
|
|
return (
|
|
PointsLedger.objects.filter(
|
|
user=user,
|
|
change_type__in=(
|
|
PointsLedger.ChangeType.CONSUME,
|
|
PointsLedger.ChangeType.REFUND,
|
|
),
|
|
)
|
|
.select_related("ref_call", "ref_call__api_key")
|
|
.order_by("-created_at", "-id")
|
|
)
|
|
|
|
|
|
def paginate_records(request, queryset):
|
|
paginator = Paginator(queryset, RECORDS_PAGE_SIZE)
|
|
page_obj = paginator.get_page(request.GET.get("page"))
|
|
return {
|
|
"paginator": paginator,
|
|
"page_obj": page_obj,
|
|
"is_paginated": page_obj.has_other_pages(),
|
|
"records_total_count": paginator.count,
|
|
"records_page_size": RECORDS_PAGE_SIZE,
|
|
}
|
|
|
|
|
|
class DashboardView(LoginRequiredMixin, TemplateView):
|
|
template_name = "portal/dashboard.html"
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context["balance"] = get_balance_snapshot(self.request.user)
|
|
context.update(get_portal_account_summary(self.request.user))
|
|
context["recent_recharge_orders"] = get_recharge_orders_for_user(
|
|
self.request.user
|
|
)[:5]
|
|
context["recent_usage_entries"] = get_usage_ledger_entries_for_user(
|
|
self.request.user
|
|
)[:5]
|
|
return context
|
|
|
|
|
|
class ApiKeyListCreateView(LoginRequiredMixin, FormView):
|
|
template_name = "portal/apikeys.html"
|
|
form_class = ApiKeyCreateForm
|
|
success_url = reverse_lazy("portal-apikeys")
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context["api_keys"] = ApiKey.objects.filter(user=self.request.user).order_by(
|
|
"-created_at",
|
|
"-id",
|
|
)
|
|
context["new_api_key"] = self.request.session.pop(NEW_API_KEY_SESSION_KEY, None)
|
|
return context
|
|
|
|
def form_valid(self, form):
|
|
name = form.cleaned_data["name"].strip()
|
|
api_key, raw_key = ApiKey.create_for_user(self.request.user, name=name)
|
|
self.request.session[NEW_API_KEY_SESSION_KEY] = {
|
|
"name": api_key.name,
|
|
"key_prefix": api_key.key_prefix,
|
|
"raw_key": raw_key,
|
|
}
|
|
messages.success(self.request, "API Key 已生成")
|
|
return super().form_valid(form)
|
|
|
|
|
|
class ApiKeyDeleteView(LoginRequiredMixin, View):
|
|
def post(self, request, pk):
|
|
api_key = get_object_or_404(ApiKey, pk=pk, user=request.user)
|
|
if api_key.status != ApiKey.Status.REVOKED:
|
|
api_key.status = ApiKey.Status.REVOKED
|
|
api_key.save(update_fields=("status", "updated_at"))
|
|
messages.success(request, "API Key 已删除")
|
|
return redirect("portal-apikeys")
|
|
|
|
|
|
class ModelCatalogView(LoginRequiredMixin, TemplateView):
|
|
template_name = "portal/models.html"
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context["models"] = get_public_model_catalog()
|
|
return context
|
|
|
|
|
|
class RechargePageView(LoginRequiredMixin, FormView):
|
|
template_name = "portal/recharge.html"
|
|
form_class = RechargeCreateForm
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
order_no = str(self.request.GET.get("order_no") or "").strip()
|
|
current_order = None
|
|
if order_no:
|
|
current_order = RechargeOrder.objects.filter(
|
|
user=self.request.user,
|
|
order_no=order_no,
|
|
).first()
|
|
context["balance"] = get_balance_snapshot(self.request.user)
|
|
context.update(get_portal_account_summary(self.request.user))
|
|
current_payment_mode = payment_callback_mode()
|
|
context["current_order"] = current_order
|
|
context["payment_callback_mode"] = current_payment_mode
|
|
context["is_mock_payment_mode"] = current_payment_mode == "mock"
|
|
context["recent_recharge_orders"] = get_recharge_orders_for_user(
|
|
self.request.user
|
|
)[:5]
|
|
return context
|
|
|
|
def form_valid(self, form):
|
|
try:
|
|
order = create_recharge_order(
|
|
user=self.request.user,
|
|
amount=form.cleaned_data["amount"],
|
|
pay_method=form.cleaned_data["pay_method"],
|
|
)
|
|
except NoExchangeRateError:
|
|
form.add_error(None, "未配置当前币种汇率，暂时无法充值。")
|
|
return self.form_invalid(form)
|
|
except RechargeOrderCreateError:
|
|
form.add_error(None, "充值下单参数错误，请检查金额和支付方式。")
|
|
return self.form_invalid(form)
|
|
except PaymentOrderCreateError:
|
|
form.add_error(None, "支付下单失败，请稍后重试。")
|
|
return self.form_invalid(form)
|
|
|
|
messages.success(self.request, "充值订单已创建，请扫码支付。")
|
|
recharge_url = reverse("portal-recharge")
|
|
return redirect(f"{recharge_url}?order_no={order.order_no}")
|
|
|
|
|
|
class RechargeRecordListView(LoginRequiredMixin, TemplateView):
|
|
template_name = "portal/recharge_records.html"
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
orders = get_recharge_orders_for_user(self.request.user)
|
|
pagination = paginate_records(self.request, orders)
|
|
context.update(get_portal_account_summary(self.request.user))
|
|
context.update(pagination)
|
|
context["recharge_orders"] = pagination["page_obj"].object_list
|
|
return context
|
|
|
|
|
|
class UsageRecordListView(LoginRequiredMixin, TemplateView):
|
|
template_name = "portal/usage_records.html"
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
entries = get_usage_ledger_entries_for_user(self.request.user)
|
|
pagination = paginate_records(self.request, entries)
|
|
context.update(get_portal_account_summary(self.request.user))
|
|
context.update(pagination)
|
|
context["usage_entries"] = pagination["page_obj"].object_list
|
|
return context
|