如何使用mongokit / pymongo填充wtform选择字段?

时间:2015-01-25 06:21:51

标签: python flask pymongo wtforms flask-wtforms

我尝试使用mongodb查询创建一个SelectField,但到目前为止我还没有成功:

# forms.py in blueprint
CATEGORIES = []
for item in db.Terms.find():
    CATEGORIES.append((item['slug'], item['name']))


class TermForm(Form):
    category = SelectField(
        choices=CATEGORIES,
        validators=[Optional()])

但我得到一个例外:

Traceback (most recent call last):
  File "/home/one/Projects/proj/manage.py", line 14, in <module>
    app = create_app(os.getenv('FLASK_CONFIG') or 'default')
  File "/home/one/Projects/proj/app/__init__.py", line 27, in create_app
    from app.term.models import Term, TermCategory
  File "/home/one/Projects/proj/app/term/__init__.py", line 5, in <module>
    from . import views
  File "/home/one/Projects/proj/app/term/views.py", line 7, in <module>
    from .forms import TermForm, CategoryForm
  File "/home/one/Projects/proj/app/term/forms.py", line 48, in <module>
    for item in db.Terms.find():
  File "/home/one/.venv/proj/lib/python3.4/site-packages/flask_mongokit.py", line 238, in __getattr__
    self.connect()
  File "/home/one/.venv/proj/lib/python3.4/site-packages/flask_mongokit.py", line 196, in connect
    host=ctx.app.config.get('MONGODB_HOST'),
AttributeError: 'NoneType' object has no attribute 'app'

如果有人能够对这个问题有所了解,我会非常感激。

2 个答案:

答案 0 :(得分:2)

看起来你正在调用一个需要应用程序上下文(db.Terms.find)的方法而不需要上下文。您可以在视图中填充选项:

# forms.py
class TermForm(Form):
    category = SelectField(validators=[Optional()])


# views.py
form = TermForm()
form.category.choices = [(item['slug'], item['name']) for item in db.Terms.find()]

答案 1 :(得分:0)

请参阅本答案的底部,了解您真正想要的内容,第一部分只是为了解释您即将发生的错误。


您正在运行依赖于此类上下文之外的应用程序上下文的代码。您需要运行在应用程序上下文中填充CATEGORIES的代码,以便Flask-MongoKit db可以获得连接。

看起来您正在使用应用程序工厂,因此请稍微重构您的代码,以便在创建应用程序时填充集合(您需要访问应用程序以设置上下文)。

将代码放在函数中,然后在上下文中导入并调用该函数。

# forms.py

CATEGORIES = []

def init():
    for item in db.Terms.find():
        CATEGORIES.append((item['slug'], item['name']))

# add to create_app function (even the import)

def create_app(conf):
    #...
    from app.term import forms

    with app.app_context():
        forms.init()
    #...

如果您正在使用蓝图,则可以添加在注册蓝图时执行的功能,因此应用工厂无需了解所有详细信息。将所有导入(例如views)移到此注册函数中。不需要从上面做出改变。

# at the bottom of app/term/__init__.py
# assuming blueprint is called bp

@bp.record_once
def register(state):
    with state.app.app_context():
        from . import forms, views

但是,在这种情况下,这些都不是你实际想要的。您似乎正在尝试将表单字段的选项动态设置为数据库中的当前值。如果您现在这样做,它将在应用程序启动时填充一次,然后即使数据库条目发生更改也不会更改。您 想要做的是在表格__init__方法中设置选项。

class TestForm(Form):
    category = SelectField()

    def __init__(self, *args, **kwargs):
        self.category.kwargs['choices'] = [(item['slug'], item['name']) for item in db.Terms.find()]
        Form.__init__(self, *args, **kwargs)