main.py

from enum import Enum
from fastapi import FastAPI, HTTPException
from pydantic import BaseModel

app = FastAPI()


class DogType(str, Enum):
    terrier = "terrier"
    bulldog = "bulldog"
    dalmatian = "dalmatian"


class Dog(BaseModel):
    name: str
    pk: int
    kind: DogType


class Timestamp(BaseModel):
    id: int
    timestamp: int


dogs_db = {
    0: Dog(name='Bob', pk=0, kind='terrier'),
    1: Dog(name='Marli', pk=1, kind="bulldog"),
    2: Dog(name='Snoopy', pk=2, kind='dalmatian'),
    3: Dog(name='Rex', pk=3, kind='dalmatian'),
    4: Dog(name='Pongo', pk=4, kind='dalmatian'),
    5: Dog(name='Tillman', pk=5, kind='bulldog'),
    6: Dog(name='Uga', pk=6, kind='bulldog')
}

post_db = [
    Timestamp(id=0, timestamp=12),
    Timestamp(id=1, timestamp=10)
]


@app.get("/")
def get_root():
    return {"Hello!"}

@app.post("/post")
def get_post():
    return post_db

@app.get("/dog")
def get_dogs(kind: DogType = None):
    if kind is not None:
        return [dog for dog in dogs_db.values() if dog.kind == kind]
    else:
        return list(dogs_db.values())

@app.post("/dog")
def create_dog(dog: Dog):
    if dog.pk in list(dogs_db.keys()):
        raise HTTPException(status_code=409,
                            detail='The specified PK already exists.')
    
    dogs_db[dog.pk] = dog
    return dog

@app.get("/dog/{pk}")
def get_dog_by_pk(pk: int):
    if pk in dogs_db:
        return dogs_db[pk]
    else:
        raise HTTPException(status_code=409,
                            detail='Dog not found')

@app.patch("/dog/{pk}")
def update_dog(pk: int, dog: Dog):
    if pk in dogs_db:
        dogs_db[pk] = dog
        return dog
    else:
        raise HTTPException(status_code=409,
                            detail='Dog not found')