Beispiel #1
0
  def Run(self, args):
    dataproc = dp.Dataproc(self.ReleaseTrack())

    template_ref = args.CONCEPTS.template.Parse()

    # Get specified version, or most recent version if no version arg provided.
    workflow_template = dataproc.GetRegionsWorkflowTemplate(
        template_ref, args.version)

    if args.destination:
      with files.FileWriter(args.destination) as stream:
        util.WriteYaml(
            message=workflow_template, stream=stream, schema_path=SCHEMA_PATH)
    else:
      util.WriteYaml(
          message=workflow_template, stream=sys.stdout, schema_path=SCHEMA_PATH)
Beispiel #2
0
    def Run(self, args):
        dataproc = dp.Dataproc(self.ReleaseTrack())

        cluster_ref = util.ParseCluster(args.name, dataproc)

        request = dataproc.messages.DataprocProjectsRegionsClustersGetRequest(
            projectId=cluster_ref.projectId,
            region=cluster_ref.region,
            clusterName=cluster_ref.clusterName)

        cluster = dataproc.client.projects_regions_clusters.Get(request)

        # Filter out Dataproc-generated labels.
        clusters.DeleteGeneratedLabels(cluster, dataproc)

        if args.destination:
            with files.FileWriter(args.destination) as stream:
                util.WriteYaml(message=cluster,
                               stream=stream,
                               schema_path=SCHEMA_PATH)
        else:
            util.WriteYaml(message=cluster,
                           stream=sys.stdout,
                           schema_path=SCHEMA_PATH)