Changeset 17:323bb4ed1810 in stamper

Timestamp:

Jul 19, 2014, 9:51:02 AM (10 years ago)

Author:

Borja Lopez <borja@…>

Branch:

default

Phase:

public

Message:

Prevent default overwrite of the destination file on conversion.
Now the script asks you for confirmation before overwriting the
destination if it is a file, exists with an error message & code
if it is a directory.

File:

• bin/stamp2json (modified) (2 diffs)

bin/stamp2json

@@ -2 +2 @@
 
 import json
+import sys
 from argparse import ArgumentParser
-from os.path import expanduser
+from os.path import expanduser, isfile, isdir
 
 
@@ -59 +60 @@
                 'action': action,
             })
+
+    if isfile(args.output):
+        print('[warning] %(out)s already exist'
+              % {'out': args.output})
+        confirm = raw_input('overwrite? (y/n) ')
+        while confirm not in ['y','n','Y','N']:
+            confirm = raw_input('overwrite? (y/n) ')
+        if confirm in ['n', 'N']:
+            print('[warning] exiting without converting')
+            sys.exit()
+
+    elif isdir(args.output):
+        print('[error] %(out)s is a directory, remove it first'
+              % {'out': args.output})
+        sys.exit(1)
+
     with open(args.output, 'w') as stamps_file:
         json.dump(stamps, stamps_file, indent=4)