Exemplo n.º 1
0
  def do(self, sys_argv):
    options, args = self._parse_args(sys_argv, 2)

    dump_path = args[1]
    target_policy = args[2]
    component = options.component

    (bucket_set, dump) = SubCommand.load_basic_files(dump_path, False)
    policy_set = PolicySet.load(SubCommand._parse_policy_list(target_policy))

    with open(SubCommand._find_prefix(dump_path) + '.maps', 'r') as maps_f:
      maps_lines = maps_f.readlines()
    PProfCommand._output(
        dump, policy_set[target_policy], bucket_set, maps_lines, component,
        sys.stdout)

    return 0
Exemplo n.º 2
0
    def do(self, sys_argv):
        options, args = self._parse_args(sys_argv, 2)

        dump_path = args[1]
        target_policy = args[2]
        component = options.component

        (bucket_set, dump) = SubCommand.load_basic_files(dump_path, False)
        policy_set = PolicySet.load(
            SubCommand._parse_policy_list(target_policy))

        with open(SubCommand._find_prefix(dump_path) + '.maps', 'r') as maps_f:
            maps_lines = maps_f.readlines()
        PProfCommand._output(dump, policy_set[target_policy], bucket_set,
                             maps_lines, component, sys.stdout)

        return 0
Exemplo n.º 3
0
    def do(self, sys_argv):
        options, args = self._parse_args(sys_argv, 2)
        dump_path = args[1]
        gs_path = args[2]

        dump_files = SubCommand._find_all_dumps(dump_path)
        bucket_files = SubCommand._find_all_buckets(dump_path)
        prefix = SubCommand._find_prefix(dump_path)
        symbol_data_sources = SymbolDataSources(prefix)
        symbol_data_sources.prepare()
        symbol_path = symbol_data_sources.path()

        handle_zip, filename_zip = tempfile.mkstemp('.zip', 'dmprof')
        os.close(handle_zip)

        try:
            file_zip = zipfile.ZipFile(filename_zip, 'w', zipfile.ZIP_DEFLATED)
            for filename in dump_files:
                file_zip.write(filename,
                               os.path.basename(os.path.abspath(filename)))
            for filename in bucket_files:
                file_zip.write(filename,
                               os.path.basename(os.path.abspath(filename)))

            symbol_basename = os.path.basename(os.path.abspath(symbol_path))
            for filename in os.listdir(symbol_path):
                if not filename.startswith('.'):
                    file_zip.write(
                        os.path.join(symbol_path, filename),
                        os.path.join(
                            symbol_basename,
                            os.path.basename(os.path.abspath(filename))))
            file_zip.close()

            returncode = UploadCommand._run_gsutil(options.gsutil, 'cp', '-a',
                                                   'public-read', filename_zip,
                                                   gs_path)
        finally:
            os.remove(filename_zip)

        return returncode
Exemplo n.º 4
0
  def do(self, sys_argv):
    options, args = self._parse_args(sys_argv, 2)
    dump_path = args[1]
    gs_path = args[2]

    dump_files = SubCommand._find_all_dumps(dump_path)
    bucket_files = SubCommand._find_all_buckets(dump_path)
    prefix = SubCommand._find_prefix(dump_path)
    symbol_data_sources = SymbolDataSources(prefix)
    symbol_data_sources.prepare()
    symbol_path = symbol_data_sources.path()

    handle_zip, filename_zip = tempfile.mkstemp('.zip', 'dmprof')
    os.close(handle_zip)

    try:
      file_zip = zipfile.ZipFile(filename_zip, 'w', zipfile.ZIP_DEFLATED)
      for filename in dump_files:
        file_zip.write(filename, os.path.basename(os.path.abspath(filename)))
      for filename in bucket_files:
        file_zip.write(filename, os.path.basename(os.path.abspath(filename)))

      symbol_basename = os.path.basename(os.path.abspath(symbol_path))
      for filename in os.listdir(symbol_path):
        if not filename.startswith('.'):
          file_zip.write(os.path.join(symbol_path, filename),
                         os.path.join(symbol_basename, os.path.basename(
                             os.path.abspath(filename))))
      file_zip.close()

      returncode = UploadCommand._run_gsutil(
          options.gsutil, 'cp', '-a', 'public-read', filename_zip, gs_path)
    finally:
      os.remove(filename_zip)

    return returncode