| Index: tools/telemetry/third_party/gsutilz/third_party/apitools/apitools/gen/gen_client.py
|
| diff --git a/tools/telemetry/third_party/gsutilz/third_party/apitools/apitools/gen/gen_client.py b/tools/telemetry/third_party/gsutilz/third_party/apitools/apitools/gen/gen_client.py
|
| deleted file mode 100755
|
| index 7b7b6487af47d2fa3e733ac6a1942ff7ec730889..0000000000000000000000000000000000000000
|
| --- a/tools/telemetry/third_party/gsutilz/third_party/apitools/apitools/gen/gen_client.py
|
| +++ /dev/null
|
| @@ -1,254 +0,0 @@
|
| -#!/usr/bin/env python
|
| -"""Command-line interface to gen_client."""
|
| -
|
| -import contextlib
|
| -import json
|
| -import logging
|
| -import os
|
| -import pkgutil
|
| -import sys
|
| -
|
| -from google.apputils import appcommands
|
| -import gflags as flags
|
| -
|
| -from apitools.base.py import exceptions
|
| -from apitools.gen import gen_client_lib
|
| -from apitools.gen import util
|
| -
|
| -flags.DEFINE_string(
|
| - 'infile', '',
|
| - 'Filename for the discovery document. Mutually exclusive with '
|
| - '--discovery_url.')
|
| -flags.DEFINE_string(
|
| - 'discovery_url', '',
|
| - 'URL (or "name.version") of the discovery document to use. '
|
| - 'Mutually exclusive with --infile.')
|
| -
|
| -flags.DEFINE_string(
|
| - 'base_package',
|
| - 'apitools.base.py',
|
| - 'Base package path of apitools (defaults to '
|
| - 'apitools.base.py)'
|
| -)
|
| -flags.DEFINE_string(
|
| - 'outdir', '',
|
| - 'Directory name for output files. (Defaults to the API name.)')
|
| -flags.DEFINE_boolean(
|
| - 'overwrite', False,
|
| - 'Only overwrite the output directory if this flag is specified.')
|
| -flags.DEFINE_string(
|
| - 'root_package', '',
|
| - 'Python import path for where these modules should be imported from.')
|
| -
|
| -
|
| -flags.DEFINE_multistring(
|
| - 'strip_prefix', [],
|
| - 'Prefix to strip from type names in the discovery document. (May '
|
| - 'be specified multiple times.)')
|
| -flags.DEFINE_string(
|
| - 'api_key', None,
|
| - 'API key to use for API access.')
|
| -flags.DEFINE_string(
|
| - 'client_json', None,
|
| - 'Use the given file downloaded from the dev. console for client_id '
|
| - 'and client_secret.')
|
| -flags.DEFINE_string(
|
| - 'client_id', '1042881264118.apps.googleusercontent.com',
|
| - 'Client ID to use for the generated client.')
|
| -flags.DEFINE_string(
|
| - 'client_secret', 'x_Tw5K8nnjoRAqULM9PFAC2b',
|
| - 'Client secret for the generated client.')
|
| -flags.DEFINE_multistring(
|
| - 'scope', [],
|
| - 'Scopes to request in the generated client. May be specified more than '
|
| - 'once.')
|
| -flags.DEFINE_string(
|
| - 'user_agent', '',
|
| - 'User agent for the generated client. Defaults to <api>-generated/0.1.')
|
| -flags.DEFINE_boolean(
|
| - 'generate_cli', True, 'If True, a CLI is also generated.')
|
| -flags.DEFINE_list(
|
| - 'unelidable_request_methods', [],
|
| - 'Full method IDs of methods for which we should NOT try to elide '
|
| - 'the request type. (Should be a comma-separated list.)')
|
| -
|
| -flags.DEFINE_boolean(
|
| - 'experimental_capitalize_enums', False,
|
| - 'Dangerous: attempt to rewrite enum values to be uppercase.')
|
| -flags.DEFINE_enum(
|
| - 'experimental_name_convention', util.Names.DEFAULT_NAME_CONVENTION,
|
| - util.Names.NAME_CONVENTIONS,
|
| - 'Dangerous: use a particular style for generated names.')
|
| -flags.DEFINE_boolean(
|
| - 'experimental_proto2_output', False,
|
| - 'Dangerous: also output a proto2 message file.')
|
| -
|
| -FLAGS = flags.FLAGS
|
| -
|
| -flags.RegisterValidator(
|
| - 'infile', lambda i: not (i and FLAGS.discovery_url),
|
| - 'Cannot specify both --infile and --discovery_url')
|
| -flags.RegisterValidator(
|
| - 'discovery_url', lambda i: not (i and FLAGS.infile),
|
| - 'Cannot specify both --infile and --discovery_url')
|
| -
|
| -
|
| -def _CopyLocalFile(filename):
|
| - with contextlib.closing(open(filename, 'w')) as out:
|
| - src_data = pkgutil.get_data(
|
| - 'apitools.base.py', filename)
|
| - if src_data is None:
|
| - raise exceptions.GeneratedClientError(
|
| - 'Could not find file %s' % filename)
|
| - out.write(src_data)
|
| -
|
| -
|
| -def _GetCodegenFromFlags():
|
| - """Create a codegen object from flags."""
|
| - if FLAGS.discovery_url:
|
| - try:
|
| - discovery_doc = util.FetchDiscoveryDoc(FLAGS.discovery_url)
|
| - except exceptions.CommunicationError:
|
| - return None
|
| - else:
|
| - infile = os.path.expanduser(FLAGS.infile) or '/dev/stdin'
|
| - discovery_doc = json.load(open(infile))
|
| - names = util.Names(
|
| - FLAGS.strip_prefix,
|
| - FLAGS.experimental_name_convention,
|
| - FLAGS.experimental_capitalize_enums)
|
| -
|
| - if FLAGS.client_json:
|
| - try:
|
| - with open(FLAGS.client_json) as client_json:
|
| - f = json.loads(client_json.read())
|
| - web = f.get('web', {})
|
| - client_id = web.get('client_id')
|
| - client_secret = web.get('client_secret')
|
| - except IOError:
|
| - raise exceptions.NotFoundError(
|
| - 'Failed to open client json file: %s' % FLAGS.client_json)
|
| - else:
|
| - client_id = FLAGS.client_id
|
| - client_secret = FLAGS.client_secret
|
| -
|
| - if not client_id:
|
| - logging.warning('No client ID supplied')
|
| - client_id = ''
|
| -
|
| - if not client_secret:
|
| - logging.warning('No client secret supplied')
|
| - client_secret = ''
|
| -
|
| - client_info = util.ClientInfo.Create(
|
| - discovery_doc, FLAGS.scope, client_id, client_secret,
|
| - FLAGS.user_agent, names, FLAGS.api_key)
|
| - outdir = os.path.expanduser(FLAGS.outdir) or client_info.default_directory
|
| - if os.path.exists(outdir) and not FLAGS.overwrite:
|
| - raise exceptions.ConfigurationValueError(
|
| - 'Output directory exists, pass --overwrite to replace '
|
| - 'the existing files.')
|
| -
|
| - root_package = FLAGS.root_package or util.GetPackage(
|
| - outdir) # pylint: disable=line-too-long
|
| - return gen_client_lib.DescriptorGenerator(
|
| - discovery_doc, client_info, names, root_package, outdir,
|
| - base_package=FLAGS.base_package,
|
| - generate_cli=FLAGS.generate_cli,
|
| - use_proto2=FLAGS.experimental_proto2_output,
|
| - unelidable_request_methods=FLAGS.unelidable_request_methods)
|
| -
|
| -
|
| -# TODO(craigcitro): Delete this if we don't need this functionality.
|
| -def _WriteBaseFiles(codegen):
|
| - with util.Chdir(codegen.outdir):
|
| - _CopyLocalFile('app2.py')
|
| - _CopyLocalFile('base_api.py')
|
| - _CopyLocalFile('base_cli.py')
|
| - _CopyLocalFile('credentials_lib.py')
|
| - _CopyLocalFile('exceptions.py')
|
| -
|
| -
|
| -def _WriteProtoFiles(codegen):
|
| - with util.Chdir(codegen.outdir):
|
| - with open(codegen.client_info.messages_proto_file_name, 'w') as out:
|
| - codegen.WriteMessagesProtoFile(out)
|
| - with open(codegen.client_info.services_proto_file_name, 'w') as out:
|
| - codegen.WriteServicesProtoFile(out)
|
| -
|
| -
|
| -def _WriteGeneratedFiles(codegen):
|
| - if codegen.use_proto2:
|
| - _WriteProtoFiles(codegen)
|
| - with util.Chdir(codegen.outdir):
|
| - with open(codegen.client_info.messages_file_name, 'w') as out:
|
| - codegen.WriteMessagesFile(out)
|
| - with open(codegen.client_info.client_file_name, 'w') as out:
|
| - codegen.WriteClientLibrary(out)
|
| - if FLAGS.generate_cli:
|
| - with open(codegen.client_info.cli_file_name, 'w') as out:
|
| - codegen.WriteCli(out)
|
| - os.chmod(codegen.client_info.cli_file_name, 0o755)
|
| -
|
| -
|
| -def _WriteInit(codegen):
|
| - with util.Chdir(codegen.outdir):
|
| - with open('__init__.py', 'w') as out:
|
| - codegen.WriteInit(out)
|
| -
|
| -
|
| -class GenerateClient(appcommands.Cmd):
|
| -
|
| - """Driver for client code generation."""
|
| -
|
| - def Run(self, _):
|
| - """Create a client library."""
|
| - codegen = _GetCodegenFromFlags()
|
| - if codegen is None:
|
| - logging.error('Failed to create codegen, exiting.')
|
| - return 128
|
| - _WriteGeneratedFiles(codegen)
|
| - _WriteInit(codegen)
|
| -
|
| -
|
| -class GenerateProto(appcommands.Cmd):
|
| -
|
| - """Generate just the two proto files for a given API."""
|
| -
|
| - def Run(self, _):
|
| - """Create proto definitions for an API."""
|
| - codegen = _GetCodegenFromFlags()
|
| - _WriteProtoFiles(codegen)
|
| -
|
| -
|
| -# pylint:disable=invalid-name
|
| -
|
| -
|
| -def run_main():
|
| - """Function to be used as setuptools script entry point."""
|
| - # Put the flags for this module somewhere the flags module will look
|
| - # for them.
|
| -
|
| - # pylint:disable=protected-access
|
| - new_name = flags._GetMainModule()
|
| - sys.modules[new_name] = sys.modules['__main__']
|
| - for flag in FLAGS.FlagsByModuleDict().get(__name__, []):
|
| - FLAGS._RegisterFlagByModule(new_name, flag)
|
| - for key_flag in FLAGS.KeyFlagsByModuleDict().get(__name__, []):
|
| - FLAGS._RegisterKeyFlagForModule(new_name, key_flag)
|
| - # pylint:enable=protected-access
|
| -
|
| - # Now set __main__ appropriately so that appcommands will be
|
| - # happy.
|
| - sys.modules['__main__'] = sys.modules[__name__]
|
| - appcommands.Run()
|
| - sys.modules['__main__'] = sys.modules.pop(new_name)
|
| -
|
| -
|
| -def main(_):
|
| - appcommands.AddCmd('client', GenerateClient)
|
| - appcommands.AddCmd('proto', GenerateProto)
|
| -
|
| -
|
| -if __name__ == '__main__':
|
| - appcommands.Run()
|
|
|