Egor fureunoir Gorbunov
376c73ba26
Fixes: 1) Remove redundant imports from admin.py; Extra: 1) Update inline classes to inherit from TabularInline; 2) Add unfold.contrib modules to INSTALLED_APPS; 3) Reorder imports in admin.py for consistency.
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
from django.contrib.admin import register
|
|
from django.db.models import QuerySet
|
|
from django.http import HttpRequest
|
|
from django.utils.translation import gettext_lazy as _
|
|
from unfold.admin import ModelAdmin, TabularInline
|
|
|
|
from engine.core.admin import ActivationActionsMixin
|
|
from engine.payments.forms import GatewayForm, TransactionForm
|
|
from engine.payments.models import Balance, Gateway, Transaction
|
|
|
|
|
|
class TransactionInline(TabularInline): # type: ignore [type-arg]
|
|
model = Transaction
|
|
form = TransactionForm
|
|
extra = 1
|
|
is_navtab = True
|
|
verbose_name = _("transaction")
|
|
verbose_name_plural = _("transactions")
|
|
|
|
def get_queryset(self, request: HttpRequest) -> QuerySet[Transaction]:
|
|
qs = super().get_queryset(request)
|
|
return qs.select_related("order")
|
|
|
|
|
|
@register(Balance)
|
|
class BalanceAdmin(ActivationActionsMixin, ModelAdmin): # type: ignore [misc, type-arg]
|
|
inlines = (TransactionInline,)
|
|
list_display = ("user", "amount")
|
|
search_fields = ("user__email",)
|
|
ordering = ("user",)
|
|
|
|
def get_queryset(self, request: HttpRequest) -> QuerySet[Balance]:
|
|
qs = super().get_queryset(request)
|
|
return qs.prefetch_related("transactions", "user")
|
|
|
|
|
|
@register(Transaction)
|
|
class TransactionAdmin(ActivationActionsMixin, ModelAdmin): # type: ignore [misc, type-arg]
|
|
list_display = ("balance", "amount", "order", "modified", "created")
|
|
search_fields = ("balance__user__email", "currency", "payment_method")
|
|
list_filter = ("currency", "payment_method")
|
|
ordering = ("balance",)
|
|
form = TransactionForm
|
|
|
|
|
|
@register(Gateway)
|
|
class GatewayAdmin(ActivationActionsMixin, ModelAdmin): # type: ignore [misc]
|
|
list_display = (
|
|
"name",
|
|
"can_be_used",
|
|
"is_active",
|
|
)
|
|
search_fields = (
|
|
"name",
|
|
"default_currency",
|
|
)
|
|
ordering = ("name",)
|
|
form = GatewayForm
|
|
|
|
def can_be_used(self, obj: Gateway) -> bool: # type: ignore[override]
|
|
return obj.can_be_used
|
|
|
|
can_be_used.boolean = True # type: ignore[attr-defined]
|
|
can_be_used.short_description = _("can be used") # type: ignore[attr-defined]
|