From 7a2175157151cf4ed27a501066c0d0604790c055 Mon Sep 17 00:00:00 2001 From: ghosx Date: Sun, 1 Dec 2019 17:14:38 +0800 Subject: [PATCH] =?UTF-8?q?refactory:=20=E5=A2=9E=E5=8A=A0=E5=B0=9D?= =?UTF-8?q?=E8=AF=95=E6=AC=A1=E6=95=B0=E9=99=90=E5=88=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- SignIn/admin.py | 2 +- SignIn/models.py | 9 +++++++-- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/SignIn/admin.py b/SignIn/admin.py index 0105974c..3f8c0e57 100644 --- a/SignIn/admin.py +++ b/SignIn/admin.py @@ -8,7 +8,7 @@ class UserAdmin(admin.ModelAdmin): class SignAdmin(admin.ModelAdmin): - list_display = ('id', 'name', 'fid', 'is_sign', 'status', 'user') + list_display = ('id', 'name', 'fid', 'is_sign', 'retry_time', 'status', 'user') search_fields = ('id', 'name', 'fid', 'status', 'user') diff --git a/SignIn/models.py b/SignIn/models.py index d229108e..75ed0557 100644 --- a/SignIn/models.py +++ b/SignIn/models.py @@ -6,7 +6,7 @@ from django.db.models import Q from SignIn.utils import utils -from constants import NOT_VALID_USER, ALREADY_UPDATE_USER, NEW_USER, API_STATUS +from constants import NOT_VALID_USER, ALREADY_UPDATE_USER, NEW_USER, API_STATUS, MAX_RETRY_TIMES class UserManager(models.Manager): @@ -123,6 +123,7 @@ class Sign(models.Model): name = models.CharField(max_length=100, verbose_name="贴吧名") fid = models.CharField(max_length=20, verbose_name="贴吧id") is_sign = models.BooleanField(default=False, verbose_name="是否签到") + retry_time = models.SmallIntegerField(default=0, verbose_name="重试次数") status = models.CharField(max_length=100, verbose_name="签到状态", default="") user = models.ForeignKey(User, on_delete=models.CASCADE, verbose_name="所属用户") objects = SignManager() @@ -146,7 +147,11 @@ def sign_callback(self, obj): self.is_sign = True else: print('签到出错', sign.name, res) - self.is_sign = False + self.retry_time += 1 + if self.retry_time > MAX_RETRY_TIMES: + self.is_sign = True + else: + self.is_sign = False self.status = res['error_msg'] self.save()