sdk work

Author: twratl

requirements.txt

@@ -16,4 +16,5 @@ mkdocs==1.3.1
 mkdocs-click==0.8.0
 twine~=4.0.1
 python-dateutil~=2.8.2
-boto3
+boto3
+jmespath

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = pybritive
-version = 0.10.2
+version = 0.11.0
 author = Britive Inc.
 author_email = support@britive.com
 description = A pure Python CLI for Britive
@@ -27,6 +27,7 @@ install_requires =
     cryptography
     python-dateutil
     britive>=2.14.0
+    jmespath
 
 [options.packages.find]
 where = src

src/pybritive/britive_cli.py

@@ -15,6 +15,7 @@
 from datetime import datetime
 import os
 import sys
+import jmespath
 
 
 default_table_format = 'fancy_grid'
@@ -596,6 +597,16 @@ def request_withdraw(self, profile):
     def clear_gcloud_auth_key_files(self):
         self.config.clear_gcloud_auth_key_files()
 
+    def sdk(self, method, parameters={}, query=None):
+        self.login()
+        func = self.b
+        for m in method.split('.'):
+            func = getattr(func, m)
+        response = func(**parameters)
+        self.print(jmespath.search(query, response) if query else response)
+
+
+
 
 
 

src/pybritive/cli_interface.py

@@ -11,6 +11,7 @@
 from .commands.cache import cache as group_cache
 from .commands.request import request as group_request
 from .commands.clear import clear as group_clear
+from .commands.sdk import sdk as command_sdk
 import sys
 import os
 
@@ -48,6 +49,7 @@ def cli(version):
 cli.add_command(group_cache)
 cli.add_command(group_request)
 cli.add_command(group_clear)
+cli.add_command(command_sdk)
 
 
 if __name__ == "__main__":

src/pybritive/commands/sdk.py

@@ -0,0 +1,45 @@
+import click
+from ..helpers.build_britive import build_britive
+from ..options.britive_options import britive_options
+
+
+@click.command(
+    context_settings=dict(
+        ignore_unknown_options=True,
+        allow_extra_args=True
+    )
+)
+@build_britive
+@britive_options(names='query,output_format,tenant,token,passphrase,federation_provider')
+@click.argument('method')
+def sdk(ctx, query, output_format, tenant, token, passphrase, federation_provider, method):
+    """Exposes the Britive Python SDK methods to the CLI.
+
+    Documentation on each SDK method can be found inside the Python SDK itself and on Github
+    (https://github.com/britive/python-sdk). The Python package `britive` is a dependency of the CLI
+    already so the SDK is available without installing any extra packages.
+
+    It is left up to the caller to provide the proper `method` and `parameters` based on the documentation
+    of the API call being performed.
+
+    The authenticated identity must have the appropriate permissions to perform the actions being requested.
+    General end users of Britive will not have these permissions. This call (and the larger SDK) is generally
+    meant for administrative functionality.
+
+    Example of use:
+
+    * pybritive sdk users.list
+
+    * pybritive sdk tags.create --name testtag --description "test tag"
+
+    * pybritive sdk users.list --query '[].email'
+
+    * pybritive sdk profiles.create --application-id <id> --name testprofile
+
+    """
+    parameters = {ctx.args[i][2:]: ctx.args[i + 1] for i in range(0, len(ctx.args), 2)}
+    ctx.obj.britive.sdk(
+        method=method,
+        parameters=parameters,
+        query=query
+    )

src/pybritive/options/britive_options.py

@@ -22,6 +22,7 @@
 from ..options.federation_provider import option as federation_provider
 from ..options.gcloud_key_file import option as gcloud_key_file
 from ..options.verbose import option as verbose
+from ..options.query import option as query
 
 options_map = {
     'tenant': tenant,
@@ -47,7 +48,8 @@
     'aws_credentials_file': aws_credentials_file,
     'federation_provider': federation_provider,
     'gcloud_key_file': gcloud_key_file,
-    'verbose': verbose
+    'verbose': verbose,
+    'query': query
 }
 
 

src/pybritive/options/query.py

@@ -0,0 +1,8 @@
+import click
+
+
+option = click.option(
+    '--query',
+    default=None,
+    help='JMESPath query to apply to the response.'
+)