#文章级别
# 关键词
A =  0
B =  1
C =  2
D =  3
E =  4

#来源
SRC_CHOICE = {
    A: '知乎',
    B: '小红书',
}

#文章级别
LEVEL_CHOICE = {
    A: '审核中',
    B: '未通过',
    C: '普通',
    D: '精品',
    E: '被删除',
}

# Create your models here.
class Article(models.Model):
    '''
    图鲁班文章模型类
    '''
    id = models.AutoField(primary_key=True)
    src_choices = ((k, v) for k,v in SRC_CHOICE.items())
    level_choices = ((k, v) for k,v in LEVEL_CHOICE.items())
    sno = models.CharField(max_length=32, verbose_name='编号', default='')
    title = models.CharField(max_length=255, verbose_name='标题', default='')
    url   = models.CharField(max_length=255, verbose_name='链接', default='')
    uid   = models.IntegerField(default=1, verbose_name='用户id')
    addtime = models.DateTimeField(auto_now_add=True, verbose_name= '添加时间')
    stype = models.SmallIntegerField(default='1', choices=src_choices, verbose_name='平台')
    slevel = models.SmallIntegerField(default='0', choices=level_choices, verbose_name='级别')
    note = models.CharField(max_length=1024, default='无', verbose_name='备注')

    def __str__(self):
        return self.sno

    class Meta:
        db_table = 'photo_article'
        verbose_name = '推广记录'
        verbose_name_plural = '推广记录'
code here...

# Register your models here.
class ArticleAdmin(admin.ModelAdmin):
    #后台展示字段
    list_display = ['sno', 'show_firm_url', 'slevel', 'stype', 'addtime', 'repay_info', 'show_kefu', 'note']

    #过滤字段
    list_filter =  ["slevel", "stype"]

    #搜索字段
    search_fields = ['title']

    #只读字段
    readonly_fields = ['sno', 'uid', 'addtime']

    @admin.display(description='返现信息', ordering='')
    def repay_info(self, obj):
        num = 0  #条数
        w_p = 0  #未结算金额
        y_p = 0  #已结算金额

        repays = Repay.objects.filter(sno=obj.sno).all()
        if repays:
            for repay in repays:
                num = num + 1
                if repay.status == 0:
                    w_p = w_p + repay.price
                if repay.status == 1:
                    y_p = y_p + repay.price

        data = {'name':'返现记录', 'url':'/admin/photo/repay/?q={}'.format(obj.sno)}
        return format_html("<a href='javascript:;' onclick='self.parent.app.openTab({data})'>共{n}条, 已结算: {y}, 未结算: {w}</a>", data=data, n=num, y=y_p, w=w_p)

    #标题带链接
    @admin.display(description='标题', ordering='')
    def show_firm_url(self, obj):
        return format_html("<a href='{url}' target='_blank'>{title}</a>", url=obj.url, title=obj.title)

    #客服回复提示
    @admin.display(description='客服回复', ordering='')
    def show_kefu(self, obj):
        text = '您的返现申请已提交,结果可通过(http://tg.tuluban.top)自助查询。查询码:{}'.format(obj.sno)
        return format_html(f"""
                    <input type="text" id="{obj.id}" value="{text}" style="position: absolute; top: -10000px">
                    <a href="#" onclick="document.getElementById('{obj.id}').select(); document.execCommand('copy')">提示</a>
                    """)


    #隐藏不需要编辑的字段
    def get_form(self, request, obj=None, **kwargs):
        self.exclude = ('sno', 'uid', 'addtime')
        form = super(ArticleAdmin, self).get_form(request, obj, **kwargs)
        return form

    #重写保存函数
    def save_model(self, request, obj, form, change):
        #新增
        if not change:
            obj.uid = request.user.id
            obj.sno = datetime.now().strftime('%Y%m%d%H%M%S') + ''.join(str(i) for i in random.sample(range(0,9),4))

        repay = Repay.objects.filter(sno=obj.sno).filter(status=0)

        if obj.slevel == 2:
            if repay:
                repay.update(price=290)
            else:
                repay = Repay()
                repay.sno = obj.sno
                repay.price = 290
                repay.uid = obj.uid
                repay.note = obj.note
                repay.save()
        elif obj.slevel == 3:
            if repay:
                repay.update(price=490)
            else:
                repay = Repay()
                repay.sno = obj.sno
                repay.price = 490
                repay.uid = obj.uid
                repay.note = obj.note
                repay.save()

        #保存
        super(ArticleAdmin, self).save_model(request, obj, form, change)