base.rb 81.2 KB
Newer Older
1 2 3 4 5
begin
  require 'psych'
rescue LoadError
end

D
Initial  
David Heinemeier Hansson 已提交
6
require 'yaml'
7
require 'set'
J
José Valim 已提交
8
require 'active_support/benchmarkable'
J
Jeremy Kemper 已提交
9
require 'active_support/dependencies'
10
require 'active_support/descendants_tracker'
11
require 'active_support/time'
12
require 'active_support/core_ext/class/attribute'
J
Jeremy Kemper 已提交
13 14
require 'active_support/core_ext/class/attribute_accessors'
require 'active_support/core_ext/class/delegating_attributes'
15
require 'active_support/core_ext/class/attribute'
J
Jeremy Kemper 已提交
16 17 18 19 20
require 'active_support/core_ext/array/extract_options'
require 'active_support/core_ext/hash/deep_merge'
require 'active_support/core_ext/hash/indifferent_access'
require 'active_support/core_ext/hash/slice'
require 'active_support/core_ext/string/behavior'
21
require 'active_support/core_ext/kernel/singleton_class'
22
require 'active_support/core_ext/module/delegation'
23
require 'active_support/core_ext/module/introspection'
24
require 'active_support/core_ext/object/duplicable'
25
require 'active_support/core_ext/object/blank'
26 27
require 'arel'
require 'active_record/errors'
28
require 'active_record/log_subscriber'
D
Initial  
David Heinemeier Hansson 已提交
29 30

module ActiveRecord #:nodoc:
31
  # = Active Record
32
  #
33 34 35
  # Active Record objects don't specify their attributes directly, but rather infer them from
  # the table definition with which they're linked. Adding, removing, and changing attributes
  # and their type is done directly in the database. Any change is instantly reflected in the
36
  # Active Record objects. The mapping that binds a given Active Record class to a certain
37 38
  # database table will happen automatically in most common cases, but can be overwritten for the uncommon ones.
  #
39
  # See the mapping rules in table_name and the full example in link:files/activerecord/README_rdoc.html for more insight.
40
  #
D
Initial  
David Heinemeier Hansson 已提交
41
  # == Creation
42
  #
43 44
  # Active Records accept constructor parameters either in a hash or as a block. The hash
  # method is especially useful when you're receiving the data from somewhere else, like an
45
  # HTTP request. It works like this:
46
  #
47
  #   user = User.new(:name => "David", :occupation => "Code Artist")
D
Initial  
David Heinemeier Hansson 已提交
48
  #   user.name # => "David"
49
  #
D
Initial  
David Heinemeier Hansson 已提交
50
  # You can also use block initialization:
51
  #
D
Initial  
David Heinemeier Hansson 已提交
52 53 54 55
  #   user = User.new do |u|
  #     u.name = "David"
  #     u.occupation = "Code Artist"
  #   end
56
  #
D
Initial  
David Heinemeier Hansson 已提交
57
  # And of course you can just create a bare object and specify the attributes after the fact:
58
  #
D
Initial  
David Heinemeier Hansson 已提交
59 60 61
  #   user = User.new
  #   user.name = "David"
  #   user.occupation = "Code Artist"
62
  #
D
Initial  
David Heinemeier Hansson 已提交
63
  # == Conditions
64
  #
65
  # Conditions can either be specified as a string, array, or hash representing the WHERE-part of an SQL statement.
D
Initial  
David Heinemeier Hansson 已提交
66
  # The array form is to be used when the condition input is tainted and requires sanitization. The string form can
67
  # be used for statements that don't involve tainted data. The hash form works much like the array form, except
68
  # only equality and range is possible. Examples:
69
  #
70
  #   class User < ActiveRecord::Base
D
Initial  
David Heinemeier Hansson 已提交
71
  #     def self.authenticate_unsafely(user_name, password)
72
  #       where("user_name = '#{user_name}' AND password = '#{password}'").first
D
Initial  
David Heinemeier Hansson 已提交
73
  #     end
74
  #
D
Initial  
David Heinemeier Hansson 已提交
75
  #     def self.authenticate_safely(user_name, password)
76
  #       where("user_name = ? AND password = ?", user_name, password).first
D
Initial  
David Heinemeier Hansson 已提交
77
  #     end
78 79
  #
  #     def self.authenticate_safely_simply(user_name, password)
80
  #       where(:user_name => user_name, :password => password).first
81
  #     end
D
Initial  
David Heinemeier Hansson 已提交
82
  #   end
83
  #
84 85
  # The <tt>authenticate_unsafely</tt> method inserts the parameters directly into the query
  # and is thus susceptible to SQL-injection attacks if the <tt>user_name</tt> and +password+
86
  # parameters come directly from an HTTP request. The <tt>authenticate_safely</tt>  and
87 88
  # <tt>authenticate_safely_simply</tt> both will sanitize the <tt>user_name</tt> and +password+
  # before inserting them in the query, which will ensure that an attacker can't escape the
89
  # query and fake the login (or worse).
90
  #
91 92 93
  # When using multiple parameters in the conditions, it can easily become hard to read exactly
  # what the fourth or fifth question mark is supposed to represent. In those cases, you can
  # resort to named bind variables instead. That's done by replacing the question marks with
94
  # symbols and supplying a hash with values for the matching symbol keys:
95
  #
96
  #   Company.where(
97
  #     "id = :id AND name = :name AND division = :division AND created_at > :accounting_date",
98
  #     { :id => 3, :name => "37signals", :division => "First", :accounting_date => '2005-01-01' }
99
  #   ).first
100
  #
101 102 103
  # Similarly, a simple hash without a statement will generate conditions based on equality with the SQL AND
  # operator. For instance:
  #
104 105
  #   Student.where(:first_name => "Harvey", :status => 1)
  #   Student.where(params[:student])
106
  #
107 108
  # A range may be used in the hash to use the SQL BETWEEN operator:
  #
109
  #   Student.where(:grade => 9..12)
110
  #
P
Pratik Naik 已提交
111 112
  # An array may be used in the hash to use the SQL IN operator:
  #
113
  #   Student.where(:grade => [9,11,12])
P
Pratik Naik 已提交
114
  #
115
  # When joining tables, nested hashes or keys written in the form 'table_name.column_name'
116
  # can be used to qualify the table name of a particular condition. For instance:
P
Pratik Naik 已提交
117
  #
118 119
  #   Student.joins(:schools).where(:schools => { :type => 'public' })
  #   Student.joins(:schools).where('schools.type' => 'public' )
P
Pratik Naik 已提交
120
  #
D
Initial  
David Heinemeier Hansson 已提交
121
  # == Overwriting default accessors
122
  #
123 124 125 126
  # All column values are automatically available through basic accessors on the Active Record
  # object, but sometimes you want to specialize this behavior. This can be done by overwriting
  # the default accessors (using the same name as the attribute) and calling
  # <tt>read_attribute(attr_name)</tt> and <tt>write_attribute(attr_name, value)</tt> to actually
127
  # change things.
128
  #
D
Initial  
David Heinemeier Hansson 已提交
129 130
  #   class Song < ActiveRecord::Base
  #     # Uses an integer of seconds to hold the length of the song
131
  #
D
Initial  
David Heinemeier Hansson 已提交
132
  #     def length=(minutes)
133
  #       write_attribute(:length, minutes.to_i * 60)
D
Initial  
David Heinemeier Hansson 已提交
134
  #     end
135
  #
D
Initial  
David Heinemeier Hansson 已提交
136
  #     def length
137
  #       read_attribute(:length) / 60
D
Initial  
David Heinemeier Hansson 已提交
138 139
  #     end
  #   end
140
  #
141
  # You can alternatively use <tt>self[:attribute]=(value)</tt> and <tt>self[:attribute]</tt>
142
  # instead of <tt>write_attribute(:attribute, value)</tt> and <tt>read_attribute(:attribute)</tt>.
143
  #
144 145 146 147
  # == Attribute query methods
  #
  # In addition to the basic accessors, query methods are also automatically available on the Active Record object.
  # Query methods allow you to test whether an attribute value is present.
148
  #
149 150 151 152 153 154 155 156 157
  # For example, an Active Record User with the <tt>name</tt> attribute has a <tt>name?</tt> method that you can call
  # to determine whether the user has a name:
  #
  #   user = User.new(:name => "David")
  #   user.name? # => true
  #
  #   anonymous = User.new(:name => "")
  #   anonymous.name? # => false
  #
158
  # == Accessing attributes before they have been typecasted
159
  #
160 161 162
  # Sometimes you want to be able to read the raw attribute data without having the column-determined
  # typecast run its course first. That can be done by using the <tt><attribute>_before_type_cast</tt>
  # accessors that all attributes have. For example, if your Account model has a <tt>balance</tt> attribute,
163
  # you can call <tt>account.balance_before_type_cast</tt> or <tt>account.id_before_type_cast</tt>.
164
  #
165 166
  # This is especially useful in validation situations where the user might supply a string for an
  # integer field and you want to display the original string back in an error message. Accessing the
167
  # attribute normally would typecast the string to 0, which isn't what you want.
168
  #
169 170
  # == Dynamic attribute-based finders
  #
171 172 173 174 175
  # Dynamic attribute-based finders are a cleaner way of getting (and/or creating) objects
  # by simple queries without turning to SQL. They work by appending the name of an attribute
  # to <tt>find_by_</tt>, <tt>find_last_by_</tt>, or <tt>find_all_by_</tt> and thus produces finders
  # like <tt>Person.find_by_user_name</tt>, <tt>Person.find_all_by_last_name</tt>, and
  # <tt>Payment.find_by_transaction_id</tt>. Instead of writing
176
  # <tt>Person.where(:user_name => user_name).first</tt>, you just do <tt>Person.find_by_user_name(user_name)</tt>.
177
  # And instead of writing <tt>Person.where(:last_name => last_name).all</tt>, you just do
178
  # <tt>Person.find_all_by_last_name(last_name)</tt>.
179
  #
180
  # It's also possible to use multiple attributes in the same find by separating them with "_and_".
181
  #
182
  #  Person.where(:user_name => user_name, :password => password).first
183
  #  Person.find_by_user_name_and_password(user_name, password) # with dynamic finder
184
  #
185
  # It's even possible to call these dynamic finder methods on relations and named scopes.
186 187 188
  #
  #   Payment.order("created_on").find_all_by_amount(50)
  #   Payment.pending.find_last_by_amount(100)
189
  #
190 191 192 193
  # The same dynamic finder style can be used to create the object if it doesn't already exist.
  # This dynamic finder is called with <tt>find_or_create_by_</tt> and will return the object if
  # it already exists and otherwise creates it, then returns it. Protected attributes won't be set
  # unless they are given in a block.
194 195 196
  #
  #   # No 'Summer' tag exists
  #   Tag.find_or_create_by_name("Summer") # equal to Tag.create(:name => "Summer")
197
  #
198 199 200
  #   # Now the 'Summer' tag does exist
  #   Tag.find_or_create_by_name("Summer") # equal to Tag.find_by_name("Summer")
  #
201 202 203
  #   # Now 'Bob' exist and is an 'admin'
  #   User.find_or_create_by_name('Bob', :age => 40) { |u| u.admin = true }
  #
204
  # Use the <tt>find_or_initialize_by_</tt> finder if you want to return a new record without
205
  # saving it first. Protected attributes won't be set unless they are given in a block.
206 207 208
  #
  #   # No 'Winter' tag exists
  #   winter = Tag.find_or_initialize_by_name("Winter")
209
  #   winter.persisted? # false
210
  #
211
  # To find by a subset of the attributes to be used for instantiating a new object, pass a hash instead of
212
  # a list of parameters.
213 214 215
  #
  #   Tag.find_or_create_by_name(:name => "rails", :creator => current_user)
  #
216
  # That will either find an existing tag named "rails", or create a new one while setting the
217
  # user that created it.
218
  #
219 220 221 222
  # Just like <tt>find_by_*</tt>, you can also use <tt>scoped_by_*</tt> to retrieve data. The good thing about
  # using this feature is that the very first time result is returned using <tt>method_missing</tt> technique
  # but after that the method is declared on the class. Henceforth <tt>method_missing</tt> will not be hit.
  #
223
  #  User.scoped_by_user_name('David')
224
  #
225
  # == Saving arrays, hashes, and other non-mappable objects in text columns
226
  #
227
  # Active Record can serialize any object in text columns using YAML. To do so, you must
228
  # specify this with a call to the class method +serialize+.
229 230
  # This makes it possible to store arrays, hashes, and other non-mappable objects without doing
  # any additional work.
231
  #
D
Initial  
David Heinemeier Hansson 已提交
232 233 234
  #   class User < ActiveRecord::Base
  #     serialize :preferences
  #   end
235
  #
236
  #   user = User.create(:preferences => { "background" => "black", "display" => large })
D
Initial  
David Heinemeier Hansson 已提交
237
  #   User.find(user.id).preferences # => { "background" => "black", "display" => large }
238
  #
239
  # You can also specify a class option as the second parameter that'll raise an exception
240
  # if a serialized object is retrieved as a descendant of a class not in the hierarchy.
241
  #
D
Initial  
David Heinemeier Hansson 已提交
242
  #   class User < ActiveRecord::Base
243
  #     serialize :preferences, Hash
D
Initial  
David Heinemeier Hansson 已提交
244
  #   end
245
  #
246
  #   user = User.create(:preferences => %w( one two three ))
D
Initial  
David Heinemeier Hansson 已提交
247
  #   User.find(user.id).preferences    # raises SerializationTypeMismatch
248
  #
249 250 251 252 253 254 255 256 257 258 259
  # When you specify a class option, the default value for that attribute will be a new
  # instance of that class.
  #
  #   class User < ActiveRecord::Base
  #     serialize :preferences, OpenStruct
  #   end
  #
  #   user = User.new
  #   user.preferences.theme_color = "red"
  #
  #
D
Initial  
David Heinemeier Hansson 已提交
260 261
  # == Single table inheritance
  #
262 263
  # Active Record allows inheritance by storing the name of the class in a column that by
  # default is named "type" (can be changed by overwriting <tt>Base.inheritance_column</tt>).
264
  # This means that an inheritance looking like this:
D
Initial  
David Heinemeier Hansson 已提交
265 266 267 268 269 270
  #
  #   class Company < ActiveRecord::Base; end
  #   class Firm < Company; end
  #   class Client < Company; end
  #   class PriorityClient < Client; end
  #
271 272
  # When you do <tt>Firm.create(:name => "37signals")</tt>, this record will be saved in
  # the companies table with type = "Firm". You can then fetch this row again using
273
  # <tt>Company.where(:name => '37signals').first</tt> and it will return a Firm object.
D
Initial  
David Heinemeier Hansson 已提交
274
  #
275 276
  # If you don't have a type column defined in your table, single-table inheritance won't
  # be triggered. In that case, it'll work just like normal subclasses with no special magic
277
  # for differentiating between them or reloading the right type with find.
278
  #
D
Initial  
David Heinemeier Hansson 已提交
279 280
  # Note, all the attributes for all the cases are kept in the same table. Read more:
  # http://www.martinfowler.com/eaaCatalog/singleTableInheritance.html
281
  #
D
Initial  
David Heinemeier Hansson 已提交
282 283
  # == Connection to multiple databases in different models
  #
284 285 286
  # Connections are usually created through ActiveRecord::Base.establish_connection and retrieved
  # by ActiveRecord::Base.connection. All classes inheriting from ActiveRecord::Base will use this
  # connection. But you can also set a class-specific connection. For example, if Course is an
287
  # ActiveRecord::Base, but resides in a different database, you can just say <tt>Course.establish_connection</tt>
P
Pratik Naik 已提交
288
  # and Course and all of its subclasses will use this connection instead.
D
Initial  
David Heinemeier Hansson 已提交
289
  #
290 291
  # This feature is implemented by keeping a connection pool in ActiveRecord::Base that is
  # a Hash indexed by the class. If a connection is requested, the retrieve_connection method
292
  # will go up the class-hierarchy until a connection is found in the connection pool.
D
Initial  
David Heinemeier Hansson 已提交
293 294
  #
  # == Exceptions
295
  #
P
Pratik Naik 已提交
296 297
  # * ActiveRecordError - Generic error class and superclass of all other errors raised by Active Record.
  # * AdapterNotSpecified - The configuration hash used in <tt>establish_connection</tt> didn't include an
D
Initial  
David Heinemeier Hansson 已提交
298
  #   <tt>:adapter</tt> key.
299
  # * AdapterNotFound - The <tt>:adapter</tt> key used in <tt>establish_connection</tt> specified a
300
  #   non-existent adapter
301
  #   (or a bad spelling of an existing one).
302
  # * AssociationTypeMismatch - The object assigned to the association wasn't of the type
303
  #   specified in the association definition.
P
Pratik Naik 已提交
304
  # * SerializationTypeMismatch - The serialized object wasn't of the class specified as the second parameter.
305
  # * ConnectionNotEstablished+ - No connection has been established. Use <tt>establish_connection</tt>
306
  #   before querying.
P
Pratik Naik 已提交
307 308 309 310 311
  # * RecordNotFound - No record responded to the +find+ method. Either the row with the given ID doesn't exist
  #   or the row didn't meet the additional restrictions. Some +find+ calls do not raise this exception to signal
  #   nothing was found, please check its documentation for further details.
  # * StatementInvalid - The database server rejected the SQL statement. The precise error is added in the message.
  # * MultiparameterAssignmentErrors - Collection of errors that occurred during a mass assignment using the
312
  #   <tt>attributes=</tt> method. The +errors+ property of this exception contains an array of
313
  #   AttributeAssignmentError
314
  #   objects that should be inspected to determine which attributes triggered the errors.
315
  # * AttributeAssignmentError - An error occurred while doing a mass assignment through the
316
  #   <tt>attributes=</tt> method.
317
  #   You can inspect the +attribute+ property of the exception object to determine which attribute
318
  #   triggered the error.
319
  #
320
  # *Note*: The attributes listed are class-level attributes (accessible from both the class and instance level).
P
Pratik Naik 已提交
321
  # So it's possible to assign a logger to the class through <tt>Base.logger=</tt> which will then be used by all
D
Initial  
David Heinemeier Hansson 已提交
322 323
  # instances in the current object space.
  class Base
E
Emilio Tagua 已提交
324
    ##
P
Pratik Naik 已提交
325
    # :singleton-method:
326 327
    # Accepts a logger conforming to the interface of Log4r or the default Ruby 1.8+ Logger class,
    # which is then passed on to any new database connections made and which can be retrieved on both
328
    # a class and instance level by calling +logger+.
329
    cattr_accessor :logger, :instance_writer => false
J
Jeremy Kemper 已提交
330

P
Pratik Naik 已提交
331 332
    ##
    # :singleton-method:
P
Pratik Naik 已提交
333 334 335 336
    # Contains the database configuration - as is typically stored in config/database.yml -
    # as a Hash.
    #
    # For example, the following database.yml...
E
Emilio Tagua 已提交
337
    #
P
Pratik Naik 已提交
338 339 340
    #   development:
    #     adapter: sqlite3
    #     database: db/development.sqlite3
E
Emilio Tagua 已提交
341
    #
P
Pratik Naik 已提交
342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357
    #   production:
    #     adapter: sqlite3
    #     database: db/production.sqlite3
    #
    # ...would result in ActiveRecord::Base.configurations to look like this:
    #
    #   {
    #      'development' => {
    #         'adapter'  => 'sqlite3',
    #         'database' => 'db/development.sqlite3'
    #      },
    #      'production' => {
    #         'adapter'  => 'sqlite3',
    #         'database' => 'db/production.sqlite3'
    #      }
    #   }
358
    cattr_accessor :configurations, :instance_writer => false
359 360
    @@configurations = {}

P
Pratik Naik 已提交
361 362
    ##
    # :singleton-method:
363 364 365
    # Accessor for the prefix type that will be prepended to every primary key column name.
    # The options are :table_name and :table_name_with_underscore. If the first is specified,
    # the Product class will look for "productid" instead of "id" as the primary column. If the
366
    # latter is specified, the Product class will look for "product_id" instead of "id". Remember
367
    # that this is a global setting for all Active Records.
368
    cattr_accessor :primary_key_prefix_type, :instance_writer => false
D
Initial  
David Heinemeier Hansson 已提交
369 370
    @@primary_key_prefix_type = nil

P
Pratik Naik 已提交
371 372
    ##
    # :singleton-method:
373 374 375
    # Accessor for the name of the prefix string to prepend to every table name. So if set
    # to "basecamp_", all table names will be named like "basecamp_projects", "basecamp_people",
    # etc. This is a convenient way of creating a namespace for tables in a shared database.
376
    # By default, the prefix is the empty string.
377
    #
378 379
    # If you are organising your models within modules you can add a prefix to the models within
    # a namespace by defining a singleton method in the parent module called table_name_prefix which
380
    # returns your chosen prefix.
381 382
    class_attribute :table_name_prefix, :instance_writer => false
    self.table_name_prefix = ""
D
Initial  
David Heinemeier Hansson 已提交
383

P
Pratik Naik 已提交
384 385
    ##
    # :singleton-method:
D
Initial  
David Heinemeier Hansson 已提交
386 387
    # Works like +table_name_prefix+, but appends instead of prepends (set to "_basecamp" gives "projects_basecamp",
    # "people_basecamp"). By default, the suffix is the empty string.
388 389
    class_attribute :table_name_suffix, :instance_writer => false
    self.table_name_suffix = ""
D
Initial  
David Heinemeier Hansson 已提交
390

P
Pratik Naik 已提交
391 392
    ##
    # :singleton-method:
393
    # Indicates whether table names should be the pluralized versions of the corresponding class names.
P
Pratik Naik 已提交
394
    # If true, the default table name for a Product class will be +products+. If false, it would just be +product+.
D
Initial  
David Heinemeier Hansson 已提交
395
    # See table_name for the full rules on table/class naming. This is true, by default.
396
    cattr_accessor :pluralize_table_names, :instance_writer => false
D
Initial  
David Heinemeier Hansson 已提交
397 398
    @@pluralize_table_names = true

P
Pratik Naik 已提交
399 400
    ##
    # :singleton-method:
401
    # Determines whether to use Time.local (using :local) or Time.utc (using :utc) when pulling
402
    # dates and times from the database. This is set to :local by default.
403
    cattr_accessor :default_timezone, :instance_writer => false
404
    @@default_timezone = :local
405

P
Pratik Naik 已提交
406 407
    ##
    # :singleton-method:
408 409
    # Specifies the format to use when dumping the database schema with Rails'
    # Rakefile.  If :sql, the schema is dumped as (potentially database-
410
    # specific) SQL statements.  If :ruby, the schema is dumped as an
411 412 413
    # ActiveRecord::Schema file which can be loaded into any database that
    # supports migrations.  Use :ruby if you want to have different database
    # adapters for, e.g., your development and test environments.
414
    cattr_accessor :schema_format , :instance_writer => false
415
    @@schema_format = :ruby
416

P
Pratik Naik 已提交
417 418
    ##
    # :singleton-method:
419
    # Specify whether or not to use timestamps for migration versions
420 421 422
    cattr_accessor :timestamped_migrations , :instance_writer => false
    @@timestamped_migrations = true

423
    # Determine whether to store the full constant name including namespace when using STI
424
    class_attribute :store_full_sti_class
425
    self.store_full_sti_class = true
426

427
    # Stores the default scope for the class
428
    class_attribute :default_scoping, :instance_writer => false
429 430
    self.default_scoping = []

J
Jeremy Kemper 已提交
431 432 433 434 435
    # Returns a hash of all the attributes that have been specified for serialization as
    # keys and their class restriction as values.
    class_attribute :serialized_attributes
    self.serialized_attributes = {}

436 437 438
    class_attribute :_attr_readonly, :instance_writer => false
    self._attr_readonly = []

D
Initial  
David Heinemeier Hansson 已提交
439
    class << self # Class methods
440 441
      delegate :find, :first, :first!, :last, :last!, :all, :exists?, :any?, :many?, :to => :scoped
      delegate :destroy, :destroy_all, :delete, :delete_all, :update, :update_all, :to => :scoped
P
Pratik Naik 已提交
442
      delegate :find_each, :find_in_batches, :to => :scoped
S
Sebastian Martinez 已提交
443
      delegate :select, :group, :order, :except, :reorder, :limit, :offset, :joins, :where, :preload, :eager_load, :includes, :from, :lock, :readonly, :having, :create_with, :to => :scoped
444
      delegate :count, :average, :minimum, :maximum, :sum, :calculate, :to => :scoped
445

P
Pratik Naik 已提交
446
      # Executes a custom SQL query against your database and returns all the results.  The results will
447
      # be returned as an array with columns requested encapsulated as attributes of the model you call
P
Pratik Naik 已提交
448 449
      # this method from.  If you call <tt>Product.find_by_sql</tt> then the results will be returned in
      # a Product object with the attributes you specified in the SQL query.
450
      #
451 452
      # If you call a complicated SQL query which spans multiple tables the columns specified by the
      # SELECT will be attributes of the model, whether or not they are columns of the corresponding
453 454
      # table.
      #
P
Pratik Naik 已提交
455
      # The +sql+ parameter is a full SQL query as a string.  It will be called as is, there will be
456 457
      # no database agnostic conversions performed.  This should be a last resort because using, for example,
      # MySQL specific terms will lock you to using that particular database engine or require you to
P
Pratik Naik 已提交
458
      # change your call if you switch engines.
459 460
      #
      # ==== Examples
P
Pratik Naik 已提交
461
      #   # A simple SQL query spanning multiple tables
462 463 464 465 466
      #   Post.find_by_sql "SELECT p.title, c.author FROM posts p, comments c WHERE p.id = c.post_id"
      #   > [#<Post:0x36bff9c @attributes={"title"=>"Ruby Meetup", "first_name"=>"Quentin"}>, ...]
      #
      #   # You can use the same string replacement techniques as you can with ActiveRecord#find
      #   Post.find_by_sql ["SELECT title FROM posts WHERE author = ? AND created > ?", author_id, start_date]
M
mjy 已提交
467
      #   > [#<Post:0x36bff9c @attributes={"title"=>"The Cheap Man Buys Twice"}>, ...]
468 469
      def find_by_sql(sql, binds = [])
        connection.select_all(sanitize_sql(sql), "#{name} Load", binds).collect! { |record| instantiate(record) }
D
Initial  
David Heinemeier Hansson 已提交
470
      end
471

472
      # Creates an object (or multiple objects) and saves it to the database, if validations pass.
473 474 475 476 477 478 479 480
      # The resulting object is returned whether the object was saved successfully to the database or not.
      #
      # The +attributes+ parameter can be either be a Hash or an Array of Hashes.  These Hashes describe the
      # attributes on the objects that are to be created.
      #
      # ==== Examples
      #   # Create a single new object
      #   User.create(:first_name => 'Jamie')
481
      #
482
      #   # Create an Array of new objects
P
Pratik Naik 已提交
483
      #   User.create([{ :first_name => 'Jamie' }, { :first_name => 'Jeremy' }])
484 485 486 487 488 489 490
      #
      #   # Create a single object and pass it into a block to set other attributes.
      #   User.create(:first_name => 'Jamie') do |u|
      #     u.is_admin = false
      #   end
      #
      #   # Creating an Array of new objects using a block, where the block is executed for each object:
P
Pratik Naik 已提交
491
      #   User.create([{ :first_name => 'Jamie' }, { :first_name => 'Jeremy' }]) do |u|
492
      #     u.is_admin = false
493
      #   end
494
      def create(attributes = nil, &block)
495
        if attributes.is_a?(Array)
496
          attributes.collect { |attr| create(attr, &block) }
497 498
        else
          object = new(attributes)
499
          yield(object) if block_given?
500 501 502
          object.save
          object
        end
D
Initial  
David Heinemeier Hansson 已提交
503 504 505
      end

      # Returns the result of an SQL statement that should only include a COUNT(*) in the SELECT part.
506
      # The use of this method should be restricted to complicated SQL queries that can't be executed
507 508
      # using the ActiveRecord::Calculations class methods.  Look into those before using this.
      #
P
Pratik Naik 已提交
509
      # ==== Parameters
510
      #
P
Pratik Naik 已提交
511
      # * +sql+ - An SQL statement which should return a count query from the database, see the example below.
512 513 514
      #
      # ==== Examples
      #
515
      #   Product.count_by_sql "SELECT COUNT(*) FROM sales s, customers c WHERE s.customer_id = c.id"
D
Initial  
David Heinemeier Hansson 已提交
516
      def count_by_sql(sql)
517
        sql = sanitize_conditions(sql)
518
        connection.select_value(sql, "#{name} Count").to_i
D
Initial  
David Heinemeier Hansson 已提交
519
      end
520

N
Neeraj Singh 已提交
521 522
      # Attributes listed as readonly will be used to create a new record but update operations will
      # ignore these fields.
523
      def attr_readonly(*attributes)
524
        self._attr_readonly = Set.new(attributes.map { |a| a.to_s }) + (self._attr_readonly || [])
D
Initial  
David Heinemeier Hansson 已提交
525
      end
526

527 528
      # Returns an array of all the attributes that have been specified as readonly.
      def readonly_attributes
529
        self._attr_readonly
D
Initial  
David Heinemeier Hansson 已提交
530 531
      end

532 533 534
      # If you have an attribute that needs to be saved to the database as an object, and retrieved as the same object,
      # then specify the name of that attribute using this method and it will be handled automatically.
      # The serialization is done through YAML. If +class_name+ is specified, the serialized object must be of that
P
Pratik Naik 已提交
535
      # class on retrieval or SerializationTypeMismatch will be raised.
D
David Heinemeier Hansson 已提交
536
      #
P
Pratik Naik 已提交
537
      # ==== Parameters
D
David Heinemeier Hansson 已提交
538
      #
P
Pratik Naik 已提交
539 540
      # * +attr_name+ - The field name that should be serialized.
      # * +class_name+ - Optional, class name that the object type should be equal to.
D
David Heinemeier Hansson 已提交
541 542 543
      #
      # ==== Example
      #   # Serialize a preferences attribute
544
      #   class User < ActiveRecord::Base
D
David Heinemeier Hansson 已提交
545 546
      #     serialize :preferences
      #   end
D
Initial  
David Heinemeier Hansson 已提交
547
      def serialize(attr_name, class_name = Object)
548 549 550 551 552 553
        coder = if [:load, :dump].all? { |x| class_name.respond_to?(x) }
                  class_name
                else
                  Coders::YAMLColumn.new(class_name)
                end

554 555 556
        # merge new serialized attribute and create new hash to ensure that each class in inheritance hierarchy
        # has its own hash of own serialized attributes
        self.serialized_attributes = serialized_attributes.merge(attr_name.to_s => coder)
D
Initial  
David Heinemeier Hansson 已提交
557
      end
558

559 560
      # Guesses the table name (in forced lower-case) based on the name of the class in the
      # inheritance hierarchy descending directly from ActiveRecord::Base. So if the hierarchy
561
      # looks like: Reply < Message < ActiveRecord::Base, then Message is used
562 563
      # to guess the table name even when called on Reply. The rules used to do the guess
      # are handled by the Inflector class in Active Support, which knows almost all common
564
      # English inflections. You can add new inflections in config/initializers/inflections.rb.
D
Initial  
David Heinemeier Hansson 已提交
565
      #
566
      # Nested classes are given table names prefixed by the singular form of
P
Pratik Naik 已提交
567 568 569
      # the parent's table name. Enclosing modules are not considered.
      #
      # ==== Examples
570 571
      #
      #   class Invoice < ActiveRecord::Base; end;
572 573
      #   file                  class               table_name
      #   invoice.rb            Invoice             invoices
574 575 576 577 578 579 580 581
      #
      #   class Invoice < ActiveRecord::Base; class Lineitem < ActiveRecord::Base; end; end;
      #   file                  class               table_name
      #   invoice.rb            Invoice::Lineitem   invoice_lineitems
      #
      #   module Invoice; class Lineitem < ActiveRecord::Base; end; end;
      #   file                  class               table_name
      #   invoice/lineitem.rb   Invoice::Lineitem   lineitems
D
Initial  
David Heinemeier Hansson 已提交
582
      #
P
Pratik Naik 已提交
583 584
      # Additionally, the class-level +table_name_prefix+ is prepended and the
      # +table_name_suffix+ is appended.  So if you have "myapp_" as a prefix,
585 586 587 588 589
      # the table name guess for an Invoice class becomes "myapp_invoices".
      # Invoice::Lineitem becomes "myapp_invoice_lineitems".
      #
      # You can also overwrite this class method to allow for unguessable
      # links, such as a Mouse class with a link to a "mice" table. Example:
D
Initial  
David Heinemeier Hansson 已提交
590 591
      #
      #   class Mouse < ActiveRecord::Base
592
      #     set_table_name "mice"
D
Initial  
David Heinemeier Hansson 已提交
593
      #   end
594
      def table_name
595 596 597
        reset_table_name
      end

598
      # Returns a quoted version of the table name, used to construct SQL statements.
P
Pratik Naik 已提交
599 600 601 602
      def quoted_table_name
        @quoted_table_name ||= connection.quote_table_name(table_name)
      end

603
      # Computes the table name, (re)sets it internally, and returns it.
D
David Heinemeier Hansson 已提交
604
      def reset_table_name #:nodoc:
605
        self.table_name = compute_table_name
D
Initial  
David Heinemeier Hansson 已提交
606 607
      end

608
      def full_table_name_prefix #:nodoc:
609 610 611
        (parents.detect{ |p| p.respond_to?(:table_name_prefix) } || self).table_name_prefix
      end

612
      # Defines the column name for use with single table inheritance. Use
N
Neeraj Singh 已提交
613
      # <tt>set_inheritance_column</tt> to set a different value.
D
Initial  
David Heinemeier Hansson 已提交
614
      def inheritance_column
A
Aaron Patterson 已提交
615
        @inheritance_column ||= "type"
D
Initial  
David Heinemeier Hansson 已提交
616 617
      end

618 619
      # Lazy-set the sequence name to the connection's default.  This method
      # is only ever called once since set_sequence_name overrides it.
D
David Heinemeier Hansson 已提交
620
      def sequence_name #:nodoc:
621 622 623
        reset_sequence_name
      end

D
David Heinemeier Hansson 已提交
624
      def reset_sequence_name #:nodoc:
625 626 627
        default = connection.default_sequence_name(table_name, primary_key)
        set_sequence_name(default)
        default
628 629
      end

630
      # Sets the table name. If the value is nil or false  then the value returned by the given
N
Neeraj Singh 已提交
631
      # block is used.
632 633 634 635
      #
      #   class Project < ActiveRecord::Base
      #     set_table_name "project"
      #   end
636
      def set_table_name(value = nil, &block)
637
        @quoted_table_name = nil
638
        define_attr_method :table_name, value, &block
639

640
        @arel_table = Arel::Table.new(table_name, arel_engine)
641
        @relation = Relation.new(self, arel_table)
642 643 644 645 646
      end
      alias :table_name= :set_table_name

      # Sets the name of the inheritance column to use to the given value,
      # or (if the value # is nil or false) to the value returned by the
647
      # given block.
648 649 650 651 652 653
      #
      #   class Project < ActiveRecord::Base
      #     set_inheritance_column do
      #       original_inheritance_column + "_id"
      #     end
      #   end
654
      def set_inheritance_column(value = nil, &block)
655 656 657 658
        define_attr_method :inheritance_column, value, &block
      end
      alias :inheritance_column= :set_inheritance_column

659 660
      # Sets the name of the sequence to use when generating ids to the given
      # value, or (if the value is nil or false) to the value returned by the
661 662
      # given block. This is required for Oracle and is useful for any
      # database which relies on sequences for primary key generation.
663
      #
664 665 666 667 668
      # If a sequence name is not explicitly set when using Oracle or Firebird,
      # it will default to the commonly used pattern of: #{table_name}_seq
      #
      # If a sequence name is not explicitly set when using PostgreSQL, it
      # will discover the sequence corresponding to your primary key for you.
669 670 671 672
      #
      #   class Project < ActiveRecord::Base
      #     set_sequence_name "projectseq"   # default would have been "project_seq"
      #   end
673
      def set_sequence_name(value = nil, &block)
674 675 676 677
        define_attr_method :sequence_name, value, &block
      end
      alias :sequence_name= :set_sequence_name

678 679
      # Indicates whether the table associated with this class exists
      def table_exists?
680
        connection.table_exists?(table_name)
681 682
      end

D
Initial  
David Heinemeier Hansson 已提交
683 684
      # Returns an array of column objects for the table associated with this class.
      def columns
A
Aaron Patterson 已提交
685
        connection_pool.columns[table_name]
D
Initial  
David Heinemeier Hansson 已提交
686
      end
687

688
      # Returns a hash of column objects for the table associated with this class.
D
Initial  
David Heinemeier Hansson 已提交
689
      def columns_hash
A
Aaron Patterson 已提交
690
        connection_pool.columns_hash[table_name]
D
Initial  
David Heinemeier Hansson 已提交
691
      end
692

D
David Heinemeier Hansson 已提交
693
      # Returns an array of column names as strings.
694
      def column_names
695
        @column_names ||= columns.map { |column| column.name }
696
      end
D
Initial  
David Heinemeier Hansson 已提交
697

698 699
      # Returns an array of column objects where the primary id, all columns ending in "_id" or "_count",
      # and columns used for single table inheritance have been removed.
D
Initial  
David Heinemeier Hansson 已提交
700
      def content_columns
701
        @content_columns ||= columns.reject { |c| c.primary || c.name =~ /(_id|_count)$/ || c.name == inheritance_column }
D
Initial  
David Heinemeier Hansson 已提交
702 703 704 705
      end

      # Returns a hash of all the methods added to query each of the columns in the table with the name of the method as the key
      # and true as the value. This makes it possible to do O(1) lookups in respond_to? to check if a given method for attribute
706
      # is available.
D
David Heinemeier Hansson 已提交
707
      def column_methods_hash #:nodoc:
708
        @dynamic_methods_hash ||= column_names.inject(Hash.new(false)) do |methods, attr|
709 710 711 712 713
          attr_name = attr.to_s
          methods[attr.to_sym]       = attr_name
          methods["#{attr}=".to_sym] = attr_name
          methods["#{attr}?".to_sym] = attr_name
          methods["#{attr}_before_type_cast".to_sym] = attr_name
D
Initial  
David Heinemeier Hansson 已提交
714 715 716
          methods
        end
      end
717

P
Pratik Naik 已提交
718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743
      # Resets all the cached information about columns, which will cause them
      # to be reloaded on the next request.
      #
      # The most common usage pattern for this method is probably in a migration,
      # when just after creating a table you want to populate it with some default
      # values, eg:
      #
      #  class CreateJobLevels < ActiveRecord::Migration
      #    def self.up
      #      create_table :job_levels do |t|
      #        t.integer :id
      #        t.string :name
      #
      #        t.timestamps
      #      end
      #
      #      JobLevel.reset_column_information
      #      %w{assistant executive manager director}.each do |type|
      #        JobLevel.create(:name => type)
      #      end
      #    end
      #
      #    def self.down
      #      drop_table :job_levels
      #    end
      #  end
744
      def reset_column_information
A
Aaron Patterson 已提交
745
        connection.clear_cache!
746
        undefine_attribute_methods
A
Aaron Patterson 已提交
747 748
        connection_pool.clear_table_cache!(table_name) if table_exists?

749
        @column_names = @content_columns = @dynamic_methods_hash = @inheritance_column = nil
750
        @arel_engine = @relation = nil
751 752
      end

753
      def clear_cache! # :nodoc:
A
Aaron Patterson 已提交
754
        connection_pool.clear_cache!
755 756
      end

757
      def attribute_method?(attribute)
J
José Valim 已提交
758
        super || (table_exists? && column_names.include?(attribute.to_s.sub(/=$/, '')))
759 760
      end

761 762
      # Set the lookup ancestors for ActiveModel.
      def lookup_ancestors #:nodoc:
763 764
        klass = self
        classes = [klass]
765 766
        return classes if klass == ActiveRecord::Base

E
Emilio Tagua 已提交
767
        while klass != klass.base_class
768 769 770 771 772
          classes << klass = klass.superclass
        end
        classes
      end

773 774 775
      # Set the i18n scope to overwrite ActiveModel.
      def i18n_scope #:nodoc:
        :activerecord
D
Initial  
David Heinemeier Hansson 已提交
776
      end
777

778 779 780 781 782 783 784
      # True if this isn't a concrete subclass needing a STI type condition.
      def descends_from_active_record?
        if superclass.abstract_class?
          superclass.descends_from_active_record?
        else
          superclass == Base || !columns_hash.include?(inheritance_column)
        end
D
Initial  
David Heinemeier Hansson 已提交
785 786
      end

787 788 789 790 791
      def finder_needs_type_condition? #:nodoc:
        # This is like this because benchmarking justifies the strange :false stuff
        :true == (@finder_needs_type_condition ||= descends_from_active_record? ? :false : :true)
      end

A
Akira Matsuda 已提交
792
      # Returns a string like 'Post(id:integer, title:string, body:text)'
793
      def inspect
794 795 796 797
        if self == Base
          super
        elsif abstract_class?
          "#{super}(abstract)"
798
        elsif table_exists?
799 800
          attr_list = columns.map { |c| "#{c.name}: #{c.type}" } * ', '
          "#{super}(#{attr_list})"
801 802
        else
          "#{super}(Table doesn't exist)"
803
        end
804 805
      end

806
      def quote_value(value, column = nil) #:nodoc:
807
        connection.quote(value,column)
808 809
      end

810
      # Used to sanitize objects before they're used in an SQL SELECT statement. Delegates to <tt>connection.quote</tt>.
811
      def sanitize(object) #:nodoc:
812
        connection.quote(object)
D
Initial  
David Heinemeier Hansson 已提交
813 814
      end

815 816 817
      # Overwrite the default class equality method to provide support for association proxies.
      def ===(object)
        object.is_a?(self)
818
      end
819

820 821 822 823
      def symbolized_base_class
        @symbolized_base_class ||= base_class.to_s.to_sym
      end

824 825 826
      # Returns the base AR subclass that this class descends from. If A
      # extends AR::Base, A.base_class will return A. If B descends from A
      # through some arbitrarily deep hierarchy, B.base_class will return A.
827 828 829
      #
      # If B < A and C < B and if A is an abstract_class then both B.base_class
      # and C.base_class would return B as the answer since A is an abstract_class.
830 831 832 833
      def base_class
        class_of_active_record_descendant(self)
      end

P
Pratik Naik 已提交
834
      # Set this to true if this is an abstract class (see <tt>abstract_class?</tt>).
835 836
      attr_accessor :abstract_class

837
      # Returns whether this class is an abstract class or not.
838
      def abstract_class?
839
        defined?(@abstract_class) && @abstract_class == true
840 841
      end

842
      def respond_to?(method_id, include_private = false)
843 844
        if match = DynamicFinderMatch.match(method_id)
          return true if all_attributes_exists?(match.attribute_names)
845 846
        elsif match = DynamicScopeMatch.match(method_id)
          return true if all_attributes_exists?(match.attribute_names)
847
        end
848

849 850 851
        super
      end

852 853 854 855
      def sti_name
        store_full_sti_class ? name : name.demodulize
      end

P
Pratik Naik 已提交
856
      def arel_table
A
Aaron Patterson 已提交
857
        Arel::Table.new(table_name, arel_engine)
858 859
      end

860 861
      def arel_engine
        @arel_engine ||= begin
862
          if self == ActiveRecord::Base
863
            ActiveRecord::Base
864
          else
865
            connection_handler.connection_pools[name] ? self : superclass.arel_engine
866 867
          end
        end
868 869
      end

870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885
      # Returns a scope for this class without taking into account the default_scope.
      #
      #   class Post < ActiveRecord::Base
      #     default_scope :published => true
      #   end
      #
      #   Post.all          # Fires "SELECT * FROM posts WHERE published = true"
      #   Post.unscoped.all # Fires "SELECT * FROM posts"
      #
      # This method also accepts a block meaning that all queries inside the block will
      # not use the default_scope:
      #
      #   Post.unscoped {
      #     limit(10) # Fires "SELECT * FROM posts LIMIT 10"
      #   }
      #
886 887
      # It is recommended to use block form of unscoped because chaining unscoped with <tt>scope</tt>
      # does not work. Assuming that <tt>published</tt> is a <tt>scope</tt> following two statements are same.
888
      #
889 890
      # Post.unscoped.published
      # Post.published
891
      def unscoped #:nodoc:
892 893 894 895 896 897 898 899
        block_given? ? relation.scoping { yield } : relation
      end

      def scoped_methods #:nodoc:
        key = :"#{self}_scoped_methods"
        Thread.current[key] = Thread.current[key].presence || self.default_scoping.dup
      end

900 901 902 903
      def before_remove_const #:nodoc:
        reset_scoped_methods
      end

J
Jon Leighton 已提交
904 905 906 907 908 909 910 911 912
      # Specifies how the record is loaded by +Marshal+.
      #
      # +_load+ sets an instance variable for each key in the hash it takes as input.
      # Override this method if you require more complex marshalling.
      def _load(data)
        record = allocate
        record.init_with(Marshal.load(data))
        record
      end
J
José Valim 已提交
913 914


915 916 917
      # Finder methods must instantiate through this method to work with the
      # single-table inheritance model that makes it possible to create
      # objects of different types from the same table.
918 919 920
      def instantiate(record)
        sti_class = find_sti_class(record[inheritance_column])
        record_id = sti_class.primary_key && record[sti_class.primary_key]
M
Marcin Raczkowski 已提交
921

922 923 924 925 926 927
        if ActiveRecord::IdentityMap.enabled? && record_id
          if (column = sti_class.columns_hash[sti_class.primary_key]) && column.number?
            record_id = record_id.to_i
          end
          if instance = IdentityMap.get(sti_class, record_id)
            instance.reinit_with('attributes' => record)
928
          else
M
Marcin Raczkowski 已提交
929
            instance = sti_class.allocate.init_with('attributes' => record)
930
            IdentityMap.add(instance)
931
          end
932 933 934 935 936
        else
          instance = sti_class.allocate.init_with('attributes' => record)
        end

        instance
937
      end
938

D
Initial  
David Heinemeier Hansson 已提交
939
      private
J
José Valim 已提交
940 941 942

        def relation #:nodoc:
          @relation ||= Relation.new(self, arel_table)
943 944 945 946 947 948

          if finder_needs_type_condition?
            @relation.where(type_condition).create_with(inheritance_column.to_sym => sti_name)
          else
            @relation
          end
D
Initial  
David Heinemeier Hansson 已提交
949
        end
950

951
        def find_sti_class(type_name)
952
          if type_name.blank? || !columns_hash.include?(inheritance_column)
953 954 955
            self
          else
            begin
956 957 958 959 960
              if store_full_sti_class
                ActiveSupport::Dependencies.constantize(type_name)
              else
                compute_type(type_name)
              end
961 962 963 964 965 966 967 968 969 970
            rescue NameError
              raise SubclassNotFound,
                "The single-table inheritance mechanism failed to locate the subclass: '#{type_name}'. " +
                "This error is raised because the column '#{inheritance_column}' is reserved for storing the class in case of inheritance. " +
                "Please rename this column if you didn't intend it to be used for storing the inheritance class " +
                "or overwrite #{name}.inheritance_column to use another column for that information."
            end
          end
        end

971
        def construct_finder_arel(options = {}, scope = nil)
972
          relation = options.is_a?(Hash) ? unscoped.apply_finder_options(options) : options
973
          relation = scope.merge(relation) if scope
974
          relation
975 976
        end

977 978
        def type_condition(table = arel_table)
          sti_column = table[inheritance_column.to_sym]
979
          sti_names  = ([self] + descendants).map { |model| model.sti_name }
980

981
          sti_column.in(sti_names)
D
Initial  
David Heinemeier Hansson 已提交
982 983 984
        end

        # Guesses the table name, but does not decorate it with prefix and suffix information.
985
        def undecorated_table_name(class_name = base_class.name)
986 987
          table_name = class_name.to_s.demodulize.underscore
          table_name = table_name.pluralize if pluralize_table_names
988
          table_name
D
Initial  
David Heinemeier Hansson 已提交
989 990
        end

991 992 993 994 995 996 997 998
        # Computes and returns a table name according to default conventions.
        def compute_table_name
          base = base_class
          if self == base
            # Nested classes are prefixed with singular parent table name.
            if parent < ActiveRecord::Base && !parent.abstract_class?
              contained = parent.table_name
              contained = contained.singularize if parent.pluralize_table_names
999
              contained += '_'
1000 1001 1002 1003 1004 1005 1006 1007
            end
            "#{full_table_name_prefix}#{contained}#{undecorated_table_name(name)}#{table_name_suffix}"
          else
            # STI subclasses always use their superclass' table.
            base.table_name
          end
        end

1008
        # Enables dynamic finders like <tt>User.find_by_user_name(user_name)</tt> and
1009 1010
        # <tt>User.scoped_by_user_name(user_name). Refer to Dynamic attribute-based finders
        # section at the top of this file for more detailed information.
1011
        #
1012
        # It's even possible to use all the additional parameters to +find+. For example, the
1013
        # full interface for +find_all_by_amount+ is actually <tt>find_all_by_amount(amount, options)</tt>.
1014
        #
1015
        # Each dynamic finder using <tt>scoped_by_*</tt> is also defined in the class after it
1016
        # is first invoked, so that future attempts to use it do not run through method_missing.
1017
        def method_missing(method_id, *arguments, &block)
1018 1019
          if match = DynamicFinderMatch.match(method_id)
            attribute_names = match.attribute_names
1020
            super unless all_attributes_exists?(attribute_names)
1021
            if match.finder?
1022
              options = arguments.extract_options!
1023
              relation = options.any? ? construct_finder_arel(options, current_scoped_methods) : scoped
1024
              relation.send :find_by_attributes, match, attribute_names, *arguments
1025
            elsif match.instantiator?
1026
              scoped.send :find_or_instantiator_by_attributes, match, attribute_names, *arguments, &block
1027
            end
1028 1029 1030 1031
          elsif match = DynamicScopeMatch.match(method_id)
            attribute_names = match.attribute_names
            super unless all_attributes_exists?(attribute_names)
            if match.scope?
1032
              self.class_eval <<-METHOD, __FILE__, __LINE__ + 1
1033 1034 1035 1036 1037
                def self.#{method_id}(*args)                                    # def self.scoped_by_user_name_and_password(*args)
                  attributes = Hash[[:#{attribute_names.join(',:')}].zip(args)] #   attributes = Hash[[:user_name, :password].zip(args)]
                                                                                #
                  scoped(:conditions => attributes)                             #   scoped(:conditions => attributes)
                end                                                             # end
1038
              METHOD
1039 1040
              send(method_id, *arguments)
            end
1041 1042 1043 1044
          else
            super
          end
        end
D
Initial  
David Heinemeier Hansson 已提交
1045

1046 1047
        # Similar in purpose to +expand_hash_conditions_for_aggregates+.
        def expand_attribute_names_for_aggregates(attribute_names)
1048
          attribute_names.map { |attribute_name|
1049
            unless (aggregation = reflect_on_aggregation(attribute_name.to_sym)).nil?
1050 1051
              aggregate_mapping(aggregation).map do |field_attr, _|
                field_attr.to_sym
1052 1053
              end
            else
1054
              attribute_name.to_sym
1055
            end
1056
          }.flatten
1057 1058
        end

1059
        def all_attributes_exists?(attribute_names)
1060 1061
          (expand_attribute_names_for_aggregates(attribute_names) -
           column_methods_hash.keys).empty?
1062
        end
1063

D
Initial  
David Heinemeier Hansson 已提交
1064
      protected
N
Neeraj Singh 已提交
1065 1066
        # with_scope lets you apply options to inner block incrementally. It takes a hash and the keys must be
        # <tt>:find</tt> or <tt>:create</tt>. <tt>:find</tt> parameter is <tt>Relation</tt> while
1067
        # <tt>:create</tt> parameters are an attributes hash.
1068 1069 1070
        #
        #   class Article < ActiveRecord::Base
        #     def self.create_with_scope
1071
        #       with_scope(:find => where(:blog_id => 1), :create => { :blog_id => 1 }) do
1072 1073 1074 1075 1076 1077 1078
        #         find(1) # => SELECT * from articles WHERE blog_id = 1 AND id = 1
        #         a = create(1)
        #         a.blog_id # => 1
        #       end
        #     end
        #   end
        #
D
David Heinemeier Hansson 已提交
1079
        # In nested scopings, all previous parameters are overwritten by the innermost rule, with the exception of
1080
        # <tt>where</tt>, <tt>includes</tt>, and <tt>joins</tt> operations in <tt>Relation</tt>, which are merged.
P
Pratik Naik 已提交
1081
        #
1082
        # <tt>joins</tt> operations are uniqued so multiple scopes can join in the same table without table aliasing
P
Pratik Naik 已提交
1083 1084
        # problems.  If you need to join multiple tables, but still want one of the tables to be uniqued, use the
        # array of strings format for your joins.
1085 1086 1087
        #
        #   class Article < ActiveRecord::Base
        #     def self.find_with_scope
1088 1089 1090
        #       with_scope(:find => where(:blog_id => 1).limit(1), :create => { :blog_id => 1 }) do
        #         with_scope(:find => limit(10)) do
        #           all # => SELECT * from articles WHERE blog_id = 1 LIMIT 10
1091
        #         end
1092 1093
        #         with_scope(:find => where(:author_id => 3)) do
        #           all # => SELECT * from articles WHERE blog_id = 1 AND author_id = 3 LIMIT 1
1094 1095 1096 1097 1098
        #         end
        #       end
        #     end
        #   end
        #
1099
        # You can ignore any previous scopings by using the <tt>with_exclusive_scope</tt> method.
1100 1101 1102
        #
        #   class Article < ActiveRecord::Base
        #     def self.find_with_exclusive_scope
1103
        #       with_scope(:find => where(:blog_id => 1).limit(1)) do
1104
        #         with_exclusive_scope(:find => limit(10)) do
1105
        #           all # => SELECT * from articles LIMIT 10
1106 1107 1108 1109
        #         end
        #       end
        #     end
        #   end
P
Pratik Naik 已提交
1110
        #
N
Neeraj Singh 已提交
1111
        # *Note*: the +:find+ scope also has effect on update and deletion methods, like +update_all+ and +delete_all+.
1112 1113 1114
        def with_scope(method_scoping = {}, action = :merge, &block)
          method_scoping = method_scoping.method_scoping if method_scoping.respond_to?(:method_scoping)

1115 1116
          if method_scoping.is_a?(Hash)
            # Dup first and second level of hash (method and params).
1117 1118 1119
            method_scoping = method_scoping.dup
            method_scoping.each do |method, params|
              method_scoping[method] = params.dup unless params == true
1120
            end
1121

1122 1123 1124 1125
            method_scoping.assert_valid_keys([ :find, :create ])
            relation = construct_finder_arel(method_scoping[:find] || {})

            if current_scoped_methods && current_scoped_methods.create_with_value && method_scoping[:create]
1126
              scope_for_create = if action == :merge
1127 1128 1129
                current_scoped_methods.create_with_value.merge(method_scoping[:create])
              else
                method_scoping[:create]
1130
              end
1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141

              relation = relation.create_with(scope_for_create)
            else
              scope_for_create = method_scoping[:create]
              scope_for_create ||= current_scoped_methods.create_with_value if current_scoped_methods
              relation = relation.create_with(scope_for_create) if scope_for_create
            end

            method_scoping = relation
          end

1142
          method_scoping = current_scoped_methods.merge(method_scoping) if current_scoped_methods && action ==  :merge
1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153

          self.scoped_methods << method_scoping
          begin
            yield
          ensure
            self.scoped_methods.pop
          end
        end

        # Works like with_scope, but discards any nested properties.
        def with_exclusive_scope(method_scoping = {}, &block)
1154
          if method_scoping.values.any? { |e| e.is_a?(ActiveRecord::Relation) }
1155 1156
            raise ArgumentError, <<-MSG
New finder API can not be used with_exclusive_scope. You can either call unscoped to get an anonymous scope not bound to the default_scope:
1157

1158 1159 1160 1161 1162 1163 1164 1165 1166
  User.unscoped.where(:active => true)

Or call unscoped with a block:

  User.unscoped do
    User.where(:active => true).all
  end

MSG
1167
          end
1168 1169 1170
          with_scope(method_scoping, :overwrite, &block)
        end

1171
        # Sets the default options for the model. The format of the
1172
        # <tt>options</tt> argument is the same as in find.
1173 1174
        #
        #   class Person < ActiveRecord::Base
1175
        #     default_scope order('last_name, first_name')
1176
        #   end
1177 1178 1179 1180 1181 1182 1183 1184
        #
        # <tt>default_scope</tt> is also applied while creating/building a record. It is not
        # applied while updating a record.
        #
        #   class Article < ActiveRecord::Base
        #     default_scope where(:published => true)
        #   end
        #
1185 1186
        #   Article.new.published    # => true
        #   Article.create.published # => true
1187
        def default_scope(options = {})
1188
          reset_scoped_methods
1189 1190
          default_scoping = self.default_scoping.dup
          self.default_scoping = default_scoping << construct_finder_arel(options, default_scoping.pop)
1191 1192
        end

D
David Heinemeier Hansson 已提交
1193
        def current_scoped_methods #:nodoc:
T
Tim Morgan 已提交
1194 1195
          method = scoped_methods.last
          if method.respond_to?(:call)
1196
            relation.scoping { method.call }
T
Tim Morgan 已提交
1197 1198 1199
          else
            method
          end
1200
        end
1201

1202 1203 1204 1205
        def reset_scoped_methods #:nodoc:
          Thread.current[:"#{self}_scoped_methods"] = nil
        end

P
Pratik Naik 已提交
1206
        # Returns the class type of the record using the current module as a prefix. So descendants of
1207
        # MyApp::Business::Account would appear as MyApp::Business::AccountSubclass.
D
Initial  
David Heinemeier Hansson 已提交
1208
        def compute_type(type_name)
1209 1210 1211
          if type_name.match(/^::/)
            # If the type is prefixed with a scope operator then we assume that
            # the type_name is an absolute reference.
1212
            ActiveSupport::Dependencies.constantize(type_name)
1213 1214 1215 1216 1217 1218 1219 1220
          else
            # Build a list of candidates to search for
            candidates = []
            name.scan(/::|$/) { candidates.unshift "#{$`}::#{type_name}" }
            candidates << type_name

            candidates.each do |candidate|
              begin
1221
                constant = ActiveSupport::Dependencies.constantize(candidate)
1222
                return constant if candidate == constant.to_s
1223 1224 1225
              rescue NameError => e
                # We don't want to swallow NoMethodError < NameError errors
                raise e unless e.instance_of?(NameError)
1226 1227
              rescue ArgumentError
              end
1228
            end
1229 1230

            raise NameError, "uninitialized constant #{candidates.first}"
D
Initial  
David Heinemeier Hansson 已提交
1231 1232 1233
          end
        end

P
Pratik Naik 已提交
1234 1235
        # Returns the class descending directly from ActiveRecord::Base or an
        # abstract class, if any, in the inheritance hierarchy.
1236
        def class_of_active_record_descendant(klass)
1237
          if klass.superclass == Base || klass.superclass.abstract_class?
1238
            klass
D
Initial  
David Heinemeier Hansson 已提交
1239 1240 1241
          elsif klass.superclass.nil?
            raise ActiveRecordError, "#{name} doesn't belong in a hierarchy descending from ActiveRecord"
          else
1242
            class_of_active_record_descendant(klass.superclass)
D
Initial  
David Heinemeier Hansson 已提交
1243 1244 1245
          end
        end

P
Pratik Naik 已提交
1246
        # Accepts an array, hash, or string of SQL conditions and sanitizes
1247
        # them into a valid SQL fragment for a WHERE clause.
1248 1249 1250
        #   ["name='%s' and group_id='%s'", "foo'bar", 4]  returns  "name='foo''bar' and group_id='4'"
        #   { :name => "foo'bar", :group_id => 4 }  returns "name='foo''bar' and group_id='4'"
        #   "name='foo''bar' and group_id='4'" returns "name='foo''bar' and group_id='4'"
1251
        def sanitize_sql_for_conditions(condition, table_name = self.table_name)
1252 1253
          return nil if condition.blank?

1254 1255
          case condition
            when Array; sanitize_sql_array(condition)
1256
            when Hash;  sanitize_sql_hash_for_conditions(condition, table_name)
1257 1258
            else        condition
          end
1259
        end
1260
        alias_method :sanitize_sql, :sanitize_sql_for_conditions
1261

P
Pratik Naik 已提交
1262
        # Accepts an array, hash, or string of SQL conditions and sanitizes
1263 1264 1265 1266 1267 1268 1269 1270 1271 1272
        # them into a valid SQL fragment for a SET clause.
        #   { :name => nil, :group_id => 4 }  returns "name = NULL , group_id='4'"
        def sanitize_sql_for_assignment(assignments)
          case assignments
            when Array; sanitize_sql_array(assignments)
            when Hash;  sanitize_sql_hash_for_assignment(assignments)
            else        assignments
          end
        end

1273 1274 1275 1276 1277
        def aggregate_mapping(reflection)
          mapping = reflection.options[:mapping] || [reflection.name, reflection.name]
          mapping.first.is_a?(Array) ? mapping : [mapping]
        end

P
Pratik Naik 已提交
1278
        # Accepts a hash of SQL conditions and replaces those attributes
1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307
        # that correspond to a +composed_of+ relationship with their expanded
        # aggregate attribute values.
        # Given:
        #     class Person < ActiveRecord::Base
        #       composed_of :address, :class_name => "Address",
        #         :mapping => [%w(address_street street), %w(address_city city)]
        #     end
        # Then:
        #     { :address => Address.new("813 abc st.", "chicago") }
        #       # => { :address_street => "813 abc st.", :address_city => "chicago" }
        def expand_hash_conditions_for_aggregates(attrs)
          expanded_attrs = {}
          attrs.each do |attr, value|
            unless (aggregation = reflect_on_aggregation(attr.to_sym)).nil?
              mapping = aggregate_mapping(aggregation)
              mapping.each do |field_attr, aggregate_attr|
                if mapping.size == 1 && !value.respond_to?(aggregate_attr)
                  expanded_attrs[field_attr] = value
                else
                  expanded_attrs[field_attr] = value.send(aggregate_attr)
                end
              end
            else
              expanded_attrs[attr] = value
            end
          end
          expanded_attrs
        end

1308
        # Sanitizes a hash of attribute/value pairs into SQL conditions for a WHERE clause.
1309 1310 1311 1312
        #   { :name => "foo'bar", :group_id => 4 }
        #     # => "name='foo''bar' and group_id= 4"
        #   { :status => nil, :group_id => [1,2,3] }
        #     # => "status IS NULL and group_id IN (1,2,3)"
1313 1314
        #   { :age => 13..18 }
        #     # => "age BETWEEN 13 AND 18"
1315 1316
        #   { 'other_records.id' => 7 }
        #     # => "`other_records`.`id` = 7"
1317 1318
        #   { :other_records => { :id => 7 } }
        #     # => "`other_records`.`id` = 7"
1319 1320 1321
        # And for value objects on a composed_of relationship:
        #   { :address => Address.new("123 abc st.", "chicago") }
        #     # => "address_street='123 abc st.' and address_city='chicago'"
1322
        def sanitize_sql_hash_for_conditions(attrs, default_table_name = self.table_name)
1323 1324
          attrs = expand_hash_conditions_for_aggregates(attrs)

1325
          table = Arel::Table.new(table_name).alias(default_table_name)
1326 1327 1328 1329
          viz = Arel::Visitors.for(arel_engine)
          PredicateBuilder.build_from_hash(arel_engine, attrs, table).map { |b|
            viz.accept b
          }.join(' AND ')
1330
        end
1331 1332 1333 1334 1335 1336
        alias_method :sanitize_sql_hash, :sanitize_sql_hash_for_conditions

        # Sanitizes a hash of attribute/value pairs into SQL conditions for a SET clause.
        #   { :status => nil, :group_id => 1 }
        #     # => "status = NULL , group_id = 1"
        def sanitize_sql_hash_for_assignment(attrs)
1337
          attrs.map do |attr, value|
1338 1339 1340
            "#{connection.quote_column_name(attr)} = #{quote_bound_value(value)}"
          end.join(', ')
        end
1341

1342
        # Accepts an array of conditions.  The array has each value
P
Pratik Naik 已提交
1343
        # sanitized and interpolated into the SQL statement.
1344
        #   ["name='%s' and group_id='%s'", "foo'bar", 4]  returns  "name='foo''bar' and group_id='4'"
1345
        def sanitize_sql_array(ary)
1346
          statement, *values = ary
1347
          if values.first.is_a?(Hash) && statement =~ /:\w+/
1348 1349
            replace_named_bind_variables(statement, values.first)
          elsif statement.include?('?')
1350
            replace_bind_variables(statement, values)
1351 1352
          elsif statement.blank?
            statement
1353
          else
1354
            statement % values.collect { |value| connection.quote_string(value.to_s) }
1355
          end
1356 1357
        end

1358 1359
        alias_method :sanitize_conditions, :sanitize_sql

D
David Heinemeier Hansson 已提交
1360
        def replace_bind_variables(statement, values) #:nodoc:
1361
          raise_if_bind_arity_mismatch(statement, statement.count('?'), values.size)
1362
          bound = values.dup
C
Comron Sattari 已提交
1363 1364
          c = connection
          statement.gsub('?') { quote_bound_value(bound.shift, c) }
1365 1366
        end

D
David Heinemeier Hansson 已提交
1367
        def replace_named_bind_variables(statement, bind_vars) #:nodoc:
1368 1369 1370 1371
          statement.gsub(/(:?):([a-zA-Z]\w*)/) do
            if $1 == ':' # skip postgresql casts
              $& # return the whole match
            elsif bind_vars.include?(match = $2.to_sym)
1372
              quote_bound_value(bind_vars[match])
1373 1374
            else
              raise PreparedStatementInvalid, "missing value for :#{match} in #{statement}"
1375 1376
            end
          end
1377 1378
        end

1379
        def expand_range_bind_variables(bind_vars) #:nodoc:
1380 1381 1382
          expanded = []

          bind_vars.each do |var|
1383 1384
            next if var.is_a?(Hash)

1385
            if var.is_a?(Range)
1386 1387
              expanded << var.first
              expanded << var.last
1388
            else
1389
              expanded << var
1390
            end
1391
          end
1392 1393

          expanded
1394 1395
        end

C
Comron Sattari 已提交
1396
        def quote_bound_value(value, c = connection) #:nodoc:
1397
          if value.respond_to?(:map) && !value.acts_like?(:string)
1398
            if value.respond_to?(:empty?) && value.empty?
C
Comron Sattari 已提交
1399
              c.quote(nil)
1400
            else
C
Comron Sattari 已提交
1401
              value.map { |v| c.quote(v) }.join(',')
1402
            end
1403
          else
C
Comron Sattari 已提交
1404
            c.quote(value)
1405 1406 1407
          end
        end

D
David Heinemeier Hansson 已提交
1408
        def raise_if_bind_arity_mismatch(statement, expected, provided) #:nodoc:
1409 1410 1411
          unless expected == provided
            raise PreparedStatementInvalid, "wrong number of bind variables (#{provided} for #{expected}) in: #{statement}"
          end
1412
        end
1413

D
David Heinemeier Hansson 已提交
1414
        def encode_quoted_value(value) #:nodoc:
1415
          quoted_value = connection.quote(value)
1416 1417
          quoted_value = "'#{quoted_value[1..-2].gsub(/\'/, "\\\\'")}'" if quoted_value.include?("\\\'") # (for ruby mode) "
          quoted_value
1418
        end
D
Initial  
David Heinemeier Hansson 已提交
1419 1420 1421 1422 1423
    end

    public
      # New objects can be instantiated as either empty (pass no construction parameter) or pre-set with
      # attributes but not yet saved (pass a hash with key names matching the associated table column names).
1424
      # In both instances, valid attribute keys are determined by the column names of the associated table --
D
Initial  
David Heinemeier Hansson 已提交
1425 1426
      # hence you can't have attributes that aren't part of the table columns.
      def initialize(attributes = nil)
1427
        @attributes = attributes_from_column_definition
1428
        @association_cache = {}
1429
        @aggregation_cache = {}
1430
        @attributes_cache = {}
1431
        @new_record = true
1432
        @readonly = false
1433 1434
        @destroyed = false
        @marked_for_destruction = false
1435 1436
        @previously_changed = {}
        @changed_attributes = {}
1437

D
Initial  
David Heinemeier Hansson 已提交
1438
        ensure_proper_type
1439
        set_serialized_attributes
P
Pratik Naik 已提交
1440

1441
        populate_with_current_scope_attributes
1442
        self.attributes = attributes unless attributes.nil?
P
Pratik Naik 已提交
1443

1444
        result = yield self if block_given?
1445
        run_callbacks :initialize
1446
        result
D
Initial  
David Heinemeier Hansson 已提交
1447
      end
1448

1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464
      # Populate +coder+ with attributes about this record that should be
      # serialized.  The structure of +coder+ defined in this method is
      # guaranteed to match the structure of +coder+ passed to the +init_with+
      # method.
      #
      # Example:
      #
      #   class Post < ActiveRecord::Base
      #   end
      #   coder = {}
      #   Post.new.encode_with(coder)
      #   coder # => { 'id' => nil, ... }
      def encode_with(coder)
        coder['attributes'] = attributes
      end

1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476
      # Initialize an empty model object from +coder+.  +coder+ must contain
      # the attributes necessary for initializing an empty model object.  For
      # example:
      #
      #   class Post < ActiveRecord::Base
      #   end
      #
      #   post = Post.allocate
      #   post.init_with('attributes' => { 'title' => 'hello world' })
      #   post.title # => 'hello world'
      def init_with(coder)
        @attributes = coder['attributes']
1477

1478
        set_serialized_attributes
1479

1480
        @attributes_cache, @previously_changed, @changed_attributes = {}, {}, {}
1481
        @association_cache = {}
1482
        @aggregation_cache = {}
1483
        @readonly = @destroyed = @marked_for_destruction = false
1484
        @new_record = false
1485 1486
        run_callbacks :find
        run_callbacks :initialize
M
Marcin Raczkowski 已提交
1487 1488

        self
1489 1490
      end

J
Jon Leighton 已提交
1491 1492 1493 1494 1495 1496 1497 1498 1499 1500
      # Specifies how the record is dumped by +Marshal+.
      #
      # +_dump+ emits a marshalled hash which has been passed to +encode_with+. Override this
      # method if you require more complex marshalling.
      def _dump(level)
        dump = {}
        encode_with(dump)
        Marshal.dump(dump)
      end

P
Pratik Naik 已提交
1501 1502 1503 1504
      # Returns a String, which Action Pack uses for constructing an URL to this
      # object. The default implementation returns this record's id as a String,
      # or nil if this record's unsaved.
      #
P
Pratik Naik 已提交
1505
      # For example, suppose that you have a User model, and that you have a
1506
      # <tt>resources :users</tt> route. Normally, +user_path+ will
P
Pratik Naik 已提交
1507
      # construct a path with the user object's 'id' in it:
P
Pratik Naik 已提交
1508 1509
      #
      #   user = User.find_by_name('Phusion')
1510
      #   user_path(user)  # => "/users/1"
P
Pratik Naik 已提交
1511
      #
P
Pratik Naik 已提交
1512 1513
      # You can override +to_param+ in your model to make +user_path+ construct
      # a path using the user's name instead of the user's id:
P
Pratik Naik 已提交
1514 1515 1516 1517 1518 1519
      #
      #   class User < ActiveRecord::Base
      #     def to_param  # overridden
      #       name
      #     end
      #   end
E
Emilio Tagua 已提交
1520
      #
P
Pratik Naik 已提交
1521
      #   user = User.find_by_name('Phusion')
1522
      #   user_path(user)  # => "/users/Phusion"
1523
      def to_param
1524
        # We can't use alias_method here, because method 'id' optimizes itself on the fly.
1525
        id && id.to_s # Be sure to stringify the id for routes
1526
      end
1527

P
Pratik Naik 已提交
1528 1529 1530
      # Returns a cache key that can be used to identify this record.
      #
      # ==== Examples
1531 1532 1533 1534 1535
      #
      #   Product.new.cache_key     # => "products/new"
      #   Product.find(5).cache_key # => "products/5" (updated_at not available)
      #   Person.find(5).cache_key  # => "people/5-20071224150000" (updated_at available)
      def cache_key
1536
        case
1537
        when new_record?
J
Jeremy Kemper 已提交
1538 1539 1540
          "#{self.class.model_name.cache_key}/new"
        when timestamp = self[:updated_at]
          "#{self.class.model_name.cache_key}/#{id}-#{timestamp.to_s(:number)}"
1541
        else
J
Jeremy Kemper 已提交
1542
          "#{self.class.model_name.cache_key}/#{id}"
1543 1544
        end
      end
1545

1546
      def quoted_id #:nodoc:
1547
        quote_value(id, column_for_attribute(self.class.primary_key))
1548
      end
1549

1550 1551 1552 1553 1554
      # Returns true if the given attribute is in the attributes hash
      def has_attribute?(attr_name)
        @attributes.has_key?(attr_name.to_s)
      end

1555
      # Returns an array of names for the attributes available on this object.
1556
      def attribute_names
1557
        @attributes.keys
D
Initial  
David Heinemeier Hansson 已提交
1558 1559 1560
      end

      # Allows you to set all the attributes at once by passing in a hash with keys
P
Pratik Naik 已提交
1561 1562 1563 1564 1565 1566
      # matching the attribute names (which again matches the column names).
      #
      # If +guard_protected_attributes+ is true (the default), then sensitive
      # attributes can be protected from this form of mass-assignment by using
      # the +attr_protected+ macro. Or you can alternatively specify which
      # attributes *can* be accessed with the +attr_accessible+ macro. Then all the
1567
      # attributes not included in that won't be allowed to be mass-assigned.
P
Pratik Naik 已提交
1568 1569 1570 1571
      #
      #   class User < ActiveRecord::Base
      #     attr_protected :is_admin
      #   end
E
Emilio Tagua 已提交
1572
      #
P
Pratik Naik 已提交
1573 1574 1575 1576
      #   user = User.new
      #   user.attributes = { :username => 'Phusion', :is_admin => true }
      #   user.username   # => "Phusion"
      #   user.is_admin?  # => false
E
Emilio Tagua 已提交
1577
      #
P
Pratik Naik 已提交
1578 1579
      #   user.send(:attributes=, { :username => 'Phusion', :is_admin => true }, false)
      #   user.is_admin?  # => true
1580
      def attributes=(new_attributes, guard_protected_attributes = true)
P
Pratik Naik 已提交
1581
        return unless new_attributes.is_a?(Hash)
1582
        attributes = new_attributes.stringify_keys
D
Initial  
David Heinemeier Hansson 已提交
1583 1584

        multi_parameter_attributes = []
1585
        attributes = sanitize_for_mass_assignment(attributes) if guard_protected_attributes
1586

1587
        attributes.each do |k, v|
1588 1589
          if k.include?("(")
            multi_parameter_attributes << [ k, v ]
J
José Valim 已提交
1590 1591
          elsif respond_to?("#{k}=")
            send("#{k}=", v)
1592
          else
J
José Valim 已提交
1593
            raise(UnknownAttributeError, "unknown attribute: #{k}")
1594
          end
D
Initial  
David Heinemeier Hansson 已提交
1595
        end
D
David Heinemeier Hansson 已提交
1596

D
Initial  
David Heinemeier Hansson 已提交
1597 1598 1599
        assign_multiparameter_attributes(multi_parameter_attributes)
      end

1600
      # Returns a hash of all the attributes with their names as keys and the values of the attributes as values.
1601
      def attributes
1602
        Hash[@attributes.map { |name, _| [name, read_attribute(name)] }]
1603 1604
      end

P
Pratik Naik 已提交
1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617
      # Returns an <tt>#inspect</tt>-like string for the value of the
      # attribute +attr_name+. String attributes are elided after 50
      # characters, and Date and Time attributes are returned in the
      # <tt>:db</tt> format. Other attributes return the value of
      # <tt>#inspect</tt> without modification.
      #
      #   person = Person.create!(:name => "David Heinemeier Hansson " * 3)
      #
      #   person.attribute_for_inspect(:name)
      #   # => '"David Heinemeier Hansson David Heinemeier Hansson D..."'
      #
      #   person.attribute_for_inspect(:created_at)
      #   # => '"2009-01-12 04:48:57"'
1618 1619 1620
      def attribute_for_inspect(attr_name)
        value = read_attribute(attr_name)

1621
        if value.is_a?(String) && value.length > 50
1622
          "#{value[0..50]}...".inspect
1623 1624
        elsif value.is_a?(Date) || value.is_a?(Time)
          %("#{value.to_s(:db)}")
1625 1626 1627 1628 1629
        else
          value.inspect
        end
      end

D
Initial  
David Heinemeier Hansson 已提交
1630
      # Returns true if the specified +attribute+ has been set by the user or by a database load and is neither
1631
      # nil nor empty? (the latter only applies to objects that respond to empty?, most notably Strings).
D
Initial  
David Heinemeier Hansson 已提交
1632
      def attribute_present?(attribute)
1633
        !_read_attribute(attribute).blank?
D
Initial  
David Heinemeier Hansson 已提交
1634 1635 1636 1637
      end

      # Returns the column object for the named attribute.
      def column_for_attribute(name)
1638
        self.class.columns_hash[name.to_s]
D
Initial  
David Heinemeier Hansson 已提交
1639
      end
1640

1641
      # Returns true if +comparison_object+ is the same exact object, or +comparison_object+
X
Xavier Noria 已提交
1642 1643 1644 1645 1646 1647 1648 1649
      # is of the same type and +self+ has an ID and it is equal to +comparison_object.id+.
      #
      # Note that new records are different from any other record by definition, unless the
      # other record is the receiver itself. Besides, if you fetch existing records with
      # +select+ and leave the ID out, you're on your own, this predicate will return false.
      #
      # Note also that destroying a record preserves its ID in the model instance, so deleted
      # models are still comparable.
D
Initial  
David Heinemeier Hansson 已提交
1650
      def ==(comparison_object)
1651
        comparison_object.equal?(self) ||
1652 1653 1654
          comparison_object.instance_of?(self.class) &&
          id.present? &&
          comparison_object.id == id
D
Initial  
David Heinemeier Hansson 已提交
1655 1656 1657 1658
      end

      # Delegates to ==
      def eql?(comparison_object)
E
Emilio Tagua 已提交
1659
        self == comparison_object
D
Initial  
David Heinemeier Hansson 已提交
1660
      end
1661

D
Initial  
David Heinemeier Hansson 已提交
1662 1663 1664
      # Delegates to id in order to allow two records of the same type and id to work with something like:
      #   [ Person.find(1), Person.find(2), Person.find(3) ] & [ Person.find(1), Person.find(4) ] # => [ Person.find(1) ]
      def hash
1665
        id.hash
D
Initial  
David Heinemeier Hansson 已提交
1666 1667
      end

1668
      # Freeze the attributes hash such that associations are still accessible, even on destroyed records.
1669
      def freeze
1670
        @attributes.freeze; self
1671
      end
1672

1673
      # Returns +true+ if the attributes hash has been frozen.
1674 1675 1676
      def frozen?
        @attributes.frozen?
      end
1677

A
Aaron Patterson 已提交
1678 1679 1680 1681 1682 1683 1684 1685 1686
      # Backport dup from 1.9 so that initialize_dup() gets called
      unless Object.respond_to?(:initialize_dup)
        def dup # :nodoc:
          copy = super
          copy.initialize_dup(self)
          copy
        end
      end

A
Aaron Patterson 已提交
1687
      # Duped objects have no id assigned and are treated as new records. Note
A
Aaron Patterson 已提交
1688 1689
      # that this is a "shallow" copy as it copies the object's attributes
      # only, not its associations. The extent of a "deep" copy is application
A
Aaron Patterson 已提交
1690 1691
      # specific and is therefore left to the application to implement according
      # to its need.
1692
      # The dup method does not preserve the timestamps (created|updated)_(at|on).
A
Aaron Patterson 已提交
1693
      def initialize_dup(other)
A
Aaron Patterson 已提交
1694 1695 1696
        cloned_attributes = other.clone_attributes(:read_attribute_before_type_cast)
        cloned_attributes.delete(self.class.primary_key)

A
Aaron Patterson 已提交
1697 1698
        @attributes = cloned_attributes

A
Aaron Patterson 已提交
1699 1700
        _run_after_initialize_callbacks if respond_to?(:_run_after_initialize_callbacks)

A
Aaron Patterson 已提交
1701 1702 1703 1704
        @changed_attributes = {}
        attributes_from_column_definition.each do |attr, orig_value|
          @changed_attributes[attr] = orig_value if field_changed?(attr, orig_value, @attributes[attr])
        end
A
Aaron Patterson 已提交
1705

1706
        @aggregation_cache = {}
1707 1708
        @association_cache = {}
        @attributes_cache = {}
1709
        @new_record  = true
A
Aaron Patterson 已提交
1710

A
Aaron Patterson 已提交
1711 1712
        ensure_proper_type
        populate_with_current_scope_attributes
1713
        clear_timestamp_attributes
1714 1715
      end

1716 1717
      # Returns +true+ if the record is read only. Records loaded through joins with piggy-back
      # attributes will be marked as read only since they cannot be saved.
1718
      def readonly?
1719
        @readonly
1720 1721
      end

1722 1723
      # Marks this record as read only.
      def readonly!
1724 1725
        @readonly = true
      end
1726

1727
      # Returns the contents of the record as a nicely formatted string.
1728
      def inspect
1729
        attributes_as_nice_string = self.class.column_names.collect { |name|
1730
          if has_attribute?(name)
1731 1732 1733
            "#{name}: #{attribute_for_inspect(name)}"
          end
        }.compact.join(", ")
1734
        "#<#{self.class} #{attributes_as_nice_string}>"
1735
      end
1736

1737 1738
    protected
      def clone_attributes(reader_method = :read_attribute, attributes = {})
1739 1740
        attribute_names.each do |name|
          attributes[name] = clone_attribute_value(reader_method, name)
1741
        end
1742
        attributes
1743 1744 1745 1746 1747 1748 1749 1750 1751
      end

      def clone_attribute_value(reader_method, attribute_name)
        value = send(reader_method, attribute_name)
        value.duplicable? ? value.clone : value
      rescue TypeError, NoMethodError
        value
      end

D
Initial  
David Heinemeier Hansson 已提交
1752 1753
    private

A
Aaron Patterson 已提交
1754 1755 1756 1757 1758 1759 1760
      def set_serialized_attributes
        (@attributes.keys & self.class.serialized_attributes.keys).each do |key|
          coder = self.class.serialized_attributes[key]
          @attributes[key] = coder.load @attributes[key]
        end
      end

1761
      # Sets the attribute used for single table inheritance to this class name if this is not the
1762
      # ActiveRecord::Base descendant.
1763 1764
      # Considering the hierarchy Reply < Message < ActiveRecord::Base, this makes it possible to
      # do Reply.new without having to set <tt>Reply[Reply.inheritance_column] = "Reply"</tt> yourself.
1765
      # No such attribute would be set for objects of the Message class in that example.
D
Initial  
David Heinemeier Hansson 已提交
1766 1767
      def ensure_proper_type
        unless self.class.descends_from_active_record?
1768
          write_attribute(self.class.inheritance_column, self.class.sti_name)
D
Initial  
David Heinemeier Hansson 已提交
1769 1770 1771
        end
      end

1772
      # The primary key and inheritance column can never be set by mass-assignment for security reasons.
1773 1774 1775
      def self.attributes_protected_by_default
        default = [ primary_key, inheritance_column ]
        default << 'id' unless primary_key.eql? 'id'
1776
        default
1777 1778
      end

1779 1780
      # Returns a copy of the attributes hash where all the values have been safely quoted for use in
      # an Arel insert/update method.
1781
      def arel_attributes_values(include_primary_key = true, include_readonly_attributes = true, attribute_names = @attributes.keys)
1782 1783 1784 1785
        attrs      = {}
        klass      = self.class
        arel_table = klass.arel_table

1786 1787 1788 1789
        attribute_names.each do |name|
          if (column = column_for_attribute(name)) && (include_primary_key || !column.primary)

            if include_readonly_attributes || (!include_readonly_attributes && !self.class.readonly_attributes.include?(name))
1790

1791 1792
              value = if coder = klass.serialized_attributes[name]
                        coder.dump @attributes[name]
1793 1794 1795 1796 1797 1798 1799 1800
                      else
                        # FIXME: we need @attributes to be used consistently.
                        # If the values stored in @attributes were already type
                        # casted, this code could be simplified
                        read_attribute(name)
                      end

              attrs[arel_table[name]] = value
1801 1802 1803 1804 1805 1806
            end
          end
        end
        attrs
      end

D
Initial  
David Heinemeier Hansson 已提交
1807
      # Quote strings appropriately for SQL statements.
1808
      def quote_value(value, column = nil)
1809
        self.class.connection.quote(value, column)
D
Initial  
David Heinemeier Hansson 已提交
1810 1811 1812 1813 1814 1815
      end

      # Instantiates objects for all attribute classes that needs more than one constructor parameter. This is done
      # by calling new on the column type or aggregation type (through composed_of) object with these parameters.
      # So having the pairs written_on(1) = "2004", written_on(2) = "6", written_on(3) = "24", will instantiate
      # written_on (a date type) with Date.new("2004", "6", "24"). You can also specify a typecast character in the
1816 1817
      # parentheses to have the parameters typecasted before they're used in the constructor. Use i for Fixnum,
      # f for Float, s for String, and a for Array. If all the values for a given attribute are empty, the
1818
      # attribute will be set to nil.
D
Initial  
David Heinemeier Hansson 已提交
1819 1820 1821 1822 1823
      def assign_multiparameter_attributes(pairs)
        execute_callstack_for_multiparameter_attributes(
          extract_callstack_for_multiparameter_attributes(pairs)
        )
      end
1824

1825
      def instantiate_time_object(name, values)
1826
        if self.class.send(:create_time_zone_conversion_attribute?, name, column_for_attribute(name))
1827
          Time.zone.local(*values)
1828
        else
1829
          Time.time_with_datetime_fallback(@@default_timezone, *values)
1830
        end
1831 1832
      end

D
Initial  
David Heinemeier Hansson 已提交
1833
      def execute_callstack_for_multiparameter_attributes(callstack)
1834
        errors = []
1835
        callstack.each do |name, values_with_empty_parameters|
1836 1837
          begin
            klass = (self.class.reflect_on_aggregation(name.to_sym) || column_for_attribute(name)).klass
1838 1839
            # in order to allow a date to be set without a year, we must keep the empty values.
            # Otherwise, we wouldn't be able to distinguish it from a date with an empty day.
1840
            values = values_with_empty_parameters.reject { |v| v.nil? }
1841

1842 1843 1844
            if values.empty?
              send(name + "=", nil)
            else
1845

1846
              value = if Time == klass
1847
                instantiate_time_object(name, values)
1848 1849
              elsif Date == klass
                begin
1850
                  values = values_with_empty_parameters.collect do |v| v.nil? ? 1 : v end
1851 1852
                  Date.new(*values)
                rescue ArgumentError => ex # if Date.new raises an exception on an invalid date
1853
                  instantiate_time_object(name, values).to_date # we instantiate Time object and convert it back to a date thus using Time's logic in handling invalid dates
1854 1855 1856 1857 1858 1859
                end
              else
                klass.new(*values)
              end

              send(name + "=", value)
1860
            end
1861 1862
          rescue => ex
            errors << AttributeAssignmentError.new("error on assignment #{values.inspect} to #{name}", ex, name)
D
Initial  
David Heinemeier Hansson 已提交
1863 1864
          end
        end
1865 1866 1867
        unless errors.empty?
          raise MultiparameterAssignmentErrors.new(errors), "#{errors.size} error(s) on assignment of multiparameter attributes"
        end
D
Initial  
David Heinemeier Hansson 已提交
1868
      end
1869

D
Initial  
David Heinemeier Hansson 已提交
1870 1871 1872 1873 1874 1875 1876 1877
      def extract_callstack_for_multiparameter_attributes(pairs)
        attributes = { }

        for pair in pairs
          multiparameter_name, value = pair
          attribute_name = multiparameter_name.split("(").first
          attributes[attribute_name] = [] unless attributes.include?(attribute_name)

1878 1879
          parameter_value = value.empty? ? nil : type_cast_attribute_value(multiparameter_name, value)
          attributes[attribute_name] << [ find_parameter_position(multiparameter_name), parameter_value ]
D
Initial  
David Heinemeier Hansson 已提交
1880 1881 1882 1883
        end

        attributes.each { |name, values| attributes[name] = values.sort_by{ |v| v.first }.collect { |v| v.last } }
      end
1884

D
Initial  
David Heinemeier Hansson 已提交
1885
      def type_cast_attribute_value(multiparameter_name, value)
1886
        multiparameter_name =~ /\([0-9]*([if])\)/ ? value.send("to_" + $1) : value
D
Initial  
David Heinemeier Hansson 已提交
1887
      end
1888

D
Initial  
David Heinemeier Hansson 已提交
1889 1890 1891
      def find_parameter_position(multiparameter_name)
        multiparameter_name.scan(/\(([0-9]*).*\)/).first.first
      end
1892

D
Initial  
David Heinemeier Hansson 已提交
1893 1894
      # Returns a comma-separated pair list, like "key1 = val1, key2 = val2".
      def comma_pair_list(hash)
1895
        hash.map { |k,v| "#{k} = #{v}" }.join(", ")
D
Initial  
David Heinemeier Hansson 已提交
1896 1897
      end

1898
      def quote_columns(quoter, hash)
1899
        Hash[hash.map { |name, value| [quoter.quote_column_name(name), value] }]
D
Initial  
David Heinemeier Hansson 已提交
1900 1901
      end

1902 1903
      def quoted_comma_pair_list(quoter, hash)
        comma_pair_list(quote_columns(quoter, hash))
D
Initial  
David Heinemeier Hansson 已提交
1904 1905
      end

1906 1907 1908 1909 1910 1911 1912 1913 1914 1915 1916 1917
      def convert_number_column_value(value)
        if value == false
          0
        elsif value == true
          1
        elsif value.is_a?(String) && value.blank?
          nil
        else
          value
        end
      end

1918 1919 1920
      def populate_with_current_scope_attributes
        if scope = self.class.send(:current_scoped_methods)
          create_with = scope.scope_for_create
1921
          create_with.each { |att,value|
J
José Valim 已提交
1922
            respond_to?("#{att}=") && send("#{att}=", value)
1923
          }
1924 1925
        end
      end
1926

1927
      # Clear attributes and changed_attributes
1928
      def clear_timestamp_attributes
1929 1930 1931
        all_timestamp_attributes_in_model.each do |attribute_name|
          self[attribute_name] = nil
          changed_attributes.delete(attribute_name)
1932 1933
        end
      end
D
Initial  
David Heinemeier Hansson 已提交
1934
  end
J
Joshua Peek 已提交
1935 1936

  Base.class_eval do
1937
    include ActiveRecord::Persistence
J
Joshua Peek 已提交
1938
    extend ActiveModel::Naming
J
Joshua Peek 已提交
1939
    extend QueryCache::ClassMethods
J
José Valim 已提交
1940
    extend ActiveSupport::Benchmarkable
1941
    extend ActiveSupport::DescendantsTracker
J
José Valim 已提交
1942

1943
    include ActiveModel::Conversion
J
Joshua Peek 已提交
1944
    include Validations
1945
    extend CounterCache
J
Joshua Peek 已提交
1946 1947
    include Locking::Optimistic, Locking::Pessimistic
    include AttributeMethods
J
Joshua Peek 已提交
1948
    include AttributeMethods::Read, AttributeMethods::Write, AttributeMethods::BeforeTypeCast, AttributeMethods::Query
1949
    include AttributeMethods::PrimaryKey
J
Joshua Peek 已提交
1950 1951
    include AttributeMethods::TimeZoneConversion
    include AttributeMethods::Dirty
1952
    include ActiveModel::MassAssignmentSecurity
1953
    include Callbacks, ActiveModel::Observing, Timestamp
J
Jon Leighton 已提交
1954
    include Associations, NamedScope
M
Marcin Raczkowski 已提交
1955
    include IdentityMap
1956
    include ActiveModel::SecurePassword
1957 1958 1959 1960

    # AutosaveAssociation needs to be included before Transactions, because we want
    # #save_with_autosave_associations to be wrapped inside a transaction.
    include AutosaveAssociation, NestedAttributes
P
Pratik Naik 已提交
1961
    include Aggregations, Transactions, Reflection, Serialization
1962

1963
    NilClass.add_whiner(self) if NilClass.respond_to?(:add_whiner)
1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974

    # Returns the value of the attribute identified by <tt>attr_name</tt> after it has been typecast (for example,
    # "2004-12-12" in a data column is cast to a date object, like Date.new(2004, 12, 12)).
    # (Alias for the protected read_attribute method).
    alias [] read_attribute

    # Updates the attribute identified by <tt>attr_name</tt> with the specified +value+.
    # (Alias for the protected write_attribute method).
    alias []= write_attribute

    public :[], :[]=
J
Joshua Peek 已提交
1975
  end
1976
end
1977 1978 1979

# TODO: Remove this and make it work with LAZY flag
require 'active_record/connection_adapters/abstract_adapter'
1980
ActiveSupport.run_load_hooks(:active_record, ActiveRecord::Base)