fix: remove repeat util

zhayujie · Nov 27, 2023 · 21ad51f · 21ad51f
1 parent 697c6d5
commit 21ad51f
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 12 deletions.
diff --git a/channel/feishu/feishu_channel.py b/channel/feishu/feishu_channel.py
@@ -19,7 +19,7 @@
 from common.expired_dict import ExpiredDict
 from bridge.context import ContextType
 from channel.chat_channel import ChatChannel, check_prefix
-from utils import file_util
+from common import utils
 import json
 import os
 
@@ -118,7 +118,7 @@ def fetch_access_token(self) -> str:
     def _upload_image_url(self, img_url, access_token):
         logger.debug(f"[WX] start download image, img_url={img_url}")
         response = requests.get(img_url)
-        suffix = file_util.get_path_suffix(img_url)
+        suffix = utils.get_path_suffix(img_url)
         temp_name = str(uuid.uuid4()) + "." + suffix
         if response.status_code == 200:
             # 将图片内容保存为临时文件

diff --git a/channel/feishu/feishu_message.py b/channel/feishu/feishu_message.py
@@ -4,7 +4,7 @@
 import requests
 from common.log import logger
 from common.tmp_dir import TmpDir
-from utils import file_util
+from common import utils
 
 
 class FeishuMessage(ChatMessage):
@@ -28,7 +28,7 @@ def __init__(self, event: dict, is_group=False, access_token=None):
             file_key = content.get("file_key")
             file_name = content.get("file_name")
 
-            self.content = TmpDir().path() + file_key + "." + file_util.get_path_suffix(file_name)
+            self.content = TmpDir().path() + file_key + "." + utils.get_path_suffix(file_name)
 
             def _download_file():
                 # 如果响应状态码是200，则将响应内容写入本地文件

diff --git a/utils/file_util.py b/utils/file_util.py