|
|
|
|
from django import forms
|
|
|
|
|
from .models import ClosedTrade
|
|
|
|
|
|
|
|
|
|
def get_all_accounts_and_strategies():
|
|
|
|
|
all_accounts = set()
|
|
|
|
|
all_strategies = set()
|
|
|
|
|
for trade in ClosedTrade.objects.all():
|
|
|
|
|
all_accounts.add(trade.account)
|
|
|
|
|
all_strategies.add(trade.strategyId)
|
|
|
|
|
return (all_accounts, all_strategies)
|
|
|
|
|
|
|
|
|
|
class NewTradeForm(forms.Form):
|
|
|
|
|
timestamp = forms.DateTimeField()
|
|
|
|
|
account = forms.CharField(max_length=256)
|
|
|
|
|
security = forms.CharField(max_length=256)
|
|
|
|
|
operation = forms.ChoiceField(choices=[('buy', 'Buy'), ('sell', 'Sell')])
|
|
|
|
|
price = forms.DecimalField()
|
|
|
|
|
quantity = forms.IntegerField()
|
|
|
|
|
volume = forms.DecimalField()
|
|
|
|
|
volumeCurrency = forms.CharField(max_length=10)
|
|
|
|
|
strategyId = forms.CharField(max_length=64)
|
|
|
|
|
signalId = forms.CharField(max_length=64)
|
|
|
|
|
|
|
|
|
|
class ClosedTradeFilterForm(forms.Form):
|
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
|
|
|
super().__init__(*args, **kwargs)
|
|
|
|
|
|
|
|
|
|
all_accounts, all_strategies = get_all_accounts_and_strategies()
|
|
|
|
|
self.fields['accounts'] = forms.MultipleChoiceField(choices=zip(list(all_accounts), list(all_accounts)), required=False)
|
|
|
|
|
self.fields['strategies'] = forms.MultipleChoiceField(choices=zip(list(all_strategies), list(all_strategies)), required=False)
|
|
|
|
|
|
