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 _add_users(users: List[dict]):
for user_dict in users:
try:
user = User(username=user_dict['username'],
password=user_dict['password'])
db_session.add(user)
db_session.commit()
except Exception as e:
print(e)
def init_db():
db.create_all()
records = [
User(username='******', password=generate_password_hash('user1')),
User(username='******', password=generate_password_hash('user2')),
Todo(user_id=1, description='Vivamus tempus'),
Todo(user_id=1, description='lorem ac odio'),
Todo(user_id=1, description='Ut congue odio'),
Todo(user_id=1, description='Sodales finibus'),
Todo(user_id=1, description='Accumsan nunc vitae'),
Todo(user_id=2, description='Lorem ipsum'),
Todo(user_id=2, description='In lacinia est'),
Todo(user_id=2, description='Odio varius gravida')
]
db.session.bulk_save_objects(records)
db.session.commit()
def upgrade():
users = [
('user1', 'user1'),
('user2', 'user2'),
('user3', 'user3'),
]
for user in users:
user = User(user[0], user[1])
db.session.add(user)
db.session.commit()
def login_POST():
username = request.form.get('username')
password = request.form.get('password')
user = User.authenticate(username, password)
if user:
session['user_id'] = user.id
session['username'] = user.username
return redirect('/todo')
flash("Invalid username or password", "danger")
return redirect('/login')
def _init_db():
# Database initialisation
db.create_all()
# Filling the database with records for User and Todo models
records = [
User(username='******',
password=generate_password_hash('user1')),
User(username='******',
password=generate_password_hash('user2')),
Todo(user_id=1, description='Vivamus tempus'),
Todo(user_id=1, description='lorem ac odio'),
Todo(user_id=1, description='Ut congue odio'),
Todo(user_id=1, description='Sodales finibus'),
Todo(user_id=1, description='Accumsan nunc vitae'),
Todo(user_id=2, description='Lorem ipsum'),
Todo(user_id=2, description='In lacinia est'),
Todo(user_id=2, description='Odio varius gravida')
]
db.session.bulk_save_objects(records)
db.session.commit()
def signup_POST():
username = request.form.get('username')
password = request.form.get('password')
user = User.query.filter_by(username=username).first()
if user:
flash('This username is already in use', category='danger')
return redirect('/signup')
if username and password:
hashed_password = bcrypt.generate_password_hash(password)
new_user = User(username=username, password=hashed_password)
db.session.add(new_user)
db.session.commit()
if new_user:
session['user'] = user_schema.dump(new_user).data
session['logged_in'] = True
return redirect('/todo')
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()
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 generate_sample_users():
with app.app_context():
g.db = connect_db()
for i in (1, 2, 3):
s = 'user{}'.format(i)
User(username=s, password=generate_password_hash(s)).save()