From 4fbe7d14b57188b75f47d411b697094479a7e5d6 Mon Sep 17 00:00:00 2001 From: mrT23 Date: Sun, 17 Sep 2023 16:41:53 +0300 Subject: [PATCH] protection for no language --- pr_agent/algo/language_handler.py | 5 +++++ pr_agent/git_providers/git_provider.py | 4 ++++ 2 files changed, 9 insertions(+) diff --git a/pr_agent/algo/language_handler.py b/pr_agent/algo/language_handler.py index 586a3161..66e85025 100644 --- a/pr_agent/algo/language_handler.py +++ b/pr_agent/algo/language_handler.py @@ -42,6 +42,11 @@ def sort_files_by_main_languages(languages: Dict, files: list): files_sorted = [] rest_files = {} + # if no languages detected, put all files in the "Other" category + if not languages: + files_sorted = [({"language": "Other", "files": list(files_filtered)})] + return files_sorted + main_extensions_flat = [] for ext in main_extensions: main_extensions_flat.extend(ext) diff --git a/pr_agent/git_providers/git_provider.py b/pr_agent/git_providers/git_provider.py index 330590a1..0911d2d2 100644 --- a/pr_agent/git_providers/git_provider.py +++ b/pr_agent/git_providers/git_provider.py @@ -132,6 +132,10 @@ def get_main_pr_language(languages, files) -> str: Get the main language of the commit. Return an empty string if cannot determine. """ main_language_str = "" + if not languages: + logging.info("No languages detected") + return main_language_str + try: top_language = max(languages, key=languages.get).lower()