首頁>技術>

0

前言

前幾天給大家分別分享了(入門篇)簡析Python web框架FastAPI——一個比Flask和Tornada更高效能的API 框架和(進階篇)Python web框架FastAPI——一個比Flask和Tornada更高效能的API 框架。今天歡迎大家來到 FastAPI 系列分享的完結篇,本文主要是對於前面文章的補充和擴充套件。

當然這些功能在實際開發中也扮演者極其重要的角色。

1

中介軟體的使用

Flask 有 鉤子函式,可以對某些方法進行裝飾,在某些全域性或者非全域性的情況下,增添特定的功能。

同樣在 FastAPI 中也存在著像鉤子函式的東西,也就是中介軟體 Middleware了。

計算回撥時間

# -*- coding: UTF-8 -*-import timefrom fastapi import FastAPIfrom starlette.requests import Requestapp = FastAPI()@app.middleware("http")async def add_process_time_header(request: Request, call_next):    start_time = time.time()    response = await call_next(request)    process_time = time.time() - start_time    response.headers["X-Process-Time"] = str(process_time)    print(response.headers)    return [email protected]("/")async def main():    return {"message": "Hello World"}if __name__ == '__main__':    import uvicorn    uvicorn.run(app, host="127.0.0.1", port=8000)

請求重定向中介軟體

from fastapi import FastAPIfrom starlette.middleware.httpsredirect import HTTPSRedirectMiddlewareapp = FastAPI()app.add_middleware(HTTPSRedirectMiddleware)# 被重定向到 [email protected]("/")async def main():    return {"message": "Hello World"}

授權允許 Host 訪問列表(支援萬用字元匹配)

from fastapi import FastAPIfrom starlette.middleware.trustedhost import TrustedHostMiddlewareapp = FastAPI()app.add_middleware(    TrustedHostMiddleware, allowed_hosts=["example.com", "*.example.com"])@app.get("/")async def main():    return {"message": "Hello World"}

跨域資源共享

from fastapi import FastAPIfrom starlette.middleware.cors import CORSMiddlewareapp = FastAPI()#允許跨域請求的域名列表(不一致的埠也會被視為不同的域名)origins = [    "https://gzky.live",    "https://google.com",    "http://localhost:5000",    "http://localhost:8000",]# 萬用字元匹配,允許域名和方法app.add_middleware(    CORSMiddleware,    allow_origins=origins,       allow_credentials=True,     allow_methods=["*"],       allow_headers=["*"],   )

在前端 ajax 請求,出現了外部連結的時候就要考慮到跨域的問題,如果不設定允許跨域,瀏覽器就會自動報錯,跨域資源 的安全問題。

所以,中介軟體的應用場景還是比較廣的,比如爬蟲,有時候在做全站爬取時抓到的 Url 請求結果為 301,302, 之類的重定向狀態碼,那就有可能是網站管理員設定了該域名(二級域名) 不在 Host 訪問列表 中而做出的重定向處理,當然如果你也是網站的管理員,也能根據中介軟體做些反爬的措施。

更多中介軟體參考 https://fastapi.tiangolo.com/advanced/middleware

2

BackgroundTasks

建立非同步任務函式,使用 async 或者普通 def 函式來對後端函式進行呼叫。

傳送訊息

# -*- coding: UTF-8 -*-from fastapi import BackgroundTasks, Depends, FastAPIapp = FastAPI()def write_log(message: str):    with open("log.txt", mode="a") as log:        log.write(message)def get_query(background_tasks: BackgroundTasks, q: str = None):    if q:        message = f"found query: {q}\\n"        background_tasks.add_task(write_log, message)    return [email protected]("/send-notification/{email}")async def send_notification(    email: str, background_tasks: BackgroundTasks, q: str = Depends(get_query)):    message = f"message to {email}\\n"    background_tasks.add_task(write_log, message)    return {"message": "Message sent"}

使用方法極其的簡單,也就不多廢話了,write_log 當成 task 方法被呼叫,先方法名,後傳參。

3

自定義 Response 狀態碼

在一些特殊場景我們需要自己定義返回的狀態碼

from fastapi import FastAPIfrom starlette import statusapp = FastAPI()# [email protected]("/201/", status_code=status.HTTP_201_CREATED)async def item201():    return {"httpStatus": 201}# [email protected]("/302/", status_code=status.HTTP_302_FOUND)async def items302():    return {"httpStatus": 302}# [email protected]("/404/", status_code=status.HTTP_404_NOT_FOUND)async def items404():    return {"httpStatus": 404}# [email protected]("/500/", status_code=status.HTTP_500_INTERNAL_SERVER_ERROR)async def items500():    return {"httpStatus": 500}if __name__ == '__main__':    import uvicorn    uvicorn.run(app, host="127.0.0.1", port=8000)

這麼一來就有趣了,設想有個人寫了這麼一段程式碼

async def getHtml(self, url, session):    try:        async with session.get(url, headers=self.headers, timeout=60, verify_ssl=False) as resp:            if resp.status in [200, 201]:                data = await resp.text()                return data    except Exception as e:        print(e)        pass

那麼就有趣了,這段獲取 Html 原始碼的函式根據 Http狀態碼 來判斷是否正常的返回。那如果根據上面的寫法,我直接返回一個 404 或者 304 的狀態碼,但是響應資料卻正常,那麼這個爬蟲豈不是什麼都爬不到了麼。所以,嘿嘿你懂的!!

4

關於部署

部署 FastAPI 應用程式相對容易

Uvicorn

FastAPI 文件推薦使用 Uvicorn 來部署應用( 其次是 hypercorn),Uvicorn 是一個基於 asyncio 開發的一個輕量級高效的 Web 伺服器框架(僅支援 python 3.5.3 以上版本)

安裝

pip install uvicorn

啟動方式

uvicorn main:app --reload --host 0.0.0.0 --port 8000

Gunicorn

如果你仍然喜歡用 Gunicorn 在部署專案的話,請看下面

安裝

pip install gunicorn

啟動方式

gunicorn -w 4 -b 0.0.0.0:5000  manage:app -D

Docker部署

採用 Docker 部署應用的好處就是不用搭建特定的執行環境(實際上就是 docker 在幫你拉取),通過 Dockerfile 構建 FastAPI 映象,啟動 Docker 容器,通過埠對映可以很輕鬆訪問到你部署的應用。

Nginx

在 Uvicorn/Gunicorn + FastAPI 的基礎上掛上一層 Nginx 服務,一個網站就可以上線了,事實上直接使用 Uvicorm 或 Gunicorn 也是沒有問題的,但 Nginx 能讓你的網站看起來更像網站。

撒花 !!!

最新評論
  • BSA-TRITC(10mg/ml) TRITC-BSA 牛血清白蛋白改性標記羅丹明
  • Android完整知識體系路線(菜鳥-資深-大牛必進之路)