async def get_user(tel_id, loop):
con, cur = await create_con(loop)
await cur.execute('select * from users where tel_id = %s', (tel_id))
user = await cur.fetchone()
con.close()
if user is None:
return None
return User(user[0], user[1], user[2], user[3], user[4], user[5])
def transform_user_data(users: dict = {}) -> dict:
output = []
for user in users:
InputUser(**user)
user_model = User(name=user['name'],
wont_eat=convert_to_dict(user['wont_eat']),
drinks=convert_to_dict(user['drinks']))
output.append(user_model)
return output
def register():
json_data = request.json
user = User(email=json_data['email'], password=json_data['password'])
try:
db.session.add(user)
db.session.commit()
status = 'success'
except:
status = 'this user is already registered'
db.session.close()
return jsonify({'result': status})
async def get_all_users(loop):
con, cur = await create_con(loop)
await cur.execute('select * from users')
users = await cur.fetchall()
con.close()
result = []
for user in users:
result.append(
User(user[0], user[1], user[2], user[3], user[4], user[5]))
return result
async def get_user(tel_id, loop):
con, cur = await create_con(loop)
await cur.execute(
'select * from users where tel_id = {0}'.format(tel_id))
u = await cur.fetchone()
con.close()
if u is None:
return None
else:
return User(u[0], u[1], u[2], u[3], u[4], u[5], u[6], u[7], u[8],
u[9], u[10], u[11], u[12])
async def get_all_users(loop):
con, cur = await create_con(loop)
await cur.execute('select * from users')
users = await cur.fetchall()
con.close()
result = []
for u in users:
result.append(
User(u[0], u[1], u[2], u[3], u[4], u[5], u[6], u[7], u[8],
u[9], u[10], u[11], u[12]))
return result
def register():
if current_user.is_authenticated:
return redirect(url_for('home'))
form = RegistrationForm()
if form.validate_on_submit():
hashed_password = bcrypt.generate_password_hash(form.password.data).decode('utf-8')
user = User(username=form.username.data, email=form.email.data, password=hashed_password)
db.session.add(user)
db.session.commit()
flash('Your account has been created! You are now able to log in', 'success')
return redirect(url_for('login'))
return render_template('register.html', title='Register', form=form)
def test_user_model(self):
user = User(name='Max',
wont_eat={
'food1': 0,
'food2': 0
},
drinks={
'drink1': 0,
'drink2': 0
})
expected = {
'name': 'Max',
'wont_eat': {
'food1': 0,
'food2': 0
},
'drinks': {
'drink1': 0,
'drink2': 0
}
}
self.assertEqual(user.dict(), expected)
def register():
form = RegistrationForm()
if form.validate_on_submit():
user = User(email=form.email.data,
username=form.username.data,
password=form.password.data)
db.session.add(user)
db.session.commit()
flash('Thanks for registering! Now you can login!')
return redirect(url_for('login'))
return render_template('register.html', form=form)
def create_admin():
"""Creates the admin user."""
db.session.add(User(email='*****@*****.**', password='******', admin=True))
db.session.commit()