xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-02 03:40:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add filter for files with bad extensions in language handler

Browse Source
This commit is contained in:
mrT23
2025-03-11 17:03:05 +02:00
parent 0f33750035
commit 9a574e0caa
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pr_agent/algo/language_handler.py
View File

@ -45,6 +45,8 @@ def sort_files_by_main_languages(languages: Dict, files: list):
else:
main_extensions.append([])
# filter out files bad extensions
files_filtered = filter_bad_extensions(files)
# sort files by their extension, put the files that are in the main extension first
# and the rest files after, map languages_sorted to their respective files
@ -53,7 +55,7 @@ def sort_files_by_main_languages(languages: Dict, files: list):
# if no languages detected, put all files in the "Other" category
if not languages:
files_sorted = [({"language": "Other", "files": list(files)})]
files_sorted = [({"language": "Other", "files": list(files_filtered)})]
return files_sorted
main_extensions_flat = []
@ -62,7 +64,7 @@ def sort_files_by_main_languages(languages: Dict, files: list):
for extensions, lang in zip(main_extensions, languages_sorted_list): # noqa: B905
tmp = []
for file in files:
for file in files_filtered:
extension_str = f".{file.filename.split('.')[-1]}"
if extension_str in extensions:
tmp.append(file)