def admin_user(): email = raw_input('输入管理员邮箱:') inp = 1 while inp == 1: pwd1 = raw_input('输入管理员密码:') pwd2 = raw_input('确认管理员密码:') if pwd1 == pwd2: inp = 0 else: print '两次输入的不一样,请重新输入' continue print '--------------------------------' print '输入的管理员邮箱为:%s' % email print '输入的管理员密码为:%s' % pwd1 print '--------------------------------' if raw_input('确定(Y)').lower() == 'y': pwd = User.make_random_passwd(email=email) user = User(email=pwd['email'], role=1, nicename=email, passwd=pwd['pwdmd5'], is_locked=0, salt=pwd['salt']) user.register_date = datetime.now(), db.session.add(user) db.session.commit() print '保存设置...[确定]' else: exit()
def test_user_check(self): pwd = User.make_random_passwd(email='*****@*****.**') user = User(email=pwd['email'], passwd=pwd['pwdmd5'], salt=pwd['salt']) db.session.add(user) db.session.commit() is_true = User.user_check('*****@*****.**', pwd['pwd']) assert is_true == True
def userchangepwd(): form = UserChangePwdForm() if form.validate_on_submit() and request.method == 'POST': flash(g.user.email) pwd = User.make_random_passwd(pwd=form.password.data, email=g.user.email) g.user.passwd = pwd['pwdmd5'] db.session.add(g.user) db.session.commit() flash(u'密码修改成功!') return redirect(url_for('usereditinfo')) return render_template('user/userchangepwd.html', title=u'修改密码', form=form)
def register(): from blog.extend.EmailHelper import register_mail form = RegisterForm(request.form) if request.method == 'POST' and form.validate(): pwd = User.make_random_passwd(email=form.email.data) user = User(email=pwd['email'], role=ROLE_USER, nicename=form.email.data, passwd=pwd['pwdmd5'], is_locked=User_LOCKED, register_ip=request.remote_addr, salt=pwd['salt']) user.register_date = datetime.datetime.now(), db.session.add(user) db.session.commit() user.passwd = pwd['pwd'] register_mail(user) flash(u'恭喜,注册成功!') Blog_info.new_user() return redirect(url_for('login')) return render_template('register.html', title=u'欢迎注册', form=form)