From e60d75228fb161e464ca59fa2526bf0765f4d902 Mon Sep 17 00:00:00 2001
From: lyg <1543117173@qq.com>
Date: 星期四, 22 五月 2025 12:35:55 +0800
Subject: [PATCH] 修改指令json生成，加入fastapi
---
knowledgebase/utils.py | 25 +++++++++++++++++++++----
1 files changed, 21 insertions(+), 4 deletions(-)
diff --git a/knowledgebase/utils.py b/knowledgebase/utils.py
index 0314db2..7ba2e2c 100644
--- a/knowledgebase/utils.py
+++ b/knowledgebase/utils.py
@@ -3,6 +3,7 @@
import os
import json
import re
+import asyncio
def get_bit_mask(start, end):
@@ -27,6 +28,7 @@
return md5_digest
+
def generate_bytes_md5(input_bytes):
# 鍒涘缓涓�涓� md5 鍝堝笇瀵硅薄
md5_hash = hashlib.md5()
@@ -39,21 +41,36 @@
return md5_digest
+
+async def get_md5_async(file: bytes):
+ md5 = await asyncio.to_thread(generate_bytes_md5, file)
+ return md5
+
+
+async def save_to_file_async(data: bytes, file_path: str):
+ await asyncio.to_thread(save_bytes_to_file, data, file_path)
+
+
def file_exists(cache_file: str):
return os.path.exists(cache_file)
-def read_from_file(cache_file: str) -> str:
- with open(cache_file, 'r', encoding='utf-8') as f:
+def read_from_file(file: str) -> str:
+ with open(file, 'r', encoding='utf-8') as f:
text = f.read()
return text
-def save_to_file(text, cache_file):
- with open(cache_file, 'w', encoding='utf-8') as f:
+def save_text_to_file(text: str, file: str):
+ with open(file, 'w', encoding='utf-8') as f:
f.write(text)
+def save_bytes_to_file(bytes_data: bytes, file: str):
+ with open(file, 'wb') as f:
+ f.write(bytes_data)
+
+
def replace_tpl_paras(tpl_text: str, data: dict):
for key, val in data.items():
if not isinstance(val, str):
--
Gitblit v1.9.1