From c2bec614e518bba563c592318aab4c59d3107a11 Mon Sep 17 00:00:00 2001 From: mrT23 Date: Wed, 8 Nov 2023 14:46:11 +0200 Subject: [PATCH] s --- pr_agent/algo/utils.py | 32 +++++++++++++++++++------------- 1 file changed, 19 insertions(+), 13 deletions(-) diff --git a/pr_agent/algo/utils.py b/pr_agent/algo/utils.py index ecaf4aa8..e5d13475 100644 --- a/pr_agent/algo/utils.py +++ b/pr_agent/algo/utils.py @@ -328,20 +328,26 @@ def set_custom_labels(variables): variables["custom_labels_examples"] = f" - {list(labels.keys())[0]}" -def get_user_labels(current_labels): - ## Only keep labels that has been added by the user - if current_labels is None: - current_labels = [] - user_labels = [] - for label in current_labels: - if label in ['Bug fix', 'Tests', 'Refactoring', 'Enhancement', 'Documentation', 'Other']: - continue - if get_settings().config.enable_custom_labels: - if label in get_settings().custom_labels: +def get_user_labels(current_labels: List[str] = None): + """ + Only keep labels that has been added by the user + """ + try: + if current_labels is None: + current_labels = [] + user_labels = [] + for label in current_labels: + if label.lower() in ['bug fix', 'tests', 'refactoring', 'enhancement', 'documentation', 'other']: continue - user_labels.append(label) - if user_labels: - get_logger().info(f"Keeping user labels: {user_labels}") + if get_settings().config.enable_custom_labels: + if label in get_settings().custom_labels: + continue + user_labels.append(label) + if user_labels: + get_logger().info(f"Keeping user labels: {user_labels}") + except Exception as e: + get_logger().exception(f"Failed to get user labels: {e}") + return current_labels return user_labels