[LABIMP-8186] Added support to import users to a project from another project

Author: ximihoque

driver.py

@@ -39,21 +39,27 @@ def get_connection(client: "LabellerrClient", connection_id: str):
     """Metaclass that combines ABC functionality with factory pattern"""
 
     def __call__(cls, client, connection_id, **kwargs):
-        # Only intercept calls to the base LabellerrConnection class
+        # Fetch connection data if not already provided
+        if "connection_data" not in kwargs:
+            connection_data = cls.get_connection(client, connection_id)
+            if connection_data is None:
+                raise InvalidConnectionError(f"Connection not found: {connection_id}")
+            kwargs["connection_data"] = connection_data
+
+        # Only intercept calls to the base LabellerrConnection class for factory behavior
         if cls.__name__ != "LabellerrConnection":
-            # For subclasses, use normal instantiation
+            # For subclasses, use normal instantiation with connection_data
             instance = cls.__new__(cls)
             if isinstance(instance, cls):
                 instance.__init__(client, connection_id, **kwargs)
             return instance
-        connection_data = cls.get_connection(client, connection_id)
-        if connection_data is None:
-            raise InvalidConnectionError(f"Connection not found: {connection_id}")
+
+        # Factory behavior for base class
+        connection_data = kwargs["connection_data"]
         connector = connection_data.get("connector")
         connection_class = cls._registry.get(connector)
         if connection_class is None:
             raise InvalidConnectionError(f"Unknown connector type: {connector}")
-        kwargs["connection_data"] = connection_data
         return connection_class(client, connection_id, **kwargs)
 
 

labellerr/core/connectors/connections.py

@@ -66,7 +66,7 @@ def test_connection(
     @staticmethod
     def create_connection(
         client: "LabellerrClient", params: GCSConnectionParams
-    ) -> str:
+    ) -> "LabellerrConnection":
         """
         Sets up GCP connector for dataset creation (quick connection).
 

labellerr/core/connectors/gcs_connection.py

@@ -65,7 +65,7 @@ def test_connection(
     @staticmethod
     def create_connection(
         client: "LabellerrClient", params: AWSConnectionParams
-    ) -> dict:
+    ) -> "LabellerrConnection":
         """
         Creates an AWS S3 connection.
 

labellerr/core/connectors/s3_connection.py

@@ -102,6 +102,65 @@ def data_type(self):
     def attached_datasets(self):
         return self.__project_data.get("attached_datasets")
 
+    def status(self) -> Dict:
+        """
+        Poll project status until completion or timeout.
+
+        Returns:
+            Final project data with status information
+
+        Examples:
+            # Get current project status
+            final_status = project.status()
+        """
+        from .utils import poll
+
+        def get_project_status():
+            unique_id = str(uuid.uuid4())
+            url = (
+                f"{constants.BASE_URL}/projects/project/{self.project_id}?client_id={self.client.client_id}"
+                f"&uuid={unique_id}"
+            )
+
+            response = self.client.make_request(
+                "GET",
+                url,
+                extra_headers={"content-type": "application/json"},
+                request_id=unique_id,
+            )
+            project_data = response.get("response", {})
+            if project_data:
+                self.__project_data = project_data
+            return project_data
+
+        def is_completed(project_data):
+            status_code = project_data.get("status_code", 500)
+            # Consider project complete when status_code is 300 (success) or >= 400 (error/failed)
+            return status_code == 300 or status_code >= 400
+
+        def on_success(project_data):
+            status_code = project_data.get("status_code", 500)
+            if status_code == 300:
+                logging.info(
+                    "Project %s processing completed successfully!", self.project_id
+                )
+            else:
+                logging.warning(
+                    "Project %s processing finished with status code: %s",
+                    self.project_id,
+                    status_code,
+                )
+            return project_data
+
+        return poll(
+            function=get_project_status,
+            condition=is_completed,
+            interval=2.0,
+            timeout=None,
+            max_retries=None,
+            on_success=on_success,
+        )
+
     def detach_dataset_from_project(self, dataset_id=None, dataset_ids=None):
         """
         Detaches one or more datasets from an existing project.
@@ -568,3 +627,19 @@ def __fetch_exports_download_url(self, project_id, uuid, export_id, client_id):
             return response.get("response")
         except Exception as e:
             raise LabellerrError(f"Failed to download export: {str(e)}")
+
+    def import_users(self, from_project: "LabellerrProject"):
+        """
+        Imports users from a source project to the current project.
+
+        :param from_project: The source project to import users from
+        :return: Dictionary containing import results
+        :raises LabellerrError: If the import fails
+        """
+        # Validate parameters using Pydantic
+        unique_id = str(uuid.uuid4())
+        url = f"{constants.BASE_URL}/users/projects/import_users?selected_project_id={self.project_id}&project_id={from_project.project_id}&client_id={self.client.client_id}&uuid={unique_id}"
+        response = self.client.make_request(
+            "POST", url, extra_headers={"Content-Type": "application/json"}
+        )
+        return response.get("response")