def createUser(userId, name, ses=Session()):
dbUser = getUserById(userId, ses)
if dbUser is not None:
return
user = User(message_count=0, karma=0, name=name, id=userId)
ses.add(user)
ses.commit()
def users_new():
print(request.form)
if 'email' not in request.form.keys():
return error('Missing email')
session = Session()
session.add(User(email=request.form['email']))
try:
session.commit()
except sqlalchemy.exc.IntegrityError as why:
return error('Email already exists')
return ok('Registered user')
def incUserMsgCount(user):
ses = Session()
dbUser = getUserById(str(user.id), ses)
if dbUser is None:
dbUser = User(message_count=1,
karma=0,
name=str(user),
id=str(user.id))
ses.add(dbUser)
else:
dbUser.message_count += 1
ses.commit()
def execute(self):
try:
user = User.get(openid=self.openid)
except DoesNotExist:
user = User(openid=self.openid, )
user.save()
return self.mk_response(UserInfo(user_id=user.id))
except:
return self.mk_response(None)
user_info = UserInfo(user_id=user.id,
openid=user.openid,
nickname=user.nickname,
head_img_url=user.head_img_url,
create_time=utils.datetime2timestamp(
user.create_time))
return self.mk_response(user_info)
def register():
form = RegistrationForm()
logging.info('form.validate_on_submit():%s' %form.validate_on_submit())
if form.validate_on_submit():
if User.query.filter_by(username=form.username.data).first() is None:
user = User(username=form.username.data,
password=form.password.data)
logging.info('username:%s' %user.username)
db.db.session.add(user)
db.db.session.commit()
token = user.generate_confirmation_token()
user.confirm(token)
#send_email(user.email, 'Confirm Your Account',
# 'auth/email/confirm', user=user, token=token)
#flash('A confirmation email has been sent to you by email.')
return redirect(url_for('login'))
else:
logging.info(u'注册IP：%s 账号 %s 已被注册，请更换用户名。' %(ip, form.username.data))
flash(u'账号 %s 已被注册，请更换用户名。' %form.username.data)
return redirect(url_for('register'))
return render_template('register.html', form=form, ip = ip)
