yamldirs command to create yaml file from directory and directory fro…

…m yaml.

docs/conf.py

@@ -48,9 +48,9 @@
 # built documents.
 #
 # The short X.Y version.
-version = '1.1.6'
+version = '1.1.7'
 # The full version, including alpha/beta/rc tags.
-release = '1.1.6'
+release = '1.1.7'
 
 # The language for content autogenerated by Sphinx. Refer to documentation
 # for a list of supported languages.

setup.py

@@ -20,7 +20,7 @@
 Topic :: Software Development :: Libraries
 """
 
-version = '1.1.6'
+version = '1.1.7'
 
 
 class PyTest(TestCommand):
@@ -55,6 +55,11 @@ def run_tests(self):
     classifiers=[line for line in classifiers.split('\n') if line],
     long_description=open('README.rst').read(),
     cmdclass={'test': PyTest},
+    entry_points={
+        'console_scripts': [
+            'yamldirs = yamldirs.yamldirs_cmd:main',
+        ]
+    },
     packages=find_packages(),
     zip_safe=False,
 )

yamldirs/__init__.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 
-__version__ = '1.1.6'
+__version__ = '1.1.7'
 
 from .filemaker import create_files
 

yamldirs/filemaker.py

@@ -120,7 +120,7 @@ def make_file(self, filename, content):
 @contextmanager
 def create_files(filedef, cleanup=True):
     """Contextmanager that creates a directory structure from a yaml
-       descripttion.
+       description.
     """
     cwd = os.getcwd()
     tmpdir = tempfile.mkdtemp()

yamldirs/yamldirs_cmd.py

@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+import os
+import sys
+
+import yaml
+from .filemaker import Filemaker
+
+
+def tree(root):
+    def _tree():
+        for rt, dirs, files in os.walk(root):
+            for name in files:
+                if name.endswith('~'): continue
+                if name.startswith('.'): continue
+                yield os.path.join(rt, name)
+    return list(sorted(_tree()))
+
+
+def directory2yaml(dirname):
+    res = {}
+    for filename in tree(dirname):
+        parts = filename.replace('\\', '/').split('/')
+        cur = res
+        for part in parts[:-1]:
+            cur.setdefault(part, {})
+            cur = cur[part]
+        cur[parts[-1]] = open(filename).read().replace('\r\n', '\n').strip()
+    print yaml.dump(res)
+
+
+def reconstitute_directory(yamlfile):
+    Filemaker(os.getcwd(), open(yamlfile).read())
+
+
+def main():
+    arg = sys.argv[1]
+    if arg.endswith('.yaml'):
+        reconstitute_directory(arg)
+    else:
+        directory2yaml(arg)