pyobfuscate-install

#!/usr/bin/env python3
# pyobfuscate - Python source code obfuscator
# 
# Copyright 2004-2007 Peter Astrand <astrand@cendio.se> for Cendio AB
# 
# 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; version 2 of the License. 
# 
# 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.
# 
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.


import os, sys

def get_origin_dir():
    startdir = os.getcwd()
    os.chdir(os.path.dirname(sys.argv[0]))
    origin = os.getcwd()
    os.chdir(startdir)
    return origin


def usage():
    print("""
Install and obfuscate a Python file in one step. 
    
Usage:
pyobfuscate-install [-m <mode>] [pyobfuscate-args] <source> <dest>

    """, file=sys.stderr)
    sys.exit(1)


def main():
    if len(sys.argv) < 2:
        usage()
    
    origin = get_origin_dir()
    mode = 0o755
    rest = []
    next_is_mode = False
    for arg in sys.argv[1:]:
        if next_is_mode:
            mode = int(arg, 8)
            next_is_mode = False
        elif arg == "-m":
            next_is_mode = True
        else:
            rest.append(arg)

    source = rest[-2]
    dest = rest[-1]
    rest = rest[:-2]

    if os.path.isdir(dest):
        dest = os.path.join(dest, os.path.basename(source))

    cmd = [os.path.join(origin, "pyobfuscate")]
    cmd.extend(rest)
    cmd.extend([source, ">", dest])
    cmd = " ".join(cmd)
    print("Calling", cmd, file=sys.stderr)
    retcode = os.system(cmd)
    os.chmod(dest, mode)
    if retcode > 254:
        retcode = 254
    sys.exit(retcode)
    

if __name__ == "__main__":
    main()