Let add/remove partition accept template_opts

twitter-archive · Jan 12, 2012 · b50a9ff · b50a9ff
1 parent 87c4af9
commit b50a9ff
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/lib/gizzard/commands.rb b/lib/gizzard/commands.rb
@@ -979,8 +979,8 @@ def run
       unless be_quiet
         transformations.each do |transformation, trees|
           puts transformation.inspect
-          puts "Applied to #{trees.length} shards:"
-          trees.keys.sort.each {|f| puts "  #{f.inspect}" }
+          puts "Applied to #{trees.length} shards"
+          #trees.keys.sort.each {|f| puts "  #{f.inspect}" }
         end
         puts ""
       end
@@ -1035,8 +1035,8 @@ def run
       unless be_quiet
         transformations.each do |transformation, trees|
           puts transformation.inspect
-          puts "Applied to #{trees.length} shards:"
-          trees.keys.sort.each {|f| puts "  #{f.inspect}" }
+          puts "Applied to #{trees.length} shards"
+          #trees.keys.sort.each {|f| puts "  #{f.inspect}" }
         end
         puts ""
       end

diff --git a/lib/gizzmo.rb b/lib/gizzmo.rb
@@ -402,6 +402,7 @@ def add_template_opts(subcommand_options, opts)
   'add-partition' => OptionParser.new do |opts|
     opts.banner = "Usage: #{zero} add-partition [options] TEMPLATE ..."
     separators(opts, DOC_STRINGS["add-partition"])
+    add_template_opts subcommand_options, opts
 
     add_scheduler_opts subcommand_options, opts
 
@@ -412,6 +413,7 @@ def add_template_opts(subcommand_options, opts)
   'remove-partition' => OptionParser.new do |opts|
     opts.banner = "Usage: #{zero} remove-partition [options] TEMPLATE ..."
     separators(opts, DOC_STRINGS["remove-partition"])
+    add_template_opts subcommand_options, opts
 
     add_scheduler_opts subcommand_options, opts