def test_get_close_db(app): with app.app_context(): db = get_db() assert db is get_db() with pytest.raises(sqlite3.ProgrammingError) as e: db.execute('SELECT 1') assert 'closed' in str(e)
def register(): if request.method == 'POST': username = request.form['username'] ## Setting password to username for speed during experiment... #password = request.form['password'] password = request.form['username'] job = request.form['job_title'] years = request.form['years_experience'] db = get_db() error = None if not username: error = 'Username is required.' elif not password: error = 'Password is required.' elif db.execute('SELECT id FROM user WHERE username = ?', (username, )).fetchone() is not None: error = 'User {} is already registered.'.format(username) if error is None: db.execute( 'INSERT INTO user (username, password, job_title, years_experience) VALUES (?, ?, ?, ?)', (username, generate_password_hash(password), job, years)) db.commit() return redirect(url_for('auth.login')) flash(error) return render_template('auth/register.html')
def app(): db_fd, db_path = tempfile.mkstemp() app = create_app({ 'TESTING': True, 'DATABASE': db_path, }) with app.app_context(): init_db() get_db().executescript(_data_sql) yield app os.close(db_fd) os.unlink(db_path)
def load_logged_in_user(): user_id = session.get('user_id') if user_id is None: g.user = None else: g.user = get_db().execute('SELECT * FROM user WHERE id = ?', (user_id, )).fetchone()
def test_register(client, app): assert client.get('/auth/register').status_code == 200 response = client.post('/auth/register', data={ 'username': '******', 'password': '******' }) assert 'http://localhost/auth/login' == response.headers['Location'] with app.app_context(): assert get_db().execute( "select * from user where username = '******'", ).fetchone() is not None
def login(): if request.method == 'POST': username = request.form['username'] password = request.form['password'] db = get_db() error = None user = db.execute('SELECT * FROM user WHERE username = ?', (username, )).fetchone() if user is None: error = 'Incorrect username.' elif not check_password_hash(user['password'], password): error = 'Incorrect password.' if error is None: session.clear() session['user_id'] = user['id'] return redirect(url_for('index')) flash(error) return render_template('auth/login.html')
def submit_table1(): if request.method == 'POST': global df #, nrfd user_id = session.get('user_id') db = get_db() error = None if db.execute( 'SELECT * FROM user WHERE id = ?', (user_id,) ).fetchone() is None: error = 'User with id {} not in database.'.format(user_id) if error is None: db.execute( 'INSERT INTO response (author_id) VALUES (?)', (user_id,) ) db.commit() response_id = db.execute('SELECT LAST_INSERT_ROWID() as r_id').fetchone()['r_id'] #print(response_id) for name in df['Name']: status = request.form[name] db.execute( 'INSERT INTO status (response_id, author_id, status, name, bed_number, t_number) VALUES (?,?,?,?,?,?)', (response_id, user_id, status, name, df[df['Name']==name]['Bed'].iloc[0],df[df['Name']==name]['T_number'].iloc[0],) ) db.commit() # if selected=="selected": # nrfd[name] = True # elif selected=="unselected": # nrfd[name] = False #testdata = db.execute('SELECT * FROM status WHERE response_id==?', (response_id,)).fetchall() #for row in testdata: # print([row[key] for key in row.keys()]) #print(nrfd) #return redirect(url_for('tables.table2')) return redirect(url_for('tables.finish'))
def register(): if request.method == 'POST': username = request.form['username'] password = request.form['password'] db = get_db() error = None if not username: error = 'Username is required.' elif not password: error = 'Password is required.' elif db.execute('SELECT id FROM user WHERE username = ?', (username, )).fetchone() is not None: error = 'User {} is already registered.'.format(username) if error is None: db.execute('INSERT INTO user (username, password) VALUES (?, ?)', (username, generate_password_hash(password))) db.commit() return redirect(url_for('auth.login')) flash(error) return render_template('auth/register.html')