feat: export/import users across workspaces (#361)

* feat: export users across workspaces

* Import users with roles

* Small fixes

* Cleanup

* Fix little things

* refactor: Use enum for user file format detection

- Replace string literals with UserFileFormat enum for better type safety
- Add TODO comment about potential --continue-on-error feature for user imports
- Addresses PR review feedback

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>

---------

Co-authored-by: Claude <[email protected]>

Author: betodealmeida

src/preset_cli/api/clients/preset.py

@@ -31,7 +31,6 @@ class Role(int, Enum):
 
 
 class PresetClient:  # pylint: disable=too-few-public-methods
-
     """
     A client for the Preset API.
     """
@@ -158,7 +157,7 @@ def export_users(self, workspace_url: URL) -> Iterator[UserType]:
                 {
                     "id": 0,
                     "username": payload["user"]["username"],
-                    "role": [],  # TODO (betodealmeida)
+                    "role": [payload["workspace_role"]["name"].lower()],
                     "first_name": payload["user"]["first_name"],
                     "last_name": payload["user"]["last_name"],
                     "email": payload["user"]["email"].lower(),

src/preset_cli/api/clients/superset.py

@@ -231,15 +231,20 @@ class OwnershipType(TypedDict):
 
 
 class SupersetClient:  # pylint: disable=too-many-public-methods
-
     """
     A client for running queries against Superset.
     """
 
-    def __init__(self, baseurl: Union[str, URL], auth: Auth):
+    def __init__(
+        self,
+        baseurl: Union[str, URL],
+        auth: Auth,
+        preset_baseurl: Union[str, URL] = "https://api.app.preset.io/",
+    ):
         # convert to URL if necessary
         self.baseurl = URL(baseurl)
         self.auth = auth
+        self.preset_baseurl = URL(preset_baseurl)
 
         self.session = auth.session
         self.session.headers.update(auth.get_headers())
@@ -360,9 +365,11 @@ def get_data(  # pylint: disable=too-many-locals, too-many-arguments
 
         # and order bys
         processed_orderbys = [
-            (orderby, not order_desc)
-            if orderby in metric_names
-            else (convert_to_adhoc_metric(orderby), not order_desc)
+            (
+                (orderby, not order_desc)
+                if orderby in metric_names
+                else (convert_to_adhoc_metric(orderby), not order_desc)
+            )
             for orderby in (order_by or [])
         ]
 
@@ -796,8 +803,7 @@ def _export_users_preset(self) -> Iterator[UserType]:
         """
         Return all users from a Preset workspace.
         """
-        # TODO (betodealmeida): remove hardcoded Manager URL
-        client = PresetClient("https://api.app.preset.io/", self.auth)
+        client = PresetClient(self.preset_baseurl, self.auth)
         return client.export_users(self.baseurl)
 
     def _export_users_superset(self) -> Iterator[UserType]: