28 lines
496 B
Python
28 lines
496 B
Python
|
from typing import Optional
|
||
|
|
|
||
|
|
from fastapi import APIRouter
|
||
|
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
from ..rotation_service import rotate as rotate_impl, status as status_impl
|
||
|
|
|
||
|
|
|
||
|
|
router = APIRouter()
|
||
|
|
|
||
|
|
|
||
|
|
class RotateRequest(BaseModel):
|
||
|
|
cityhash: Optional[str] = None
|
||
|
|
num: Optional[int] = None
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/rotate")
|
||
|
|
def rotate(req: RotateRequest):
|
||
|
|
result = rotate_impl(cityhash=req.cityhash, num=req.num)
|
||
|
|
return result
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/status")
|
||
|
|
def get_status():
|
||
|
|
return status_impl()
|
||
|
|
|
||
|
|
|