--- a/bake/__init__.py Tue Nov 01 09:29:06 2011 +0100
+++ b/bake/__init__.py Tue Nov 01 10:30:41 2011 +0100
@@ -1,4 +1,6 @@
from Dependencies import Dependencies,DependencyUnmet,CycleDetected
from Bake import Bake
from Exceptions import MetadataError
+def main(args):
+ Bake().main(args)
--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/generate-binary.py Tue Nov 01 10:30:41 2011 +0100
@@ -0,0 +1,93 @@
+#!/usr/bin/env python
+
+def collect_source(source):
+ import os
+ sources = []
+ for root, dirs, files in os.walk(source):
+ for f in files:
+ if f.endswith('.py'):
+ sources.append(os.path.join(root, f))
+ return sources
+
+def calculate_hash(sources):
+ import hashlib
+ m = hashlib.md5()
+ for source in sources:
+ f = open(source, 'r')
+ for line in f:
+ m.update(line)
+ f.close()
+ md5hash = m.hexdigest()
+ return md5hash
+
+def generate_zip(sources):
+ import zipfile
+ import tempfile
+ import os
+ (handle, pathname) = tempfile.mkstemp()
+ f = zipfile.ZipFile(pathname, 'w')
+ for source in sources:
+ f.write(source)
+ f.close()
+ os.close(handle)
+ return pathname
+
+def generate_binary(source_dir, output):
+ import base64
+ import StringIO
+ sources = collect_source(source_dir)
+ sources_md5 = calculate_hash(sources)
+ zipfile = generate_zip(sources)
+ zipdata = StringIO.StringIO()
+ base64.encode(open(zipfile, 'r'), zipdata)
+
+ f = open(output, 'w')
+ f.write("""#!/usr/bin/env python
+sources = [%s]
+sources_md5 = "%s"
+zipdata = \"\"\"%s\"\"\"
+
+def decompress(output):
+ import StringIO
+ import zipfile
+ import base64
+ decoded = StringIO.StringIO()
+ base64.decode(StringIO.StringIO(zipdata), decoded)
+ f = zipfile.ZipFile(decoded, 'r')
+ f.extractall(pathname)
+ f.close()
+
+
+import os
+import sys
+pathname = os.path.join('.bake', sources_md5)
+if not os.path.exists(pathname):
+ os.makedirs(pathname)
+ decompress(pathname)
+elif not os.path.isdir(pathname):
+ import tempfile
+ pathname = tempfile.mkdtemp()
+ decompress(pathname)
+sys.path.append(pathname)
+import %s as source
+source.main(sys.argv)
+
+""" % (','.join(["'%s'" % source for source in sources]), sources_md5, zipdata.getvalue(),
+ source_dir))
+ f.close()
+
+if __name__ == '__main__':
+ import optparse
+ parser = optparse.OptionParser()
+ parser.add_option('-s', '--source', dest='source', type='string', action='store', default='bake',
+ help='Source directory to embedd')
+ parser.add_option('-o', '--output', dest='output', type='string', action='store', default='bake.binary',
+ help='Executable to generate')
+ (options, args) = parser.parse_args()
+ generate_binary(options.source, options.output)
+
+
+
+
+
+