import logging from rio import Component from src.ez_lan_manager.components.LoginBox import LoginBox from src.ez_lan_manager.components.UserInfoBox import UserInfoBox from src.ez_lan_manager.types.SessionStorage import SessionStorage logger = logging.getLogger(__name__.split(".")[-1]) class UserInfoAndLoginBox(Component): def build(self) -> Component: if self.session[SessionStorage].user_id is None: return LoginBox(status_change_cb=self.force_refresh) else: return UserInfoBox(status_change_cb=self.force_refresh)