turingdb 0.2.2__py3-none-any.whl → 0.2.3__py3-none-any.whl

turingdb/__init__.py

@@ -1,150 +1,3 @@
-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)
+from .turingdb import TuringDB, TuringDBException
 
+__all__ = ["TuringDB", "TuringDBException"]

turingdb/turingdb.py

@@ -0,0 +1,151 @@
+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: int | str):
+        if isinstance(change, int):
+            change = f"{change:x}"
+        self._params["change"] = change
+
+    def checkout(self, change: int | 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.2.2.dist-info → turingdb-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: turingdb
-Version: 0.2.2
+Version: 0.2.3
 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
@@ -8,6 +8,7 @@ License-File: LICENSE.txt
 Requires-Python: >=3.10
 Requires-Dist: httpx>=0.28.1
 Requires-Dist: orjson>=3.11.1
+Requires-Dist: pandas-stubs>=2.3.0.250703
 Requires-Dist: pandas>=2.3.1
 Description-Content-Type: text/markdown
 

turingdb-0.2.3.dist-info/RECORD

@@ -0,0 +1,8 @@
+turingdb/__init__.py,sha256=zOhbnDfi_6Zf2aLn7mZho-xHIydeLd8-1TIT2_046Kg,95
+turingdb/__init__.pyi,sha256=O6ZMBGuBgXInR2anKZI3ycMZECIuhlsTUanZiHjZEqs,790
+turingdb/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+turingdb/turingdb.py,sha256=zT1ms_avuTmTA7xg8DfOQbGnGQ4p0awuQXcu43ItPhc,4615
+turingdb-0.2.3.dist-info/METADATA,sha256=vtyv4ImbtOrYk1GuQnVi-46D3s2Nng8WP0l7cm9hRII,519
+turingdb-0.2.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+turingdb-0.2.3.dist-info/licenses/LICENSE.txt,sha256=PgwD9hjeoyWO93yZQnxCyt0s-PlhXWLgqdCPpB91GhY,1078
+turingdb-0.2.3.dist-info/RECORD,,

turingdb-0.2.2.dist-info/RECORD

@@ -1,6 +0,0 @@
-turingdb/__init__.py,sha256=jDI3-GTmSmWvEhFlV4y7l1UTSt9L-6952D5WOzW55cY,4533
-turingdb/__init__.pyi,sha256=O6ZMBGuBgXInR2anKZI3ycMZECIuhlsTUanZiHjZEqs,790
-turingdb-0.2.2.dist-info/METADATA,sha256=wMW-ulQiFRq9wOCMNbMErxdkZfzjdGdLJwrCPfO-QVw,477
-turingdb-0.2.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-turingdb-0.2.2.dist-info/licenses/LICENSE.txt,sha256=PgwD9hjeoyWO93yZQnxCyt0s-PlhXWLgqdCPpB91GhY,1078
-turingdb-0.2.2.dist-info/RECORD,,