from django.contrib import admin
# Register your models here.from index.models import QrCode

class QrCodeAdmin(admin.ModelAdmin):
    def save_model(self, request, obj, form, change):
        content = form.cleaned_data.get("content")  #  獲取頁面表單填寫的信息，是個(gè)字典類型        try:            now = json.loads(content)        except Exception as e:            raise TypeError("二維碼規(guī)則配置必須是可以json格式化的")        if change:  #   （如果資料時(shí)新增時(shí)，change是False，修改資料時(shí)是True）            old = self.get_object(request, obj.id)  #  這個(gè)方法可以獲取沒保存之前的對(duì)象            old = old.content            old = json.loads(old)            if old != now:  # 如果修改二維碼參數(shù)                # 獲取appid                pt_appid_ = AppletServer.objects.filter(qr_code=obj.id).values_list("app_id", flat=True)  # 第三方平臺(tái)appid                set_pt_appid = set(pt_appid_)                apps_id = MiniApp.objects.filter(AppId__in=set_pt_appid).values_list("app_id", flat=True)                for appid in apps_id:                    add_qr_code(app_id=appid, json_kwargs=now)        obj.save()  #  保存結(jié)果                admin.site.register(QrCode, QrCodeAdmin)  #  將模型注冊(cè)到 admin

class AllCustomerManager(models.Manager):    # 改變查詢的結(jié)果集    def all(self):        # 1.調(diào)用父類的all, 獲取所有數(shù)據(jù)        customers = super().all()        # 2. 返回的books是QuerySet集合，還可以繼續(xù)使用所有查詢        return customers

class CustomerManager(models.Manager):    # 改變查詢的結(jié)果集    def get_queryset(self):        return super(CustomerManager, self).get_queryset().filter(is_default_wx=False)

class Customer(models.Model, CustomerMixin):    user = models.OneToOneField(BaseUser, on_delete=models.PROTECT, related_name="customer", null=True)    is_default_wx = models.BooleanField(verbose_name="微信默認(rèn)用戶", default=True)
    class Meta:        ordering = ["-id", ]        db_table = "customer"        verbose_name = "客戶"
    # def __str__(self):    #     return self.company_name
    all_objects = AllCustomerManager()  # 自定義一個(gè)客戶的管理類對(duì)象    objects = CustomerManager()  # 自定義一個(gè)客戶的管理類對(duì)象