Class: DataSeeder::Config

Inherits:
Object
  • Object
show all
Defined in:
lib/data_seeder/config.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ Config

Returns a new instance of Config.



7
8
9
10
11
12
13
14
 
# File 'lib/data_seeder/config.rb', line 7

def initialize
  @seed_dirs        = ['db/seed'].freeze
  @loaders          = default_loaders
  @verbose          = true
  @is_default       = true
  @indent_level     = 0
  @indent           = ''
end

Instance Attribute Details

#loaders ⇒ Object

Returns the value of attribute loaders.



4
5
6
 
# File 'lib/data_seeder/config.rb', line 4

def loaders
  @loaders
end

#logger ⇒ Object 



16
17
18
19
20
21
22
 
# File 'lib/data_seeder/config.rb', line 16

def logger
  @logger ||= begin
    logger = Logger.new($stdout)
    logger.formatter = ->(severity, datetime, progname, msg) { "#{@indent}#{msg}\n" }
    logger
  end
end

#seed_dirs ⇒ Object

Returns the value of attribute seed_dirs.



3
4
5
 
# File 'lib/data_seeder/config.rb', line 3

def seed_dirs
  @seed_dirs
end

#verbose ⇒ Object

Returns the value of attribute verbose.



3
4
5
 
# File 'lib/data_seeder/config.rb', line 3

def verbose
  @verbose
end

Instance Method Details

#add_loader(ext, loader) ⇒ Object 



42
43
44
 
# File 'lib/data_seeder/config.rb', line 42

def add_loader(ext, loader)
  @loaders[ext] = loader
end

#add_loaders(loaders) ⇒ Object 



38
39
40
 
# File 'lib/data_seeder/config.rb', line 38

def add_loaders(loaders)
  @loaders = @loaders.merge(loaders)
end

#add_seed_dir(seed_dir) ⇒ Object 



54
55
56
57
58
59
60
 
# File 'lib/data_seeder/config.rb', line 54

def add_seed_dir(seed_dir)
  if @seed_dirs.frozen?
    @seed_dirs = [seed_dir]
  else
    @seed_dirs << seed_dir
  end
end

#default_loaders ⇒ Object 



24
25
26
27
28
29
30
31
32
 
# File 'lib/data_seeder/config.rb', line 24

def default_loaders
  {
    'csv'  => Loader::CSV,
    'json' => Loader::JSON,
    'txt'  => Loader::Txt,
    'yaml' => Loader::YAML,
    'yml'  => Loader::YAML,
  }
end

#log_indent(&block) ⇒ Object 



62
63
64
65
66
67
68
69
 
# File 'lib/data_seeder/config.rb', line 62

def log_indent(&block)
  @indent_level += 1
  @indent = '  ' * @indent_level
  yield
ensure
  @indent_level -= 1
  @indent = '  ' * @indent_level
end

#seed_dir ⇒ Object 



50
51
52
 
# File 'lib/data_seeder/config.rb', line 50

def seed_dir
  @seed_dirs.first
end

#seed_dir=(seed_dir) ⇒ Object 



46
47
48
 
# File 'lib/data_seeder/config.rb', line 46

def seed_dir=(seed_dir)
  @seed_dirs = [seed_dir]
end