Commit cb016d5c by Jack Danger Canty

fixing that the method names weren't preceeded by 'self.'

parent 903ae812
module AnnotateModels module AnnotateModels
class << self
MODEL_DIR = "app/models" MODEL_DIR = "app/models"
FIXTURE_DIR = "test/fixtures" FIXTURE_DIR = "test/fixtures"
PREFIX = "== Schema Information" PREFIX = "== Schema Information"
# Simple quoting for the default column value # Simple quoting for the default column value
def quote(value) def self.quote(value)
case value case value
when NilClass then "NULL" when NilClass then "NULL"
when TrueClass then "TRUE" when TrueClass then "TRUE"
...@@ -21,7 +22,7 @@ module AnnotateModels ...@@ -21,7 +22,7 @@ module AnnotateModels
# to create a comment block containing a line for # to create a comment block containing a line for
# each column. The line contains the column name, # each column. The line contains the column name,
# the type (and length), and any optional attributes # the type (and length), and any optional attributes
def get_schema_info(klass, header) def self.get_schema_info(klass, header)
info = "# #{header}\n#\n" info = "# #{header}\n#\n"
info << "# Table name: #{klass.table_name}\n#\n" info << "# Table name: #{klass.table_name}\n#\n"
...@@ -48,7 +49,7 @@ module AnnotateModels ...@@ -48,7 +49,7 @@ module AnnotateModels
# a schema info block (a comment starting # a schema info block (a comment starting
# with "Schema as of ..."), remove it first. # with "Schema as of ..."), remove it first.
def annotate_one_file(file_name, info_block) def self.annotate_one_file(file_name, info_block)
if File.exist?(file_name) if File.exist?(file_name)
content = File.read(file_name) content = File.read(file_name)
...@@ -65,7 +66,7 @@ module AnnotateModels ...@@ -65,7 +66,7 @@ module AnnotateModels
# on the columns and their types) and put it at the front # on the columns and their types) and put it at the front
# of the model and fixture source files. # of the model and fixture source files.
def annotate(klass, file, header) def self.annotate(klass, file, header)
info = get_schema_info(klass, header) info = get_schema_info(klass, header)
model_file_name = File.join(MODEL_DIR, file) model_file_name = File.join(MODEL_DIR, file)
...@@ -80,7 +81,7 @@ module AnnotateModels ...@@ -80,7 +81,7 @@ module AnnotateModels
# the underscore or CamelCase versions of model names. # the underscore or CamelCase versions of model names.
# Otherwise we take all the model files in the # Otherwise we take all the model files in the
# app/models directory. # app/models directory.
def get_model_files def self.get_model_files
models = ARGV.dup models = ARGV.dup
models.shift models.shift
...@@ -95,7 +96,7 @@ module AnnotateModels ...@@ -95,7 +96,7 @@ module AnnotateModels
# Retrieve the classes belonging to the model names we're asked to process # Retrieve the classes belonging to the model names we're asked to process
# Check for namespaced models in subdirectories as well as models # Check for namespaced models in subdirectories as well as models
# in subdirectories without namespacing. # in subdirectories without namespacing.
def get_model_class(file) def self.get_model_class(file)
model = file.gsub(/\.rb$/, '').camelize model = file.gsub(/\.rb$/, '').camelize
parts = model.split('::') parts = model.split('::')
begin begin
...@@ -109,7 +110,7 @@ module AnnotateModels ...@@ -109,7 +110,7 @@ module AnnotateModels
# ActiveRecord models. If we can find the class, and # ActiveRecord models. If we can find the class, and
# if its a subclass of ActiveRecord::Base, # if its a subclass of ActiveRecord::Base,
# then pas it to the associated block # then pas it to the associated block
def do_annotations def self.do_annotations
header = PREFIX.dup header = PREFIX.dup
version = ActiveRecord::Migrator.current_version rescue 0 version = ActiveRecord::Migrator.current_version rescue 0
if version > 0 if version > 0
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment