diff --git a/lib/data_migrate/config.rb b/lib/data_migrate/config.rb index 78ec34d2..67533b7b 100644 --- a/lib/data_migrate/config.rb +++ b/lib/data_migrate/config.rb @@ -12,10 +12,11 @@ def config end class Config - attr_accessor :data_migrations_path, :db_configuration, :spec_name + attr_accessor :data_migrations_gen_path, :data_migrations_path, :db_configuration, :spec_name def initialize @data_migrations_path = "db/data/" + @data_migrations_gen_path = "db/data/" @db_configuration = nil @spec_name = nil end diff --git a/lib/generators/data_migration/data_migration_generator.rb b/lib/generators/data_migration/data_migration_generator.rb index f8279545..ab1546af 100644 --- a/lib/generators/data_migration/data_migration_generator.rb +++ b/lib/generators/data_migration/data_migration_generator.rb @@ -35,7 +35,7 @@ def data_migrations_file_path end def data_migrations_path - DataMigrate.config.data_migrations_path + DataMigrate.config.data_migrations_gen_path || DataMigrate.config.data_migrations_path end end end