def register():
form = RegisterForm()
Role.insert_roles()
if current_user.is_active():
return redirect(url_for('landing_blueprint.index'))
else:
if form.validate_on_submit():
user = User(username=request.form['username'],
email=request.form['email'],
password=request.form['password'],
role_id=3)
db.session.add(user)
db.session.commit()
# Add same info to session for new user as per /login route
session["current_user"] = {
"first_name": user.first_name,
"id": user.id,
"num_received_requests": 0,
"num_sent_requests": 0,
"num_total_requests": 0
}
flash('Log In')
return redirect(url_for('auth_blueprint.login'))
return render_template('users/registration.html', form=form)
def register():
form = RegisterForm()
Role.insert_roles()
if current_user.is_active():
return redirect(url_for('landing_blueprint.index'))
else:
if form.validate_on_submit():
user = User(username=request.form['username'],
email=request.form['email'],
password=request.form['password'],
role_id=3)
db.session.add(user)
db.session.commit()
flash('Log In')
return redirect(url_for('auth_blueprint.login'))
return render_template('users/registration.html', form=form)
from model import db, User, Role db.create_all() Role.insert_roles()
""" Script to add sample data in the database """ from model import User, Role, Item, Category from catalog import db # Add roles Role.insert_roles() # Add users user1 = User(name='John', email='*****@*****.**', password='******') user2 = User(name='Mary', email='*****@*****.**', password='******') user3 = User(name='Bob', email='*****@*****.**', password='******') user4 = User(name='Mike', email='*****@*****.**', password='******') user5 = User(name='Estella', email='*****@*****.**', password='******') # Add users in the db session db.session.add(user1) db.session.add(user2) db.session.add(user3) db.session.add(user4) db.session.add(user5) # Add Categories category1 = Category(name='Exercise and Fitness') category2 = Category(name='Cricket') category3 = Category(name='Camping and Hiking') category4 = Category(name='Cycling') category5 = Category(name='Running')
