Merge pull request #791 from phenobarbital/scylla-connector

new scylladb connector
phenobarbital · Sep 25, 2023 · 78e01c7 · 78e01c7
2 parents 2565c99 + 52a69c4
commit 78e01c7
Show file tree

Hide file tree

Showing 8 changed files with 1,115 additions and 10 deletions.
diff --git a/asyncdb/drivers/cassandra.py b/asyncdb/drivers/cassandra.py
@@ -10,7 +10,6 @@
 import asyncio
 from typing import (
     Any,
-    List,
     Union
 )
 from ssl import PROTOCOL_TLSv1
@@ -55,7 +54,10 @@ def pandas_factory(colnames, rows):
 
 
 def record_factory(colnames, rows):
-    return Recordset(result=[dict(zip(colnames, values)) for values in rows], columns=colnames)
+    return Recordset(
+        result=[dict(zip(colnames, values)) for values in rows],
+        columns=colnames
+    )
 
 
 class cassandra(InitDriver):

diff --git a/asyncdb/drivers/influx.py b/asyncdb/drivers/influx.py
@@ -447,6 +447,7 @@ async def write(self, data: Union[list, dict], bucket: str, **kwargs):
             ) from err
 
     save = write
+    copy = write
 
     async def query(self, sentence: str, frmt: str = 'native', params: dict = None, **kwargs):
         self._result = None