gitlab

dagster-io · May 7, 2024 · b5206d2 · b5206d2
1 parent 4e1ad1b
commit b5206d2
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/python_modules/dagster/dagster/_core/definitions/metadata/source_code.py b/python_modules/dagster/dagster/_core/definitions/metadata/source_code.py
@@ -204,14 +204,21 @@ def _build_github_url(url: str, branch: str) -> str:
     return f"{url}/tree/{branch}"
 
 
+def _build_gitlab_url(url: str, branch: str) -> str:
+    return f"{url}/-/tree/{branch}"
+
+
 @experimental
 def link_to_source_control(
     assets_defs: Sequence[Union["AssetsDefinition", "SourceAsset", "CacheableAssetsDefinition"]],
     source_control_url: str,
     source_control_branch: str,
     repository_root_absolute_path: Union[Path, str],
 ) -> Sequence[Union["AssetsDefinition", "SourceAsset", "CacheableAssetsDefinition"]]:
-    if source_control_url and "github.com/" in source_control_url:
+    if "gitlab.com" in source_control_url:
+        source_control_url = _build_gitlab_url(source_control_url, source_control_branch)
+    else:
+        # assume GitHub URL scheme
         source_control_url = _build_github_url(source_control_url, source_control_branch)
 
     return [