CODEWITHSUNDEEP
pythonhttprequestfastapistarlette
0x55b1E06FF
0x55b1E06FF

Reputation: 842

How to share variables between HTTP requests in FastAPI?

How can I share the value of variables between HTTP requests in FastAPI? For instance, I have a POST request in which I get some audio files and then I convert their info into a Pandas Dataframe. I would like to send that Dataframe in a GET request, but I can't access the Dataframe on the GET request scope.

@app.post(
    path="/upload-audios/",
    status_code=status.HTTP_200_OK
)
async def upload_audios(audios: list[UploadFile] = File(...)):
    filenames = [audio.filename for audio in audios]
    audio_data = [audio.file for audio in audios]
    new_data = []
    final_data = []
    header = ["name", "file"]
    for i in range(len(audios)):
        new_data = [filenames[i], audio_data[i]]
        final_data.append(new_data)
    new_df = pd.DataFrame(final_data, columns=header)
    return f"You have uploaded {len(audios)} audios which names are: {filenames}"

@app.get("/get-dataframe/")
async def get_dataframe():
    pass

Upvotes: 12

Views: 17109

Answers (1)

Chris
Chris

Reputation: 34045

If you need read-only access to that variable, and/or you never expect it to be modified by some other request before reading it (in other words, you never expect to serve more than one client), as well as your app does not use several workers—since each worker has its own things, variables and memory)—you could either (as mentioned by @MatsLindh in the comments above) declare a dictionary, e.g., foo = {}, outside the endpoints (in other words, globally) and assign a key to it inside the endpoint, e.g., foo['pd'] = df, (which you can later retrieve when another request arrives), or declare your variable as global (as described here), or, preferably, store the variable on the app instance, which allows you to store arbitrary extra state using the generic app.state attribute—see this answer on how to do this before the application starts up. For example:

from fastapi import FastAPI

app = FastAPI()

@app.post("/create")
async def create_dataframe():
    # df = ...  # create the df here
    app.state.df = df

Then, inside the get_dataframe endpoint, you can retrieve the df like this:

@app.get("/get")
async def get_dataframe():
    df = app.state.df

or, if the app instance is not available in the file from which you are working (let's say you have your endpoints defined in submodules, separately from the main module, as described here), you could get the app instance from the Request object:

from fastapi import Request

@app.get('/get')
async def get_dataframe(request: Request):
    df = request.app.state.df

Otherwise, if you needed that variable/object to be shared among different clients, as well as among multiple processes/workers, that may also require read/write access to it, you should rather use a database storage, such as PostgreSQL, SQLite, MongoDB, etc., or Key-Value stores (Caches), such as Redis, Memcached, etc. You may want to have a look at this answer as well.

Upvotes: 17

Related Questions