Gitlab repo sync ETL handler

backend/analytics_server/mhq/exapi/gitlab.py

@@ -18,6 +18,20 @@ def __init__(self, access_token: str, domain="gitlab.com"):
         self.base_url = f"https://{domain}/api/v4"
         self.headers = {"Authorization": f"Bearer {self._token}"}
 
+    def check_pat(self) -> bool:
+        """
+        Checks if PAT is Valid
+        :returns:
+        :raises HTTPError: If the request fails and status code is not 200
+        """
+        url = f"{self.base_url}/user"
+        try:
+            response = requests.get(url, headers=self.headers)
+        except Exception as e:
+            raise Exception(f"Error in PAT validation, Error: {e.data}")
+
+        return response.status_code == 200
+
     def _handle_error(self, response):
         if response.status_code != 200:
             error = response.json().get("error", "")

backend/analytics_server/mhq/service/code/integration.py

@@ -5,6 +5,7 @@
 
 CODE_INTEGRATION_BUCKET = [
     UserIdentityProvider.GITHUB.value,
+    UserIdentityProvider.GITLAB.value,
 ]
 
 

backend/analytics_server/mhq/service/code/sync/etl_code_factory.py

@@ -1,3 +1,4 @@
+from mhq.service.code.sync.etl_gitlab_handler import get_gitlab_etl_handler
 from mhq.service.code.sync.etl_github_handler import get_github_etl_handler
 from mhq.service.code.sync.etl_provider_handler import CodeProviderETLHandler
 from mhq.store.models.code import CodeProvider
@@ -10,4 +11,8 @@ def __init__(self, org_id: str):
     def __call__(self, provider: str) -> CodeProviderETLHandler:
         if provider == CodeProvider.GITHUB.value:
             return get_github_etl_handler(self.org_id)
+
+        if provider == CodeProvider.GITLAB.value:
+            return get_gitlab_etl_handler(self.org_id)
+
         raise NotImplementedError(f"Unknown provider - {provider}")