FateWithShellmiaoComment/Fate/python/fate_client/flow_client/flow_cli/commands/privilege.py

#
#  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="Privilege Operations")
@click.pass_context
def privilege(ctx):
    """
    \b
    Provides numbers of privilege operational commands, including grant, query and delete.
    For more details, please check out the help text.
    """
    pass


@privilege.command("grant", short_help="Grant Privilege Command")
@cli_args.CONF_PATH
@click.pass_context
def grant(ctx, **kwargs):
    """
    - DESCRIPTION:


    \b
    grant component | dataset privilege

    \b
    - USAGE:
        flow privilege grant -c fateflow/examples/permission/grant.json
    """
    config_data, dsl_data = preprocess(**kwargs)
    access_server('post', ctx, 'permission/grant', config_data)


@privilege.command("delete", short_help="Delete Privilege Command")
@cli_args.CONF_PATH
@click.pass_context
def delete(ctx, **kwargs):
    """
    - DESCRIPTION:


    \b
    delete component | dataset privilege

    \b
    - USAGE:
        flow privilege delete  -c fateflow/examples/permission/delete.json
    """
    config_data, dsl_data = preprocess(**kwargs)
    access_server('post', ctx, 'permission/delete', config_data)


@privilege.command("query", short_help="Query Privilege Command")
@cli_args.PARTYID_REQUIRED
@click.pass_context
def query(ctx, **kwargs):
    """
    - DESCRIPTION:


    \b
    query component | dataset privilege

    \b
    - USAGE:
        flow privilege query -p 10000
    """
    config_data, dsl_data = preprocess(**kwargs)
    access_server('post', ctx, 'permission/query', config_data)