瀏覽代碼

fix: remove repeat util

master
zhayujie 1 年之前
父節點
當前提交
21ad51ffbf
共有 3 個檔案被更改,包括 4 行新增12 行删除
  1. +2
    -2
      channel/feishu/feishu_channel.py
  2. +2
    -2
      channel/feishu/feishu_message.py
  3. +0
    -8
      utils/file_util.py

+ 2
- 2
channel/feishu/feishu_channel.py 查看文件

@@ -19,7 +19,7 @@ from config import conf
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 @@ class FeiShuChanel(ChatChannel):
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:
# 将图片内容保存为临时文件


+ 2
- 2
channel/feishu/feishu_message.py 查看文件

@@ -4,7 +4,7 @@ import json
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 @@ class FeishuMessage(ChatMessage):
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,则将响应内容写入本地文件


+ 0
- 8
utils/file_util.py 查看文件

@@ -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('.')

Loading…
取消
儲存