rowan-python 2.1.9__py3-none-any.whl → 2.1.10__py3-none-any.whl

rowan/folder.py

@@ -58,6 +58,7 @@ class Folder(BaseModel):
 
         return self
 
+
     def update(
         self,
         name: str | None = None,
@@ -110,6 +111,15 @@ class Folder(BaseModel):
             response = client.delete(f"/folder/{self.uuid}")
             response.raise_for_status()
 
+    def print_folder_tree(self, max_depth: int = 10, show_uuids: bool = False) -> None:
+        """
+        Retrieves a folder tree from the API.
+
+        :param max_depth: The maximum depth of the folder tree.
+        :param show_uuids: Whether to show the UUIDs of the folders.
+        :raises HTTPError: If the API request fails.
+        """
+        print_folder_tree(self.uuid, max_depth, show_uuids)
 
 def retrieve_folder(uuid: str) -> Folder:
     """
@@ -125,19 +135,6 @@ def retrieve_folder(uuid: str) -> Folder:
         return Folder(**response.json())
 
 
-def home_folder() -> Folder:
-    """
-    Retrieves the home folder from the API.
-
-    :return: A Folder object representing the home folder.
-    :raises HTTPError: If the API request fails.
-    """
-    with api_client() as client:
-        response = client.get("/user/me/root_folders")
-        response.raise_for_status()
-        return Folder(**response.json()["user_root"])
-
-
 def list_folders(
     parent_uuid: str | None = None,
     name_contains: str | None = None,
@@ -210,3 +207,22 @@ def create_folder(
         response.raise_for_status()
         folder_data = response.json()
     return Folder(**folder_data)
+
+def print_folder_tree(uuid: str, max_depth: int = 10, show_uuids: bool = False) -> None:
+    """
+    Retrieves a folder tree from the API.
+
+    :param uuid: The UUID of the root of the folder tree.
+    :param max_depth: The maximum depth of the folder tree.
+    :param show_uuids: Whether to show the UUIDs of the folders.
+    :raises HTTPError: If the API request fails.
+    """
+    params: dict[str, Any] = {
+        "max_depth": max_depth,
+        "show_uuids": show_uuids,
+    }
+    with api_client() as client:
+        response = client.get(f"/folder/{uuid}/folder_tree", params=params)
+        response.raise_for_status()
+        folder_data = response.json()
+    print(folder_data)

rowan/workflow.py

@@ -408,6 +408,33 @@ def retrieve_workflow(uuid: str) -> Workflow:
         return Workflow(**response.json())
 
 
+def retrieve_workflows(uuids: list[str]) -> list[Workflow]:
+    """
+    Retrieves a list of workflows from the API.
+
+    :param uuids: The UUIDs of the workflows to retrieve.
+    :return: A list of Workflow objects representing the retrieved workflows.
+    :raises HTTPError: If the API request fails.
+    """
+    with api_client() as client:
+        response = client.post("/workflow/batch_retrieve", json={"uuids": uuids})
+        response.raise_for_status()
+        return [Workflow(**workflow_data) for workflow_data in response.json()]
+
+def batch_poll_status(uuids: list[str]) -> list[Workflow]:
+    """
+    Polls the status of a list of workflows.
+
+    :param uuids: The UUIDs of the workflows to poll.
+    :return: A dictionary of statuses and the count of workflows in that status.
+    :raises HTTPError: If the API request fails.
+    """
+    with api_client() as client:
+        response = client.post("/workflow/batch_status", json={"uuids": uuids})
+        response.raise_for_status()
+        return response.json()
+
+
 def retrieve_calculation_molecules(
     uuid: str, return_frequencies: bool = False
 ) -> list[dict[str, Any]]:

{rowan_python-2.1.9.dist-info → rowan_python-2.1.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rowan-python
-Version: 2.1.9
+Version: 2.1.10
 Summary: Rowan Python Library
 Project-URL: Homepage, https://github.com/rowansci/rowan-client
 Project-URL: Bug Tracker, https://github.com/rowansci/rowan-client/issues
@@ -11,7 +11,7 @@ Requires-Dist: httpx
 Requires-Dist: nest-asyncio
 Requires-Dist: rdkit
 Requires-Dist: setuptools
-Requires-Dist: stjames>=0.0.125
+Requires-Dist: stjames>=0.0.128
 Description-Content-Type: text/markdown
 
 # Rowan Python Library

{rowan_python-2.1.9.dist-info → rowan_python-2.1.10.dist-info}/RECORD

@@ -1,15 +1,15 @@
 rowan/__init__.py,sha256=2rz6dW0l9DzawiFi6S0WSv8XlZq1CoTBKJrsJ1uesvk,171
 rowan/constants.py,sha256=emCH4m9OL2Hm5E-6mJGM_FgzrK_JrZT-FiKJ6pMNQ4Y,84
-rowan/folder.py,sha256=n9WkjHMweQLtVcWUvCttOrmezvUdbFxam_eDEMzLF_A,6791
+rowan/folder.py,sha256=MF3SU7uG6Hl2SJLFxbPmbhosS-pPEHwbTyummaaRdzM,7509
 rowan/project.py,sha256=ALPPkMa_cg7w5OkXno1cs6acCofw8AOUYRSeWgr3L0o,3774
 rowan/protein.py,sha256=mFSVCr-08bSikXBUtJtSWzjKcVAuBmTeRckn7JUHYSE,8810
 rowan/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rowan/user.py,sha256=Dl--NPUPATKCs2VmILsW8HnLiunG0Lxr0n6mKuHm21U,3891
 rowan/utils.py,sha256=64II-cPOe_SFJK302Bm8hP62d_3_CgnTVYCbn3zKT7U,3334
-rowan/workflow.py,sha256=ovhjVUaoI0K-pepk3u_xGnymBOq9cEEUEoyzsFr36Sc,60331
+rowan/workflow.py,sha256=l21KygojLRsM6JI2b0H-hcbOnaFsDZGhbqmLy3sexjA,61360
 rowan/rowan_rdkit/__init__.py,sha256=EATX2VRzywzKxqkpCUMTf7RNQLkWsfi5VcCNDW6EIiw,503
 rowan/rowan_rdkit/chem_utils.py,sha256=sKCzul2e0ldVYTBImhTwso7ddNgPKmvS-OmvCEjVJH0,34788
-rowan_python-2.1.9.dist-info/METADATA,sha256=Teu23XCt-9sjkXXVcsjC5PID3Pgp5bR-conmLvaPiFE,1600
-rowan_python-2.1.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-rowan_python-2.1.9.dist-info/licenses/LICENSE,sha256=i05z7xEhyrg6f8j0lR3XYjShnF-MJGFQ-DnpsZ8yiVI,1084
-rowan_python-2.1.9.dist-info/RECORD,,
+rowan_python-2.1.10.dist-info/METADATA,sha256=8AzU2Im1apsPTrfFyZ3OYfPYaI9V_vwmRjKOfOy6xiI,1601
+rowan_python-2.1.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+rowan_python-2.1.10.dist-info/licenses/LICENSE,sha256=i05z7xEhyrg6f8j0lR3XYjShnF-MJGFQ-DnpsZ8yiVI,1084
+rowan_python-2.1.10.dist-info/RECORD,,