Class: CrawlStation::Command::Generate

Inherits:
Thor
  • Object
show all
Defined in:
lib/crawl_station/command/generate.rb

Instance Method Summary collapse

Instance Method Details

#migration(args) ⇒ Object 



7
8
9
10
11
12
13
14
15
 
# File 'lib/crawl_station/command/generate.rb', line 7

def migration(args)
  raise "generate #{args} error" if !args.is_a?(Array) || args.size < 2
  module_name, file_name = args.shift, args.shift

  file_path = dest_path(module_name, file_name)
  template_path = utils.template_filepath('generate/migration.erb')
  render(file_path, template_path, class_name: file_name.camelize)
  logs "generate migration #{module_name}:#{file_name} done"
end

#new_module(args) ⇒ Object 



18
19
20
21
22
23
24
25
26
27
 
# File 'lib/crawl_station/command/generate.rb', line 18

def new_module(args)
  raise "create module #{args} error" if !args.is_a?(Array) || args.empty?
  module_name = args.shift
  m_path = utils.module_path(module_name)
  return logs("#{module_name} module exist!") if File.exist?(m_path)
  logs "create new module #{module_name}"
  template_m_path = "#{utils.templates_path}/generate/module"
  FileUtils.copy_entry template_m_path, m_path
  logs "create #{module_name} done"
end

#parser(args) ⇒ Object 



30
31
32
33
34
35
36
37
38
39
40
41
42
 
# File 'lib/crawl_station/command/generate.rb', line 30

def parser(args)
  raise "generate parser #{args} error" if !args.is_a?(Array) || args.size < 2
  module_name, parser_name = args.shift, args.shift
  logs "create #{module_name} parser #{parser_name}"
  template_parser_path = utils.template_filepath('generate/parser.erb')
  template_item_path = utils.template_filepath('generate/item.erb')
  opts = {
    module_class_name: module_name.camelize,
    class_name: parser_name.camelize
  }
  render(parser_path(module_name, parser_name, 'parser'), template_parser_path, opts)
  render(parser_path(module_name, parser_name, 'item'), template_item_path, opts)
end

#parsers(args) ⇒ Object 



45
46
47
48
49
50
51
52
 
# File 'lib/crawl_station/command/generate.rb', line 45

def parsers(args)
  raise "generate parser #{args} error" if !args.is_a?(Array) || args.size < 2
  module_name = args.shift
  loop do
    break if args.empty?
    parser([module_name, args.shift])
  end
end