feat: make it simple

Codium-ai · Aug 22, 2024 · 055c4cd · 055c4cd
1 parent a5bb130
commit 055c4cd
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pr_agent/servers/gitlab_webhook.py b/pr_agent/servers/gitlab_webhook.py
@@ -124,12 +124,13 @@ async def inner(data: dict):
             return JSONResponse(status_code=status.HTTP_200_OK, content=jsonable_encoder({"message": "success"}))
 
         log_context["sender"] = sender
+        should_skip_draft = get_settings().get("GITLAB.SKIP_DRAFT_MR", False)
         if data.get('object_kind') == 'merge_request' and data['object_attributes'].get('action') in ['open', 'reopen']:
             url = data['object_attributes'].get('url')
             draft = data['object_attributes'].get('draft')
             get_logger().info(f"New merge request: {url}")
 
-            if draft:
+            if draft and should_skip_draft:
                 get_logger().info(f"Skipping draft MR: {url}")
                 return JSONResponse(status_code=status.HTTP_200_OK, content=jsonable_encoder({"message": "success"}))
 
@@ -139,7 +140,7 @@ async def inner(data: dict):
                 mr = data['merge_request']
                 url = mr.get('url')
                 draft = mr.get('draft')
-                if draft:
+                if draft and should_skip_draft:
                     get_logger().info(f"Skipping draft MR: {url}")
                     return JSONResponse(status_code=status.HTTP_200_OK, content=jsonable_encoder({"message": "success"}))