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()