Continue implementing live risk checks
This commit is contained in:
@@ -116,6 +116,7 @@ class Account(models.Model):
|
||||
risk_model = models.ForeignKey(
|
||||
"core.RiskModel", on_delete=models.SET_NULL, null=True, blank=True
|
||||
)
|
||||
initial_balance = models.FloatField(default=0)
|
||||
|
||||
def __str__(self):
|
||||
name = f"{self.name} ({self.exchange})"
|
||||
@@ -167,6 +168,10 @@ class Account(models.Model):
|
||||
def get_by_id(cls, account_id, user):
|
||||
return cls.objects.get(id=account_id, user=user)
|
||||
|
||||
@classmethod
|
||||
def get_by_id_no_user_check(cls, account_id):
|
||||
return cls.objects.get(id=account_id)
|
||||
|
||||
|
||||
class Session(models.Model):
|
||||
user = models.ForeignKey(User, on_delete=models.CASCADE)
|
||||
|
||||
Reference in New Issue
Block a user