在Flask中获取当前用户ID

时间:2016-12-12 03:08:48

标签: python flask flask-wtforms flask-login

我是Python的新手(而且,老实说,一般编程)。我目前正在开发一种待办事项列表,我需要它将待办事项放入适当的课程(这些都与教育相关)。所以,问题非常简单。我将此作为Flask驱动的路线:

@app.route('/add_course', methods=('GET', 'POST'))
@login_required
def course():
    form = forms.CourseForm()
    if form.validate_on_submit():
        models.Course.create(teacher=g.user._get_current_object(),
                             name=form.name.data.strip(),
                             difficulty=form.level.data.strip(),
                             description=form.description.data.strip())
        flash("Course successfully created!", "success")
        return redirect(url_for('index'))
    return render_template('add_course.html', form=form)

然后我在forms.py中有这个。你可以看到我用来表明我的问题所在的聪明标记。它说THE_PROBLEM

def courses():
    try:
        courses = models.Course.select(models.Course.id,
                                   models.Course.name,
                                   models.Course.difficulty).where(THE_PROBLEM)
        course_list = []
        for course in courses:
            course_list.append((str(course.id), course.name + ' - ' + course.difficulty.title()))
        return course_list
    except models.DoesNotExist:
        return [('', 'No courses')]


class ToDoForm(FlaskForm):
    name = StringField("What's up?", validators=[
    DataRequired()
    ])
    due_date = DateTimeField('When?', format='%Y-%m-%d %H-%M')
    course = SelectField('Course', choices=courses())
    priority = SelectField('Priority',
                           choices=[('high', 'High priority'),
                                    ('medium', 'Normal priority'),
                                    ('low', 'Low priority')],
                           validators=[
                               DataRequired()
                           ])
    description = TextAreaField('Description')

所以,是的,我正在寻找的是传递当前登录的所有者(在这种情况下是老师)的ID的方法。我使用此函数courses()来构建元组列表对于course中字段ToDoForm的choices属性。我需要将当前登录教师的id传递给此函数,以便它可以评估此​​传递的教师是否有任何与其id相匹配的课程。我尝试使用current_user._get_current_object()和其他任何东西,但它只是造成了很多错误。

任何帮助,建议或建议将不胜感激。我真的希望我在这里所说的(以及我想要实现的目标)是可以理解的

1 个答案:

答案 0 :(得分:-4)

如果有人遇到同样令人沮丧的问题,解决方案非常简单。将外部类包装到函数中并将所需的变量作为参数传递,然后将它们用于您心中的内容。 只是不要忘记返回班级的实例。