Files
ai-agent-gemini/functions/get_files_info.py
T

20 lines
883 B
Python

import os
def get_files_info(working_directory, directory="."):
abs_path = os.path.abspath(working_directory)
target_dir = os.path.normpath(os.path.join(abs_path, directory))
# Will be True or False
valid_target_dir = os.path.commonpath([abs_path, target_dir]) == abs_path
if not valid_target_dir:
return f'Error: Cannot list "{directory}" as it is outside the permitted working directory'
if type(directory) is not str:
return f'Error: "{directory}" is not a directory'
try:
dir_list = os.listdir(target_dir)
dir_string = ""
for item in dir_list:
dir_string += f"- {item}: file_size={os.path.getsize(f"{target_dir}/{item}")}, is_dir={os.path.isdir(f"{target_dir}/{item}")}\n"
return dir_string
except Exception as e:
return f"Error: encountered during loop over dir_list: {e}"