当前位置: 代码迷 >> python >> 装饰器:flask / flask_login,使用login_required
  详细解决方案

装饰器:flask / flask_login,使用login_required

热度:93   发布时间:2023-06-14 08:44:46.0

我想在有和没有烧瓶login_required装饰器的情况下都使用渲染功能。 我的尝试如下:定义基本功能,然后分配给一个新的变量,并用login_required包装:

import pymongo
from flask import render_template
from flask_login import login_required

def base_page(collection='posts'):
    db = pymongo.MongoClient().www
    title = collection.capitalize()
    page = db[collection].find().sort("created", pymongo.DESCENDING)
    return render_template('home/home.html', **locals())

base_page_login_required = login_required(base_page)

然后,在我的app.add_url_rule调用中使用它们:

app.add_url_rule('/blog', view_func=base_page, defaults={'collection': 'posts'})
app.add_url_rule('/users', view_func=base_page_login_required, defaults={'collection': 'users'})

但显然我出了点问题:

Traceback (most recent call last):\r, referer: http://localhost/
   File "C:/www/flask/wsgi.py", line 26, in <module>\r, referer: http://localhost/
     app.add_url_rule('/users', view_func=base_page_login_required, defaults={'collection': 'users'})\r, referer: http://localhost/
   File "C:\\bin\\Python34\\lib\\site-packages\\flask\\app.py", line 62, in wrapper_func\r, referer: http://localhost/
     return f(self, *args, **kwargs)\r, referer: http://localhost/
   File "C:\\bin\\Python34\\lib\\site-packages\\flask\\app.py", line 984, in add_url_rule\r, referer: http://localhost/
     'existing endpoint function: %s' % endpoint)\r, referer: http://localhost/
 AssertionError: View function mapping is overwriting an existing endpoint function: base_page\r, referer: http://localhost/

谁能帮助我了解此错误? 我认为这是特定于flask和add_url_rule 我知道我可以将第一个功能代码复制到第二个并正常使用装饰器,但是我想遵循DRY。

Flask使用端点名称注册路由; 您可以在url_for()使用此名称来生成URL。

默认情况下,Flask为此使用函数名称,即function.__name__的值。 将函数分配给另一个名称(即使使用正确构造的装饰器进行装饰)也不会为函数对象提供新名称。

您可以使用endpoint关键字参数来为您的第二个注册赋予不同的名称:

app.add_url_rule('/blog', view_func=base_page, 
                 defaults={'collection': 'posts'})
app.add_url_rule('/users', view_func=base_page_login_required,
                 endpoint='users',
                 defaults={'collection': 'users'})