使用Flask開發簡單接口(2)--POST請求接口

今天咱們繼續學習如何使用Flask開發POST接口：用戶註冊接口和用戶登陸接口。

request接收參數

當咱們在頁面發出一個POST請求，請求傳到服務器時，須要如何拿到當前請求的數據呢？在Flask可使用 request 來接收請求參數。

request導入方式

from flask import request

使用 request 來接收參數的方式有多種，好比：

request.form：用於接收表單參數
request.args：用於接收GET參數
request.json：用於接收JSON參數
request.values：獲取全部參數（表單參數+GET參數）
request.file：用於接收文件

固然 request 還有很多其餘屬性，這裏就不一一列出了，咱們在這裏使用了 request.json 和 request.values 來開發POST接口，若是要拿到具體某個參數的值，則是經過 request.json.get("key") 或 request.values.get("key") 來完成。

用戶註冊接口

@app.route("/register", methods=['POST'])
def user_register():
    username = request.json.get("username").strip() # 用戶名
    password = request.json.get("password").strip() # 密碼
    sex = request.json.get("sex", "0").strip() # 性別，默認爲0(男性)
    telephone = request.json.get("telephone", "").strip() # 手機號，默認爲空串
    address = request.json.get("address", "").strip() # 地址，默認爲空串
    if username and password and telephone:
        if username == "wintest":
            return jsonify({"code": 2002, "msg": "用戶名已存在！！！"})
        elif not (sex == "0" or sex == "1"):
            return jsonify({"code": 2003, "msg": "輸入的性別只能是 0(男) 或 1(女)！！！"})
        elif not (len(telephone) == 11 and re.match("^1[3,5,7,8]\d{9}$", telephone)):
            return jsonify({"code": 2004, "msg": "手機號格式不正確！！！"})
        else:
            return jsonify({"code": 0, "msg": "恭喜，註冊成功！"})
    else:
        return jsonify({"code": 2001, "msg": "用戶名/密碼/手機號不能爲空，請檢查！！！"})

在這裏，咱們接收參數是經過 request.json.get("key") 來實現，在請求的時候，由於直接在瀏覽器上輸入網址訪問資源都是GET方式，因此咱們能夠經過其餘工具來模擬，如Postman、Fiddler均可以。

正則表達式 "^1[3,5,7,8]\d{9}$" 簡單分析：

^ 表示一個字符串的開始
1 表示第一個字符是數字1
[3,5,7,8] 表示第二個字符只是數字 3/5/7/8 中的一個
\d{9} 表示後面後面只能匹配數字，且只能有9位數字
$ 表示一個字符串的結束

注意：該正則表達式是一個不徹底的手機號驗證，在這裏只能較簡單的驗證手機號的合法性。

經過Fiddler發送請求模擬：

注意，在這裏由於咱們沒有操做數據庫，因此只是返回了註冊成功的信息，沒有作其餘處理。後面引入數據庫的時候，應該還須要作其餘處理，好比將註冊用戶信息插入到數據庫中，檢查 用戶/手機號 是否已註冊。

用戶登陸接口

@app.route("/login", methods=['POST'])
def user_login():
    username = request.values.get("username")
    password = request.values.get("password")
    if username and password:
        if username == "wintest" and password == "123456":
            return jsonify({"code": 0, "msg": "恭喜，登陸成功！"})
        return jsonify({"code": 1002, "msg": "用戶名或密碼錯誤！！！"})
    else:
        return jsonify({"code": 1001, "msg": "用戶名或密碼不能爲空！！！"})

在這裏，咱們接收參數是經過 request.values.get("key") 來實現，由於使用了 request.values，那麼咱們的請求頭參數中，Content-Type參數就不能用 application/json 了，咱們可使用表單提交的格式：application/x-www-form-urlencoded ，而 body 中參數的格式爲：key1=value1&key2=value2 。

這裏也能夠把請求數據寫到請求路徑下：

OK，以上咱們使用Flask開發了2個簡單的POST請求接口，但並無鏈接到數據庫進行操做，好比用戶註冊/登陸接口中，驗證的用戶名都是固定的，不夠靈活。下一節，咱們將引入數據庫來進一步完善。

源碼附上：

from flask import Flask, jsonify, request

app = Flask(__name__)
app.config["JSON_AS_ASCII"] = False  # jsonify返回的中文正常顯示

@app.route("/register", methods=['POST'])
def user_register():
    username = request.json.get("username").strip() # 用戶名
    password = request.json.get("password").strip() # 密碼
    sex = request.json.get("sex", "0").strip() # 性別，默認爲0(男性)
    telephone = request.json.get("telephone", "").strip() # 手機號，默認爲空
    address = request.json.get("telphone", "").strip() # 地址，默認爲空
    if username and password and telephone:
        import re
        if username == "wintest":
            return jsonify({"code": 2002, "msg": "用戶名已存在！！！"})
        elif not (sex == "0" or sex == "1"):
            return jsonify({"code": 2003, "msg": "輸入的性別只能是 0(男) 或 1(女)！！！"})
        elif not (len(telephone) == 11 and re.match("^1[3,5,7,8]\d{9}$", telephone)):
            return jsonify({"code": 2004, "msg": "手機號格式不正確！！！"})
        else:
            return jsonify({"code": 0, "msg": "恭喜，註冊成功！"})
    else:
        return jsonify({"code": 2001, "msg": "用戶名/密碼/手機號不能爲空，請檢查！！！"})

@app.route("/login", methods=['POST'])
def user_login():
    username = request.values.get("username")
    password = request.values.get("password")
    if username and password:
        if username == "wintest" and password == "123456":
            return jsonify({"code": 0, "msg": "恭喜，登陸成功！"})
        return jsonify({"code": 1002, "msg": "用戶名或密碼錯誤！！！"})
    else:
        return jsonify({"code": 1001, "msg": "用戶名或密碼不能爲空！！！"})


if __name__ == '__main__':
    app.run()