def error(rq, status_code=404):
headers = {
'Content-Type': 'text/html',
}
body = '<h3>Not Found</h3>'
response_msg = make_response_msg(status_code, headers, body)
return response_msg
def index(rq):
user = current_user(rq)
if user is None:
username = '******'
else:
username = user.username
body = template(env, 'index.html', username=username)
response_msg = make_response_msg(body=body)
return response_msg
def index(rq):
user = current_user(rq)
if user is None:
username = '******'
else:
username = user.username
tweets = Tweet.get_all()
body = template(env, 'index.html', username=username, tweets=tweets)
response_msg = make_response_msg(body=body)
return response_msg
def register(rq):
if rq.method == 'POST':
d = rq.form()
user = User(d)
if user.validate_register() is not None:
response_msg = redirect('/login')
else:
response_msg = redirect('/register')
else:
body = template(env, 'register.html')
response_msg = make_response_msg(body=body)
return response_msg
else:
body = template(env, 'register.html')
response_msg = make_response_msg(body=body)
return response_msg
def login(rq):
if rq.method == 'POST':
d = rq.form()
user = User(d)
if (u := user.validate_login()) is not None:
session_id = make_session_id()
session[session_id] = u.id
headers = {
'Set-Cookie': f'session_id={session_id}',
}
response_msg = redirect('/', headers)
log('登录成功')
else:
response_msg = redirect('/login')
log('账号或密码错误')
else:
body = template(env, 'login.html')
response_msg = make_response_msg(body=body)
return response_msg
route_dict = {
'/register': register,
'/login': login,
}
def edit(rq):
tweet_id = int(rq.query.get('id', -1))
t = Tweet.find(tweet_id)
body = template(env, 'edit.html', tweet=t)
response_msg = make_response_msg(body=body)
return response_msg
def edit(rq):
cmt_id = int(rq.query.get('id', -1))
cmt = Comment.find(cmt_id)
body = template(env, 'edit.html', comment=cmt)
response_msg = make_response_msg(body=body)
return response_msg