lib/generators/pbw/token/token_generator.rb in pbw-0.0.4 vs lib/generators/pbw/token/token_generator.rb in pbw-0.0.5

@@ -1,3 +1,10 @@
-class Pbw::TokenGenerator < Rails::Generators::NamedBase
+require 'generators/pbw/scaffold_generator'
+
+class Pbw::TokenGenerator < Pbw::Generators::ScaffoldGenerator
   source_root File.expand_path('../templates', __FILE__)
+
+  protected
+	def base_model_class
+		"Pbw::Token"
+	end
 end