修复 Gitlab 不同版本 mr 请求区别

Author: xin.xu1

gitlab_integration/gitlab_fetcher.py

@@ -202,4 +202,16 @@ def _build_authenticated_url(self, repo_url):
         elif repo_url.startswith("http://"):
             return f"http://oauth2:{token}@{repo_url[7:]}"
         else:
-            raise ValueError("Unsupported URL scheme")
+            raise ValueError("Unsupported URL scheme")
+
+def is_merge_request_opened(gitlab_payload) -> bool:
+    """
+    判断是否是merge request打开事件
+    """
+    try:
+        gitlab_merge_request_old  = gitlab_payload.get("object_attributes").get("state") == "opened" and gitlab_payload.get("object_attributes").get("merge_status") == "preparing"
+        gitlab_merge_request_new = gitlab_payload.get("object_attributes").get("state") == "merged" and gitlab_payload.get("object_attributes").get("merge_status") == "can_be_merged"
+        return gitlab_merge_request_old or gitlab_merge_request_new
+    except Exception as e:
+        log.error(f"判断是否是merge request打开事件失败: {e}")
+        return False

gitlab_integration/webhook_listener.py

@@ -7,7 +7,7 @@
 from response_module.response_controller import ReviewResponse
 from review_engine.review_engine import ReviewEngine
 from utils.logger import log
-
+from gitlab_integration.gitlab_fetcher import is_merge_request_opened
 
 class WebhookListener:
     def __init__(self):
@@ -53,7 +53,7 @@ def handle_merge_request(self, gitlab_payload, reply):
         """
         处理合并请求事件
         """
-        if gitlab_payload.get("object_attributes").get("state") == "opened" and gitlab_payload.get("object_attributes").get("merge_status") == "preparing":
+        if is_merge_request_opened(gitlab_payload):
             log.info("首次merge_request ", gitlab_payload)
             project_id = gitlab_payload.get('project')['id']
             merge_request_iid = gitlab_payload.get("object_attributes")["iid"]