From f74a1bfa95ce184e73a84c14313296b8a7d08153 Mon Sep 17 00:00:00 2001 From: hehesheng Date: Tue, 18 Jun 2024 22:52:26 +0800 Subject: [PATCH] fix: sign generate --- backend/TgFileSystemClientManager.py | 4 ++++ backend/api.py | 1 - frontend/search.py | 2 +- 3 files changed, 5 insertions(+), 2 deletions(-) diff --git a/backend/TgFileSystemClientManager.py b/backend/TgFileSystemClientManager.py index 22c54a9..bcc8f02 100644 --- a/backend/TgFileSystemClientManager.py +++ b/backend/TgFileSystemClientManager.py @@ -113,10 +113,14 @@ class TgFileSystemClientManager(object): need_encrypt_bin = need_encrypt_str.encode() sign_bin = rsa.encrypt(need_encrypt_bin, self.public_key) sign = base64.b64encode(sign_bin).decode() + sign = sign.replace("+", "-") + sign = sign.replace("/", "_") logger.info(f"generate {sign_type.name} sign: {sign}") return sign def parse_sign(self, sign: str) -> dict[str, any] | None: + sign = sign.replace("-", "+") + sign = sign.replace("_", "/") try: res_dict = {} sign_bin = base64.b64decode(sign) diff --git a/backend/api.py b/backend/api.py index ee31508..5849010 100644 --- a/backend/api.py +++ b/backend/api.py @@ -59,7 +59,6 @@ async def verify_post_sign(body: TgToFileListRequestBody): async def verify_get_sign(sign: str): clients_mgr = TgFileSystemClientManager.get_instance() - sign = sign.replace(" ", "+") if not clients_mgr.verify_sign(sign): raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=f"{sign}") return sign diff --git a/frontend/search.py b/frontend/search.py index 056c4b4..ebdb089 100644 --- a/frontend/search.py +++ b/frontend/search.py @@ -45,7 +45,7 @@ def loop(sign: str): st.session_state.chat_select_list = st.multiselect("**Search in**", chat_list, default=chat_list) with columns[2]: st.text("Sort") - st.query_params.is_order = st.toggle("Time🔼", value=isorder) + st.query_params.is_order = st.toggle("Time 🔼", value=isorder) search_limit_container = st.container() with search_limit_container: