feat: opt admin

apps/sspanel/admin.py

@@ -1,6 +1,6 @@
 from typing import Any
 
-from django.contrib import admin
+from django.contrib import admin, messages
 from django.contrib.auth.models import Group
 from django.db.models.query import QuerySet
 from django.http.request import HttpRequest
@@ -85,10 +85,44 @@ class PurchaseHistoryAdmin(admin.ModelAdmin):
 class InviteCodeAdmin(admin.ModelAdmin):
     list_display = ["code", "created_at", "used", "code_type"]
     search_fields = ["code"]
+    list_per_page = 10
+    actions = ["batch_add"]
+    list_filter = ["used"]
+
+    def batch_add(self, request, queryset):
+        # todo support params
+        amount = request.POST.get("amount", 10)
+        number = request.POST.get("number", 10)
+        messages.add_message(
+            request,
+            messages.SUCCESS,
+            f"create code:{amount} x {number}",
+        )
+        models.InviteCode.batch_create(number, 1)
+
+    batch_add.short_description = "批量添加"
+    batch_add.type = "danger"
 
 
 class MoneyCodeAdmin(admin.ModelAdmin):
-    list_display = ["user", "code", "isused"]
+    list_display = ["code", "number", "isused", "user"]
+    list_per_page = 10
+    actions = ["batch_add"]
+    list_filter = ["isused"]
+
+    def batch_add(self, request, queryset):
+        # todo support params
+        amount = request.POST.get("amount", 10)
+        number = request.POST.get("number", 10)
+        messages.add_message(
+            request,
+            messages.SUCCESS,
+            f"create code:={amount}x{number}",
+        )
+        models.MoneyCode.batch_create(amount, number)
+
+    batch_add.short_description = "批量添加"
+    batch_add.type = "danger"
 
 
 class DonateAdmin(admin.ModelAdmin):
@@ -169,13 +203,15 @@ def save_model(
 
 
 # Register your models here.
+admin.site.register(models.MoneyCode, MoneyCodeAdmin)
+admin.site.register(models.InviteCode, InviteCodeAdmin)
+
+
 admin.site.register(models.User, UserAdmin)
 admin.site.register(models.UserOrder, UserOrderAdmin)
 admin.site.register(models.UserCheckInLog, UserCheckInAdmin)
 admin.site.register(models.UserRefLog, UserRefLogAdmin)
-admin.site.register(models.InviteCode, InviteCodeAdmin)
 admin.site.register(models.Donate, DonateAdmin)
-admin.site.register(models.MoneyCode, MoneyCodeAdmin)
 admin.site.register(models.Goods, GoodsAdmin)
 admin.site.register(models.PurchaseHistory, PurchaseHistoryAdmin)
 admin.site.register(models.Announcement)

apps/sspanel/models.py

@@ -605,6 +605,11 @@ class InviteCode(models.Model):
     used = models.BooleanField(verbose_name="是否使用", default=False)
     created_at = models.DateTimeField(editable=False, auto_now_add=True)
 
+    @classmethod
+    def batch_create(cls, number, code_type):
+        models = [cls(code_type=code_type) for _ in range(number)]
+        cls.objects.bulk_create(models)
+
     def __str__(self):
         return f"<{self.user_id}>-<{self.code}>"
 
@@ -727,7 +732,7 @@ def get_most_donated_user_by_count(cls, count):
 class MoneyCode(models.Model):
     """充值码"""
 
-    user = models.CharField(verbose_name="用户名", max_length=128, blank=True, null=True)
+    user = models.CharField(verbose_name="使用人", max_length=128, blank=True, null=True)
     time = models.DateTimeField("捐赠时间", editable=False, auto_now_add=True)
     code = models.CharField(
         verbose_name="充值码",
@@ -737,7 +742,7 @@ class MoneyCode(models.Model):
         default=get_long_random_string,
     )
     number = models.DecimalField(
-        verbose_name="捐赠金额",
+        verbose_name="金额",
         decimal_places=2,
         max_digits=10,
         default=10,
@@ -766,6 +771,11 @@ def __str__(self):
     def isused_cn(self):
         return "已使用" if self.isused else "未使用"
 
+    @classmethod
+    def batch_create(cls, number, amount):
+        models = [cls(number=amount) for _ in range(number)]
+        cls.objects.bulk_create(models)
+
 
 class Goods(models.Model):
     """商品"""