def welcome(request: HTTPRequest) -> HTTPResponse:
cookie_header = request.headers.get("Cookie", None)
# Cookieが送信されてきていなければ、ログインしていないとみなして/loginへリダイレクト
if not cookie_header:
return HTTPResponse(status_code=302, headers={"Location": "/login"})
# str から list へ変換
# ex) "name1=value1; name2=value2" => ["name1=value1", "name2=value2"]
cookie_strings = cookie_header.split("; ")
# list から dict へ変換
# ex) ["name1=value1", "name2=value2"] => {"name1": "value1", "name2": "value2"}
cookies = {}
for cookie_string in cookie_strings:
name, value = cookie_string.split("=", maxsplit=1)
cookies[name] = value
# Cookieにusernameが含まれていなければ、ログインしていないとみなして/loginへリダイレクト
if "username" not in cookies:
return HTTPResponse(status_code=302, headers={"Location": "/login"})
# Welcome画面を表示
body = render("welcome.html", context={"username": cookies["username"]})
return HTTPResponse(body=body)
def now(request: HTTPRequest) -> HTTPResponse:
"""
現在時刻を表示するHTMLを生成する
"""
context = {"now": datetime.now()}
body = render("now.html", context)
return HTTPResponse(body=body)
def now(request: HTTPRequest) -> HTTPResponse:
"""
現在時刻を表示するHTMLを生成する
"""
context = {"now": datetime.now()}
body = render("now.html", context)
content_type = "text/html; charset=UTF-8"
return HTTPResponse(body=body, content_type=content_type, status_code=200)
def show_request(request: HTTPRequest) -> HTTPResponse:
"""
HTTPリクエストの内容を表示するHTMLを生成する
"""
context = {
"request": request,
"headers": pformat(request.headers),
"body": request.body.decode("utf-8", "ignore")
}
body = render("show_request.html", context)
return HTTPResponse(body=body)
def login(request: HTTPRequest) -> HTTPResponse:
if request.method == "GET":
body = render("login.html", {})
return HTTPResponse(body=body)
elif request.method == "POST":
post_params = urllib.parse.parse_qs(request.body.decode())
username = post_params["username"][0]
headers = {
"Location": "/welcome",
"Set-Cookie": f"username={username}"
}
return HTTPResponse(status_code=302, headers=headers)
def welcome(request: HTTPRequest) -> HTTPResponse:
# Cookieにusernameが含まれていなければ、ログインしていないとみなして/loginへリダイレクト
if "username" not in request.cookies:
return HTTPResponse(status_code=302, headers={"Location": "/login"})
# Welcome画面を表示
username = request.cookies["username"]
email = request.cookies["email"]
body = render("welcome.html",
context={
"username": username,
"email": email
})
return HTTPResponse(body=body)
def parameters(request: HTTPRequest) -> HTTPResponse:
"""
POSTパラメータを表示するHTMLを表示する
"""
# GETリクエストの場合は、405を返す
if request.method == "GET":
body = b"<html><body><h1>405 Method Not Allowed</h1></body></html>"
return HTTPResponse(body=body, status_code=405)
elif request.method == "POST":
context = {"params": urllib.parse.parse_qs(request.body.decode())}
body = render("parameters.html", context)
return HTTPResponse(body=body)
def login(request: HTTPRequest) -> HTTPResponse:
if request.method == "GET":
body = render("login.html", {})
return HTTPResponse(body=body)
elif request.method == "POST":
post_params = urllib.parse.parse_qs(request.body.decode())
username = post_params["username"][0]
email = post_params["email"][0]
cookies = [
Cookie(name="username", value=username, max_age=30),
Cookie(name="email", value=email, max_age=30),
]
return HTTPResponse(status_code=302,
headers={"Location": "/welcome"},
cookies=cookies)
def user_profile(request: HTTPRequest) -> HTTPResponse:
context = {"user_id": request.params["user_id"]}
body = render("user_profile.html", context)
return HTTPResponse(body=body)