h-shibuya commited on
Commit
e84b2bd
·
verified ·
1 Parent(s): d8240d8

Update public_src/app.py

Browse files
Files changed (1) hide show
  1. public_src/app.py +6 -3
public_src/app.py CHANGED
@@ -125,15 +125,18 @@ async def auth(request: Request):
125
 
126
  userinfo = access_token.get("userinfo")
127
  id_token = access_token.get('id_token')
128
- session_id = request.cookies.get("session_id")
129
  session_store.set(session_id, {"user": userinfo, "id_token": id_token})
130
-
 
 
 
131
  # ログインユーザー ログ出力
132
  filename = f"loginuser_{datetime.today().strftime('%Y%m')}"
133
  data = {"datetime": datetime.now(JST).strftime('%Y-%m-%d %H:%M:%S'), "email": userinfo.get('email')}
134
  write_log(filename, json.dumps(data))
135
 
136
- return RedirectResponse(url='/main')
137
 
138
  app.mount("/assets", StaticFiles(directory="public_src/assets"), name="assets")
139
 
 
125
 
126
  userinfo = access_token.get("userinfo")
127
  id_token = access_token.get('id_token')
128
+ session_id = request.cookies.get("session_id") or generate_session_id()
129
  session_store.set(session_id, {"user": userinfo, "id_token": id_token})
130
+
131
+ response = RedirectResponse(url='/main')
132
+ response.set_cookie(key="session_id", value=session_id, httponly=True)
133
+
134
  # ログインユーザー ログ出力
135
  filename = f"loginuser_{datetime.today().strftime('%Y%m')}"
136
  data = {"datetime": datetime.now(JST).strftime('%Y-%m-%d %H:%M:%S'), "email": userinfo.get('email')}
137
  write_log(filename, json.dumps(data))
138
 
139
+ return response
140
 
141
  app.mount("/assets", StaticFiles(directory="public_src/assets"), name="assets")
142