Вернуться к URL, если есть ошибка в сигнале Django - PullRequest
Новая
       58

Вернуться к URL, если есть ошибка в сигнале Django

0 голосов
/ 12 декабря 2018

У меня есть поле с именем keyChecker в модели mt десятины.KeyChecker является уникальным и генерируется только после отправки формы (с помощью сигнала pre_save). Теперь проблема в том, что все сохраняется, когда ключ не существует, но когда ключ существует, он генерирует IntegrityError, что ожидается.Однако я хочу перенаправлять пользователей на URL всякий раз, когда возникает эта ошибка.

Models.py 

class Tithe(models.Model):
YEAR = []
for r in range((datetime.datetime.now().year), (datetime.datetime.now().year+10)):
    YEAR.append((r,r))
MONTHS = (
    ('January', 'January'),
    ('February', 'February'),
    ('March', 'March'),
    ('April', 'April'),
    ('May', 'May'),
    ('June', 'June'),
    ('July', 'July'),
    ('August', 'August'),
    ('September', 'September'),
    ('October', 'October'),
    ('November', 'November'),
    ('December', 'December'),
)
year = models.IntegerField( choices=YEAR, default=datetime.datetime.now().year)
month = models.CharField(max_length = 50, choices = MONTHS, null=True, blank=True)
week1 = models.DecimalField(max_digits=10, decimal_places=2, default = 0)
week2 = models.DecimalField(max_digits=10, decimal_places=2, default = 0)
week3 = models.DecimalField(max_digits=10, decimal_places=2, default = 0)
week4 = models.DecimalField(max_digits=10, decimal_places=2, default = 0)
total = models.DecimalField(max_digits=10, decimal_places=2, default = 0)
keyChecker = models.CharField(max_length=500,unique=True, null=True, blank=True)
member = models.ForeignKey('Member', on_delete = models.CASCADE)

def __str__(self):
    return "{0}_{1}_{2}".format(self.member, self.year, self.month) 

def get_absolute_url(self):
    return reverse('tithe', kwargs={'slug': datetime.datetime.now().year, 'slug2':datetime.datetime.now().month})

def get_key_checker(instance, new_slug = None):
memberpk = instance.member_id
member = Member.objects.get(pk = memberpk)

fname = member.fname
lname = member.lname
contact = member.contact
year = instance.year
month = instance.month

keyChecker = "{0}_{1}_{2}_{3}_{4}".format(fname,lname,contact,year,month)

return keyChecker

def pre_save_tithe_receiver(sender, instance, *args, **kwargs):
instance.keyChecker = get_key_checker(instance)

pre_save.connect(pre_save_tithe_receiver, sender=Tithe)
...