Python webargs 模塊的簡單使用

Posted on by

webargs是一個用於解析和驗證HTTP請求對象的Python庫，內置瞭對流行web框架的支持，包括Flask、Django、Bottle、Tornado、Pyramid、webapp2、Falcon和aiohttp。下面看下Python webargs 模塊詳解。

一、安裝

python3 -m pip install webargs

文檔

二、基礎特性

# encoding=utf-8
from flask import Flask
from webargs import fields
from webargs.flaskparser import use_args
app = Flask(__name__)
app.route("/")
@use_args({
    "name": fields.Str(required=True),
    "age": fields.Int(required=True),
}, location='query')
def index(args):
    print('args', args)
    return "Hello " + args["name"]
if __name__ == "__main__":
    app.run(debug=1)

2.1. 使用

2.1.1 通過裝飾器

@use_args({
    "name": fields.Str(required=True),
    "age": fields.Int(required=True),
}, location='query')
  • 第一個參數是需要獲取的字段名，類型，是否必須等的定義
  • location是指從哪裡獲取這些參數，默認是json，可選：
  • 'querystring' (same as 'query')
  • 'json'
  • 'form'
  • 'headers'
  • 'cookies'
  • 'files'

解析完後，把所有參數放在字典裡面，傳給下層函數

2.1.2 通過函數

args = parser.parse(user_args, request)

參數和裝飾器一樣，多瞭一傳request

2.2 參數檢查

from webargs import fields, validate

args_1 = {
    # 必須參數，字符串類型
    "username": fields.Str(required=True),
    # validate
    "password": fields.Str(validate=lambda p: len(p) >= 6),
    "password": fields.Str(validate=validate.Length(min=6)),
    # Default value when argument is missing
    "display_per_page": fields.Int(missing=10),
    # Repeated parameter, e.g. "/?nickname=Fred&nickname=Freddie"
    "nickname": fields.List(fields.Str()),
    # Delimited list, e.g. "/?languages=python,javascript"
    "languages": fields.DelimitedList(fields.Str()),
    # When value is keyed on a variable-unsafe name
    # or you want to rename a key
    "user_type": fields.Str(data_key="user-type"),
    "start_day": fields.DateTime(required=True, format='%Y-%m-%d %X'), 
    "bool": fields.Bool(),
    "email": fields.Email(),
    "ip": fields.IP(),
    "type": fields.Constant(constant='COMMON_TYPE'),
    "money": fields.Decimal(),
    "weight": fields.Float(),
    "url": fields.URL(),
    "uuid": fields.UUID(),
    "raw": fields.Raw(),
}
  • fields.Str 表示接收字符串參數
  • required=True 表示必傳
  • validate=lambda p: len(p) >= 6 表示自定義檢查函數。會把參數傳遞給該函數，該函數返回True表示檢查通過，返回False或者拋出異常表示檢查不通過
    • 如果要對多個參數進行聯合檢查，需要在裝飾器層架validate參數：@use_args(args_1, validate=lambda args: len(args["username"]) < len(args["password"]))
    • 異常需要是from webargs import ValidationError這個異常，不然會當程序異常處理
  • 也可以用validate庫裡面的內置檢查函數
  • missing=10 表示如果沒有入參，設置為默認值
  • fields.List(fields.Str()) 表示列表型參數，列表的元素是字符串
  • fields.DelimitedList(fields.Str()) 表示逗號型的列表參數
  • data_key="user-type" 表示字段名修改，入參是user-type，在args字典會改為user_type
  • fields.DateTime(required=True, format='%Y-%m-%d %X') 表示接收日期類型，並且格式需要符合，參數值會轉換為datetime類型
  • "bool": fields.Bool() 表示佈爾類型，傳1,0，true，false都能識別
  • fields.Email() 隻接收email，估計裡面會有正則檢查
  • fields.IP() 隻接收IP
  • fields.Constant(constant='COMMON_TYPE') 常量參數，無論入參是什麼值，type永遠等於COMMON_TYPE
  • fields.Decimal() 轉換為Decimal類型
  • fields.Float() 轉換為float類型
  • fields.URL() fields.UUID() 正則檢查url格式或者uuid格式
  • fields.Raw 不檢查參數類型

內置參數檢查

  • validate=validate.Length(min=1,max=10) 檢查字符串長度需要在某個區間
  • validate=validate.OneOf(['male', 'female']) 入參需要在枚舉裡面

2.3 檢查失敗處理

如果參數檢查失敗，會返回422響應，但是不會提示哪個參數有問題。我們可以通過Flask的異常處理機制，捕獲這個異常，然後構造我們想要的返回

@app.errorhandler(422) # 捕獲422和400的異常碼
@app.errorhandler(400)
def handle_error(err):
    headers = err.data.get("headers", None)
    messages = err.data.get("messages", ["Invalid request."])
    print(headers)
    print(messages)  # {'json': {'password': ['Shorter than minimum length 6.']}}
    return json.dumps({'err_code': 10000, 'err_msg': messages['json']})
  • 從err裡面獲取信息，headers不知道有什麼用的，message會有異常信息，例如不滿足validate=validate.Length(min=6)檢查，就會返回{'json': {'password': ['Shorter than minimum length 6.']}}
  • 如果是主動拋出的ValidationError異常，message會包含ValidationError異常的內容
  • 我們可以把這個參數檢查信息返回給前端，提示前端哪個參數錯誤瞭。
  • messages['json'] 的json是location的key

2.4 嵌套參數

對於一些復雜的，有多重嵌套的參數

"name": fields.Nested(
    {"first": fields.Str(required=True), "last": fields.Str(required=True)}
)
  • 表示name是一個嵌套參數，也就是字典
  • 然後裡面需要要first key和last key

三、高級特性

3.1 自定義location

上面說瞭location支持query，json這些，也可以自定義

@parser.location_loader("data")
def load_data(request, schema):
    data = {}
    data.update({k: request.args.get(k) for k in request.args})
    if request.json:
        data.update({k: request.json.get(k) for k in request.json})
    print(data, 'dataaaaa')
    return data

parser.location = 'data' # 設置默認的location為data
  • 上面定義瞭一個data的location，會合並args和json入參
  • 把默認的location修改為data

也可以這樣，這個是官方推薦方法：

@parser.location_loader("args_and_json")
def load_data(request, schema):
    from webargs.multidictproxy import MultiDictProxy
    newdata = request.args.copy()
    if request.json:
        newdata.update(request.json)
    return MultiDictProxy(newdata, schema)

3.2 定義schema

除瞭可以通過字典定義args，也可以通過類：

from marshmallow import Schema
class UserSchema(Schema):
    name = fields.Str(required=True)
    age = fields.Int()
@app.route("/")
@use_args(UserSchema())
def index1(args):
    print('args', args)
    return "Hello "

3.3 未定義的參數處理

如果入參有未定義的參數，webargs默認會拋出422異常

from webargs.flaskparser import parser
import marshmallow
parser.unknown = marshmallow.EXCLUDE  # 如果有未定義參數，不放到args參數，不拋異常
parser.unknown = marshmallow.INCLUDE  # 如果有未定義參數，放到args參數，不拋異常
  • 可以修改parse.unknown來修改策略。
  • 也可以精確設置不同location的unknown策略

3.4 Flask的url參數

@app.route("/<int:id>/")
@use_args(UserSchema())
def index1(args, id):
    print('args', args, id)
    return "Hello "

如果需要用到Flask的url參數，就需要這樣傳遞參數

到此這篇關於Python webargs 模塊的文章就介紹到這瞭,更多相關Python webargs 模塊內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：