rmmsafe

files/file_version.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 import os, argparse, sys
-from shutil import copyfile
+from shutil import copyfile, copytree
 from datetime import datetime
 
 def get_options():
@@ -12,13 +12,13 @@ def get_options():
         action = "store_false",
         dest = "move",
         default = True,
-        help = "Copy file instead of moving",
+        help = "Copy path instead of moving",
     )
     parser.add_argument(
         '-d',
         action = "store_true",
         dest = "date",
-        help = "Use file's date+time as version number",
+        help = "Use paths date+time as version number",
         default = False
     )
     parser.add_argument(
@@ -105,8 +105,6 @@ def test_existing(name, overwrite):
 
 if __name__ == "__main__":
     opts = get_options()
-    if os.path.isdir(opts.file):
-        raise ValueError("Can not handle directories")
     if opts.date:
         new_name = get_date_name(opts.file)
     else:
@@ -117,4 +115,7 @@ if __name__ == "__main__":
     if opts.move:
         os.rename(opts.file, new_name)
     else:
-        copyfile(opts.file, new_name)
+        if os.path.isdir(opts.file):
+            copytree(opts.file, new_name)
+        else:
+            copyfile(opts.file, new_name)