kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-07-20 05:27:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: remove repeat util

Browse Source
This commit is contained in:
zhayujie
2023-11-27 14:24:26 +08:00
parent 697c6d5fbe
commit 21ad51ffbf
3 changed files with 4 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
channel/feishu/feishu_channel.py
View File

@@ -19,7 +19,7 @@ from config import conf
from common.expired_dict import ExpiredDict from common.expired_dict import ExpiredDict
from bridge.context import ContextType from bridge.context import ContextType
from channel.chat_channel import ChatChannel, check_prefix from channel.chat_channel import ChatChannel, check_prefix
from utils import file_util from common import utils
import json import json
import os import os
@@ -118,7 +118,7 @@ class FeiShuChanel(ChatChannel):
def _upload_image_url(self, img_url, access_token): def _upload_image_url(self, img_url, access_token):
logger.debug(f"[WX] start download image, img_url={img_url}") logger.debug(f"[WX] start download image, img_url={img_url}")
response = requests.get(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 temp_name = str(uuid.uuid4()) + "." + suffix
if response.status_code == 200: if response.status_code == 200:
# 将图片内容保存为临时文件 # 将图片内容保存为临时文件

4
channel/feishu/feishu_message.py
View File

@@ -4,7 +4,7 @@ import json
import requests import requests
from common.log import logger from common.log import logger
from common.tmp_dir import TmpDir from common.tmp_dir import TmpDir
from utils import file_util from common import utils
class FeishuMessage(ChatMessage): class FeishuMessage(ChatMessage):
@@ -28,7 +28,7 @@ class FeishuMessage(ChatMessage):
file_key = content.get("file_key") file_key = content.get("file_key")
file_name = content.get("file_name") 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(): def _download_file():
# 如果响应状态码是200，则将响应内容写入本地文件 # 如果响应状态码是200，则将响应内容写入本地文件

8
utils/file_util.py
View File

@@ -1,8 +0,0 @@
from urllib.parse import urlparse
import os
# 获取url后缀
def get_path_suffix(path):
path = urlparse(path).path
return os.path.splitext(path)[-1].lstrip('.')