From f241e2ba27cdae3b99d7d15e303021ee3cce7cae Mon Sep 17 00:00:00 2001 From: Herman Schaaf Date: Wed, 2 Aug 2023 11:13:02 +0100 Subject: [PATCH] Fix docs signature --- cloudquery/sdk/serve/plugin.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/cloudquery/sdk/serve/plugin.py b/cloudquery/sdk/serve/plugin.py index 7842425..d412a74 100644 --- a/cloudquery/sdk/serve/plugin.py +++ b/cloudquery/sdk/serve/plugin.py @@ -1,16 +1,16 @@ import argparse -import structlog from concurrent import futures import grpc +import structlog import sys - from cloudquery.discovery_v1 import discovery_pb2_grpc from cloudquery.plugin_v3 import plugin_pb2_grpc + from cloudquery.sdk.docs.generator import Generator from cloudquery.sdk.internal.servers.discovery_v1.discovery import DiscoveryServicer from cloudquery.sdk.internal.servers.plugin_v3 import PluginServicer -from cloudquery.sdk.plugin.plugin import Plugin +from cloudquery.sdk.plugin.plugin import Plugin, TableOptions DOC_FORMATS = ["json", "markdown"] @@ -105,6 +105,13 @@ def stop(self): def _generate_docs(self, args): print("Generating docs in format: " + args.format) generator = Generator( - self._plugin.name(), self._plugin.get_tables(tables=["*"], skip_tables=[]) + self._plugin.name(), + self._plugin.get_tables( + options=TableOptions( + tables=["*"], + skip_tables=[], + skip_dependent_tables=False, + ) + ), ) generator.generate(args.directory, args.format)