| 9 | |
| 10 | |
| 11 | class SessionState(reflex_local_auth.LocalAuthState): |
| 12 | @rx.cached_var |
| 13 | def my_userinfo_id(self) -> str | None: |
| 14 | if self.authenticated_user_info is None: |
| 15 | return None |
| 16 | return self.authenticated_user_info.id |
| 17 | |
| 18 | @rx.cached_var |
| 19 | def my_user_id(self) -> str | None: |
| 20 | if self.authenticated_user.id < 0: |
| 21 | return None |
| 22 | return self.authenticated_user.id |
| 23 | |
| 24 | @rx.cached_var |
| 25 | def authenticated_username(self) -> str | None: |
| 26 | if self.authenticated_user.id < 0: |
| 27 | return None |
| 28 | return self.authenticated_user.username |
| 29 | |
| 30 | @rx.cached_var |
| 31 | def authenticated_user_info(self) -> UserInfo | None: |
| 32 | if self.authenticated_user.id < 0: |
| 33 | return None |
| 34 | with rx.session() as session: |
| 35 | result = session.exec( |
| 36 | sqlmodel.select(UserInfo).where( |
| 37 | UserInfo.user_id == self.authenticated_user.id |
| 38 | ), |
| 39 | ).one_or_none() |
| 40 | if result is None: |
| 41 | return None |
| 42 | # database lookup |
| 43 | # result.user |
| 44 | # user_obj = result.user |
| 45 | # print(result.user) |
| 46 | return result |
| 47 | |
| 48 | def on_load(self): |
| 49 | if not self.is_authenticated: |
| 50 | return reflex_local_auth.LoginState.redir |
| 51 | print(self.is_authenticated) |
| 52 | print(self.authenticated_user_info) |
| 53 | |
| 54 | def perform_logout(self): |
| 55 | self.do_logout() |
| 56 | return rx.redirect("/") |
| 57 | |
| 58 | |
| 59 |
nothing calls this directly
no outgoing calls
no test coverage detected