def index(): session_id = request.get_cookie('session_id') if session_id: session = Session() data = session.get(session_id) uid = data.get('uid', None) if uid: storage = UserStorage() user = storage.get_user(uid) html = """<!DOCTYPE html> <html> <body> <h2>Welcome. %s</h2> <p>you from %s, and your social id is %s<p> <img src="%s" /> <p><a href="/logout">Logout</a></p> </body> </html>""" % (user['name'], user['site_name'], user['uid'], user['avatar']) return html if not session_id: response.set_cookie('session_id', gen_session_id()) html = """<html> <body><a href="/login">Login</a></body> </html>""" return html
def logout(): session_id = request.get_cookie('session_id') if not session_id: redirect('/') session = Session() data = session.get(session_id) session.rem(session_id) uid = data.get('uid', None) if uid: # 重置其session_id Session.refresh_session_id(uid) response.set_cookie('session_id', '') redirect('/')