archive.py

#!/usr/bin/python
# -*- coding: UTF-8 -*-
#
# Copyright © 2014 by Adam Hellberg <adam.hellberg@sharparam.com>.
#

import io
import os
import re
import sys
import json
import fnmatch
import zipfile
from httplib import HTTPConnection
from mimetools import choose_boundary

SCRIPT = sys.argv[0]

if len(sys.argv) < 3:
    print '{0}: expected args: name, build number'.format(SCRIPT)
    sys.exit(1)

name = None
build = None

try:
    name = str(sys.argv[1]).strip()
    build = int(sys.argv[2].strip())
except ValueError:
    print "{0}: expected build number argument of type integer".format(SCRIPT)
    sys.exit(1)

print "{0}: Cleaning build directory".format(SCRIPT)
if os.path.isdir('build'):
    for f in os.listdir('build'):
        fp = os.path.join('build', f)
        try:
            if os.path.isfile(fp):
                os.unlink(fp)
        except Exception, e:
            print "{0}: EXCEPTION: Failed to delete {1}: {2}".format(SCRIPT, fp, e)
elif os.path.isfile('build'):
    os.unlink('build')
    os.makedirs('build')
else:
    os.makedirs('build')
print "{0}: Build directory cleanup completed!".format(SCRIPT)

ignored = ['/.*', SCRIPT]

try:
    if os.path.isfile('.gitignore'):
        print "Reading ignore rules from .gitignore file"
        for line in open('.gitignore').read().splitlines():
            ignored.append(line)
    else:
        print ".gitignore not found, skipping"
except Exception, e:
    print "{0}: failed to read .gitignore: {1}".format(SCRIPT, e)

def is_file_ignored(file):
    for pattern in ignored:
        if os.name == 'nt':
            pattern = pattern.replace('/', '\\')
        if re.search(fnmatch.translate(pattern), file):
            return True
    return False

def make_zip(src, dst):
    ignored.append(dst)
    print "{0}: Zipping {1} into {2}".format(SCRIPT, src, dst)
    zf = zipfile.ZipFile(dst, 'w')
    abs_src = os.path.abspath(src)
    for dirname, subdirs, files in os.walk(src):
        for filename in files:
            absname = os.path.abspath(os.path.join(dirname, filename))
            if not is_file_ignored(absname):
                arcname = os.path.join(name, absname[len(abs_src) + 1:])
                print "{0} -> {1}".format(os.path.join(dirname, filename), arcname)
                zf.write(absname, arcname)
    zf.close()
    print "{0}: DONE! Zip successfully created: {1}!".format(SCRIPT, dst)

zipname = "build/{0}_b{1}.zip".format(name, build)

make_zip('.', zipname)