1class User(UserMixin, db.Model):
2 id = db.Column(db.Integer, primary_key = True)
3 name = db.Column(db.String(30))
4 def __init__(self, id, name):
5 self.id = id
6 self.name = name
7@login_manager.user_loader
8def firstuser(id):
9 return User.query.get(int(id))
10@app.route('/')
11
12def index():
13 u1 = User.query.filter_by(id = 112).first()
14 login_user(u1)
15 return current_user.name + 'logged in'
16@app.route('/logout')
17@login_required
18def logout():
19 logout_user()
20 return 'logged out'
21@app.route('/home')
22@login_required
23def home():
24 return "current user is " + current_user.name
25
26def init_db():
27 db.create_all()
28 new_user = User(112,'Rose')
29 new_user2 = User(113,'lily')
30 db.session.add(new_user)
31 db.session.add(new_user2)
32 db.session.commit()
33if __name__ == '__main__':
34 init_db()
35 app.run(debug = True)