Implement closing positions

This commit is contained in:
Mark Veidemanis 2022-12-02 07:20:37 +00:00
parent 5aac60a7ee
commit 1ce6c3fafa
Signed by: m
GPG Key ID: 5ACFCEED46C0904F
5 changed files with 68 additions and 6 deletions

View File

@ -143,6 +143,11 @@ urlpatterns = [
positions.Positions.as_view(),
name="positions",
),
path(
"positions/close/<str:account_id>/<str:side>/<str:symbol>/",
positions.PositionAction.as_view(),
name="position_action",
),
path(
"positions/<str:type>/<str:account_id>/<str:symbol>/",
positions.PositionAction.as_view(),

View File

@ -218,6 +218,10 @@ class BaseExchange(ABC):
def get_all_positions(self):
pass
@abstractmethod
def close_position(self, side, symbol):
pass
@abstractmethod
def close_all_positions(self):
pass

View File

@ -130,3 +130,9 @@ class AlpacaExchange(BaseExchange):
item["unrealized_pl"] = float(item["unrealized_pl"])
items.append(item)
return items
def close_position(self, side, symbol):
pass
def close_all_positions(self):
pass

View File

@ -518,7 +518,7 @@ class LongOrderFillTransaction(BaseModel):
longPositionCloseout: LongPositionCloseout
class OrderTransation(BaseModel):
class OrderTransaction(BaseModel):
id: str
accountID: str
userID: int
@ -531,8 +531,8 @@ class OrderTransation(BaseModel):
timeInForce: str | None
positionFill: str | None
reason: str
longPositionCloseout: LongPositionCloseout
longOrderFillTransaction: dict
longPositionCloseout: LongPositionCloseout | None
longOrderFillTransaction: dict | None
class PositionClose(BaseModel):
@ -578,7 +578,7 @@ class TradeDetailsTrade(BaseModel):
dividendAdjustment: str
closeTime: str
averageClosePrice: str
clientExtensions: ClientExtensions
clientExtensions: ClientExtensions | None
class TradeDetails(BaseModel):

View File

@ -3,6 +3,7 @@ import uuid
from django.contrib.auth.mixins import LoginRequiredMixin
from django.http import HttpResponseBadRequest
from django.shortcuts import render
from django.urls import reverse
from django.views import View
from rest_framework.parsers import FormParser
from two_factor.views.mixins import OTPRequiredMixin
@ -64,12 +65,20 @@ class Positions(LoginRequiredMixin, OTPRequiredMixin, View):
def get(self, request, type, account_id=None):
if type not in self.allowed_types:
return HttpResponseBadRequest
template_name = f"wm/{type}.html"
self.template_name = f"wm/{type}.html"
unique = str(uuid.uuid4())[:8]
items = get_positions(request.user, account_id)
annotate_positions(items, request.user, return_order_ids=False)
orig_type = type
if type == "page":
type = "modal"
cast = {
"type": orig_type,
}
if account_id:
cast["account_id"] = account_id
list_url = reverse("positions", kwargs={**cast})
context = {
"title": f"Positions ({type})",
"unique": unique,
@ -79,8 +88,17 @@ class Positions(LoginRequiredMixin, OTPRequiredMixin, View):
"type": type,
"page_title": self.page_title,
"page_subtitle": self.page_subtitle,
"list_url": list_url,
"context_object_name_singular": "position",
"context_object_name": "positions",
}
return render(request, template_name, context)
# Return partials for HTMX
if self.request.htmx:
if orig_type == "page":
self.template_name = self.list_template
else:
context["window_content"] = self.list_template
return render(request, self.template_name, context)
class PositionAction(LoginRequiredMixin, OTPRequiredMixin, View):
@ -103,6 +121,10 @@ class PositionAction(LoginRequiredMixin, OTPRequiredMixin, View):
annotate_positions([info], request.user, return_order_ids=True)
)
# Remove some fields from the info dict
del info["long"]
del info["short"]
if type == "page":
type = "modal"
context = {
@ -115,3 +137,28 @@ class PositionAction(LoginRequiredMixin, OTPRequiredMixin, View):
}
return render(request, template_name, context)
def delete(self, request, account_id, side, symbol):
"""
Close a position.
"""
template_name = "partials/notify.html"
account = Account.get_by_id(account_id, request.user)
try:
api_response = account.client.close_position(side, symbol)
except GenericAPIError as e:
context = {"message": e, "class": "danger"}
return render(request, template_name, context)
if "longOrderCreateTransaction" in api_response:
context = {
"message": f"Long position closed on {symbol}",
"class": "success",
}
elif "shortOrderCreateTransaction" in api_response:
context = {
"message": f"Short position closed on {symbol}",
"class": "success",
}
response = render(request, template_name, context)
response["HX-Trigger"] = "positionEvent"
return response