def test_add_multiple(db_session):
db_session.add_all([
User(name="John", age=90),
User(name="Philo", age=98),
User(name="Paris", age=95)
])
db_session.commit()
result = db_session.query(User).filter(User.age >= 90)
assert result.count() == 3
def verifica_user():
user_json = session['barzinga_user']
user = User.query().filter(User.email == user_json["email"]).get()
dx = user_json['email'].endswith('@dextra-sw.com')
mutant = user_json['email'].endswith('@mutantbr.com')
if not dx and not mutant:
raise Exception('Invalid domain %s' % user_json['email'])
if not user:
# user = User(name = user_json['name'], email=user_json['email'], photo_url=user_json['picture'], money=0.0, admin=False, rfid=str(''))
user = User(name = '', email=user_json['email'], photo_url=user_json['picture'], money=0.0, admin=False, infra=False, rfid=str(''), active = True)
user.put()
user_document = search.Document(
fields=[
search.TextField(name='name', value=user.name),
search.TextField(name='email', value=user.email)
])
search.Index(name='user').put(user_document)
elif not user.photo_url:
user.photo_url = user_json['picture']
user.put()
identity = Identity(user.key.id())
identity_changed.send(current_app._get_current_object(), identity=identity)
def register():
if request.method == 'POST':
username = request.form.get('username')
password = request.form.get('password')
gender = request.form.get('gender')
age = request.form.get('age')
city = request.form.get('city')
phone = request.form.get('phone')
try:
user = User.query.filter_by(username=username).one()
if user:
return '用户名已存在'
except Exception:
user = User(username=username,
password=password,
gender=gender,
age=age,
city=city,
phone=phone)
db.session.add(user)
db.session.commit()
return redirect('/user/login')
else:
return render_template('register.html')
def setup_base():
Base.metadata.drop_all(bind=db, tables=[User.__table__, Address.__table__])
Base.metadata.create_all(db)
session = Session()
change_user = User(name="Mike", age=44)
session.add(change_user)
session.commit()
session.close()
def test_db_relationships(db_session):
new_user = User(name="Lomothy", age=25)
new_user.addresses = [
Address(email="*****@*****.**"),
Address(email="*****@*****.**")
]
db_session.add(new_user)
db_session.commit()
lomothy = db_session.query(User).filter(User.name == "Lomothy").first()
assert len(lomothy.addresses) == 2
def register():
if request.method == "GET":
return render_template('register.html')
if request.method == "POST":
username = request.form.get('username')
password = request.form.get("password")
if not all([username, password]):
return render_template('register.html')
# 保存注册信息
user = User()
user.username = username
user.password = password
db.session.add(user)
db.session.commit()
return render_template('login.html')
def add():
name = request.form['name']
email = request.form['email']
rfid = request.form['rfid']
user = User(name=name,
email=email,
admin=False,
photo_url='',
money=0,
rfid=rfid)
user.put()
user_document = search.Document(fields=[
search.TextField(name='name', value=user.name),
search.TextField(name='email', value=user.email)
])
search.Index(name='user').put(user_document)
return '', 204
def verifica_user():
user_json = session['barzinga_user']
user = User.query().filter(User.email == user_json["email"]).get()
if not user:
# user = User(name = user_json['name'], email=user_json['email'], photo_url=user_json['picture'], money=0.0, admin=False, rfid=str(''))
user = User(name = '', email=user_json['email'], photo_url=user_json['picture'], money=0.0, admin=False, rfid=str(''))
user.put()
user_document = search.Document(
fields=[
search.TextField(name='name', value=user.name),
search.TextField(name='email', value=user.email)
])
search.Index(name='user').put(user_document)
elif not user.photo_url:
user.photo_url = user_json['picture']
user.put()
identity = Identity(user.key.id())
identity_changed.send(current_app._get_current_object(), identity=identity)
def test_add_to_db(db_session):
new_user = User(name="Dec", age=55)
db_session.add(new_user)
db_session.commit()
assert True
