turingdb 0.1.0__py3-none-any.whl

turingdb/__init__.py

@@ -0,0 +1,150 @@
+from typing import Optional
+
+
+class TuringDBException(Exception):
+    def __init__(self, message: str):
+        super().__init__(message)
+
+
+class TuringDB:
+    DEFAULT_HEADERS = {
+        "Accept": "application/json",
+        "Content-Type": "application/json",
+    }
+
+    def __init__(
+        self,
+        instance_id: str = "",
+        auth_token: str = "",
+        host: str = "https://engines.turingdb.ai/sdk",
+    ):
+        import httpx
+        import copy
+
+        self.host = host
+
+        self._client = httpx.Client(auth=None, verify=False)
+
+        self._params = {
+            "graph": "default",
+        }
+
+        self._headers = copy.deepcopy(TuringDB.DEFAULT_HEADERS)
+
+        if instance_id != "":
+            self._headers["Turing-Instance-Id"] = instance_id
+
+        if auth_token != "":
+            self._headers["Authorization"] = f"Bearer {auth_token}"
+
+    def list_available_graphs(self) -> list[str]:
+        return self._send_request("list_avail_graphs")["data"]
+
+    def list_loaded_graphs(self) -> list[str]:
+        return self._send_request("list_loaded_graphs")["data"][0][0]
+
+    def load_graph(self, graph_name: str, raise_if_loaded: bool = True):
+        try:
+            return self._send_request("load_graph", params={"graph": graph_name})
+        except TuringDBException as e:
+            if raise_if_loaded or e.__str__() != "GRAPH_ALREADY_EXISTS":
+                raise e
+
+    def create_graph(self, graph_name: str):
+        return self.query(f"create graph {graph_name}")
+
+    def query(self, query: str):
+        json = self._send_request("query", data=query, params=self._params)
+
+        if not isinstance(json, dict):
+            raise TuringDBException("Invalid response from the server")
+
+        return self._parse_chunks(json)
+
+    def set_commit(self, commit: str):
+        self._params["commit"] = commit
+
+    def set_change(self, change: str):
+        self._params["change"] = change
+
+    def checkout(self, change: str = "main", commit: str = "HEAD"):
+        if change == "main":
+            if "change" in self._params:
+                del self._params["change"]
+        else:
+            self.set_change(change)
+
+        if commit == "HEAD":
+            if "commit" in self._params:
+                del self._params["commit"]
+        else:
+            self.set_commit(commit)
+
+    def set_graph(self, graph_name: str):
+        self._params["graph"] = graph_name
+
+    def _send_request(
+        self,
+        path: str,
+        data: Optional[dict | str] = None,
+        params: Optional[dict] = None,
+    ):
+        import orjson
+
+        if data is None:
+            data = ""
+
+        url = f"{self.host}/{path}"
+
+        if isinstance(data, dict):
+            response = self._client.post(
+                url, json=data, params=params, headers=self._headers
+            )
+        else:
+            response = self._client.post(
+                url, content=data, params=params, headers=self._headers
+            )
+        response.raise_for_status()
+
+        json = orjson.loads(response.text)
+
+        if isinstance(json, dict):
+            err = json.get("error")
+            if err is not None:
+                raise TuringDBException(err)
+
+        return json
+
+    def _parse_chunks(self, json: dict):
+        import pandas as pd
+        import numpy as np
+
+        data_header = json["header"]
+        column_names: list[str] = data_header["column_names"]
+        column_types: list[str] = data_header["column_types"]
+        dtypes: list[np.dtype] = []
+        columns: list[list] = []
+
+        for chunk in json["data"]:
+            for i, col in enumerate(chunk):
+                if i >= len(columns):
+                    match column_types[i]:
+                        case "String":
+                            dtypes.append(np.dtype(np.dtypes.StringDType))
+                        case "Int64":
+                            dtypes.append(np.dtype(np.dtypes.Int64DType))
+                        case "UInt64":
+                            dtypes.append(np.dtype(np.dtypes.UInt64DType))
+                        case "Double":
+                            dtypes.append(np.dtype(np.dtypes.Float64DType))
+                        case "Boolean":
+                            dtypes.append(np.dtype(np.dtypes.BoolDType))
+
+                    columns.append([])
+
+                columns[i] = columns[i] + col
+
+        arrays = [np.array(columns[i], dtype=dtypes[i]) for i in range(len(columns))]
+
+        return pd.DataFrame(dict(zip(column_names, arrays)), index=None)
+

turingdb-0.1.0.dist-info/METADATA

@@ -0,0 +1,15 @@
+Metadata-Version: 2.4
+Name: turingdb
+Version: 0.1.0
+Summary: TuringDB python sdk
+Project-URL: Homepage, https://github.com/turing-db/turingdb-sdk-python
+Project-URL: Repository, https://github.com/turing-db/turingdb-sdk-python
+License-File: LICENSE.txt
+Requires-Python: >=3.9
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: orjson>=3.11.1
+Requires-Dist: pandas>=2.3.1
+Description-Content-Type: text/markdown
+
+# turingdb-sdk-python
+Python SDK of the TuringDB graph database engine

turingdb-0.1.0.dist-info/RECORD

@@ -0,0 +1,5 @@
+turingdb/__init__.py,sha256=jDI3-GTmSmWvEhFlV4y7l1UTSt9L-6952D5WOzW55cY,4533
+turingdb-0.1.0.dist-info/METADATA,sha256=G_24yqlVrl6DoJ4sZQ_gBSj8jKXgfhrmRN7-uskF42U,476
+turingdb-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+turingdb-0.1.0.dist-info/licenses/LICENSE.txt,sha256=PgwD9hjeoyWO93yZQnxCyt0s-PlhXWLgqdCPpB91GhY,1078
+turingdb-0.1.0.dist-info/RECORD,,

turingdb-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.27.0
+Root-Is-Purelib: true
+Tag: py3-none-any

turingdb-0.1.0.dist-info/licenses/LICENSE.txt

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Turing Biosystems Ltd
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.