style: applies formatting (#172)

Author: tdstein

examples/connect/dash/app.py

@@ -37,8 +37,12 @@ def update_page(_):
     Dash example application that shows user information and
     the first few rows from a table hosted in Databricks.
     """
-    session_token = flask.request.headers.get("Posit-Connect-User-Session-Token")
-    credentials_provider = viewer_credentials_provider(user_session_token=session_token)
+    session_token = flask.request.headers.get(
+        "Posit-Connect-User-Session-Token"
+    )
+    credentials_provider = viewer_credentials_provider(
+        user_session_token=session_token
+    )
 
     def get_greeting():
         cfg = Config(

examples/connect/flask/app.py

@@ -28,7 +28,9 @@ def get_fares():
     global rows
 
     session_token = request.headers.get("Posit-Connect-User-Session-Token")
-    credentials_provider = viewer_credentials_provider(user_session_token=session_token)
+    credentials_provider = viewer_credentials_provider(
+        user_session_token=session_token
+    )
 
     if rows is None:
         query = "SELECT * FROM samples.nyctaxi.trips LIMIT 10;"

examples/connect/shiny-python/app.py

@@ -24,8 +24,12 @@ def server(input: Inputs, output: Outputs, session: Session):
     Shiny for Python example application that shows user information and
     the first few rows from a table hosted in Databricks.
     """
-    session_token = session.http_conn.headers.get("Posit-Connect-User-Session-Token")
-    credentials_provider = viewer_credentials_provider(user_session_token=session_token)
+    session_token = session.http_conn.headers.get(
+        "Posit-Connect-User-Session-Token"
+    )
+    credentials_provider = viewer_credentials_provider(
+        user_session_token=session_token
+    )
 
     @render.data_frame
     def result():
@@ -40,7 +44,9 @@ def result():
             with connection.cursor() as cursor:
                 cursor.execute(query)
                 rows = cursor.fetchall()
-                df = pd.DataFrame(rows, columns=[col[0] for col in cursor.description])
+                df = pd.DataFrame(
+                    rows, columns=[col[0] for col in cursor.description]
+                )
                 return df
 
     @render.text

examples/connect/streamlit/app.py

@@ -16,11 +16,17 @@
 DATABRICKS_HOST_URL = f"https://{DATABRICKS_HOST}"
 SQL_HTTP_PATH = os.getenv("DATABRICKS_PATH")
 
-session_token = _get_websocket_headers().get("Posit-Connect-User-Session-Token")
+session_token = _get_websocket_headers().get(
+    "Posit-Connect-User-Session-Token"
+)
 
-credentials_provider = viewer_credentials_provider(user_session_token=session_token)
+credentials_provider = viewer_credentials_provider(
+    user_session_token=session_token
+)
 
-cfg = Config(host=DATABRICKS_HOST_URL, credentials_provider=credentials_provider)
+cfg = Config(
+    host=DATABRICKS_HOST_URL, credentials_provider=credentials_provider
+)
 databricks_user = CurrentUserAPI(ApiClient(cfg)).me()
 st.write(f"Hello, {databricks_user.display_name}!")
 

pyproject.toml

@@ -32,6 +32,9 @@ addopts = ["--import-mode=importlib"]
 [tool.setuptools_scm]
 version_file = "src/posit/_version.py"
 
+[tool.ruff]
+line-length = 79
+
 [tool.ruff.format]
 docstring-code-format = true
 docstring-code-line-length = "dynamic"

src/posit/connect/bundles.py

@@ -88,7 +88,9 @@ def size(self) -> int | None:
 
     @property
     def metadata(self) -> BundleMetadata:
-        return BundleMetadata(self.config, self.session, **self.get("metadata", {}))
+        return BundleMetadata(
+            self.config, self.session, **self.get("metadata", {})
+        )
 
     # CRUD Methods
 
@@ -99,7 +101,9 @@ def delete(self) -> None:
 
 
 class Bundles(Resources):
-    def __init__(self, config: Config, session: Session, content_guid: str) -> None:
+    def __init__(
+        self, config: Config, session: Session, content_guid: str
+    ) -> None:
         super().__init__(config, session)
         self.content_guid = content_guid
 

src/posit/connect/cursors.py

@@ -16,7 +16,9 @@ class CursorPage:
 
 
 class CursorPaginator:
-    def __init__(self, session: requests.Session, url: str, params: dict = {}) -> None:
+    def __init__(
+        self, session: requests.Session, url: str, params: dict = {}
+    ) -> None:
         self.session = session
         self.url = url
         self.params = params

src/posit/connect/errors.py

@@ -1,6 +1,10 @@
 class ClientError(Exception):
     def __init__(
-        self, error_code: int, error_message: str, http_status: int, http_message: str
+        self,
+        error_code: int,
+        error_message: str,
+        http_status: int,
+        http_message: str,
     ):
         self.error_code = error_code
         self.error_message = error_message

src/posit/connect/external/databricks.py

@@ -36,9 +36,9 @@ def auth_type(self) -> str:
 
     def __call__(self, *args, **kwargs) -> HeaderFactory:
         def inner() -> Dict[str, str]:
-            access_token = self.posit_oauth.get_credentials(self.user_session_token)[
-                "access_token"
-            ]
+            access_token = self.posit_oauth.get_credentials(
+                self.user_session_token
+            )["access_token"]
             return {"Authorization": f"Bearer {access_token}"}
 
         return inner
@@ -68,7 +68,9 @@ def viewer_credentials_provider(
             "The user-session-token is required for viewer authentication."
         )
 
-    return PositOAuthIntegrationCredentialsProvider(client.oauth, user_session_token)
+    return PositOAuthIntegrationCredentialsProvider(
+        client.oauth, user_session_token
+    )
 
 
 def service_account_credentials_provider(client: Optional[Client] = None):

src/posit/connect/hooks.py

@@ -12,7 +12,9 @@ def handle_errors(response: Response, *args, **kwargs) -> Response:
             message = data["error"]
             http_status = response.status_code
             http_status_message = responses[http_status]
-            raise ClientError(error_code, message, http_status, http_status_message)
+            raise ClientError(
+                error_code, message, http_status, http_status_message
+            )
         except JSONDecodeError:
             # No JSON error message from Connect, so just raise
             response.raise_for_status()