Python pyaml.dump方法代码示例

# 需要导入模块: import pyaml [as 别名]
# 或者: from pyaml import dump [as 别名]
def test_pretty_dumping(self):
        "Use pyaml to dump nicer"
        # pyaml only runs on 2.7 and above
        if sys.version_info > (2, 6) and pyaml is not None:

            with codecs.open("tests/unpretty.md", "r", "utf-8") as f:
                data = f.read()

            post = frontmatter.load("tests/unpretty.md")
            yaml = pyaml.dump(post.metadata)

            # the unsafe dumper gives you nicer output, for times you want that
            dump = frontmatter.dumps(post, Dumper=pyaml.UnsafePrettyYAMLDumper)

            self.assertEqual(dump, data)
            self.assertTrue(yaml in dump) 

# 需要导入模块: import pyaml [as 别名]
# 或者: from pyaml import dump [as 别名]
def process_playbook(playbook_path):
    print('Processing playbook %s' % playbook_path)
    with open(playbook_path, 'r') as fh:
        data = yaml.load(fh)


    for task in data:
        task_keys = frozenset(task.keys())
        for key in task_keys:
            if key.startswith('netscaler_'):
                module_name = key
        print('module name is %s' % module_name)
        module_arguments = task[module_name]
        del module_arguments['nsip']
        del module_arguments['nitro_user']
        del module_arguments['nitro_pass']
        module_arguments['mas_ip'] = '{{ nsip }}'
        module_arguments['mas_auth_token'] = '{{ mas_login_result.nitro_auth_token }}'
        module_arguments['mas_proxy_call'] = 'true'
        module_arguments['instance_ip'] = '{{ instance_ip }}'

    with open(playbook_path, 'w') as fh:
        pyaml.dump(data,fh, indent=2) 

# 需要导入模块: import pyaml [as 别名]
# 或者: from pyaml import dump [as 别名]
def test_dumping_with_custom_delimiters(self):
        "dump with custom delimiters"
        post = frontmatter.load("tests/hello-world.markdown")
        dump = frontmatter.dumps(post, start_delimiter="+++", end_delimiter="+++")

        self.assertTrue("+++" in dump) 

# 需要导入模块: import pyaml [as 别名]
# 或者: from pyaml import dump [as 别名]
def test_dump_to_file(self):
        "dump post to filename"
        post = frontmatter.load("tests/hello-world.markdown")

        tempdir = tempfile.mkdtemp()
        filename = os.path.join(tempdir, "hello.md")
        frontmatter.dump(post, filename)

        with open(filename) as f:
            self.assertEqual(f.read(), frontmatter.dumps(post))

        # cleanup
        shutil.rmtree(tempdir) 

# 需要导入模块: import pyaml [as 别名]
# 或者: from pyaml import dump [as 别名]
def sanity_check(self, filename, handler_type):
        "Ensure we can load -> dump -> load"
        post = frontmatter.load(filename)

        self.assertIsInstance(post.handler, handler_type)

        # dump and reload
        repost = frontmatter.loads(frontmatter.dumps(post))

        self.assertEqual(post.metadata, repost.metadata)
        self.assertEqual(post.content, repost.content)
        self.assertEqual(post.handler, repost.handler) 

# 需要导入模块: import pyaml [as 别名]
# 或者: from pyaml import dump [as 别名]
def transform(self, app_config):
        converted_app_config = self._spec_factory.transform(app_config, strip_defaults=True)
        return pyaml.dump(converted_app_config) 

# 需要导入模块: import pyaml [as 别名]
# 或者: from pyaml import dump [as 别名]
def test_config_from_file(self, key, attr, value, tmpdir):
        config_file = tmpdir.join("config.yaml")
        with config_file.open("w") as fobj:
            pyaml.dump({key: value}, fobj, safe=True, default_style='"')
        config = Configuration(["--config-file", config_file.strpath])
        assert getattr(config, attr) == value 

# 需要导入模块: import pyaml [as 别名]
# 或者: from pyaml import dump [as 别名]
def test_host_rewrite_rules_from_file(self, tmpdir):
        args = ("pattern=value", r"(\d+)\.\example\.com=$1.example.net", "www.([a-z]+.com)={env}.$1")
        config_file = tmpdir.join("config.yaml")
        with config_file.open("w") as fobj:
            pyaml.dump({"host-rewrite-rule": args}, fobj, safe=True, default_style='"')
        config = Configuration(["--config-file", config_file.strpath])
        assert config.host_rewrite_rules == [HostRewriteRule(arg) for arg in args] 

# 需要导入模块: import pyaml [as 别名]
# 或者: from pyaml import dump [as 别名]
def test_behavior_for_undefined_flags_in_config_yml(self, tmpdir):
        config_flags = {
            "undefined_configuration_parameter": "a value",
            "debug": "true",
        }
        config_file = tmpdir.join("config.yaml")
        with config_file.open("w") as fobj:
            pyaml.dump(config_flags, fobj, safe=True, default_style='"')
        config = Configuration(["--config-file", config_file.strpath])
        # Defined configuration flags should be set
        assert config.debug is True
        # Undefined configuration flags should not be available
        with pytest.raises(AttributeError):
            config.undefined_configuration_parameter 

# 需要导入模块: import pyaml [as 别名]
# 或者: from pyaml import dump [as 别名]
def yaml(self):
        logs_list = None
        if self.logs is not None:
            logs_list = [lm.yaml() for lm in self.logs]

        data = OrderedDict([
            ('service_name', self.name),
            ('team_name', self.team_name),
            ('port', self.port),
            ('healthcheck_url', self.healthcheck_url),
            ('logs', logs_list),
            ('code_deploy_logs', self.code_deploy_logs),
            ('environments', [e.yaml() for e in self.environments])
        ])
        return pyaml.dump(OrderedDict((k, v) for k, v in data.items() if v is not None)) 

# 需要导入模块: import pyaml [as 别名]
# 或者: from pyaml import dump [as 别名]
def generate_use_stub(resource_definition, output_yaml):
    #print(json.dumps(usage_playbook_template, indent=4))
    playbook = copy.deepcopy(usage_playbook_template)
    playbook[0]['tasks'][0]['vars'] = copy.deepcopy(resource_definition)
    delete_resource_definition= copy.deepcopy(resource_definition)
    del delete_resource_definition['resource_attributes']
    del delete_resource_definition['resource_non_updateable_attributes']

    playbook[0]['tasks'][1]['vars'] = delete_resource_definition
    #print(json.dumps(playbook, indent=4))
    with open(output_yaml, 'w') as fh:
        pyaml.dump(playbook, fh, indent=2)
    pass 

# 需要导入模块: import pyaml [as 别名]
# 或者: from pyaml import dump [as 别名]
def generate_skeleton(args):

    role_path = os.path.join(args.dir_path, 'citrix_adc_nitro_resource')
    if not os.path.exists(role_path):
        os.makedirs(role_path)

    paths = [
        os.path.join(role_path, 'tasks'),
        os.path.join(role_path, 'defaults'),
        os.path.join(role_path, 'vars'),
        os.path.join(role_path, 'tests/nitro/tasks'),
    ]
    for path in paths:
        if not os.path.exists(path):
            os.makedirs(path)

    with open(os.path.join(role_path, 'tasks', 'main.yaml'), 'w') as fh:
        fh.write(MAIN_ROLE_FILE)

    with open(os.path.join(role_path, 'tasks', 'nitro.yaml'), 'w') as fh:
        fh.write(NITRO_ROLE_FILE)

    with open(os.path.join(role_path, 'defaults', 'main.yaml'), 'w') as fh:
        fh.write(DEFAUTLS_VARIABLES_FILE)

    with open(os.path.join(role_path, 'tests/nitro/tasks', 'nitro_resource_task.yaml'), 'w') as fh:
        fh.write(NITRO_RESOURCE_TASK_FILE)


    workflow_source = os.path.join(HERE, '../../nitro_resource_utils/workflows.yaml')
    with open(workflow_source, 'r') as fh:
        workflow = yaml.load(fh)

    with open(os.path.join(role_path, 'vars', 'main.yaml'), 'w') as fh:
        pyaml.dump(workflow, fh) 

# 需要导入模块: import pyaml [as 别名]
# 或者: from pyaml import dump [as 别名]
def save_test(args, playbook_name, playbook):
    playbook_file = os.path.join(args.dir_path, 'citrix_adc_nitro_resource', 'tests', 'nitro', '%s.yaml' % playbook_name)
    with open(playbook_file, 'w') as fh:
        pyaml.dump(playbook, fh, vspacing=[1,0]) 

# 需要导入模块: import pyaml [as 别名]
# 或者: from pyaml import dump [as 别名]
def yamlstr_to_yaml(output_yamlfile, yamlstr):
    with open(output_yamlfile, 'w') as fw:
        load_data = yaml.load(yamlstr)
        yaml_data = pyaml.dump(load_data)
        fw.write(yaml_data) 

# 需要导入模块: import pyaml [as 别名]
# 或者: from pyaml import dump [as 别名]
def python_to_yaml(output_yamlfile, pydata):
    with open(output_yamlfile, 'w') as fw:
        yaml_data = pyaml.dump(pydata)
        fw.write(yaml_data)