def register():
username = request.form.get('username')
password = request.form.get('password')
confirm_password = request.form.get('confirm_password')
if request.method == 'POST':
valid = True
user = User.query.filter_by(username=username).first()
print(user)
if username is None :
flash('Please enter a username')
valid = False
if user is not None:
flash('This username is already taken. Please choose another')
valid = False
if password is None or confirm_password is None:
flash('Please enter a password')
valid = False
if password != confirm_password:
flash('Password dont match')
valid = False
if valid:
user = User(username=username)
user.set_password(password)
db.session.add(user)
db.session.commit()
return redirect('/login')
else:
return render_template('register.html')
else:
return render_template('register.html')
def register():
form = RegistrationForm(request.form)
if form.validate_on_submit():
user = User(form.username.data)
user.set_password(form.password.data)
db_session.add(user)
db_session.commit()
flash(
'Congratulations, you are now registered with the username: %s' %
form.username.data, 'success')
return redirect(url_for('login'))
return render_template('register.html', form=form)
def db_seed():
with open('resources/default_data.json') as jf:
data = json.load(jf)
for u in data.get('users'):
user = User(username=u['username'])
user.set_password(u['password'])
db.session.add(user)
for t in u['todos']:
db.session.add(
Todo(description=t['description'],
completed=False,
user=user))
db.session.commit()
def populate_users():
user1 = User(username='******', password='******')
user1.set_password('user1')
user2 = User(username='******', password='******')
user2.set_password('user2')
user3 = User(username='******', password='******')
user3.set_password('user3')
db.session.add_all([user1, user2, user3])
db.session.commit()
def _create_seed():
user1 = User(username='******')
user1.set_password('user1')
db.session.add(user1)
user2 = User(username='******')
user2.set_password('user2')
db.session.add(user2)
user3 = User(username='******')
user3.set_password('user3')
db.session.add(user3)
print(User.query.count(), "Users added")
db.session.add(Todo(user_id=1, description='Vivamus tempus'))
db.session.add(Todo(user_id=1, description='lorem ac odio'))
db.session.add(Todo(user_id=1, description='Ut congue odio'))
db.session.add(Todo(user_id=1, description='Sodales finibus'))
db.session.add(Todo(user_id=1, description='Accumsan nunc vitae'))
db.session.add(Todo(user_id=2, description='Lorem ipsum'))
db.session.add(Todo(user_id=2, description='In lacinia est'))
db.session.add(Todo(user_id=2, description='Odio varius gravida'))
print(Todo.query.count(), "Todos added")
db.session.commit()