Make account required for trade information

This commit is contained in:
2022-12-08 07:20:46 +00:00
parent 312ddb4dc1
commit 633894ae75
5 changed files with 15 additions and 7 deletions

View File

@@ -235,12 +235,16 @@ class Trade(models.Model):
return cls.objects.get(id=trade_id, user=user)
@classmethod
def get_by_id_or_order(cls, trade_id, user):
def get_by_id_or_order(cls, trade_id, account_id, user):
try:
return cls.objects.get(id=trade_id, user=user)
account = Account.objects.get(id=account_id, user=user)
except Account.DoesNotExist:
return None
try:
return cls.objects.get(id=trade_id, account=account, user=user)
except cls.DoesNotExist:
try:
return cls.objects.get(order_id=trade_id, user=user)
return cls.objects.get(order_id=trade_id, account=account, user=user)
except cls.DoesNotExist:
return None