lib/generators/persistence/persistence_generator.rb in amee-data-persistence-2.0.1 vs lib/generators/persistence/persistence_generator.rb in amee-data-persistence-2.0.2

@@ -1,10 +1,13 @@
 require 'rails/generators/migration'
 require 'rails/generators/active_record'
 
 class PersistenceGenerator < Rails::Generators::Base
 
+  # Get method from command line - default is metadata
+  argument :method, :type => :string, :desc => "The storage method to use; everything, metadata, or outputs", :default => 'metadata'
+
   include Rails::Generators::Migration
 
   desc "Generates a migration for the AMEE Organisation models"
   def self.source_root
     @source_root ||= File.dirname(__FILE__) + '/templates'
@@ -19,23 +22,16 @@
     migration_template 'db/migrate/002_add_unit_columns.rb', 'db/migrate/add_unit_columns'
     migration_template 'db/migrate/003_add_value_types.rb', 'db/migrate/add_value_types'
   end
 
   def manifest
-    record do |m|
-      
-      ########################################
-      # persistence level configuration file #
-      ########################################
+    ########################################
+    # persistence level configuration file #
+    ########################################
 
-      # Get method from command line - default is metadata
-      method = args[0] || 'metadata'
-      # Make sure there is a config directory
-      m.directory File.join("config")
-      # Create persistence.yml file
-      m.template File.join("config","persistence.yml.erb"),
-                 File.join("config","persistence.yml"),
-                 :assigns => {:method => method}
+    # Create persistence.yml file
+    template File.join("config","persistence.yml.erb"),
+             File.join("config","persistence.yml"),
+             :assigns => {:method => self.method}
 
-    end
   end
 end
\ No newline at end of file