from django import forms from django.contrib.auth.forms import UserCreationForm from django.core.exceptions import FieldDoesNotExist from django.forms import ModelForm from .models import Account, Hook, Signal, Strategy, Trade, TradingTime, User # flake8: noqa: E501 class RestrictedFormMixin: """ This mixin is used to restrict the queryset of a form to the current user. The request object is passed from the view.""" def __init__(self, *args, **kwargs): self.request = kwargs.pop("request") super().__init__(*args, **kwargs) for field in self.fields: # Check it's not something like a CharField which has no queryset if not hasattr(self.fields[field], "queryset"): continue model = self.fields[field].queryset.model # Check if the model has a user field try: model._meta.get_field("user") # Add the user to the queryset filters self.fields[field].queryset = model.objects.filter( user=self.request.user ) except FieldDoesNotExist: pass class NewUserForm(UserCreationForm): email = forms.EmailField(required=True) class Meta: model = User fields = ( "username", "email", "first_name", "last_name", "password1", "password2", ) def save(self, commit=True): user = super(NewUserForm, self).save(commit=False) user.email = self.cleaned_data["email"] if commit: user.save() return user class CustomUserCreationForm(UserCreationForm): class Meta: model = User fields = "__all__" class HookForm(RestrictedFormMixin, ModelForm): class Meta: model = Hook fields = ( "name", "hook", ) help_texts = { "name": "Name of the hook. Informational only.", "hook": "The URL slug to use for the hook. Make it unique.", } class SignalForm(RestrictedFormMixin, ModelForm): class Meta: model = Signal fields = ( "name", "signal", "hook", "direction", ) help_texts = { "name": "Name of the signal. Informational only.", "signal": "The name of the signal in Drakdoo. Copy it from there.", "hook": "The hook this signal belongs to.", "direction": "The direction of the signal. This is used to determine if the signal is a buy or sell.", } class AccountForm(RestrictedFormMixin, ModelForm): class Meta: model = Account fields = ( "name", "exchange", "api_key", "api_secret", "sandbox", ) help_texts = { "name": "Name of the account. Informational only.", "exchange": "The exchange to use for this account.", "api_key": "The API key or username for the account.", "api_secret": "The API secret or password/token for the account.", "sandbox": "Whether to use the sandbox/demo or not.", } class StrategyForm(RestrictedFormMixin, ModelForm): class Meta: model = Strategy fields = ( "name", "description", "account", "trading_times", "order_type", "time_in_force", "entry_signals", "exit_signals", "enabled", "take_profit_percent", "stop_loss_percent", "trailing_stop_loss_percent", "price_slippage_percent", "callback_price_deviation_percent", "trade_size_percent", ) help_texts = { "name": "Name of the strategy. Informational only.", "description": "Description of the strategy. Informational only.", "account": "The account to use for this strategy.", "trading_times": "When the strategy will place new trades.", "order_type": "Market: Buy/Sell at the current market price. Limit: Buy/Sell at a specified price. Limits protect you more against market slippage.", "time_in_force": "The time in force controls how the order is executed.", "entry_signals": "The entry signals to attach to this strategy. Callbacks received to these signals will trigger a trade.", "exit_signals": "The exit signals to attach to this strategy. Callbacks received to these signals will close all trades for the symbol on the account.", "enabled": "Whether the strategy is enabled.", "take_profit_percent": "The take profit will be set at this percentage above/below the entry price.", "stop_loss_percent": "The stop loss will be set at this percentage above/below the entry price.", "trailing_stop_loss_percent": "The trailing stop loss will be set at this percentage above/below the entry price. A trailing stop loss will follow the price as it moves in your favor.", "price_slippage_percent": "The price slippage is the maximum percent the price can move against you before the trade is cancelled. Limit orders will be set at this percentage against your favour. Market orders will have a price bound set if this is supported.", "callback_price_deviation_percent": "The callback price deviation is the maximum percent the price can change between receiving the callback and acting on it. This protects against rogue or delayed callbacks. Keep it low.", "trade_size_percent": "Percentage of the account balance to use for each trade.", } entry_signals = forms.ModelMultipleChoiceField( queryset=Signal.objects.all(), widget=forms.CheckboxSelectMultiple, help_text=Meta.help_texts["entry_signals"], required=False, ) exit_signals = forms.ModelMultipleChoiceField( queryset=Signal.objects.all(), widget=forms.CheckboxSelectMultiple, help_text=Meta.help_texts["exit_signals"], required=False, ) trading_times = forms.ModelMultipleChoiceField( queryset=TradingTime.objects.all(), widget=forms.CheckboxSelectMultiple ) class TradeForm(RestrictedFormMixin, ModelForm): class Meta: model = Trade fields = ( "account", "symbol", "type", "time_in_force", "amount", "price", "stop_loss", "trailing_stop_loss", "take_profit", "direction", ) help_texts = { "account": "The account to use for this trade.", "symbol": "The symbol to trade.", "type": "Market: Buy/Sell at the current market price. Limit: Buy/Sell at a specified price. Limits protect you more against market slippage.", "time_in_force": "The time in force controls how the order is executed.", "amount": "The amount to trade in the quote currency (the second part of the symbol if applicable, otherwise your account base currency).", "price": "The price to trade at. Sets this price as the trigger price for limit orders. Sets a price bound (if supported) for market orders.", "stop_loss": "The stop loss price. This will be set at the specified price.", "trailing_stop_loss": "The trailing stop loss price. This will be set at the specified price and will follow the price as it moves in your favor.", "take_profit": "The take profit price. This will be set at the specified price.", "direction": "The direction of the trade. This is used to determine if the trade is a buy or sell.", } class TradingTimeForm(RestrictedFormMixin, ModelForm): class Meta: model = TradingTime fields = ( "name", "description", "start_day", "start_time", "end_day", "end_time", ) help_texts = { "name": "Name of the trading time. Informational only.", "description": "Description of the trading time. Informational only.", "start_day": "The day of the week to start trading.", "start_time": "The time of day to start trading.", "end_day": "The day of the week to stop trading.", "end_time": "The time of day to stop trading.", }