tasks: convert to pytho3/invoke

diff --git a/fabfile.py b/fabfile.py
deleted file mode 100644 (file)
index f5c9300..0000000
--- a/fabfile.py
+++ /dev/null
@@ -1,71 +0,0 @@
-from fabric.api import *
-
-import os
-import glob
-import hashlib
-import tempfile
-import shutil
-
-os.umask(0022)
-env.shell = "/bin/sh -c"
-env.command_prefixes = [ 'export PATH=$HOME/.virtualenvs/hyde/bin:$PATH',
-                         'export VIRTUAL_ENV=$HOME/.virtualenvs/hyde' ]
-
-def _hyde(args):
-    return local('python ../hyde/h %s' % args)
-
-def gen():
-    """Generate dev content"""
-    _hyde('gen')
-
-def serve():
-    """Serve dev content"""
-    _hyde('serve -a 0.0.0.0')
-
-def build():
-    """Build production content"""
-    # Generate the website from scratch
-    local("rm -rf deploy")
-    conf = "site-production.yaml"
-    _hyde('gen -c %s' % conf)
-
-    # Compute hash for media files
-    with lcd("deploy"):
-        for p in [ 'media/js/*.js',
-                   'media/css/*.css' ]:
-            files = glob.glob("%s/%s" % (env.lcwd, p))
-            for f in files:
-                # Compute hash
-                md5 = hashlib.md5()
-                md5.update(file(f).read())
-                md5 = md5.hexdigest()[:8]
-                f = f[len(env.lcwd)+1:]
-                print "[+] MD5 hash for %s is %s" % (f, md5)
-                # New name
-                root, ext = os.path.splitext(f)
-                newname = "%s.%s%s" % (root, md5, ext)
-                # Symlink
-                local("cp %s %s" % (f, newname))
-                # Fix HTML
-                local(r"find . -name '*.html' -type f -print0 | xargs -r0 sed -i "
-                      '"'
-                      r"s@\([\"']\)\([^\"']*\)%s\1@\1\2%s\1@g"
-                      '"' % (f, newname))
-
-    lldpdir = os.getcwd()
-    tempdir = tempfile.mkdtemp()
-    try:
-        with lcd(tempdir):
-            local("git clone %s -b gh-pages ." % lldpdir)
-            local("rsync -ac --exclude=.git %s/deploy/ ." % lldpdir)
-            local("git add .")
-            local("git diff --stat HEAD")
-            answer = prompt("More diff?", default="yes")
-            if answer.lower().startswith("y"):
-                local("git diff --word-diff HEAD")
-            answer = prompt("Keep?", default="yes")
-            if answer.lower().startswith("y"):
-                local('git commit -a -m "Update generated copy of website"')
-                local('git push origin')
-    finally:
-        shutil.rmtree(tempdir)

diff --git a/tasks.py b/tasks.py
new file mode 100644 (file)
index 0000000..ca86056
--- /dev/null
+++ b/tasks.py
@@ -0,0 +1,96 @@
+from invoke import task
+
+import os
+import sys
+import glob
+import hashlib
+import tempfile
+import shutil
+
+os.environ["PATH"] = os.path.expanduser('~/.virtualenvs/hyde/bin') \
+    + os.pathsep + os.environ["PATH"]
+
+
+def confirm(question, default=False):
+    if default:
+        suffix = "Y/n"
+    else:
+        suffix = "y/N"
+    while True:
+        response = input("{0} [{1}] ".format(question, suffix))
+        response = response.lower().strip()  # Normalize
+        # Default
+        if not response:
+            return default
+        if response in ["y", "yes"]:
+            return True
+        if response in ["n", "no"]:
+            return False
+        err = "I didn't understand you. Please specify '(y)es' or '(n)o'."
+        print(err, file=sys.stderr)
+
+
+@task
+def gen(c):
+    """Generate dev content"""
+    c.run('hyde -x gen')
+
+
+@task(post=[gen])
+def regen(c):
+    """Regenerate dev content"""
+    c.run('rm -rf deploy')
+
+
+@task
+def serve(c):
+    """Serve dev content"""
+    c.run('hyde -x serve -a 0.0.0.0')
+
+
+@task
+def build(c):
+    """Build production content"""
+    # Generate the website from scratch
+    c.run("rm -rf deploy")
+    conf = "site-production.yaml"
+    c.run('hyde gen -c %s' % conf)
+
+    # Compute hash for media files
+    with c.cd("deploy"):
+        for p in ['media/js/*.js',
+                  'media/css/*.css']:
+            files = glob.glob("%s/%s" % (c.cwd, p))
+            for f in files:
+                # Compute hash
+                md5 = hashlib.md5()
+                md5.update(open(f, 'rb').read())
+                md5 = md5.hexdigest()[:8]
+                f = f[len(c.cwd)+1:]
+                print("[+] MD5 hash for %s is %s" % (f, md5))
+                # New name
+                root, ext = os.path.splitext(f)
+                newname = "%s.%s%s" % (root, md5, ext)
+                # Symlink
+                c.run("cp %s %s" % (f, newname))
+                # Fix HTML
+                c.run(r"find . -name '*.html' -type f -print0 | xargs -r0 sed -i "
+                      '"'
+                      r"s@\([\"']\)\([^\"']*\)%s\1@\1\2%s\1@g"
+                      '"' % (f, newname))
+
+    lldpdir = os.getcwd()
+    tempdir = tempfile.mkdtemp()
+    try:
+        with c.cd(tempdir):
+            c.run("git clone %s -b gh-pages ." % lldpdir)
+            c.run("rsync -ac --exclude=.git %s/deploy/ ." % lldpdir)
+            c.run("git add .")
+            c.run("git diff --stat HEAD || true", pty=True)
+            if confirm("More diff?", default=True):
+                c.run("git diff --word-diff HEAD || true", pty=True)
+            if confirm("Keep?", default=True):
+                c.run('git commit -a -m "Update generated copy of website"')
+                c.run('git push origin')
+    finally:
+        shutil.rmtree(tempdir)