klmp200/Sith
1
0
Fork 0
mirror of https://github.com/ae-utbm/sith.git synced 2025-07-11 04:19:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve generation of account id

Browse Source
This commit is contained in:
Skia
2016-08-05 00:50:25 +02:00
parent 44a6621aac
commit 260a17ae4f
4 changed files with 30 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
counter/models.py
View File

@ -6,7 +6,8 @@ from django.core.urlresolvers import reverse
from django.forms import ValidationError
from datetime import timedelta
from random import randrange
import random
import string
from club.models import Club
from accounting.models import CurrencyField
@ -26,12 +27,17 @@ class Customer(models.Model):
class Meta:
verbose_name = _('customer')
verbose_name_plural = _('customers')
ordering = ['account_id',]
def __str__(self):
return self.user.username
def generate_account_id():
return randrange(0, 4000) # TODO: improve me!
def generate_account_id(number):
number = str(number)
letter = random.choice(string.ascii_lowercase)
while Customer.objects.filter(account_id=number+letter).exists():
letter = random.choice(string.ascii_lowercase)
return number+letter
def save(self, *args, **kwargs):
if self.amount < 0: