add key cli

Signed-off-by: zhihuiwan <[email protected]>

python/fate_client/flow_client/flow.py

@@ -20,7 +20,7 @@
 from ruamel import yaml
 
 from flow_client.flow_cli.commands import checkpoint, tracking, component, data, job, model, queue, table, tag, task, \
-    provider, server, service, resource, privilege, test, template
+    provider, server, service, resource, privilege, test, template, key
 from flow_client.flow_cli.utils.cli_utils import prettify
 
 
@@ -160,6 +160,7 @@ def initialization(**kwargs):
 flow_cli.add_command(checkpoint.checkpoint)
 flow_cli.add_command(test.test)
 flow_cli.add_command(template.template)
+flow_cli.add_command(key.key)
 
 
 if __name__ == '__main__':

python/fate_client/flow_client/flow_cli/commands/key.py

@@ -0,0 +1,86 @@
+#
+#  Copyright 2019 The FATE Authors. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+#
+import click
+from flow_client.flow_cli.utils import cli_args
+from flow_client.flow_cli.utils.cli_utils import preprocess, access_server
+
+
+@click.group(short_help="Key Operations")
+@click.pass_context
+def key(ctx):
+    """
+    \b
+    Provides numbers of key operational commands, including save, query and delete.
+    For more details, please check out the help text.
+    """
+    pass
+
+
+@key.command("save", short_help="Save Public Key Command")
+@cli_args.CONF_PATH
+@click.pass_context
+def save(ctx, **kwargs):
+    """
+    - DESCRIPTION:
+
+
+    \b
+    save other site public key
+
+    \b
+    - USAGE:
+        flow key save -c fateflow/examples/key/save_public_key.json
+    """
+    config_data, dsl_data = preprocess(**kwargs)
+    access_server('post', ctx, 'key/public/save', config_data)
+
+
+@key.command("delete", short_help="Delete Public Key Command")
+@cli_args.PARTYID_REQUIRED
+@click.pass_context
+def delete(ctx, **kwargs):
+    """
+    - DESCRIPTION:
+
+
+    \b
+    delete other site public key
+
+    \b
+    - USAGE:
+        flow key delete  -p 10000
+    """
+    config_data, dsl_data = preprocess(**kwargs)
+    access_server('post', ctx, 'key/public/delete', config_data)
+
+
+@key.command("query", short_help="Query Public Key Command")
+@cli_args.PARTYID_REQUIRED
+@click.pass_context
+def query(ctx, **kwargs):
+    """
+    - DESCRIPTION:
+
+
+    \b
+     query site public key
+
+    \b
+    - USAGE:
+        flow key query -p 10000
+    """
+    config_data, dsl_data = preprocess(**kwargs)
+    access_server('post', ctx, 'key/query', config_data)