X-Git-Url: http://club.cc.cmu.edu/~cmccabe/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=names_to_numbers.rb;h=c41c3161d377c6deb86229b2405e2ae9f4d87d09;hb=7c32ed021cf6b47090c0ce879dbf35639e60c544;hp=689e6de61298ff8ce59822602f16df33be8992a0;hpb=48535204969cc0fdb31be8f397b9dcdb8ed86763;p=cmccabe-bin

diff --git a/names_to_numbers.rb b/names_to_numbers.rb
index 689e6de..c41c316 100755
--- a/names_to_numbers.rb
+++ b/names_to_numbers.rb
@@ -19,7 +19,9 @@ class MyOptions
     opts.dry_run = false
     opts.num_digits = 2
     opts.extension = nil
+    opts.starting_number = 1
     $fu_args = { :verbose => true }
+    opts.preserve_names = false
 
     # Fill in $opts values
     parser = OptionParser.new do |myparser|
@@ -38,6 +40,14 @@ class MyOptions
               "The file extension for the files to rename.") do |e|
         opts.extension = e
       end
+      myparser.on("--preserve-names", "-p",
+              "Preserve the names while changing the numbers.") do |e|
+        opts.preserve_names = true
+      end
+      myparser.on("--starting-number NUMBER", "-N",
+              "The starting number (defaults to 1)") do |e|
+        opts.starting_number = e.to_i
+      end
     end
 
     parser.parse!(args)
@@ -67,11 +77,26 @@ def count_files(file)
 end
 
 def get_file_name(num)
-  return sprintf("%0#{$opts.num_digits}d.#{$opts.extension}", num)
+  return sprintf("%0#{$opts.num_digits}d", num)
 end
 
 def rename_files(file)
-  FileUtils.mv(file, get_file_name(1 + $total_files), $fu_args)
+  dst="#{get_file_name($opts.starting_number + $total_files)}.#{$opts.extension}"
+  if (file != dst) then
+    FileUtils.mv(file, dst, $fu_args)
+  end
+  $total_files = $total_files + 1
+end
+
+def rename_files_keep_names(file)
+  proper_file = ""
+  if (file =~ /^[0-9. -]*(.*)$/) then
+    proper_file = $1
+  else
+    raise "can't find proper name for #{file}"
+  end
+  full_name = "#{get_file_name(1 + $total_files)} - #{proper_file}"
+  FileUtils.mv(file, full_name, $fu_args)
   $total_files = $total_files + 1
 end
 
@@ -95,6 +120,10 @@ end
 
 # rename files
 $total_files = 0
-file_iter { |f| rename_files(f) }
+if ( $opts.preserve_names ) then
+  file_iter { |f| rename_files_keep_names(f) }
+else
+  file_iter { |f| rename_files(f) }
+end
 
 exit 0