def testWrite(self):
# Load file without processing includes
data = yaml_load('foo.yml', include=False)
self.assertIsInstance(data['c'], IncludeYamlFile)
self.assertIsInstance(data['d'], IncludeYamlFile)
self.assertIsInstance(data['e'], IncludeYamlFile)
self.assertIsInstance(data['f'], IncludeYamlFile)
# Write data out
yaml_write(self._tmp, data)
self.assertTrue(os.path.isfile(self._tmp))
# Load the newly written file and process includes
# (the results should be as in TestYamlLoad.testInclude)
data = yaml_load(self._tmp)
self.assertEqual(data['a'], 1)
self.assertEqual(data['b'], [1.43, 543.55])
self.assertEqual(data['c'][0], 3.6)
self.assertEqual(data['c'][1], [1, 2, 3])
self.assertEqual(data['c'][2], 'item')
self.assertEqual(data['c'][3], 'other')
# Test that output is exact as input
with open('foo.yml', 'r') as fid:
in_data = fid.read()
with open(self._tmp, 'r') as fid:
out_data = fid.read()
self.assertEqual(in_data, out_data)
def testError(self):
with self.assertRaises(IOError) as e:
yaml_load('unkown.yml')
self.assertIn("No such file or directory: 'unkown.yml'", str(e.exception))
with self.assertRaises(IOError) as e:
yaml_load('foo_error.yml')
self.assertIn("Unknown include file 'unknown.yml' on line 5 of foo_error.yml",
str(e.exception))
def testError(self):
with self.assertRaises(IOError) as e:
yaml_load('unkown.yml')
self.assertIn("No such file or directory: 'unkown.yml'", str(e.exception))
with self.assertRaises(IOError) as e:
yaml_load('foo_error.yml')
self.assertIn("Unknown include file 'unknown.yml' on line 5 of foo_error.yml",
str(e.exception))
def load_config(filename, **kwargs):
"""
Read the config.yml file and create the Translator object.
"""
config = yaml_load(filename, root=MooseDocs.ROOT_DIR)
# Replace 'default' and 'disable' key in Extensions to allow for recursive_update to accept command line
for key in config.get('Extensions', dict()).keys():
if config['Extensions'][key] == 'default':
config['Extensions'][key] = dict()
if config['Extensions'][key] == 'disable':
config['Extensions'][key] = dict(active=False)
# Apply command-line key value pairs
recursive_update(config, kwargs)
extensions = _yaml_load_extensions(config)
reader = _yaml_load_object('Reader', config, DEFAULT_READER)
renderer = _yaml_load_object('Renderer', config, DEFAULT_RENDERER)
content = _yaml_load_content(config, reader.EXTENSIONS)
executioner = _yaml_load_object('Executioner', config, DEFAULT_EXECUTIONER)
translator = _yaml_load_object('Translator', config, DEFAULT_TRANSLATOR,
content, reader, renderer, extensions, executioner)
return translator, config
def testInclude(self):
data = yaml_load('foo.yml')
self.assertEqual(data['a'], 1)
self.assertEqual(data['b'], [1.43, 543.55])
self.assertEqual(data['c'][0], 3.6)
self.assertEqual(data['c'][1], [1,2,3])
self.assertEqual(data['c'][2], 'item')
self.assertEqual(data['c'][3], 'other')
def defaultConfig():
packages = yaml_load(os.path.join(MooseDocs.MOOSE_DIR, 'framework', 'doc', 'packages.yml'))
config = command.CommandExtension.defaultConfig()
config['packages'] = (packages, "A dict of packages by name.")
config['link'] = (r'http://www.mooseframework.org/moose_packages',
"Location of packages.")
return config
def testInclude(self):
data = yaml_load('foo.yml')
self.assertEqual(data['a'], 1)
self.assertEqual(data['b'], [1.43, 543.55])
self.assertEqual(data['c'][0], 3.6)
self.assertEqual(data['c'][1], [1, 2, 3])
self.assertEqual(data['c'][2], 'item')
self.assertEqual(data['c'][3], 'other')
def defaultConfig():
packages = yaml_load(
os.path.join(MooseDocs.MOOSE_DIR, 'framework', 'doc',
'packages.yml'))
config = command.CommandExtension.defaultConfig()
config['packages'] = (packages, "A dict of packages by name.")
config['link'] = (r'http://www.mooseframework.org/moose_packages',
"Location of packages.")
return config
def defaultConfig():
packages_config = yaml_load(
os.path.join(MooseDocs.MOOSE_DIR, 'framework', 'doc',
'packages_config.yml'))
config = command.CommandExtension.defaultConfig()
# Assign a key/value for every item in packages_config.yml
for k, v in packages_config.items():
config[k] = (v, 'Default version for %s' % (k))
return config
def load_config(filename):
"""
Read the config.yml file and create the Translator object.
"""
config = yaml_load(filename, root=MooseDocs.ROOT_DIR)
content = _yaml_load_content(config)
extensions = _yaml_load_extensions(config)
reader = _yaml_load_object('Reader', config, DEFAULT_READER)
renderer = _yaml_load_object('Renderer', config, DEFAULT_RENDERER)
translator = _yaml_load_object('Translator', config, DEFAULT_TRANSLATOR,
content, reader, renderer, extensions)
return translator, config
def _loadYamlFiles(self, filenames):
"""Load the removed/alias yml files"""
content = dict()
if filenames is not None:
for fname in filenames:
yml_file = mooseutils.eval_path(fname)
if not os.path.isdir(yml_file):
yml_file = os.path.join(self.working_dir, yml_file)
if not os.path.isfile(yml_file):
raise NotADirectoryError("YAML file is not a directory: {}".format(fname))
content.update({fname:yaml_load(yml_file)})
return content
def load_config(filename, **kwargs):
"""
Read the config.yml file and create the Translator object.
"""
config = yaml_load(filename, root=MooseDocs.ROOT_DIR)
config.update(kwargs)
extensions = _yaml_load_extensions(config)
reader = _yaml_load_object('Reader', config, DEFAULT_READER)
renderer = _yaml_load_object('Renderer', config, DEFAULT_RENDERER)
content = _yaml_load_content(config, reader.EXTENSIONS)
executioner = _yaml_load_object('Executioner', config, DEFAULT_EXECUTIONER)
translator = _yaml_load_object('Translator', config, DEFAULT_TRANSLATOR,
content, reader, renderer, extensions, executioner)
return translator, config
def load_config(filename, **kwargs):
"""
Read the config.yml file and create the Translator object.
"""
config = yaml_load(filename, root=MooseDocs.ROOT_DIR)
config.update(kwargs)
extensions = _yaml_load_extensions(config)
reader = _yaml_load_object('Reader', config, DEFAULT_READER)
renderer = _yaml_load_object('Renderer', config, DEFAULT_RENDERER)
content = _yaml_load_content(config, reader.EXTENSIONS)
executioner = _yaml_load_object('Executioner', config, DEFAULT_EXECUTIONER)
translator = _yaml_load_object('Translator', config, DEFAULT_TRANSLATOR,
content, reader, renderer, extensions, executioner)
return translator, config
def __init__(self, translator, options, *args, **kwargs):
super(MooseDocsWatcher, self).__init__(*args, **kwargs)
self._options = options
self._translator = translator
self._config = yaml_load(options.config, root=MooseDocs.ROOT_DIR)
# Determine the directories to watch
roots = set()
self._items = common.get_items(self._config.get('Content'))
for root, _ in common.get_files(self._items, self._translator.reader.EXTENSIONS):
roots.add(root)
for root in roots:
self.watch(root, self.build, delay=1)
def __init__(self, translator, options, *args, **kwargs):
super(MooseDocsWatcher, self).__init__(*args, **kwargs)
self._options = options
self._translator = translator
self._config = yaml_load(options.config, root=MooseDocs.ROOT_DIR)
# Determine the directories to watch
roots = set()
self._items = common.get_items(self._config.get('Content'))
for root, _ in common.get_files(self._items, self._translator.reader.EXTENSIONS):
roots.add(root)
for root in roots:
self.watch(root, self.build, delay=1)
def defaultConfig():
packages_config = yaml_load(
os.path.join(MooseDocs.MOOSE_DIR, 'framework', 'doc',
'packages_config.yml'))
config = command.CommandExtension.defaultConfig()
# Assign a key/value for every item in packages_config.yml
for k, v in packages_config.iteritems():
if k != 'moose_packages':
config[k] = (v, 'Default version for %s' % (k))
else:
config[k] = (v, 'MOOSE Environment installer package')
config['link'] = (r'http://www.mooseframework.org/moose_packages',
"Location of packages.")
return config
def defaultConfig():
packages_config = yaml_load(os.path.join(MooseDocs.MOOSE_DIR,
'framework',
'doc',
'packages_config.yml'))
config = command.CommandExtension.defaultConfig()
# Assign a key/value for every item in packages_config.yml
for k, v in packages_config.iteritems():
if k != 'moose_packages':
config[k] = (v, 'Default version for %s' % (k))
else:
config[k] = (v, 'MOOSE Environment installer package')
config['link'] = (r'http://www.mooseframework.org/moose_packages',
"Location of packages.")
return config
def testLoad(self):
data = yaml_load('bar.yml')
self.assertEqual(data[0], 3.6)
self.assertEqual(data[1], [1,2,3])
self.assertEqual(data[2], 'item')
self.assertEqual(data[3], 'other')
def testIncludeWithKey(self):
data = yaml_load('foo.yml')
self.assertEqual(data['d'], 1980)
self.assertEqual(data['e'], ['Edward', 'Bonnie'])
self.assertEqual(data['f'], [1906])
def testLoad(self):
data = yaml_load('bar.yml')
self.assertEqual(data[0], 3.6)
self.assertEqual(data[1], [1, 2, 3])
self.assertEqual(data[2], 'item')
self.assertEqual(data[3], 'other')
def testIncludeWithKey(self):
data = yaml_load('foo.yml')
self.assertEqual(data['d'], 1980)
self.assertEqual(data['e'], ['Edward', 'Bonnie'])
self.assertEqual(data['f'], [1906])
