Allow to pass prefix to generate_code, too (#523)

This commit is contained in:
Tobias Kunze
2017-06-18 18:30:38 +02:00
committed by Raphael Michel
parent 260ba19e44
commit f7de2ead40

View File

@@ -20,9 +20,9 @@ def _generate_random_code(prefix=None):
return get_random_string(length=settings.ENTROPY['voucher_code'], allowed_chars=charset)
def generate_code():
def generate_code(prefix=None):
while True:
code = _generate_random_code()
code = _generate_random_code(prefix=prefix)
if not Voucher.objects.filter(code=code).exists():
return code