class YooLottoUser(models.Model):
name = models.CharField(max_length=255, blank=True, null=True)
email = models.CharField(max_length=255, blank=True, null=True, unique=True)
password = models.CharField(max_length=255, null=True, blank=True)
referral = models.CharField(max_length=64, null=True, blank=True)
identifier = models.CharField(max_length=32, null=True, blank=True, unique=True)
added_at = models.DateTimeField(auto_now_add=True)
def get_wallet(self):
try:
return self.wallet
except CoinWallet.DoesNotExist:
self.wallet = CoinWallet(user=self)
self.wallet.save()
return self.wallet
def get_identifier(self):
if not self.identifier:
self.identifier = uuid.uuid4().hex
self.save()
return self.identifier
def __unicode__(self):
return "%s:%s" % (self.email, self.pk)
class Meta:
db_table = u"user"
def get_wallet(self):
try:
return self.wallet
except CoinWallet.DoesNotExist:
self.wallet = CoinWallet(user=self)
self.wallet.save()
return self.wallet
