gitscope
/
← marzneshin / app/routes/inbounds.py / blame
Files Commits 446 Branches 1 Tags 43 Markers 4 README License
https://dev.kronoz.win/r/marzneshin.git
app/routes/inbounds.py · v0.7.4
Source
1 d4ec019 Dawsh 2y ago from fastapi import APIRouter, Depends, Query
2 2de9d72 Dawsh 2y ago from fastapi import HTTPException
3 d4ec019 Dawsh 2y ago from fastapi_pagination.ext.sqlalchemy import paginate
4 d4ec019 Dawsh 2y ago from fastapi_pagination.links import Page
5 2de9d72 Dawsh 2y ago
6 2de9d72 Dawsh 2y ago from app.db import crud
7 d4ec019 Dawsh 2y ago from app.db.models import InboundHost as DBInboundHost, Inbound as DBInbound
8 2de9d72 Dawsh 2y ago from app.dependencies import DBDep, sudo_admin
9 2de9d72 Dawsh 2y ago from app.models.proxy import Inbound, InboundHost, InboundHostResponse
10 2de9d72 Dawsh 2y ago
11 03b3dbc Mardin 2y ago HOST_NOT_FOUND_ERROR_MSG = "Host not found"
12 03b3dbc Mardin 2y ago
13 01cdf2e Dawsh 2y ago router = APIRouter(
14 01cdf2e Dawsh 2y ago prefix="/inbounds", dependencies=[Depends(sudo_admin)], tags=["Inbounds"]
15 01cdf2e Dawsh 2y ago )
16 2de9d72 Dawsh 2y ago
17 2de9d72 Dawsh 2y ago
18 d4ec019 Dawsh 2y ago @router.get("", response_model=Page[Inbound])
19 d4ec019 Dawsh 2y ago def get_inbounds(db: DBDep, tag: str = Query(None)):
20 2de9d72 Dawsh 2y ago """
21 2de9d72 Dawsh 2y ago Get all inbounds
22 2de9d72 Dawsh 2y ago """
23 d4ec019 Dawsh 2y ago query = db.query(DBInbound)
24 d4ec019 Dawsh 2y ago if tag:
25 d4ec019 Dawsh 2y ago query = query.filter(DBInbound.tag.ilike(f"%{tag}%"))
26 2de9d72 Dawsh 2y ago
27 d4ec019 Dawsh 2y ago return paginate(db, query)
28 2de9d72 Dawsh 2y ago
29 2de9d72 Dawsh 2y ago
30 7428e12 Mrdn 2y ago @router.get("/hosts", response_model=Page[InboundHostResponse])
31 2de9d72 Dawsh 2y ago def get_hosts(db: DBDep):
32 d4ec019 Dawsh 2y ago return paginate(db.query(DBInboundHost))
33 2de9d72 Dawsh 2y ago
34 2de9d72 Dawsh 2y ago
35 4056c66 Dawsh 1y ago @router.post("/hosts", response_model=InboundHostResponse)
36 4056c66 Dawsh 1y ago def create_unbound_host(host: InboundHost, db: DBDep):
37 4056c66 Dawsh 1y ago """
38 4056c66 Dawsh 1y ago Add a host without an inbound
39 4056c66 Dawsh 1y ago """
40 4056c66 Dawsh 1y ago return crud.add_host(db, None, host)
41 4056c66 Dawsh 1y ago
42 4056c66 Dawsh 1y ago
43 03b3dbc Mardin 2y ago @router.get("/hosts/{id}", response_model=InboundHostResponse)
44 03b3dbc Mardin 2y ago def get_host(id: int, db: DBDep):
45 03b3dbc Mardin 2y ago """
46 03b3dbc Mardin 2y ago Get a host
47 03b3dbc Mardin 2y ago """
48 03b3dbc Mardin 2y ago host = crud.get_host(db, id)
49 03b3dbc Mardin 2y ago if not host:
50 03b3dbc Mardin 2y ago raise HTTPException(status_code=404, detail=HOST_NOT_FOUND_ERROR_MSG)
51 03b3dbc Mardin 2y ago
52 03b3dbc Mardin 2y ago return host
53 03b3dbc Mardin 2y ago
54 03b3dbc Mardin 2y ago
55 2de9d72 Dawsh 2y ago @router.put("/hosts/{id}", response_model=InboundHostResponse)
56 2de9d72 Dawsh 2y ago def update_host(id: int, host: InboundHost, db: DBDep):
57 2de9d72 Dawsh 2y ago """
58 2de9d72 Dawsh 2y ago Modify a host
59 2de9d72 Dawsh 2y ago """
60 2de9d72 Dawsh 2y ago
61 2de9d72 Dawsh 2y ago db_host = crud.get_host(db, id)
62 2de9d72 Dawsh 2y ago if not db_host:
63 03b3dbc Mardin 2y ago raise HTTPException(status_code=404, detail=HOST_NOT_FOUND_ERROR_MSG)
64 2de9d72 Dawsh 2y ago
65 2de9d72 Dawsh 2y ago return crud.update_host(db, db_host, host)
66 2de9d72 Dawsh 2y ago
67 2de9d72 Dawsh 2y ago
68 2de9d72 Dawsh 2y ago @router.delete("/hosts/{id}")
69 2de9d72 Dawsh 2y ago def delete_host(id: int, db: DBDep):
70 2de9d72 Dawsh 2y ago """
71 2de9d72 Dawsh 2y ago Remove a host
72 2de9d72 Dawsh 2y ago """
73 2de9d72 Dawsh 2y ago db_host = crud.get_host(db, id)
74 2de9d72 Dawsh 2y ago if not db_host:
75 03b3dbc Mardin 2y ago raise HTTPException(status_code=404, detail=HOST_NOT_FOUND_ERROR_MSG)
76 2de9d72 Dawsh 2y ago
77 2de9d72 Dawsh 2y ago db.delete(db_host)
78 2de9d72 Dawsh 2y ago db.commit()
79 2de9d72 Dawsh 2y ago return {}
80 2de9d72 Dawsh 2y ago
81 2de9d72 Dawsh 2y ago
82 2de9d72 Dawsh 2y ago @router.get("/{id}", response_model=Inbound)
83 2de9d72 Dawsh 2y ago def get_inbound(id: int, db: DBDep):
84 2de9d72 Dawsh 2y ago """
85 2de9d72 Dawsh 2y ago Get a specific inbound
86 2de9d72 Dawsh 2y ago """
87 2de9d72 Dawsh 2y ago inbound = crud.get_inbound(db, id)
88 2de9d72 Dawsh 2y ago if not inbound:
89 2de9d72 Dawsh 2y ago raise HTTPException(status_code=404, detail="Inbound not found")
90 2de9d72 Dawsh 2y ago
91 2de9d72 Dawsh 2y ago return inbound
92 2de9d72 Dawsh 2y ago
93 2de9d72 Dawsh 2y ago
94 d4ec019 Dawsh 2y ago @router.get("/{id}/hosts", response_model=Page[InboundHostResponse])
95 f4fecee Dawsh 2y ago def get_inbound_hosts(id: int, db: DBDep):
96 2de9d72 Dawsh 2y ago """
97 2de9d72 Dawsh 2y ago Get hosts of a specific inbound
98 2de9d72 Dawsh 2y ago """
99 2de9d72 Dawsh 2y ago inbound = crud.get_inbound(db, id)
100 2de9d72 Dawsh 2y ago if not inbound:
101 2de9d72 Dawsh 2y ago raise HTTPException(status_code=404, detail="Inbound not found")
102 2de9d72 Dawsh 2y ago
103 f1ab16c Dawsh 2y ago return paginate(
104 f1ab16c Dawsh 2y ago db, db.query(DBInboundHost).filter(DBInboundHost.inbound_id == id)
105 f1ab16c Dawsh 2y ago )
106 2de9d72 Dawsh 2y ago
107 2de9d72 Dawsh 2y ago
108 2de9d72 Dawsh 2y ago @router.post("/{id}/hosts", response_model=InboundHostResponse)
109 2de9d72 Dawsh 2y ago def create_host(id: int, host: InboundHost, db: DBDep):
110 2de9d72 Dawsh 2y ago """
111 2de9d72 Dawsh 2y ago Add a host to the inbound
112 2de9d72 Dawsh 2y ago """
113 2de9d72 Dawsh 2y ago inbound = crud.get_inbound(db, id)
114 2de9d72 Dawsh 2y ago if not inbound:
115 2de9d72 Dawsh 2y ago raise HTTPException(status_code=404, detail="Inbound not found")
116 2de9d72 Dawsh 2y ago
117 2de9d72 Dawsh 2y ago return crud.add_host(db, inbound, host)