diff options
Diffstat (limited to 'labbot/addons')
| -rw-r--r-- | labbot/addons/merge-label.py | 73 |
1 files changed, 73 insertions, 0 deletions
diff --git a/labbot/addons/merge-label.py b/labbot/addons/merge-label.py new file mode 100644 index 0000000..e980e6c --- /dev/null +++ b/labbot/addons/merge-label.py @@ -0,0 +1,73 @@ +import os +import re + +title_regex = r"^(?:#|)(\d+)\s*" +word_regex = r"^#(\d+)$" +relation_keywords = [ + "related" +] +relation_distance = 2 + +state_label = { + "opened": "Code-Review", + "merged": "Testing", + "closed": "In Progress", +} +async def merge_request_hook(event, gl, *args, **kwargs): + state = event.object_attributes["state"] + related_issues = [] + + match = re.search(title_regex, event.object_attributes["title"]) + if match: + related_issues.append(match.group(1)) + + for line in event.object_attributes["description"].split("\\n"): + line = line.lower() + line_list = line.split(" ") + + for keyword in relation_keywords: + try: + keyword_index = line_list.index(keyword) + + min_pos = keyword_index - relation_distance + if min_pos < 0: + min_pos = 0 + + max_pos = keyword_index + relation_distance + if max_pos >= len(line_list): + max_pos = len(line_list)-1 + + for word in line_list[min_pos:max_pos+1]: + match = re.search(word_regex, word) + if match: + related_issues.append(match.group(1)) + + except ValueError: + pass + + for issue in related_issues: + if not issue: continue + + base_url = f"/projects/{event.project_id}/issues/{issue}" + + delete_labels = list(state_label.values()) + + try: + label = state_label[state] + if label in delete_labels: + delete_labels.remove(label) + + remove_labels = ",".join(delete_labels) + + await gl.put(base_url, data={ + "add_labels": label, + "remove_labels": remove_labels, + }) + + except KeyError: + # unknown state + pass + +def setup(bot): + bot.register(merge_request_hook, "Merge Request Hook") + pass
\ No newline at end of file |