Как зарегистрировать Thor :: Group в качестве подкоманды с аргументами - PullRequest
Новая
       37

Как зарегистрировать Thor :: Group в качестве подкоманды с аргументами

6 голосов
/ 18 февраля 2012

Этот вопрос возник здесь . Но значительно изменился, когда я узнал больше о Торе.

Я пытаюсь создать подкоманду Thor :: Group, которая принимает аргумент. Как ни странно, это работает, если нет аргументов.

Могу ли я использовать Thor :: Group в качестве подкоманды?

Это работает, когда я печатаю: foo counter

Foo / bin / Foo 

module Foo
  class CLI < Thor
    register(Counter, 'counter', 'counter', 'Count up from the 1.')
  end

  class Counter < Thor::Group
    desc "Prints 1 2"

    def one
      puts 1
    end

    def two
      puts 2
    end

  end

end

Foo::CLI.start

Но это не работает, когда я набираю: foo counter 5 

module Foo
  class CLI < Thor
    register(Counter, 'counter', 'counter <number>', 'Count up from the input.')
  end

  class Counter < Thor::Group
    argument :number, :type => :numeric, :desc => "The number to start counting"
    desc "Prints 2 numbers based on input"

    def one
      puts number + 0
    end

    def two
      puts number + 1
    end

  end


end

Foo::CLI.start

Ответы: counter was called incorrectly. Call as foo counter number

1 Ответ

5 голосов
/ 18 февраля 2012

У меня есть решение. Вместо использования Thor :: Group я использую Invocations

bin / foo выглядит так: 

#!/usr/bin/env ruby

require 'foo'

Foo::CLI.start

lib / cli.rb - регистрирует 'generate' как подзадачу базовой команды, foo: 

module Foo
  class CLI < Thor
    register(Generate, 'generate', 'generate [something]', 'Type foo generate for more help.')
  end
end

lib / generate.rb выглядит так: 

module Foo

  class Generate < Thor

    desc "project [name]", "Prints the project making step"
    def project(name)
      puts "making first project file #{name}"
      invoke :config
      invoke :project_sub
    end

    desc "config [name]", "Prints the config making step"
    def config(name)
      puts "making first config file #{name}"
      invoke :project_sub
    end

    desc "project_sub [name]", "Prints the project_sub making step"
    def project_sub(name)
      puts "making subsystem file #{name}"
    end

    def self.banner(task, namespace = false, subcommand = true)
       task.formatted_usage(self, true, subcommand).split(':').join(' ')
    end

  end

end

Теперь я могу напечатать: foo generate project fred

и он выдаст: 

> making first project file fred
> making first config file fred
> making subsystem file fred

Обратите внимание на переопределение баннера. Это означает, что ввод: foo generate project с недействительными или отсутствующими аргументами даст правильное сообщение справки: 

"project" was called incorrectly. Call as "foo generate project [name]".

в отличие от 

"project" was called incorrectly. Call as "foo project [name]".
...