Minor updates to setup.py

setup.py

@@ -218,8 +218,12 @@ class CleanCommand(Command):
         pass
 
     def run(self):
-        sys.stdout.write("removing %s%s" %(VERSION_FILE, os.linesep))
-        os.remove(VERSION_FILE)
+        sys.stdout.write("removing %s\n" % (VERSION_FILE))
+
+        try:
+            os.remove(VERSION_FILE)
+        except OSError as e:
+            sys.stderr.write("failed to remove file %s: %s\n" % (VERSION_FILE, str(e)))
 
         try:
             remove_tree("build")
@@ -292,13 +296,19 @@ install_data_files = []
 for data_dir in ["magic", "config", "plugins", "modules", "core"]:
     install_data_files.append("%s%s*" % (data_dir, os.path.sep))
 
+# If doing a build or installation, then create a version.py file
+# which defines the current binwalk version. This file is excluded
+# from git in the .gitignore file.
 if 'install' in ' '.join(sys.argv) or 'build' in ' '.join(sys.argv):
-    # Create a version.py file which defines the current binwalk version.
-    # This file is excluded from git in the .gitignore file.
-    sys.stdout.write("creating %s%s" % (VERSION_FILE, os.linesep))
-    with open(VERSION_FILE, "w") as fp:
-        fp.write("# This file has been auto-generated by setup.py\n")
-        fp.write('__version__ = "%s"' % MODULE_VERSION)
+    sys.stdout.write("creating %s\n" % (VERSION_FILE))
+
+    try:
+        with open(VERSION_FILE, "w") as fp:
+            fp.write("# This file has been auto-generated by setup.py\n")
+            fp.write('__version__ = "%s"' % MODULE_VERSION)
+    except IOError as e:
+        sys.stderr.write("failed to create file %s: %s\n" % (VERSION_FILE, str(e)))
+        sys.exit(1)
 
 # Install the module, script, and support files
 setup(