FastAPI_Mini/Query/schema_3.py at main · Abdo-Mohammed-10/FastAPI_Mini · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
from pydantic import BaseModel
from enum import Enum

class GenreChoice(str, Enum):
    ROCK = "rock"
    POP = "pop"
    JAZZ = "jazz"
    CLASSICAL = "classical"
    HIPHOP = "hip-hop"
    COUNTRY = "country"
    ELECTRONIC = "electronic"
    REGGAE = "reggae"
    BLUES = "blues"
    METAL = "metal"

class Album(BaseModel):
    id: int
    title: str
    band_id: int
    year: int

class Band(BaseModel):
    id: int
    name: str
    genre: str
    albums: list[Album] | None = None

    class Config:
        orm_mode = True