1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
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
|
