支持将文件路径/文件内容/字符串转成base64编码格式,方便用例执行过程中调用
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user