Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
38 changes: 19 additions & 19 deletions payments/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -95,6 +95,25 @@ class BasePayment(models.Model):
class Meta:
abstract = True

def __str__(self):
return self.variant

def save(self, **kwargs):
if not self.token:
tries = {} # Stores a set of tried values
while True:
token = str(uuid4())
if (
token in tries and len(tries) >= 100
): # After 100 tries we are impliying an infinite loop
raise SystemExit("A possible infinite loop was detected")
if not self.__class__._default_manager.filter(token=token).exists():
self.token = token
break
tries.add(token)

return super().save(**kwargs)

def change_status(self, status: PaymentStatus | str, message=""):
"""
Updates the Payment status and sends the status_changed signal.
Expand All @@ -119,25 +138,6 @@ def change_fraud_status(self, status: PaymentStatus, message="", commit=True):
if commit:
self.save()

def save(self, **kwargs):
if not self.token:
tries = {} # Stores a set of tried values
while True:
token = str(uuid4())
if (
token in tries and len(tries) >= 100
): # After 100 tries we are impliying an infinite loop
raise SystemExit("A possible infinite loop was detected")
if not self.__class__._default_manager.filter(token=token).exists():
self.token = token
break
tries.add(token)

return super().save(**kwargs)

def __str__(self):
return self.variant

def get_form(self, data=None):
"""Return a form to be rendered to complete this payment.

Expand Down
1 change: 1 addition & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -93,6 +93,7 @@ extend-select = [
"B",
# "BLE",
"C4",
"DJ",
"E",
"ERA",
# "FBT", # TODO
Expand Down
Loading