models.py 740 B

1234567891011121314151617181920
  1. from django.db import models
  2. from hashlib import sha256
  3. # Create your models here.
  4. class User(models.Model):
  5. status = models.BooleanField(default=True, blank=True)
  6. first_name = models.CharField(max_length=255, default="", blank=False)
  7. last_name = models.CharField(max_length=255, default="", blank=False)
  8. pin = models.CharField(max_length=255, null=False, blank=False, unique=True)
  9. def save(self, *args, **kwargs):
  10. self.pin = sha256(self.pin.encode('utf-8')).hexdigest()
  11. user = User.objects.filter(pin=self.pin).first()
  12. if user:
  13. super(User, self).save(*args, **kwargs)
  14. def __str__(self):
  15. return "%s, %s (Enabled: %s)" % (self.last_name, self.first_name, self.status)