From 0d629621ac8ed6b7ad33daea334b16fc1f876f93 Mon Sep 17 00:00:00 2001
From: yym68686 <yym68686@outlook.com>
Date: Thu, 9 Nov 2023 14:03:37 +0800
Subject: [PATCH] =?UTF-8?q?=E5=B0=86=E6=8C=87=E5=AE=9A=E8=8C=83=E5=9B=B4?=
 =?UTF-8?q?=E5=86=85=E7=9A=84API=E5=AD=97=E7=AC=A6=E6=9B=BF=E6=8D=A2?=
 =?UTF-8?q?=E4=B8=BA=E6=98=9F=E5=8F=B7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 bot.py           | 18 ++++++++++--------
 test/test_API.py |  6 ++++++
 2 files changed, 16 insertions(+), 8 deletions(-)
 create mode 100644 test/test_API.py

diff --git a/bot.py b/bot.py
index 768aacba..51b2357c 100644
--- a/bot.py
+++ b/bot.py
@@ -260,6 +260,8 @@ async def delete_message(update, context, messageid, delay=10):
 if os.environ.get('GOOGLE_API_KEY', None) == None and os.environ.get('GOOGLE_CSE_ID', None) == None:
     first_buttons[1][1] = InlineKeyboardButton("google已关闭", callback_data="google")
 
+def replace_with_asterisk(string, start=10, end=45):
+    return string[:start] + '*' * (end - start) + string[end:]
 
 banner = "👇下面可以随时更改默认 gpt 模型："
 async def button_press(update, context):
@@ -269,7 +271,7 @@ async def button_press(update, context):
         f"**Default engine:** `{config.GPT_ENGINE}`\n"
         f"**temperature:** `{config.temperature}`\n"
         f"**API_URL:** `{config.API_URL}`\n\n"
-        f"**API:** `{config.API}`\n\n"
+        f"**API:** `{replace_with_asterisk(config.API)}`\n\n"
         f"**WEB_HOOK:** `{config.WEB_HOOK}`\n\n"
     )
     callback_query = update.callback_query
@@ -286,7 +288,7 @@ async def button_press(update, context):
                 f"**Default engine:** `{config.GPT_ENGINE}`\n"
                 f"**temperature:** `{config.temperature}`\n"
                 f"**API_URL:** `{config.API_URL}`\n\n"
-                f"**API:** `{config.API}`\n\n"
+                f"**API:** `{replace_with_asterisk(config.API)}`\n\n"
                 f"**WEB_HOOK:** `{config.WEB_HOOK}`\n\n"
             )
             message = await callback_query.edit_message_text(
@@ -320,7 +322,7 @@ async def button_press(update, context):
             f"**Default engine:** `{config.GPT_ENGINE}`\n"
             f"**temperature:** `{config.temperature}`\n"
             f"**API_URL:** `{config.API_URL}`\n\n"
-            f"**API:** `{config.API}`\n\n"
+            f"**API:** `{replace_with_asterisk(config.API)}`\n\n"
             f"**WEB_HOOK:** `{config.WEB_HOOK}`\n\n"
         )
         message = await callback_query.edit_message_text(
@@ -340,7 +342,7 @@ async def button_press(update, context):
             f"**Default engine:** `{config.GPT_ENGINE}`\n"
             f"**temperature:** `{config.temperature}`\n"
             f"**API_URL:** `{config.API_URL}`\n\n"
-            f"**API:** `{config.API}`\n\n"
+            f"**API:** `{replace_with_asterisk(config.API)}`\n\n"
             f"**WEB_HOOK:** `{config.WEB_HOOK}`\n\n"
         )
 
@@ -363,7 +365,7 @@ async def button_press(update, context):
             f"**Default engine:** `{config.GPT_ENGINE}`\n"
             f"**temperature:** `{config.temperature}`\n"
             f"**API_URL:** `{config.API_URL}`\n\n"
-            f"**API:** `{config.API}`\n\n"
+            f"**API:** `{replace_with_asterisk(config.API)}`\n\n"
             f"**WEB_HOOK:** `{config.WEB_HOOK}`\n\n"
         )
         message = await callback_query.edit_message_text(
@@ -383,7 +385,7 @@ async def button_press(update, context):
             f"**Default engine:** `{config.GPT_ENGINE}`\n"
             f"**temperature:** `{config.temperature}`\n"
             f"**API_URL:** `{config.API_URL}`\n\n"
-            f"**API:** `{config.API}`\n\n"
+            f"**API:** `{replace_with_asterisk(config.API)}`\n\n"
             f"**WEB_HOOK:** `{config.WEB_HOOK}`\n\n"
         )
         message = await callback_query.edit_message_text(
@@ -403,7 +405,7 @@ async def button_press(update, context):
             f"**Default engine:** `{config.GPT_ENGINE}`\n"
             f"**temperature:** `{config.temperature}`\n"
             f"**API_URL:** `{config.API_URL}`\n\n"
-            f"**API:** `{config.API}`\n\n"
+            f"**API:** `{replace_with_asterisk(config.API)}`\n\n"
             f"**WEB_HOOK:** `{config.WEB_HOOK}`\n\n"
         )
         message = await callback_query.edit_message_text(
@@ -419,7 +421,7 @@ async def info(update, context):
         f"**Default engine:** `{config.GPT_ENGINE}`\n"
         f"**temperature:** `{config.temperature}`\n"
         f"**API_URL:** `{config.API_URL}`\n\n"
-        f"**API:** `{config.API}`\n\n"
+        f"**API:** `{replace_with_asterisk(config.API)}`\n\n"
         f"**WEB_HOOK:** `{config.WEB_HOOK}`\n\n"
     )
     message = await context.bot.send_message(chat_id=update.message.chat_id, text=escape(info_message), reply_markup=InlineKeyboardMarkup(first_buttons), parse_mode='MarkdownV2', disable_web_page_preview=True)
diff --git a/test/test_API.py b/test/test_API.py
new file mode 100644
index 00000000..f548b178
--- /dev/null
+++ b/test/test_API.py
@@ -0,0 +1,6 @@
+def replace_with_asterisk(string, start=15, end=40):
+    return string[:start] + '*' * (end - start) + string[end:]
+
+original_string = "sk-zIuWeeuWY8vNCVhhHCXLroNmA6QhBxnv0ARMFcODVQwwqGRg"
+result = replace_with_asterisk(original_string)
+print(result)