add suite class

pyenzyme/__init__.py

@@ -7,6 +7,7 @@
 
 from .model import *  # noqa: F403
 from .sbml import to_sbml, read_sbml  # noqa: F401
+from .suite import EnzymeMLSuite  # noqa: F401
 from .tabular import to_pandas, read_csv, read_excel, from_dataframe  # noqa: F401
 
 __all__ = [

pyenzyme/suite.py

@@ -0,0 +1,72 @@
+import json
+
+import httpx
+import rich
+
+import pyenzyme as pe
+
+
+class EnzymeMLSuite:
+    """
+    A suite for interacting with the EnzymeML service.
+
+    Attributes:
+        client (httpx.Client): The HTTP client for making requests to the EnzymeML service.
+    """
+
+    def __init__(self):
+        """
+        Initializes the EnzymeMLSuite with a base URL for the EnzymeML service.
+        """
+        self.client = httpx.Client(base_url="http://localhost:13452")
+
+    def get_current(self) -> pe.EnzymeMLDocument:
+        """
+        Retrieves the current EnzymeML document from the service.
+
+        Returns:
+            EnzymeMLDocument: The current EnzymeML document.
+
+        Raises:
+            httpx.HTTPStatusError: If the request to the service fails.
+        """
+
+        try:
+            response = self.client.get("/docs/:current")
+        except httpx.ConnectError:
+            raise ConnectionError(
+                "Could not connect to the EnzymeML suite. Make sure it is running."
+            )
+
+        response.raise_for_status()
+
+        content = response.json()["data"]["content"]
+
+        if not isinstance(content["references"], list):
+            content["references"] = []
+
+        return pe.EnzymeMLDocument.model_validate(content)
+
+    def update_current(self, doc: pe.EnzymeMLDocument):
+        """
+        Updates the current EnzymeML document on the service.
+
+        Args:
+            doc (EnzymeMLDocument): The EnzymeML document to update.
+
+        Raises:
+            httpx.HTTPStatusError: If the request to the service fails.
+        """
+        try:
+            response = self.client.put(
+                "/docs/:current",
+                json=json.loads(doc.model_dump_json()),
+            )
+        except httpx.ConnectError:
+            raise ConnectionError(
+                "Could not connect to the EnzymeML suite. Make sure it is running."
+            )
+
+        response.raise_for_status()
+
+        rich.print("[bold green]Document updated successfully![/]")

pyproject.toml

@@ -18,6 +18,7 @@ loguru = "^0.7.2"
 rdflib = "7.0.0"
 sympy = "^1.12.1"
 pymetadata = "^0.4.4"
+httpx = "^0.27.2"
 
 [tool.poetry.group.neo4j.dependencies]
 neo4j = "^5.21.0"