cloudquery-plugin-sdk 0.1.23__tar.gz → 0.1.24__tar.gz

{cloudquery_plugin_sdk-0.1.23 → cloudquery_plugin_sdk-0.1.24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cloudquery-plugin-sdk
-Version: 0.1.23
+Version: 0.1.24
 Summary: CloudQuery Plugin SDK for Python
 Home-page: https://github.com/cloudquery/plugin-sdk-python
 Author: CloudQuery LTD

{cloudquery_plugin_sdk-0.1.23 → cloudquery_plugin_sdk-0.1.24}/cloudquery/sdk/internal/memdb/memdb.py

@@ -1,14 +1,22 @@
+import json
+
 from cloudquery.sdk import plugin
 from cloudquery.sdk import message
 from cloudquery.sdk import schema
 from typing import List, Generator, Dict
 import pyarrow as pa
 from cloudquery.sdk.types import JSONType
+from dataclasses import dataclass, field
 
 NAME = "memdb"
 VERSION = "development"
 
 
+@dataclass
+class Spec:
+    abc: str = field(default="abc")
+
+
 class MemDB(plugin.Plugin):
     def __init__(self) -> None:
         super().__init__(
@@ -73,6 +81,12 @@ class MemDB(plugin.Plugin):
             ),
         }
 
+    def init(self, spec, no_connection: bool = False):
+        if no_connection:
+            return
+        self._spec_json = json.loads(spec)
+        self._spec = Spec(**self._spec_json)
+
     def get_tables(self, options: plugin.TableOptions = None) -> List[plugin.Table]:
         tables = list(self._tables.values())
         return schema.filter_dfs(tables, options.tables, options.skip_tables)

{cloudquery_plugin_sdk-0.1.23 → cloudquery_plugin_sdk-0.1.24}/cloudquery/sdk/internal/servers/plugin_v3/plugin.py

@@ -32,7 +32,9 @@ class PluginServicer(plugin_pb2_grpc.PluginServicer):
         return plugin_pb2.GetSpecSchema.Response(json_schema=self._plugin.json_schema())
 
     def Init(self, request: plugin_pb2.Init.Request, context):
-        self._plugin.init(request.spec, no_connection=request.no_connection)
+        self._plugin.init(
+            sanitize_spec(request.spec), no_connection=request.no_connection
+        )
         return plugin_pb2.Init.Response()
 
     def GetTables(self, request: plugin_pb2.GetTables.Request, context):
@@ -113,3 +115,7 @@ class PluginServicer(plugin_pb2_grpc.PluginServicer):
     def Close(self, request, context):
         self._plugin.close()
         return plugin_pb2.Close.Response()
+
+
+def sanitize_spec(spec=None):
+    return b"{}" if spec is None or spec == b"" or spec == b"null" else spec

{cloudquery_plugin_sdk-0.1.23 → cloudquery_plugin_sdk-0.1.24}/cloudquery_plugin_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cloudquery-plugin-sdk
-Version: 0.1.23
+Version: 0.1.24
 Summary: CloudQuery Plugin SDK for Python
 Home-page: https://github.com/cloudquery/plugin-sdk-python
 Author: CloudQuery LTD

{cloudquery_plugin_sdk-0.1.23 → cloudquery_plugin_sdk-0.1.24}/setup.py

@@ -51,7 +51,7 @@ packages = [
 ]
 setuptools.setup(
     name=name,
-    version="0.1.23",
+    version="0.1.24",
     description=description,
     long_description=long_description,
     author="CloudQuery LTD",