add DVC_STUDIO_URL envvar support (#9340)

add DVC_STUDIO_URL envvar
iterative · Apr 18, 2023 · dd3deb6 · dd3deb6
1 parent 7841020
commit dd3deb6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/dvc/env.py b/dvc/env.py
@@ -10,3 +10,5 @@
 DVC_EXP_AUTO_PUSH = "DVC_EXP_AUTO_PUSH"
 DVC_NO_ANALYTICS = "DVC_NO_ANALYTICS"
 DVC_SHOW_TRACEBACK = "DVC_SHOW_TRACEBACK"
+DVC_STUDIO_TOKEN = "DVC_STUDIO_TOKEN"  # noqa: S105 # nosec B105
+DVC_STUDIO_URL = "DVC_STUDIO_URL"
diff --git a/dvc/repo/experiments/push.py b/dvc/repo/experiments/push.py
@@ -14,6 +14,7 @@
 from funcy import compact, group_by
 from scmrepo.git.backend.base import SyncStatus
 
+from dvc.env import DVC_STUDIO_TOKEN, DVC_STUDIO_URL
 from dvc.exceptions import DvcException
 from dvc.repo import locked
 from dvc.repo.scm_context import scm_context
@@ -47,7 +48,7 @@ def notify_refs_to_studio(
         return None
 
     token = (
-        os.environ.get("DVC_STUDIO_TOKEN")
+        os.environ.get(DVC_STUDIO_TOKEN)
         or os.environ.get("STUDIO_TOKEN")
         or config.get("token")
     )
@@ -60,7 +61,7 @@ def notify_refs_to_studio(
     from dvc.utils import studio
 
     _, repo_url = get_remote_repo(repo.scm.dulwich.repo, git_remote)
-    studio_url = config.get("url")
+    studio_url = os.environ.get(DVC_STUDIO_URL) or config.get("url")
     d = studio.notify_refs(repo_url, token, base_url=studio_url, **refs)
     return d.get("url")