Page 526 - MDP2022-3
P. 526
from models import db, Myuser, Todo
app = Flask(__name__)
app.register_blueprint(api_v1, url_prefix='/api/v1')
@app.route('/', methods=['GET'])
def homepage():
userid = session.get('userid')
todos = []
if userid:
myuser = Myuser.query.filter_by(userid=userid).first()
todos = Todo.query.filter_by(myuser_id=myuser.id)
return render_template('index.html', userid=userid, todos=todos)
else:
return redirect('/error')
@app.route('/homepage', methods=['GET'])
def todo():
return render_template('index.html')
@app.route('/login', methods=['GET', 'POST'])
def login():
form = LoginForm()
if form.validate_on_submit():
session['userid'] = form.data.get('userid')
return redirect('/')
return render_template('login.html', form=form)
@app.route('/logout', methods=['GET'])
def logout():
session.pop('userid', None)
return redirect('/')
@app.route('/register', methods=['GET', 'POST'])
def register():
form = RegisterForm()
if form.validate_on_submit():
myuser = Myuser()
myuser.userid = form.data.get('userid')
myuser.password = form.data.get('password')