create_archive.py

#!/usr/bin/env python

"""
Tools for Energy Model Optimization and Analysis (Temoa):
An open source framework for energy systems optimization modeling

Copyright (C) 2015,  NC State University

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

A complete copy of the GNU General Public License v2 (GPLv2) is available
in LICENSE.txt.  Users uncompressing this from an archive may not have
received this license file.  If not, see <http://www.gnu.org/licenses/>.
"""

# This script creates the 'temoa.py' zip archive/executable using Python's
# PyZipFile interface.  It accepts no arguments.

import os
import stat

from zipfile import PyZipFile, ZIP_DEFLATED

# Ensure compatibility with Python 2.7 and 3
try:
    from cStringIO import StringIO
    temoa_pkg = StringIO()
except ImportError:
    from io import BytesIO
    temoa_pkg = BytesIO()

with PyZipFile(temoa_pkg, mode='w', compression=ZIP_DEFLATED) as zf:
    zf.debug = 3
    zf.writepy('temoa_model/')

fname = 'temoa.py'
with open(fname, 'wb') as f:
    f.write(temoa_pkg.getvalue())

os.chmod(fname, stat.S_IRWXU)