一个endpointfastapi中的多种功能

Question

我试图创建一个音频服务器，我可以上传各种音频文件，我有一个要求，我只能创建一个endpoint创建，我想出了这个，但它确实显示了输入数据的请求表单。

class AudioType(str, Enum):
    Song = "Song"
    Podcast = "Podcast"
    Audiobook = "Audiobook"


@app.post("/{audio_type}", status_code=status.HTTP_200_OK)
def audio(audio_type: AudioType):
    if audio_type == AudioType.Song:
        def create_song(request: schemas.Song, db: Session = Depends(database.get_db)):
            new_song = models.Song(name=request.name, duration=request.duration, uploadTime=request.uploadTime)
            db.add(new_song)
            db.commit()
            db.refresh(new_song)
            return new_song
    elif audio_type == AudioType.Podcast:
        def create_podcast(request: schemas.Podcast, db: Session = Depends(database.get_db)):
            new_podcast = models.Podcast(name=request.name,  duration=request.duration, uploadTime=request.uploadTime,                                         host=request.host)
            db.add(new_podcast)
            db.commit()
            db.refresh(new_podcast)
            return new_podcast
    elif audio_type == AudioType.Audiobook:
        def create_audiobook(request: schemas.Audiobook, db: Session = Depends(database.get_db)):
            new_audiobook = models.Audiobook(titile=request.title, author=request.author, narrator=request.narrator,                                              duration=request.duration, uploadTime=request.uploadTime)
            db.add(new_audiobook)
            db.commit()
            db.refresh(new_audiobook)
            return new_audiobook

Answer 1

您的方法不接受< code >请求对象，而只接受< code>audio_type。< br >另外，根据我对您的代码的理解，您可能有多个请求体(您所指的< code > schema )

你想要什么有两个选择:

>

您需要按如下方式声明endpoint：

from typing import Union

@app.post("/{audio_type}", status_code=status.HTTP_200_OK)
def audio(
    request: Union[schemas.Song, schemas.Podcast, 
    schemas.Audiobook], audio_type: AudioType
):
    ... Your method ...

但是fastapi的自动炫耀不会提供模式示例，您必须手动提供示例（这可能是可能的，也可能是不可能的，我不知道，也没有尝试过：/）

或者，您可以有一个架构，该架构可以接受所有内容作为可选和audio_type参数：

from typing import Optional


class EndpointSchema(BaseModel):
    audio_type: AudioType
    song: Optional[schemas.Song]
    podcast: Optional[schemas.Podcast]
    audiobook: Optional[schemas.Audiobook]


@app.post("/audio", status_code=status.HTTP_200_OK)
def audio(request_body: EndpointSchema):
    if request_body.audio_type == AudioType.Song:
        ... Continue with your request processing ...

最后，非常重要：您声明了之后不再调用的内部方法（create_song等），因此您的代码将不执行任何操作。您不需要这样做，直接在if，elif ...块中使用要创建歌曲，播客或有声读物的代码！