resource.py 1.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162
  1. #
  2. # Copyright 2019 The FATE Authors. All Rights Reserved.
  3. #
  4. # Licensed under the Apache License, Version 2.0 (the "License");
  5. # you may not use this file except in compliance with the License.
  6. # You may obtain a copy of the License at
  7. #
  8. # http://www.apache.org/licenses/LICENSE-2.0
  9. #
  10. # Unless required by applicable law or agreed to in writing, software
  11. # distributed under the License is distributed on an "AS IS" BASIS,
  12. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13. # See the License for the specific language governing permissions and
  14. # limitations under the License.
  15. #
  16. import click
  17. from flow_client.flow_cli.utils import cli_args
  18. from flow_client.flow_cli.utils.cli_utils import preprocess, access_server
  19. @click.group(short_help="Resource Manager")
  20. @click.pass_context
  21. def resource(ctx):
  22. """
  23. \b
  24. Provides numbers of resource operational commands, including query and return.
  25. For more details, please check out the help text.
  26. """
  27. pass
  28. @resource.command("query", short_help="Query Resource Command")
  29. @click.pass_context
  30. def query(ctx, **kwargs):
  31. """
  32. \b
  33. - DESCRIPTION:
  34. Query Resource Information.
  35. \b
  36. - USAGE:
  37. flow resource query
  38. """
  39. config_data, dsl_data = preprocess(**kwargs)
  40. access_server('post', ctx, 'resource/query', config_data)
  41. @resource.command("return", short_help="Return Job Resource Command")
  42. @cli_args.JOBID
  43. @click.pass_context
  44. def resource_return(ctx, **kwargs):
  45. """
  46. \b
  47. - DESCRIPTION:
  48. Return Job Resource Command
  49. \b
  50. - USAGE:
  51. flow resource return -j $JobId
  52. """
  53. config_data, dsl_data = preprocess(**kwargs)
  54. access_server('post', ctx, 'resource/return', config_data)