公开可重用函数来处理 HTTP POST 方法

Question

我正在使用 Flask 和 Python 编写各种 API 方法。例如，一种方法检查数据库以确定用户名是否已被使用。有点像这样：

@app.route('/register/checkuser', methods=['POST'])
def checkuser():

if request.method == "POST":

    conn = Connection('localhost', 27017)
    db = conn['user-data']
    userTable = db["logins"]

    userToCheck = request.form['usertocheck']

    #search for user to check if it already exists
    doesExist = str(userTable.find_one({"username": userToCheck}))
    conn.close()

    if doesExist == "None":
        return "Username is available"
    elif doesExist.find("ObjectId") != -1:
        return "Username already taken."
    else:
        return "Error"

简而言之，我希望能够从我的 Flask 应用程序的其他地方调用 checkuser() 函数（可能跟随其他装饰器）。例如，在我创建用户帐户之前。

我该怎么做呢？

谢谢。

Answer 1

将其包装在另一个函数中并将请求发送给该函数：

def checkuser(request):
    if request.method == "POST":
        conn = Connection('localhost', 27017)
        db = conn['user-data']
        userTable = db["logins"]

        userToCheck = request.form['usertocheck']

        #search for user to check if it already exists
        doesExist = str(userTable.find_one({"username": userToCheck}))
        conn.close()

        if doesExist == "None":
            return "Username is available"
        elif doesExist.find("ObjectId") != -1:
            return "Username already taken."
        else:
            return "Error"


@app.route('/register/checkuser', methods=['POST'])
def func():
    return checkUser(request)