aboutsummaryrefslogtreecommitdiff
path: root/labbot/addons/approve-merge.py
blob: 7a9e54cf9bd6af35fbd1a4ce2e18bc138e159aa6 (plain)
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
"""
automatically merges merge requests if a certain number of approvals
is reached.
"""
import logging

from labbot.config import Config

log = logging.getLogger(__name__)

config = Config(__name__)
config.set_global_data(
    required_approval_count = 2
)

@config.config_decorator()
async def merge_label_hook(event, gl, *args, **kwargs):
    title = event.object_attributes["title"]
    state = event.object_attributes["state"]
    merge_status = event.object_attributes["merge_status"]
    iid = event.object_attributes['iid']

    if state != "opened" or title.lower().startswith("draft"):
        return

    approvals_url = f"/projects/{event.project_id}/merge_requests/{iid}/approvals"
    merge_url = f"/projects/{event.project_id}/merge_requests/{iid}/merge"

    data = await gl.getitem(approvals_url)

    approval_count = len(data["approved_by"])

    if approval_count >= config["required_approval_count"]:
        if merge_status == "can_be_merged":
            await gl.put(merge_url)
        else:
            log.debug(f"Cannot merge !{iid}({event.project_id}) because of its merge_status `{merge_status}`")

def setup(bot):
    config.setup(__name__, bot.name)
    bot.register_merge_hook(merge_label_hook)