Python yaml.safe_dump_all函数代码示例

def sanitise_report(report_file, semaphore):
    match = re.search("^" + re.escape(settings.reports_directory) + "(.*)", report_file)

    # read report file
    report = Report(report_file)
    report_header = report.header
    report_header["report_file"] = match.group(1)

    report_filename = os.path.split(report_file)[-1]
    report_filename_sanitised = os.path.join(settings.sanitised_directory, report_filename)

    if os.path.isfile(report_filename_sanitised):
        log.info("Sanitised report name already exists, overwriting: %s" % report_filename_sanitised)
    else:
        log.info("New report file: %s" % report_filename_sanitised)

    report_file_sanitised = open(report_filename_sanitised, "w")

    safe_dump(report_header, report_file_sanitised, explicit_start=True, explicit_end=True)

    safe_dump_all(report, report_file_sanitised, explicit_start=True, explicit_end=True, default_flow_style=False)

    log.info("Moving original unsanitised file %s to archive" % report_file)

    archive_report(report_file)

    report_file_sanitised.close()
    report.close()

    os.remove(report_file)

    semaphore.release()

def readpage(page):
	news=get_news(page)
	if len(news)>0:
		with open('meneame.yml', 'a') as outfile:
			yaml.safe_dump_all(news, outfile, default_flow_style=False,allow_unicode=True, default_style=None)
			outfile.write("---\n")
	else:
		print "SKIP"

def read(id):
	url=STR+id
	response = requests.get(url)
	soup = bs4.BeautifulSoup(response.text,"lxml")
	body=soup.select("div.news-body")
	if not body or len(body)==0:
		print "ERROR: "+url
		return True
	a=body[0]
	cerrado=a.select("#a-va-"+id+" span.closed")
	if len(cerrado)==0:
		print "ABIERTA: "+url
		return False
	div=a.select("div.news-shakeit")
	if len(div)==0:
		print "SIN SHAKE: "+url
		return False
	status=div[0].attrs['class'][-1].split('-')[-1]
	h1=soup.find("h1")
	if not h1:
		print "SIN h1: "+url
		return False
	dates=[int(d.attrs['data-ts'].strip()) for d in a.select("div.news-submitted span.ts")]
	new={
		'id': int(id),
		'title': h1.get_text().strip(), 
		'url': h1.a.attrs["href"],
		'author': a.select("div.news-submitted a")[0].attrs["href"].split("/")[-1],
		'body': " ".join([t.string.strip() for t in a.find_all(text=True,recursive=False)]).strip(), 
		'sub': a.select("div.news-details span.tool a")[0].get_text().strip(),
		'story': get_meta(soup,"property","og:url"),
		'karma': int(a.select("#a-karma-"+id)[0].get_text().strip()),
		'sent': dates[0], 
		'status': status,
		'votes': {
			'users': int(a.select("#a-usu-"+id)[0].get_text().strip()),
			'anonymous': int(a.select("#a-ano-"+id)[0].get_text().strip()),
			'negative':int(a.select("#a-neg-"+id)[0].get_text().strip())
		}
	}
	if status == "published":
		new['comments']=dates[1]
	counter=a.select("span.comments-counter span.counter")
	if len(counter)>0:
		new['comments']=int(counter[0].get_text().strip())
	else:
		new['comments']=0
	tags=[t.get_text().strip() for t in a.select("span.news-tags a") if len(t.get_text().strip())>0]
	if len(tags)>0:
		new['tags']=tags

	with open(OUT, 'a') as outfile:
		yaml.safe_dump_all([new], outfile, default_flow_style=False,allow_unicode=True, default_style=None)
		outfile.write("---\n")

	return True

def main(archive_file):
    if not os.path.isfile(archive_file):
        print "Archive file does not exist: " + archive_file
        return 1

    # we only support tar.gz and .gz
    if archive_file.endswith("tar.gz"):
        archive_tar = tarfile.open(archive_file)
        for element in archive_tar:
            f = archive_tar.extractfile(element)
            fp, report_file = tempfile.mkstemp()
            while True:
                data = f.read()
                if not data:
                    break
                os.write(fp, data)
            f.close()
    elif archive_file.endswith(".gz"):
        fp, report_file = tempfile.mkstemp()
        f = gzip.open(archive_file, 'rb')
        while True:
            data = f.read()
            if not data:
                break
            os.write(fp, data)
        f.close()

    report = Report(report_file, action="sanitise")
    report_filename = generate_filename(report.header)
    report_filename_sanitised = os.path.join(settings.sanitised_directory,
                                             report_filename)
    report.header['report_file'] = "%s/%s" % (report.header['probe_cc'],
                                              report_filename)

    report_file_sanitised = open(report_filename_sanitised, "w")

    safe_dump(
        report.header,
        report_file_sanitised,
        explicit_start=True,
        explicit_end=True)
    safe_dump_all(
        report,
        report_file_sanitised,
        explicit_start=True,
        explicit_end=True,
        default_flow_style=False)
    delete_existing_report_entries(report.header)

    public_report_file = os.path.join(
        settings.public_directory, report.header['probe_cc'], report_filename)
    if os.path.isfile(public_report_file):
        os.remove(public_report_file)

    report_file_sanitised.close()
    os.remove(report_file)

def main():
    parser = optparse.OptionParser()
    parser.add_option("-H", "--hash", action="store",
                      help="verify that stdin hash given sha256 hash")
    options, args = parser.parse_args()
    if options.hash:
        gen = process_package_with_hash(sys.stdin, options.hash)
    else:
        gen = process_package(sys.stdin)
    yaml.safe_dump_all(gen, sys.stdout)

def write(fname, *items):
  try:
    written = False
    with closing(_open_userfile(fname, 'w')) as f:
      if written:
        f.write(SEPARATOR)
      yaml.safe_dump_all(items, f)
  except Exception as e:
    print("Can't write filename", fname, e.message)
    if PROPAGATE_EXCEPTIONS:
      raise

def make_adf(archive=None, canonical=False, out=None, pretty=False):
    """
    >>> meta = Meta('zip', 'aes-256-ctr', created='now')
    >>> archive = Archive('title', meta)
    >>> cert = Certificate(chr(255)*16)
    >>> make_adf([archive, cert], out=sys.stdout, pretty=True)
    !archive
    meta: !meta
      cipher: aes-256-ctr
      created: now
      format: zip
    title: title
    --- !certificate
    key: !!binary |
      /////////////////////w==
    >>> make_adf([archive, cert], out=sys.stdout)
    !archive
    meta: !meta {cipher: aes-256-ctr, created: now, format: zip}
    title: title
    --- !certificate
    key: !!binary |
      /////////////////////w==
    """

    if not hasattr(archive, '__getitem__'):
        archive = [archive]
    if pretty:
        out.write("--- ".join(map(pyaml.dump, archive)))
        return
    return yaml.safe_dump_all(archive, out, canonical=canonical)

    def CompileReport(self, analysis_mediator):
        """Compiles a report of the analysis.

    Args:
      analysis_mediator: The analysis mediator object (instance of
                         AnalysisMediator).

    Returns:
      The analysis report (instance of AnalysisReport).
    """
        report = event.AnalysisReport(self.NAME)

        if self._output_format == u"yaml":
            lines_of_text = []
            lines_of_text.append(yaml.safe_dump_all(self._service_collection.services))
        else:
            lines_of_text = [u"Listing Windows Services"]
            for service in self._service_collection.services:
                lines_of_text.append(self._FormatServiceText(service))
                # Separate services with a blank line.
                lines_of_text.append(u"")

        report.SetText(lines_of_text)

        return report

    def test_deserializer_validate_existing_assignments(self):
        assignment1 = coremodels.Assignment.objects.create(
            title='Existing1', text='text1')
        assignment2 = coremodels.Assignment.objects.create(
            title='Existing2', text='text2')
        duck1000tag = coremodels.Tag.objects.create(tag='duck1000')
        assignment1.tags.add(duck1000tag)
        assignment2.tags.add(duck1000tag)
        deserializer = multiassignment_serialize.Deserializer(yaml.safe_dump_all([
            {'id': assignment1.id, 'title': 'Updated1', 'text': 'updatedText1',
             'tags': ['oblig1']},
            {'title': 'New'},
            {'id': assignment2.id, 'title': 'Updated2', 'text': 'updatedText2',
             'tags': ['duck1000', 'oblig2', 'week3']},
        ]), course_tag='duck1000')

        assignments_by_tag = {}
        existing_assignments = deserializer._validate_existing_assignments(assignments_by_tag)
        self.assertEquals(len(existing_assignments), 2)
        self.assertEquals(
            set(assignments_by_tag.keys()),
            set(['duck1000', 'oblig1', 'oblig2', 'week3']))
        self.assertEquals(
            assignments_by_tag['duck1000'], [assignment2, assignment1])
        self.assertEquals(
            assignments_by_tag['oblig2'], [assignment2])

def pass_insert_host_yaml(host, yaml_docs):
    value = yaml.safe_dump_all(
        yaml_docs,
        # use multiline "block" style instead of {a:b} ("flow" style)
        default_flow_style=False
    )

    pass_insert( host_to_pass_key(host), value )

def dump(arg):
    """Dump a sequence of dictionaries as YAML for editing.
    """
    return yaml.safe_dump_all(
        arg,
        allow_unicode=True,
        default_flow_style=False,
    )

def write_yaml(input, filename):
    with open(filename, "w") as f:
        if isinstance(input, list):
            f.write(yaml.safe_dump_all(input, default_flow_style=False))
        elif isinstance(input, dict):
            f.write(yaml.safe_dump(input, default_flow_style=False))
        else:
            raise Exception("cannot dump $s objects to yaml." % str(type(input)))

 def test_deserializer_validate_existing_assignments_invalid(self):
     assignment1 = coremodels.Assignment.objects.create(
         title='Existing1', text='text1')
     assignment1.tags.create(tag='duck1000')
     deserializer = multiassignment_serialize.Deserializer(yaml.safe_dump_all([
         {'id': assignment1.id},
     ]), course_tag='duck1000')
     with self.assertRaises(multiassignment_serialize.DeserializerValidationErrors):
         deserializer._validate_existing_assignments(assignments_by_tag={})

def main(argv=sys.argv):

    parser = arg_parser()

    args = parser.parse_args()

    print("Reading %s" % args.infile)
    infile = open( args.infile )
    d, report = csv2dict(infile)


    print("Writing to %s" % args.outfile)
    outfile = open( args.outfile, 'w' )

    if args.fmt == 'json':
        import json
        outfile.write( json.dumps( d ) )
    elif args.fmt == 'yaml':
        import yaml
        yaml.safe_dump_all( d, outfile, default_flow_style=False )

  def FormatArtifacts(self, artifacts):
    """Formats artifacts to desired output format.

    Args:
      artifacts (list[ArtifactDefinition]): artifact definitions.

    Returns:
      str: formatted string of artifact definition.
    """
    # TODO: improve output formatting of yaml
    artifact_definitions = [artifact.AsDict() for artifact in artifacts]
    yaml_data = yaml.safe_dump_all(artifact_definitions)
    return yaml_data