gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: Add input validation and error handling for paths

Browse source
This commit is contained in:
lxobr 2024-11-19 09:10:04 +01:00
parent 8bc26bba97
commit 263ecb9149
1 changed files with 22 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
cognee/tasks/repo_processor/local_script_dependencies.py
View file

@ -101,14 +101,29 @@ async def get_local_script_dependencies(script_path: str, repo_path: Optional[st
""" """
Extract and return a list of unique module paths that the script depends on. Extract and return a list of unique module paths that the script depends on.
""" """
if repo_path: try:
repo_path_resolved = str(Path(repo_path).resolve()) script_path = Path(script_path).resolve(strict=True)
with add_sys_path(repo_path_resolved): except (FileNotFoundError, PermissionError) as e:
dependencies = await _extract_dependencies(script_path) logger.error(f"Error resolving script path: {e}")
dependencies = [path for path in dependencies if path.startswith(repo_path_resolved)] return []
else:
if not repo_path:
return await _extract_dependencies(script_path)
try:
repo_path = Path(repo_path).resolve(strict=True)
except (FileNotFoundError, PermissionError) as e:
logger.warning(f"Error resolving repo path: {e}. Proceeding without repo_path.")
return await _extract_dependencies(script_path)
if not script_path.is_relative_to(repo_path):
logger.warning(f"Script {script_path} not in repo {repo_path}. Proceeding without repo_path.")
return await _extract_dependencies(script_path)
with add_sys_path(str(repo_path)):
dependencies = await _extract_dependencies(script_path) dependencies = await _extract_dependencies(script_path)
return dependencies
return [path for path in dependencies if path.startswith(str(repo_path))]
if __name__ == "__main__": if __name__ == "__main__":