Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- @app.before_request
- def before_request():
- g.user = current_user
- user = User.query.filter_by(email=form.email.data,password=form.password.data).first()
- from app import db,unicode,UserMixin
- ROLE_USER = 0
- ROLE_ADMIN = 1
- class User(UserMixin,db.Model):
- id = db.Column(db.Integer, primary_key = True)
- username = db.Column(db.String(64), index = True, unique = True)
- email = db.Column(db.String(120), index = True, unique = True)
- password = db.Column(db.String(120), index = True)
- role = db.Column(db.SmallInteger, default=ROLE_USER)
- def __init__(self, id, username, email, password, role):
- self.id = id
- self.username = username
- self.email = email
- self.password = password
- self.role = role
- def get_id(self):
- return self.id
- def is_authenticated(self):
- return True
- def is_active(self):
- return True
- def is_anonymous(self):
- return False
- def __repr__(self):
- return 'ID: %r ' % (self.id) + ' USERNAME: %r ' % (self.username) + ' EMAIL: %r ' % (self.email) + ' PASSWORD: %r ' % (self.password) + ' ROLE: %r ' % (self.role)
- @lm.user_loader
- def load_user(id):
- return User.query.get(id)
- @app.route('/login', methods=('GET', 'POST'))
- def login():
- if g.user is not None and g.user.is_authenticated:
- return redirect(url_for('index', title=g.user.username))
- form = LoginForm()
- if form.validate_on_submit():
- user = User.query.filter_by(email=form.email.data,password=form.password.data).first()
- print('USER :' + str(user))
- print('G.USER :' + str(g.user))
- print('USER LOGIN IS : ' + str(login_user(user)))
- login_user(user)
- return redirect(url_for('index'))
- return render_template('login.html', form=form)
- class LoginForm(Form):
- email = StringField('Email', validators=[DataRequired()])
- password = PasswordField('New Password', validators=[DataRequired()])
- remember_me = BooleanField('Remember me', default=False)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement