class Bootstrap::Generators::ThemedGenerator

def retrieve_columns
  if defined?(ActiveRecord)
    rescue_block ActiveRecord::StatementInvalid do
      @model_name.constantize.columns
    end
  else
    rescue_block do
      @model_name.constantize.fields.map {|c| c[1] }
    end
  end
end