diff --git a/utils/data_utils/data_handle.py b/utils/data_utils/data_handle.py index cb81857..4dd9801 100644 --- a/utils/data_utils/data_handle.py +++ b/utils/data_utils/data_handle.py @@ -20,7 +20,7 @@ from requests.utils import dict_from_cookiejar from utils.data_utils.faker_handle import FakerData from utils.data_utils.eval_data_handle import eval_data from utils.data_utils.data_tools import zip_test_step -from utils.files_utils.files_handle import file_to_base64, get_files +from utils.files_utils.files_handle import file_to_base64, filepath_to_base64, get_files from config.path_config import FILES_DIR @@ -252,6 +252,43 @@ def none_to_null(target): return target +def get_file_base64(file_name): + """ + 返回文文件内容的base64编码 + """ + file_path = os.path.join(FILES_DIR, file_name) + if os.path.exists(file_path): + # 如果文件是一个真实存在的路径,则返回base64编码内容 + return file_to_base64(file_path=file_path) + else: + logger.warning(f"找不到该文件,将返回空, 传入的文件名:{file_name}") + return None + + +def get_filepath_base64(file_name): + """ + 返回文件路径的base64编码 + """ + file_path = os.path.join(FILES_DIR, file_name) + if os.path.exists(file_path): + # 如果文件是一个真实存在的路径,则返回base64编码内容 + return filepath_to_base64(file_path=file_path) + + else: + logger.warning(f"找不到该文件,将返回空, 传入的文件名:{file_name}") + return None + + +def get_base64_content(input_string: str): + """ + 获取base64编码内容 + """ + byte_string = input_string.encode('utf-8') + base64_bytes = base64.b64encode(byte_string) + base64_string = base64_bytes.decode('utf-8') + return base64_string + + # 声明data_handle方法,这样外部就可以直接import data_handle来使用了 data_handle = DataHandle().data_handle