Merge pull request #87 from dhellmann/retry-81

Improve error message for IOErrors
python-virtualenvwrapper · Dec 21, 2023 · 0b60b4d · 0b60b4d
2 parents 513621b + 389610e
commit 0b60b4d
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/virtualenvwrapper/hook_loader.py b/virtualenvwrapper/hook_loader.py
@@ -146,14 +146,16 @@ def main():
         log.debug('Saving sourcable %s hooks to %s',
                   hook, options.script_filename)
         options.sourcing = True
-        output = open(options.script_filename, "w")
         try:
-            output.write('# %s\n' % hook)
-            # output.write('echo %s\n' % hook)
-            # output.write('set -x\n')
-            run_hooks(hook + '_source', options, args, output)
-        finally:
-            output.close()
+            with open(options.script_filename, "w") as output:
+                output.write('# %s\n' % hook)
+                # output.write('echo %s\n' % hook)
+                # output.write('set -x\n')
+                run_hooks(hook + '_source', options, args, output)
+        except (IOError, OSError) as e:
+            log.error('Error while writing to %s: \n %s',
+                      options.script_filename, e)
+            sys.exit(1)
 
     return 0