Python Flask WTForms:动态SelectField返回“无效选择”

时间:2019-11-01 01:57:16

标签: python flask-wtforms

首先,我知道这里有很多类似的问题。我查看了大多数(如果不是全部的话),而且没有一个帮助我找到解决方案。以下问题是最相似的,但是我对“ dynamic”的实现与他们的实现有些不同(下面对此有更多介绍): Flask / Python / WTForms validation and dynamically set SelectField choices

简而言之:

我有一个表格,用于从我构建的网络监视工具中请求报告。该工具跟踪各种无线网络的所有不同种类的统计信息。这是表单类定义。我的动态字段是ssidFilter selectField。

class RequestReportForm(FlaskForm):
    startDate = DateField('Start Date', validators=[DataRequired(), validate_startDate])
    startTime = TimeField('Start Time', format='%H:%M', validators=[DataRequired()])
    endDate = DateField('End Date', format='%Y-%m-%d', validators=[DataRequired(), validate_endDate])
    endTime = TimeField('End Time', format='%H:%M', validators=[DataRequired(), validate_allDates])
    ssidFilter = SelectField('SSID', default=('All', 'All'))
    reportType = SelectField('Report Type', validators = [DataRequired()], choices=[
                                                                ('rssi', 'RSSI vs. Time'), 
                                                                ('snr', 'SNR vs. Time'),
                                                                ('ClientCount', 'Client Count vs. Time'),

    ])
    selectLocation = SelectField('Locations', validators = [DataRequired()], choices=[ 
                                                                ('All','All'),
                                                                ('mainLobby', 'Main Lobby'), 
                                                                ('level1', 'Level 1'), 
                                                                ('level2', 'Level 2'),
                                                                ])
    submit = SubmitField('Generate Report')

我已经实现了Javascript来接受用户输入的startDate和endDate字段,并通过“提取”应用程序中的另一个烧瓶路由来返回数据库中所有无线网络(SSID)的列表,从而对数据库运行查询在他们输入的日期范围内使用。这是那条路线:

@app.route('/updateSSIDs/<startDate>/<endDate>', methods=['GET'])
def updateSSIDs(startDate, endDate):
    startDate = datetime.strptime(startDate, '%Y-%m-%d')
    endDate = datetime.strptime(endDate, '%Y-%m-%d')
    # Get a list of unique SSIDs that we have data for between the start and end dates selected on the form.
    SSIDs = getSSIDs(startDate, endDate)
    SSIDArray = []
    for ssid_tuple in SSIDs:
        ssidObj = {}
        ssidObj['id'] = ssid_tuple[0]
        ssidObj['ssid'] = ssid_tuple[0]
        SSIDArray.append(ssidObj)
    return jsonify({'SSIDs' : SSIDArray})

变量SSIDArray在进行jsonify之前看起来像这样:

[{'id': 'Example Network 1', 'ssid': 'Example Network 1'}, {'id': 'Staff', 'ssid': 'Staff'}, ... ]

这是我实例化表单的方式:

@app.route('/requestReport', methods=['GET', 'POST'])
def requestReport():
    form = RequestReportForm()
    form.ssidFilter.choices = getSSIDs(datetime.now(), datetime.now())

    if form.validate_on_submit():
        print("Valid form data:")
        print(form.data)
        flash(f'Received request for report from {form.startDate.data} at {form.startTime.data} through {form.endDate.data} at {form.endTime.data}', 'success')
        startDate = form.startDate.data
        startTime = form.startTime.data
        endDate = form.endDate.data
        endTime = form.endTime.data

        reportType = form.reportType.data
        locations = form.selectLocation.data
        ssid = form.ssidFilter.data

        # Put requested times into datetime objects
        startDateTime = datetime(startDate.year, startDate.month, startDate.day, startTime.hour, startTime.minute)
        endDateTime = datetime(endDate.year, endDate.month, endDate.day, endTime.hour, endTime.minute)

        # Generate report and redirect client to report.
        reportParameters = rpt.prepareReport_single(startDateTime, endDateTime, reportType, locations, ssid)        
        report = rpt.buildReport_singleLocation(reportParameters)       
        report = Markup(report)
        return render_template('viewReport.html', value=report)

请注意,我正在通过调用响应我的Javascript提取调用的同一form.ssidFilter.choices函数来填充我的动态字段getSSIDs,但是我同时传递了datetime.now()开始和结束日期。最初是为了向用户显示当前正在使用的无线网络列表,但是一旦它们更改了日期,该列表就会使用一组不同的网络进行更新。

其中存在一个问题:如何设置可接受选项(form.ssidFilter.choices)的列表,以包含客户输入报告日期后返回的网络列表?

我正在探索的可能解决方案:

  • 选择日期后重新加载页面,以使用动态数据实例化新表单。

  • 首先保留所有可用选项的巨大列表,然后当用户更改表单上的日期时,这些选项将通过JS动态筛选。

哦,如果所选的SSID恰好是form.ssidFilter.choices = getSSIDs(datetime.now(), datetime.now())语句列表中的SSID,则该格式可以正常工作。仅当选择的项目最初不在选择列表中时才发生此问题(这很有意义-我只是不知道如何解决该问题)。

谢谢您的时间。

编辑/解决方案:

由于@SuperShoot的回答,我得以解决此问题。对我来说,关键是让Flask路由区分HTTP请求的类型-GET还是POST。由于我知道GET方法仅用于检索表单,而POST方法仅用于提交填写的表单,因此我可以从用户中提取startDate和endDate选择,运行查询以获取数据并更新我的表单类中的choices字段。

我还必须做一些额外的验证,就像@SuperShoot也提到的那样,但是我做了一些不同的事情。由于修改结束日期后,我的JavaScript代码会从Flask应用程序中调用一条单独的路线,因此该表格不负责验证所选日期。我在其他Flask路线中进行了一些验证。

这是我修改过的Flask requestReport路线:

@app.route('/requestReport', methods=['GET', 'POST'])
def requestReport():
    form = RequestReportForm()
    form.ssidFilter.choices = getSSIDs(datetime.now(), datetime.now())

    if request.method == 'POST':
        startDate = datetime(form.startDate.data.year, form.startDate.data.month, form.startDate.data.day)
        endDate = datetime(form.endDate.data.year, form.endDate.data.month, form.endDate.data.day)
        # Update acceptable choices for the SSIDs on the form if the form is submitted.
        form.ssidFilter.choices = getSSIDs(startDate, endDate)

    if form.validate_on_submit():
        flash(f'Received request for report from {form.startDate.data} at {form.startTime.data} through {form.endDate.data} at {form.endTime.data}', 'success')
        startDate = form.startDate.data
        startTime = form.startTime.data
        endDate = form.endDate.data
        endTime = form.endTime.data

        reportType = form.reportType.data
        locations = form.selectLocation.data
        ssid = form.ssidFilter.data

        # Put requested times into datetime objects
        startDateTime = datetime(startDate.year, startDate.month, startDate.day, startTime.hour, startTime.minute)
        endDateTime = datetime(endDate.year, endDate.month, endDate.day, endTime.hour, endTime.minute)

        # Generate report and redirect client to report.
        reportParameters = rpt.prepareReport_single(startDateTime, endDateTime, reportType, locations, ssid)        
        report = rpt.buildReport_singleLocation(reportParameters)       
        report = Markup(report)
        return render_template('viewReport.html', value=report)

    else:
        return render_template('requestReport.html', title='Report Request', form=form)

这是我更新的updateSSIDs路由,当表单的结束日期更改时,该路由通过Javascript调用:

@app.route('/updateSSIDs/<startDate>/<endDate>', methods=['GET'])
def updateSSIDs(startDate, endDate):

    startDate = datetime.strptime(startDate, '%Y-%m-%d')
    endDate = datetime.strptime(endDate, '%Y-%m-%d')

    # Validate startDate and endDate
    emptyDataSet = {'SSIDs' : {'id ': 'All', 'ssid' : 'All'}}
    if startDate > endDate:
        return jsonify(emptyDataSet)

    if startDate >= datetime.now():
        return jsonify(emptyDataSet)

    if startDate.year not in range(2019, 2029) or endDate.year not in range(2019, 2029):
        return jsonify(emptyDataSet)

    # Get a list of unique SSIDs that we have data for between the start and end dates selected on the form.
    SSIDs = getSSIDs(startDate, endDate)
    SSIDArray = []
    for ssid_tuple in SSIDs:
        ssidObj = {}
        ssidObj['id'] = ssid_tuple[0]
        ssidObj['ssid'] = ssid_tuple[0]

        SSIDArray.append(ssidObj)
    return jsonify({'SSIDs' : SSIDArray})

此路线正在做一些基本的检查,以确保在尝试通过getSSIDs从数据库检索数据之前提交的日期不是完全荒谬的,但是我在getSSIDs中做了一些更彻底的验证功能。

1 个答案:

答案 0 :(得分:1)

您可以根据路由是处理GET还是POST请求,以不同的方式实例化表单:

@app.route('/requestReport', methods=['GET', 'POST'])
def requestReport():
    form = RequestReportForm()
    if request.method == "GET":
        start = end = datetime.now()
    else:
        # validate start and end dates here first?
        start, end = form.startDate.data, form.endDate.data
    form.ssidFilter.choices = getSSIDs(start, end)
    ...

...尽管在POST情况下,它使用开始日期和结束日期在验证之前。因此,一种选择是首先在“ POST”条件处理(我在其中放置了评论)中内联验证它们,或者另一种方法是覆盖.validate()上的RequestReportForm方法。

这是Form.validate()的文档字符串:

        """
        Validates the form by calling `validate` on each field.
        :param extra_validators:
            If provided, is a dict mapping field names to a sequence of
            callables which will be passed as extra validators to the field's
            `validate` method.
        Returns `True` if no errors occur.
        """

可能的实现方式是:

class RequestReportForm(FlaskForm):
    ...

    def validate(self, *args, **kwargs):
        """Ensure ssidFilter field choices match input startDate and endDate"""
        if not (self.startDate.validate(self) and self.endDate.validate(self)):
            return False
        self.ssidFilter.choices = getSSIDs(self.startDate.data, self.endDate.data)
        return super().validate(*args, **kwargs)

FlaskForm.validate_on_submit()首先检查表单是否已提交,然后将调用自定义.validate()方法。该方法首先确保开始日期和结束日期有效,然后使用它们填充ssidFilter的预期可能值,然后再将验证委托给MRO。

我还没有运行这段代码,所以让我知道是否有任何错误,但是希望我对这个想法有足够的了解,以便您在合适的情况下使用它。