Method: Entity#update_model_files

Defined in:
lib/entity.rb

#update_model_files(start_entity, end_model_line) ⇒ Object 



96
97
98
99
100
101
102
103
104
105
106
 
# File 'lib/entity.rb', line 96

def update_model_files(start_entity, end_model_line)
  return if end_model_line.empty?

  polymorphic_end = one_polymorphic_names?(start_entity)

  if polymorphic_end
    ProjectFile.update_line(table.name, 'model', /belongs_to :#{start_entity.name}/, end_model_line)
  else
    ProjectFile.add_belong_line(table.name, end_model_line)
  end
end