def register():
data = request.json
log(data)
u = User(**data)
u.save()
session = Session.new(u)
return session.response()
def login():
data = request.json
if User.validate_login(data):
session = Session.new(User.get_by_name(data['username']))
return session.response()
else:
return json_response({'status': 'fail'})
def user_login(request):
# header = 'HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n'
# 仅添加content-type字段
session = None
if request.method == 'POST':
form = request.form()
username = form.get('username')
password = form.get('password')
if User.validate_login(username, password):
u = User.find_by(username=username)
# 向headers中添加set-cookie字段
# 通过设置一个随机字符串生成session_id
session_id = random_str()
form = dict(
session_id=session_id,
user_id=u.id,
)
s = Session.new(form)
s.save()
session = {'Set-Cookie': 'session_id={}'.format(
session_id
)}
result = '登录成功'
else:
result = '用户名或者密码错误'
u = User.guest()
else:
result = ''
u = current_user(request)
body = template(
'login.html',
username=u.username,
result=result
)
return http_response(body, session)
