Merge pull request #2 from crazyh/master

Fix broken pip compatibility, hardcode python2.7 dependency

Author: typeundefined

output.py

@@ -1,4 +1,4 @@
-#!/usr/bin/env python
+#!/usr/bin/env python2.7
 # -*-mode: python; coding: utf-8 -*-
 #
 # pyobfuscate - Python source code obfuscator
@@ -36,7 +36,43 @@ import re
 import ast
 
 import astor
-from output import BaseWriter, StdoutWriter, FileWriter
+
+import os
+import shutil
+from tempfile import mkstemp
+
+
+class BaseWriter(object):
+    def write(self, raw_str):
+        raise NotImplemented()
+
+    def close(self):
+        pass
+
+
+class StdoutWriter(BaseWriter):
+    def write(self, raw_str):
+        sys.stdout.write(raw_str)
+
+
+class FileWriter(BaseWriter):
+    def __init__(self, file_name):
+        self.target_filename = file_name
+        _, self.temp_filename = mkstemp()
+        self.file_obj = open(self.temp_filename, 'w')
+
+    def write(self, raw_str):
+        self.file_obj.write(raw_str)
+
+    def close(self):
+        self.file_obj.close()
+        try:
+            shutil.copyfile(self.temp_filename, self.target_filename)
+        except IOError:
+            print >> sys.stderr, 'Failed to write to the file. ' \
+                                 'The obfuscated content is left in temporary file %s' % self.temp_filename
+            raise
+        os.unlink(self.temp_filename)
 
 TOKENBLANKS=1
 MAX_STRING_RECURSION = 10

pyobfuscate

@@ -1,4 +1,4 @@
-#!/usr/bin/env python
+#!/usr/bin/env python2.7
 # pyobfuscate - Python source code obfuscator
 # 
 # Copyright 2004-2007 Peter Astrand <astrand@cendio.se> for Cendio AB

pyobfuscate-install

@@ -1,4 +1,4 @@
-#!/usr/bin/env python
+#!/usr/bin/env python2.7
 #
 # Execute the file with the same name as myself minus the extension.
 # Author: Peter Astrand <astrand@cendio.se>

pyobfuscate.py

@@ -1,4 +1,4 @@
-#!/usr/bin/env python
+#!/usr/bin/env python2.7
 
 DESCRIPTION = """\
 Python source code obfuscator