支持将文件路径/文件内容/字符串转成base64编码格式,方便用例执行过程中调用

This commit is contained in:
floraachy
2024-01-26 09:58:07 +08:00
parent 1c657dd264
commit 2eb41318ab

View File

@@ -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