108 lines
3.1 KiB
Python
108 lines
3.1 KiB
Python
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 Product, Contract
|
|
|
|
router = APIRouter(prefix="/admin", tags=["admin"])
|
|
|
|
EXCHANGES = ["CZCE", "DCE", "SHFE", "CFFEX", "INE"]
|
|
|
|
|
|
@router.get("/", response_class=HTMLResponse)
|
|
def admin_page(request: Request, db: Session = Depends(get_db)):
|
|
products = db.query(Product).order_by(Product.code).all()
|
|
|
|
product_data = []
|
|
for p in products:
|
|
contracts = (
|
|
db.query(Contract)
|
|
.filter(Contract.product_id == p.id)
|
|
.order_by(Contract.code)
|
|
.all()
|
|
)
|
|
product_data.append({
|
|
"id": p.id,
|
|
"code": p.code,
|
|
"name": p.name,
|
|
"exchange": p.exchange,
|
|
"contracts": [
|
|
{"id": c.id, "code": c.code, "name": c.name, "is_active": c.is_active}
|
|
for c in contracts
|
|
],
|
|
})
|
|
|
|
template = request.app.state.templates.get_template("admin.html")
|
|
return HTMLResponse(
|
|
template.render(
|
|
request=request,
|
|
active_nav="admin",
|
|
products=product_data,
|
|
exchanges=EXCHANGES,
|
|
)
|
|
)
|
|
|
|
|
|
@router.post("/product")
|
|
def create_product(
|
|
request: Request,
|
|
code: str = Form(...),
|
|
name: str = Form(...),
|
|
exchange: str = Form(...),
|
|
db: Session = Depends(get_db),
|
|
):
|
|
existing = db.query(Product).filter(Product.code == code.upper()).first()
|
|
if not existing:
|
|
p = Product(code=code.upper(), name=name, exchange=exchange)
|
|
db.add(p)
|
|
db.commit()
|
|
return RedirectResponse("/admin/", status_code=303)
|
|
|
|
|
|
@router.post("/contract")
|
|
def create_contract(
|
|
request: Request,
|
|
product_id: int = Form(...),
|
|
code: str = Form(...),
|
|
name: str = Form(...),
|
|
db: Session = Depends(get_db),
|
|
):
|
|
existing = db.query(Contract).filter(Contract.code == code.upper()).first()
|
|
if not existing:
|
|
c = Contract(
|
|
product_id=product_id,
|
|
code=code.upper(),
|
|
name=name,
|
|
is_active=True,
|
|
)
|
|
db.add(c)
|
|
db.commit()
|
|
return RedirectResponse("/admin/", status_code=303)
|
|
|
|
|
|
@router.post("/contract/{contract_id}/toggle")
|
|
def toggle_contract(contract_id: int, db: Session = Depends(get_db)):
|
|
c = db.query(Contract).filter(Contract.id == contract_id).first()
|
|
if c:
|
|
c.is_active = not c.is_active
|
|
db.commit()
|
|
return RedirectResponse("/admin/", status_code=303)
|
|
|
|
|
|
@router.post("/contract/{contract_id}/delete")
|
|
def delete_contract(contract_id: int, db: Session = Depends(get_db)):
|
|
c = db.query(Contract).filter(Contract.id == contract_id).first()
|
|
if c:
|
|
db.delete(c)
|
|
db.commit()
|
|
return RedirectResponse("/admin/", status_code=303)
|
|
|
|
|
|
@router.post("/product/{product_id}/delete")
|
|
def delete_product(product_id: int, db: Session = Depends(get_db)):
|
|
p = db.query(Product).filter(Product.id == product_id).first()
|
|
if p:
|
|
db.delete(p)
|
|
db.commit()
|
|
return RedirectResponse("/admin/", status_code=303)
|