Added support for interaction with Posit Connect deployments in SPCS (#660)

Author: edavidaja

.github/workflows/main.yml

@@ -17,6 +17,7 @@ permissions:
 jobs:
   test:
     strategy:
+      fail-fast: false
       matrix:
         os: [ubuntu-latest]
         python-version: ['3.8', '3.9', '3.10', '3.11', '3.12']
@@ -254,5 +255,3 @@ jobs:
               name: cypress-screenshots_${{ matrix.PY_VERSION }}_native
               path: test/connect-rsconnect-python/cypress/screenshots
               if-no-files-found: ignore
-
-

CHANGELOG.md

@@ -8,6 +8,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Added
 
+- Added support for interaction with Posit Connect deployments
+  hosted in Snowpark Container Services.
 - `rsconnect` now detects Python interpreter version requirements from
   `.python-version`, `pyproject.toml` and `setup.cfg`
 - `--python` and `--override-python-version` options are now deprecated

pyproject.toml

@@ -38,6 +38,7 @@ test = [
     "twine",
     "types-Flask",
 ]
+snowflake = ["snowflake-cli"]
 
 [project.urls]
 Repository = "http://github.com/posit-dev/rsconnect-python"

rsconnect/actions.py

@@ -172,6 +172,14 @@ def test_rstudio_server(server: api.PositServer):
             raise RSConnectException("Failed to verify with {} ({}).".format(server.remote_name, exc))
 
 
+def test_spcs_server(server: api.SPCSConnectServer):
+    with api.RSConnectClient(server) as client:
+        try:
+            client.me()
+        except RSConnectException as exc:
+            raise RSConnectException("Failed to verify with {} ({}).".format(server.remote_name, exc))
+
+
 def test_api_key(connect_server: api.RSConnectServer) -> str:
     """
     Test that an API Key may be used to authenticate with the given Posit Connect server.

rsconnect/actions_content.py

@@ -9,11 +9,11 @@
 import traceback
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from datetime import datetime, timedelta
-from typing import Iterator, Literal, Optional, Sequence, cast
+from typing import Iterator, Literal, Optional, Sequence, cast, Union
 
 import semver
 
-from .api import RSConnectClient, RSConnectServer, emit_task_log
+from .api import RSConnectServer, SPCSConnectServer, RSConnectClient, emit_task_log
 from .exception import RSConnectException
 from .log import logger
 from .metadata import ContentBuildStore, ContentItemWithBuildState
@@ -33,7 +33,7 @@ def content_build_store() -> ContentBuildStore:
     return _content_build_store
 
 
-def ensure_content_build_store(connect_server: RSConnectServer) -> ContentBuildStore:
+def ensure_content_build_store(connect_server: Union[RSConnectServer, SPCSConnectServer]) -> ContentBuildStore:
     global _content_build_store
     if not _content_build_store:
         logger.info("Initializing ContentBuildStore for %s" % connect_server.url)
@@ -42,7 +42,7 @@ def ensure_content_build_store(connect_server: RSConnectServer) -> ContentBuildS
 
 
 def build_add_content(
-    connect_server: RSConnectServer,
+    connect_server: Union[RSConnectServer, SPCSConnectServer],
     content_guids_with_bundle: Sequence[ContentGuidWithBundle],
 ):
     """
@@ -85,7 +85,7 @@ def _validate_build_rm_args(guid: Optional[str], all: bool, purge: bool):
 
 
 def build_remove_content(
-    connect_server: RSConnectServer,
+    connect_server: Union[RSConnectServer, SPCSConnectServer],
     guid: Optional[str],
     all: bool,
     purge: bool,
@@ -109,20 +109,20 @@ def build_remove_content(
     return guids
 
 
-def build_list_content(connect_server: RSConnectServer, guid: str, status: Optional[str]):
+def build_list_content(connect_server: Union[RSConnectServer, SPCSConnectServer], guid: str, status: Optional[str]):
     build_store = ensure_content_build_store(connect_server)
     if guid:
         return [build_store.get_content_item(g) for g in guid]
     else:
         return build_store.get_content_items(status=status)
 
 
-def build_history(connect_server: RSConnectServer, guid: str):
+def build_history(connect_server: Union[RSConnectServer, SPCSConnectServer], guid: str):
     return ensure_content_build_store(connect_server).get_build_history(guid)
 
 
 def build_start(
-    connect_server: RSConnectServer,
+    connect_server: Union[RSConnectServer, SPCSConnectServer],
     parallelism: int,
     aborted: bool = False,
     error: bool = False,
@@ -251,7 +251,9 @@ def build_start(
             build_monitor.shutdown()
 
 
-def _monitor_build(connect_server: RSConnectServer, content_items: list[ContentItemWithBuildState]):
+def _monitor_build(
+    connect_server: Union[RSConnectServer, SPCSConnectServer], content_items: list[ContentItemWithBuildState]
+):
     """
     :return bool: True if the build completed without errors, False otherwise
     """
@@ -296,7 +298,9 @@ def _monitor_build(connect_server: RSConnectServer, content_items: list[ContentI
     return True
 
 
-def _build_content_item(connect_server: RSConnectServer, content: ContentItemWithBuildState, poll_wait: int):
+def _build_content_item(
+    connect_server: Union[RSConnectServer, SPCSConnectServer], content: ContentItemWithBuildState, poll_wait: int
+):
     build_store = ensure_content_build_store(connect_server)
     with RSConnectClient(connect_server) as client:
         # Pending futures will still try to execute when ThreadPoolExecutor.shutdown() is called
@@ -351,7 +355,7 @@ def write_log(line: str):
 
 
 def emit_build_log(
-    connect_server: RSConnectServer,
+    connect_server: Union[RSConnectServer, SPCSConnectServer],
     guid: str,
     format: str,
     task_id: Optional[str] = None,
@@ -369,7 +373,7 @@ def emit_build_log(
         raise RSConnectException("Log file not found for content: %s" % guid)
 
 
-def download_bundle(connect_server: RSConnectServer, guid_with_bundle: ContentGuidWithBundle):
+def download_bundle(connect_server: Union[RSConnectServer, SPCSConnectServer], guid_with_bundle: ContentGuidWithBundle):
     """
     :param guid_with_bundle: models.ContentGuidWithBundle
     """
@@ -387,7 +391,7 @@ def download_bundle(connect_server: RSConnectServer, guid_with_bundle: ContentGu
         return client.download_bundle(guid_with_bundle.guid, guid_with_bundle.bundle_id)
 
 
-def get_content(connect_server: RSConnectServer, guid: str | list[str]):
+def get_content(connect_server: Union[RSConnectServer, SPCSConnectServer], guid: str | list[str]):
     """
     :param guid: a single guid as a string or list of guids.
     :return: a list of content items.
@@ -401,7 +405,7 @@ def get_content(connect_server: RSConnectServer, guid: str | list[str]):
 
 
 def search_content(
-    connect_server: RSConnectServer,
+    connect_server: Union[RSConnectServer, SPCSConnectServer],
     published: bool,
     unpublished: bool,
     content_type: Sequence[str],