compile_resources.py

#!/usr/bin/env python
import subprocess
import shutil
import glob
import os
import re
import sys

import logging
import ajenti.compat
import ajenti.log


def check_output(*args, **kwargs):
    try:
        return subprocess.check_output(*args, **kwargs)
    except Exception, e:
        logging.error('Call failed')
        logging.error(' '.join(args[0]))
        logging.error(str(e))


ajenti.log.init()

def compile_coffeescript(inpath):
    outpath = '%s.js' % inpath
    logging.info('Compiling CoffeeScript: %s' % inpath)

    check_output('coffee -o compiler-output -c "%s"' % inpath, shell=True)
    shutil.move(glob.glob('./compiler-output/*.js')[0], outpath)
    shutil.rmtree('compiler-output')


def compile_less(inpath):
    outpath = '%s.css' % inpath
    logging.info('Compiling LESS %s' % inpath)
    out = check_output('lessc "%s" "%s"' % (inpath, outpath), shell=True)
    if out:
        logging.error(out)
    #print subprocess.check_output('recess --compile "%s" > "%s"' % (inpath, outpath), shell=True)

compilers = {
    r'.+\.coffee$': compile_coffeescript,
    r'.+[^i]\.less$': compile_less,
}


def compress_js(inpath):
    outpath = os.path.splitext(inpath)[0] + '.c.js'
    if not do_compress:
        return shutil.copy(inpath, outpath)
    logging.info('Compressing JS: %s' % inpath)
    cmd = 'yui-compressor "%s"' % inpath
    out = check_output(cmd, shell=True)
    open(outpath, 'w').write(out)


def compress_css(inpath):
    outpath = os.path.splitext(inpath)[0] + '.c.css'
    return shutil.copy(inpath, outpath)
    #if not do_compress:
    #   return shutil.copy(inpath, outpath)
    logging.info('Compressing CSS: %s' % inpath)
    check_output('yui-compressor -o "%s" "%s"' % (outpath, inpath), shell=True)

compressors = {
    r'.+[^\.][^mci]\.js$': compress_js,
    r'.+[^\.][^mci]\.css$': compress_css,
}


def traverse(fx):
    plugins_path = './ajenti/plugins'
    for plugin in os.listdir(plugins_path):
        path = os.path.join(plugins_path, plugin, 'content')
        if os.path.exists(path):
            for (dp, dn, fn) in os.walk(path):
                for name in fn:
                    file_path = os.path.join(dp, name)
                    fx(file_path)


def compile(file_path):
    for pattern in compilers:
        if re.match(pattern, file_path):
            compilers[pattern](file_path)


def compress(file_path):
    for pattern in compressors:
        if re.match(pattern, file_path):
            compressors[pattern](file_path)


do_compress = True
if len(sys.argv) > 1 and sys.argv[1] == 'nocompress':
    do_compress = False


traverse(compile)
traverse(compress)