First version

Author: raymondbutcher

.gitignore

@@ -0,0 +1,4 @@
+*.pyc
+build/
+dist/
+terrafile.egg-info/

Makefile

@@ -0,0 +1,34 @@
+$(eval NAME := $(shell python setup.py --name))
+$(eval PY_NAME := $(shell python setup.py --name | sed 's/-/_/g'))
+$(eval VERSION := $(shell python setup.py --version))
+
+SDIST := dist/$(NAME)-$(VERSION).tar.gz
+WHEEL := dist/$(PY_NAME)-$(VERSION)-py2.py3-none-any.whl
+
+.PHONY: all
+all: build
+
+$(SDIST):
+	python setup.py sdist
+
+$(WHEEL):
+	python setup.py bdist_wheel
+
+.PHONY: build
+build: $(SDIST) $(WHEEL)
+
+.PHONY: install
+install: $(WHEEL)
+	pip install --user $(WHEEL)
+
+.PHONY: uninstall
+uninstall:
+	pip uninstall $(NAME)
+
+.PHONY: upload
+upload: $(SDIST) $(WHEEL)
+	twine upload $(SDIST) $(WHEEL)
+
+.PHONY: clean
+clean:
+	rm -rf build dist *.egg-info

README.md

@@ -1,2 +1,19 @@
 # python-terrafile
-Manages external Terraform modules
+
+Manages external Terraform modules, controlled by a `Terrafile`.
+
+This is basically a Python version of the tool described at [http://bensnape.com/2016/01/14/terraform-design-patterns-the-terrafile/](http://bensnape.com/2016/01/14/terraform-design-patterns-the-terrafile/)
+
+## Installation
+
+```shell
+pip install terrafile
+```
+
+## Usage
+
+```shell
+terrafile [path]
+```
+
+If `path` is provided, it must be the path to a `Terrafile` file, or a directory containing one. If not provided, it looks for the file in the current working directory.

bin/terrafile

@@ -0,0 +1,2 @@
+#!/bin/sh
+python -m terrafile "$@"

setup.cfg

@@ -0,0 +1,2 @@
+[wheel]
+universal = 1

setup.py

@@ -0,0 +1,22 @@
+#!/usr/bin/env python
+
+from setuptools import setup
+
+setup(
+    name='terrafile',
+    version='0.1',
+    description='Manages external Terraform modules.',
+    author='Raymond Butcher',
+    author_email='[email protected]',
+    url='https://github.com/claranet/python-terrafile',
+    license='MIT License',
+    packages=(
+        'terrafile',
+    ),
+    scripts=(
+        'bin/terrafile',
+    ),
+    install_requires=(
+        'pyyaml',
+    ),
+)

terrafile/__init__.py

@@ -0,0 +1,71 @@
+import errno
+import os
+import shutil
+import subprocess
+import sys
+import yaml
+
+
+def run(*args, **kwargs):
+    proc = subprocess.Popen(args, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, **kwargs)
+    stdout, stderr = proc.communicate()
+    return (stdout, proc.returncode)
+
+
+def get_terrafile_path(path):
+    if os.path.isdir(path):
+        return os.path.join(path, 'Terrafile')
+    else:
+        return path
+
+
+def read_terrafile(path):
+    try:
+        with open(path) as open_file:
+            terrafile = yaml.load(open_file)
+        if not terrafile:
+            raise ValueError('{} is empty'.format(path)) 
+    except IOError as error:
+        sys.stderr.write('Error loading Terrafile: {}\n'.format(error.strerror))
+        sys.exit(1)
+    except ValueError as error:
+        sys.stderr.write('Error loading Terrafile: {}\n'.format(error))
+        sys.exit(1)
+    else:
+        return terrafile
+
+
+def has_git_tag(path, tag):
+    tags = set()
+    if os.path.isdir(path):
+        output, returncode = run('git', 'tag', '--points-at=HEAD', cwd=path)
+        if returncode == 0:
+            tags.update(output.split())
+    return tag in tags
+
+
+def update_modules(path):
+    terrafile_path = get_terrafile_path(path)
+    module_path = os.path.dirname(terrafile_path)
+    module_path_name = os.path.basename(os.path.abspath(module_path))
+
+    terrafile = read_terrafile(terrafile_path)
+
+    for name, repository_details in sorted(terrafile.items()):
+        source = repository_details['source']
+        version = repository_details['version']
+        target = os.path.join(module_path, name)
+
+        # Skip this module if it has already been checked out.
+        if has_git_tag(path=target, tag=version):
+            print('Fetched {}/{}'.format(module_path_name, name))
+            continue
+
+        print('Fetching {}/{}'.format(module_path_name, name))
+
+        # Delete the old directory and clone it from scratch.
+        shutil.rmtree(target, ignore_errors=True)
+        output, returncode = run('git', 'clone', '--branch={}'.format(version), source, target)
+        if returncode != 0:
+            sys.stderr.write(output)
+            sys.exit(returncode)

terrafile/__main__.py

@@ -0,0 +1,11 @@
+import os
+import sys
+
+from terrafile import update_modules
+
+if len(sys.argv) > 1:
+    path = sys.argv[1]
+else:
+    path = os.getcwd()
+
+update_modules(path)