You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
from typing import List
|
|
|
|
from fastapi import FastAPI, Depends, HTTPException
|
|
from sqlalchemy.orm import Session
|
|
|
|
from .database import SessionLocal, engine
|
|
from . import models, schemas, crud
|
|
|
|
models.Base.metadata.create_all(bind=engine)
|
|
app = FastAPI()
|
|
|
|
# Dependency
|
|
def get_db():
|
|
db = SessionLocal()
|
|
try:
|
|
yield db
|
|
finally:
|
|
db.close()
|
|
|
|
@app.get("/hosts/", response_model=List[schemas.Host])
|
|
def get_hosts(db: Session = Depends(get_db)):
|
|
hosts = crud.get_hosts(db)
|
|
|
|
return hosts
|
|
|
|
@app.get("/get_host_by_hostname/{hostname}")
|
|
def get_host_by_hostname(hostname: str, db: Session = Depends(get_db)):
|
|
return crud.get_host_by_hostname(db, hostname=hostname)
|
|
|
|
@app.get("/get_host_id_for_hostname/{hostname}")
|
|
def get_host_id_for_hostname(hostname: str, db: Session = Depends(get_db)):
|
|
return crud.get_host_id_for_hostname(db, hostname=hostname)
|
|
|
|
@app.post("/add_host/", response_model=schemas.HostCreate, status_code=201)
|
|
def create_host(host: schemas.HostCreate, db: Session = Depends(get_db)):
|
|
db_host = crud.get_host_by_hostname(db, hostname=host.hostname)
|
|
if db_host:
|
|
raise HTTPException(status_code=400, detail="Host already exists")
|
|
|
|
return crud.create_host(db=db, host=host)
|
|
|
|
@app.delete("/delete_host/{host_id}")
|
|
def delete_host(host_id: int, db: Session = Depends(get_db)):
|
|
return crud.delete_host(db, host_id=host_id)
|
|
|
|
@app.delete("/delete_host_by_hostname/{hostname}")
|
|
def delete_host_by_hostname(hostname: str, db: Session = Depends(get_db)):
|
|
return crud.delete_host_by_hostname(db, hostname=hostname)
|
|
|
|
@app.get("/get_cpus_by_host_id/{host_id}")
|
|
def get_cpus_by_host_id(host_id: int, db: Session = Depends(get_db)) -> List[schemas.CPU]:
|
|
return crud.get_cpus_by_host_id(db, host_id=host_id)
|
|
|
|
@app.post("/add_cpu/", response_model=schemas.CPUCreate, status_code=201)
|
|
def create_cpu(cpu: schemas.CPUCreate, db: Session = Depends(get_db)):
|
|
db_cpus = crud.get_cpus_by_host_id(db, host_id=cpu.host_id)
|
|
for db_cpu in db_cpus:
|
|
if cpu.socket == db_cpu.socket:
|
|
raise HTTPException(status_code=400, detail="CPU already exists")
|
|
|
|
return crud.create_cpu(db, cpu=cpu)
|