Linting, sorting

Author: 1pxone

src/rushdb/__init__.py

@@ -3,9 +3,9 @@
 Exposes the RushDB class.
 """
 
-from .client import RushDB
 from .api.query import QueryAPI
 from .api.relationships import RelationsAPI
+from .client import RushDB
 from .common import RushDBError
 from .models.property import Property
 from .models.record import Record

src/rushdb/api/__init__.py

@@ -1,15 +1,15 @@
-from .records import RecordsAPI
-from .properties import PropertiesAPI
 from .labels import LabelsAPI
-from .transactions import TransactionsAPI
+from .properties import PropertiesAPI
 from .query import QueryAPI
+from .records import RecordsAPI
 from .relationships import RelationsAPI
+from .transactions import TransactionsAPI
 
 __all__ = [
-	"RecordsAPI",
-	"PropertiesAPI",
-	"LabelsAPI",
-	"TransactionsAPI",
-	"QueryAPI",
-	"RelationsAPI",
+    "RecordsAPI",
+    "PropertiesAPI",
+    "LabelsAPI",
+    "TransactionsAPI",
+    "QueryAPI",
+    "RelationsAPI",
 ]

src/rushdb/api/query.py

@@ -1,7 +1,7 @@
 from typing import Any, Dict, Optional
 
-from .base import BaseAPI
 from ..models.transaction import Transaction
+from .base import BaseAPI
 
 
 class QueryAPI(BaseAPI):
@@ -38,6 +38,4 @@ def raw(
             Dict[str, Any]: API response including raw driver result in 'data'.
         """
         headers = Transaction._build_transaction_header(transaction)
-        return self.client._make_request(
-            "POST", "/query/raw", body, headers
-        )
+        return self.client._make_request("POST", "/query/raw", body, headers)

src/rushdb/api/records.py

@@ -560,7 +560,9 @@ def import_csv(
                 "newline",
             }
             payload["parseConfig"] = {
-                k: v for k, v in parse_config.items() if k in allowed_keys and v is not None
+                k: v
+                for k, v in parse_config.items()
+                if k in allowed_keys and v is not None
             }
 
         return self.client._make_request(

src/rushdb/api/relationships.py

@@ -177,7 +177,9 @@ def create_many(
             payload["direction"] = direction
         if many_to_many is not None:
             payload["manyToMany"] = many_to_many
-        return self.client._make_request("POST", "/relationships/create-many", payload, headers)
+        return self.client._make_request(
+            "POST", "/relationships/create-many", payload, headers
+        )
 
     def delete_many(
         self,
@@ -201,4 +203,6 @@ def delete_many(
             payload["direction"] = direction
         if many_to_many is not None:
             payload["manyToMany"] = many_to_many
-        return self.client._make_request("POST", "/relationships/delete-many", payload, headers)
+        return self.client._make_request(
+            "POST", "/relationships/delete-many", payload, headers
+        )

src/rushdb/client.py

@@ -13,10 +13,10 @@
 
 from .api.labels import LabelsAPI
 from .api.properties import PropertiesAPI
-from .api.records import RecordsAPI
-from .api.transactions import TransactionsAPI
 from .api.query import QueryAPI
+from .api.records import RecordsAPI
 from .api.relationships import RelationsAPI
+from .api.transactions import TransactionsAPI
 from .common import RushDBError
 from .utils.token_prefix import extract_mixed_properties_from_token