新增期权交易记录功能,支持C/P类型和行权价
Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,108 @@
|
||||
from datetime import date
|
||||
from fastapi import APIRouter, Depends, Form, Request
|
||||
from fastapi.responses import HTMLResponse, RedirectResponse
|
||||
from sqlalchemy.orm import Session
|
||||
from app.database import get_db
|
||||
from app.models import OptionTrade, Contract, Product
|
||||
|
||||
router = APIRouter(prefix="/options", tags=["options"])
|
||||
|
||||
OPTION_TYPES = [("C", "C 看涨"), ("P", "P 看跌")]
|
||||
|
||||
|
||||
def get_product_contracts(db: Session) -> dict:
|
||||
result: dict[str, list[str]] = {}
|
||||
products = db.query(Product).order_by(Product.code).all()
|
||||
for p in products:
|
||||
codes = [c.code for c in p.contracts if c.is_active]
|
||||
if codes:
|
||||
result[p.code] = codes
|
||||
return result
|
||||
|
||||
|
||||
def today_str() -> str:
|
||||
return date.today().isoformat()
|
||||
|
||||
|
||||
@router.get("/", response_class=HTMLResponse)
|
||||
def option_page(request: Request, db: Session = Depends(get_db)):
|
||||
view = request.query_params.get("view", "")
|
||||
open_trades = (
|
||||
db.query(OptionTrade).filter(OptionTrade.status == "open")
|
||||
.order_by(OptionTrade.open_date.desc()).all()
|
||||
)
|
||||
closed_trades = (
|
||||
db.query(OptionTrade).filter(OptionTrade.status == "closed")
|
||||
.order_by(OptionTrade.close_date.desc()).limit(50).all()
|
||||
)
|
||||
|
||||
template = request.app.state.templates.get_template("options.html")
|
||||
return HTMLResponse(
|
||||
template.render(
|
||||
request=request,
|
||||
active_nav="options",
|
||||
product_contracts=get_product_contracts(db),
|
||||
option_types=OPTION_TYPES,
|
||||
open_trades=open_trades,
|
||||
closed_trades=closed_trades,
|
||||
today=today_str(),
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
@router.post("/open")
|
||||
def open_trade(
|
||||
request: Request,
|
||||
contract_code: str = Form(...),
|
||||
option_type: str = Form(...),
|
||||
strike_price: float = Form(...),
|
||||
open_date: str = Form(...),
|
||||
open_price: float = Form(...),
|
||||
open_fee: float = Form(0.0),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
code = contract_code.upper()
|
||||
contract = db.query(Contract).filter(Contract.code == code).first()
|
||||
product_code = contract.product.code if contract else code[:2]
|
||||
|
||||
t = OptionTrade(
|
||||
product_code=product_code,
|
||||
contract_code=code,
|
||||
option_type=option_type,
|
||||
strike_price=strike_price,
|
||||
open_date=date.fromisoformat(open_date),
|
||||
open_price=open_price,
|
||||
open_fee=open_fee,
|
||||
status="open",
|
||||
)
|
||||
db.add(t)
|
||||
db.commit()
|
||||
return RedirectResponse("/options/", status_code=303)
|
||||
|
||||
|
||||
@router.post("/{trade_id}/close")
|
||||
def close_trade(
|
||||
request: Request,
|
||||
trade_id: int,
|
||||
close_date: str = Form(...),
|
||||
close_price: float = Form(...),
|
||||
close_fee: float = Form(0.0),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
t = db.query(OptionTrade).filter(OptionTrade.id == trade_id).first()
|
||||
if t and t.status == "open":
|
||||
t.close_date = date.fromisoformat(close_date)
|
||||
t.close_price = close_price
|
||||
t.close_fee = close_fee
|
||||
t.status = "closed"
|
||||
db.commit()
|
||||
return RedirectResponse("/options/", status_code=303)
|
||||
|
||||
|
||||
@router.post("/{trade_id}/delete")
|
||||
def delete_trade(trade_id: int, db: Session = Depends(get_db)):
|
||||
t = db.query(OptionTrade).filter(OptionTrade.id == trade_id).first()
|
||||
if t:
|
||||
db.delete(t)
|
||||
db.commit()
|
||||
return RedirectResponse("/options/", status_code=303)
|
||||
Reference in New Issue
Block a user