Skip to content
This repository has been archived by the owner on Jul 16, 2020. It is now read-only.

Commit

Permalink
refactory: 增加尝试次数限制
Browse files Browse the repository at this point in the history
  • Loading branch information
ghosx committed Dec 1, 2019
1 parent 70d74b2 commit 7a21751
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 3 deletions.
2 changes: 1 addition & 1 deletion SignIn/admin.py
Original file line number Diff line number Diff line change
Expand Up @@ -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')


Expand Down
9 changes: 7 additions & 2 deletions SignIn/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down Expand Up @@ -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()
Expand All @@ -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()

Expand Down

0 comments on commit 7a21751

Please sign in to comment.