remove old labeler & prompt REST endpoints

This commit is contained in:
Andreas Köpf
2022-12-16 15:23:49 +01:00
parent 05b677c16f
commit 869c8ebcba
13 changed files with 3 additions and 350 deletions
+1 -3
View File
@@ -1,10 +1,8 @@
# -*- coding: utf-8 -*-
from app.api.v1 import labelers, prompts, tasks, tasks2
from app.api.v1 import tasks, tasks2
from fastapi import APIRouter
api_router = APIRouter()
api_router.include_router(labelers.router, prefix="/labelers", tags=["labelers"])
api_router.include_router(prompts.router, prefix="/prompts", tags=["prompts"])
api_router.include_router(tasks.router, prefix="/tasks", tags=["tasks"])
api_router.include_router(tasks2.router, prefix="/task2", tags=["task2"]) # temporary
-114
View File
@@ -1,114 +0,0 @@
# -*- coding: utf-8 -*-
from typing import Any, List
from app import crud, schemas
from app.api import deps
from fastapi import APIRouter, Depends, HTTPException
from fastapi.security.api_key import APIKey
from sqlmodel import Session
from starlette.status import HTTP_400_BAD_REQUEST, HTTP_404_NOT_FOUND
router = APIRouter()
@router.get("/", response_model=List[schemas.Labeler])
def read_labelers(
db: Session = Depends(deps.get_db),
api_key: APIKey = Depends(deps.get_api_key),
begin_id: int = 0,
limit: int = 100,
) -> Any:
"""
Retrieve labelers.
"""
deps.api_auth(api_key, db)
if limit > 10000:
raise HTTPException(status_code=HTTP_400_BAD_REQUEST, detail="Bad request")
labelers = crud.labeler.get_multi(db, begin_id=begin_id, limit=limit)
return labelers
@router.post("/", response_model=schemas.Labeler)
def create_labeler(
*,
db: Session = Depends(deps.get_db),
api_key: APIKey = Depends(deps.get_api_key),
item_in: schemas.LabelerCreate,
) -> Any:
"""
Create new labeler.
"""
deps.api_auth(api_key, db)
item = crud.labeler.create(db=db, obj_in=item_in)
return item
@router.put("/{id}", response_model=schemas.Labeler)
def update_labeler(
*,
db: Session = Depends(deps.get_db),
api_key: APIKey = Depends(deps.get_api_key),
id: int,
item_in: schemas.LabelerUpdate,
) -> Any:
"""
Update a labeler.
"""
deps.api_auth(api_key, db)
item = crud.labeler.get(db=db, id=id)
if not item:
raise HTTPException(status_code=HTTP_404_NOT_FOUND, detail="Item not found")
item = crud.labeler.update(db=db, db_obj=item, obj_in=item_in)
return item
@router.get("/by-username", response_model=schemas.Labeler)
def read_labeler_by_username(
*,
db: Session = Depends(deps.get_db),
api_key: APIKey = Depends(deps.get_api_key),
discord_username: str,
) -> Any:
"""
Get labeler by ID.
"""
deps.api_auth(api_key, db)
item = crud.labeler.get_by_discord_username(db=db, discord_username=discord_username)
if not item:
raise HTTPException(status_code=404, detail="Item not found")
return item
@router.get("/{id}", response_model=schemas.Labeler)
def read_labeler(
*,
db: Session = Depends(deps.get_db),
api_key: APIKey = Depends(deps.get_api_key),
id: int,
) -> Any:
"""
Get labeler by ID.
"""
deps.api_auth(api_key, db)
item = crud.labeler.get(db=db, id=id)
if not item:
raise HTTPException(status_code=HTTP_404_NOT_FOUND, detail="Item not found")
return item
@router.delete("/{id}", response_model=schemas.Labeler)
def delete_labeler(
*,
db: Session = Depends(deps.get_db),
api_key: APIKey = Depends(deps.get_api_key),
id: int,
) -> Any:
"""
Delete a labeler.
"""
deps.api_auth(api_key, db)
labeler = crud.labeler.get(db=db, id=id)
if not labeler:
raise HTTPException(status_code=HTTP_404_NOT_FOUND, detail="Item not found")
labeler = crud.labeler.remove(db=db, id=id)
return labeler
-91
View File
@@ -1,91 +0,0 @@
# -*- coding: utf-8 -*-
from typing import Any, List
from app import crud, schemas
from app.api import deps
from fastapi import APIRouter, Depends, HTTPException
from fastapi.security.api_key import APIKey
from sqlmodel import Session
from starlette.status import HTTP_400_BAD_REQUEST, HTTP_401_UNAUTHORIZED, HTTP_404_NOT_FOUND
router = APIRouter()
@router.get("/", response_model=List[schemas.Prompt])
def read_prompts(
db: Session = Depends(deps.get_db),
api_key: APIKey = Depends(deps.get_api_key),
begin_id: int = 0,
limit: int = 1000,
) -> Any:
"""
Retrieve prompts.
"""
deps.api_auth(api_key, db)
if limit > 10000:
raise HTTPException(status_code=HTTP_400_BAD_REQUEST, detail="Bad request")
return crud.prompt.get_multi(db, begin_id=begin_id, limit=limit)
@router.post("/", response_model=schemas.Prompt)
def create_prompt(
*,
db: Session = Depends(deps.get_db),
api_key: APIKey = Depends(deps.get_api_key),
item_in: schemas.PromptCreate,
) -> Any:
"""
Create new prompt.
"""
deps.api_auth(api_key, db)
if item_in.labeler_id is None:
if item_in.discord_username is None:
raise HTTPException(status_code=HTTP_400_BAD_REQUEST, detail="Bad request")
labeler = crud.labeler.get_by_discord_username(db=db, discord_username=item_in.discord_username)
else:
labeler = crud.labeler.get(db=db, id=item_in.labeler_id)
if labeler is None:
raise HTTPException(status_code=HTTP_404_NOT_FOUND, detail="Invalid labeler user name")
if not labeler.is_enabled:
raise HTTPException(status_code=HTTP_401_UNAUTHORIZED, detail="Labeler disabled")
item_in.labeler_id = labeler.id
item_in.discord_username = None
item = crud.prompt.create(db=db, obj_in=item_in)
return item
@router.get("/{id}", response_model=schemas.Prompt)
def read_prompt(
*,
db: Session = Depends(deps.get_db),
api_key: APIKey = Depends(deps.get_api_key),
id: int,
) -> Any:
"""
Get prompt by ID.
"""
deps.api_auth(api_key, db)
item = crud.prompt.get(db=db, id=id)
if not item:
raise HTTPException(status_code=HTTP_404_NOT_FOUND, detail="Item not found")
return item
@router.delete("/{id}", response_model=schemas.Prompt)
def delete_prompt(
*,
db: Session = Depends(deps.get_db),
api_key: APIKey = Depends(deps.get_api_key),
id: int,
) -> Any:
"""
Delete a prompt.
"""
deps.api_auth(api_key, db)
item = crud.prompt.get(db=db, id=id)
if not item:
raise HTTPException(status_code=HTTP_404_NOT_FOUND, detail="Item not found")
item = crud.prompt.remove(db=db, id=id)
return item