diff options
| -rw-r--r-- | src/infini/input.py | 19 |
1 files changed, 11 insertions, 8 deletions
diff --git a/src/infini/input.py b/src/infini/input.py index 0bbb3f2f..2a926e2a 100644 --- a/src/infini/input.py +++ b/src/infini/input.py @@ -13,16 +13,19 @@ class Input(Generic[T]): self.plain_data = plain_data self.variables = variables or {} - def get_user_id(self) -> Optional[str]: - return self.variables.get("user_id") + def get_user_id(self) -> str: + return self.variables.get("user_id", "0") def get_session_id(self) -> str: - if session_id := self.variables.get("session_id"): - return session_id - - user_id = self.variables.get("user_id", "unknown") - group_id = self.variables.get("group_id", "unknown") - return f"session_{group_id}_{user_id}" + return ( + self.variables.get("group_id") + or self.variables.get("session_id") + or self.variables.get("user_id") + or "0" + ) + + def is_tome(self) -> bool: + return bool(self.variables.get("is_tome")) def get_plain_text(self) -> str: return str(self.plain_data) |