def populate():
session = Session()
user = User(u'Renatopp', u'*****@*****.**')
user.password = hashlib.md5(u'123456'+g.SALT).hexdigest()
user.realname = u'Renato Pereira'
session.add(user)
transaction.commit()
def populate():
session = Session()
user = User(u'Renatopp', u'*****@*****.**')
user.password = hashlib.md5(u'123456' + g.SALT).hexdigest()
user.realname = u'Renato Pereira'
session.add(user)
transaction.commit()
def regist():
if request.method == 'GET':
return render_template('regist.html')
else:
telephone = request.form.get('telephone')
username = request.form.get('username')
password1 = request.form.get('password1')
password2 = request.form.get('password2')
# 查看手机号码是否已经被注册
user = User.query.filter(User.telephone == telephone).first()
if user:
print('该手机号码已经被注册！请更换手机号码或者登录！！！')
return render_template('regist.html')
else:
# 两次密码必须相同才能成功注册
if password1 != password2:
print('两次密码不相同！请重新输入!!!')
return render_template('regist.html')
else:
user = User(telephone=telephone,
username=username,
password=password1)
db.session.add(user)
db.session.commit()
return redirect(url_for('index.login'))
def register():
form = RegisterForm()
if form.validate_on_submit():
username = form.username.data
email = form.email.data
password = form.password1.data
user = User(username=username, email=email, password=password)
db.session.add(user)
db.session.commit()
return redirect(url_for('blog.login',form=form))
return render_template('auth/register.html', form=form)
def client():
# 设置测试数据库路径和测试环境
app = create_app('test')
# 初始化测试数据库
with app.test_client() as client:
with app.app_context():
db.drop_all()
db.create_all()
db.session.add(User(username='******', password='******'))
db.session.commit()
db.session.close()
yield client
def fake_users(count):
admin = User(username='******',
password='******',
email='*****@*****.**',
create_time=fake.unix_time(),
info=fake.text(20),
nickname=fake.name(),
phone_number=fake.phone_number())
db.session.add(admin)
for i in range(int(count / 2)):
username_test = fake.user_name()
user = User(username=username_test,
password=username_test,
email=fake.email(),
create_time=fake.unix_time(),
info=fake.text(20),
nickname=fake.name(),
phone_number=fake.phone_number())
db.session.add(user)
db.session.commit()
click.echo('Created {} users'.format(int(count / 2 + 1)))
def register():
"""
Handle requests to the /register route
Add an employee to the database through the registration form
"""
form = RegistrationForm()
if form.validate_on_submit():
employee = User(email=form.email.data,
username=form.username.data,
first_name=form.first_name.data,
last_name=form.last_name.data,
password=form.password.data)
# add employee to the database
db.session.add(employee)
db.session.commit()
flash('You have successfully registered! You may now login.')
# redirect to the login page
return redirect(url_for('auth.login'))
# load registration template
return render_template('auth/register.html', form=form, title='Register')
def test_password_hash(setup): # 测试密码加密
user = User(telephone='11111111111', password='******')
assert user.check_password('111111') is True
assert user.check_password('11') is False