base.rb 79.7 KB
Newer Older
D
Initial  
David Heinemeier Hansson 已提交
1
require 'yaml'
2
require 'set'
J
José Valim 已提交
3
require 'active_support/benchmarkable'
J
Jeremy Kemper 已提交
4
require 'active_support/dependencies'
5
require 'active_support/descendants_tracker'
6
require 'active_support/time'
7
require 'active_support/core_ext/class/attribute'
J
Jeremy Kemper 已提交
8 9
require 'active_support/core_ext/class/attribute_accessors'
require 'active_support/core_ext/class/delegating_attributes'
10
require 'active_support/core_ext/class/attribute'
J
Jeremy Kemper 已提交
11 12 13 14 15
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'
16
require 'active_support/core_ext/kernel/singleton_class'
17
require 'active_support/core_ext/module/delegation'
18
require 'active_support/core_ext/module/introspection'
19
require 'active_support/core_ext/object/duplicable'
20
require 'active_support/core_ext/object/blank'
21 22
require 'arel'
require 'active_record/errors'
23
require 'active_record/log_subscriber'
D
Initial  
David Heinemeier Hansson 已提交
24 25

module ActiveRecord #:nodoc:
26
  # = Active Record
27
  #
28 29 30
  # 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
31
  # Active Record objects. The mapping that binds a given Active Record class to a certain
32 33
  # database table will happen automatically in most common cases, but can be overwritten for the uncommon ones.
  #
34
  # See the mapping rules in table_name and the full example in link:files/activerecord/README_rdoc.html for more insight.
35
  #
D
Initial  
David Heinemeier Hansson 已提交
36
  # == Creation
37
  #
38 39
  # 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
40
  # HTTP request. It works like this:
41
  #
42
  #   user = User.new(:name => "David", :occupation => "Code Artist")
D
Initial  
David Heinemeier Hansson 已提交
43
  #   user.name # => "David"
44
  #
D
Initial  
David Heinemeier Hansson 已提交
45
  # You can also use block initialization:
46
  #
D
Initial  
David Heinemeier Hansson 已提交
47 48 49 50
  #   user = User.new do |u|
  #     u.name = "David"
  #     u.occupation = "Code Artist"
  #   end
51
  #
D
Initial  
David Heinemeier Hansson 已提交
52
  # And of course you can just create a bare object and specify the attributes after the fact:
53
  #
D
Initial  
David Heinemeier Hansson 已提交
54 55 56
  #   user = User.new
  #   user.name = "David"
  #   user.occupation = "Code Artist"
57
  #
D
Initial  
David Heinemeier Hansson 已提交
58
  # == Conditions
59
  #
60
  # Conditions can either be specified as a string, array, or hash representing the WHERE-part of an SQL statement.
D
Initial  
David Heinemeier Hansson 已提交
61
  # The array form is to be used when the condition input is tainted and requires sanitization. The string form can
62
  # be used for statements that don't involve tainted data. The hash form works much like the array form, except
63
  # only equality and range is possible. Examples:
64
  #
65
  #   class User < ActiveRecord::Base
D
Initial  
David Heinemeier Hansson 已提交
66
  #     def self.authenticate_unsafely(user_name, password)
67
  #       where("user_name = '#{user_name}' AND password = '#{password}'").first
D
Initial  
David Heinemeier Hansson 已提交
68
  #     end
69
  #
D
Initial  
David Heinemeier Hansson 已提交
70
  #     def self.authenticate_safely(user_name, password)
71
  #       where("user_name = ? AND password = ?", user_name, password).first
D
Initial  
David Heinemeier Hansson 已提交
72
  #     end
73 74
  #
  #     def self.authenticate_safely_simply(user_name, password)
75
  #       where(:user_name => user_name, :password => password).first
76
  #     end
D
Initial  
David Heinemeier Hansson 已提交
77
  #   end
78
  #
79 80
  # 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+
81
  # parameters come directly from an HTTP request. The <tt>authenticate_safely</tt>  and
82 83
  # <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
84
  # query and fake the login (or worse).
85
  #
86 87 88
  # 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
89
  # symbols and supplying a hash with values for the matching symbol keys:
90
  #
91
  #   Company.where(
92
  #     "id = :id AND name = :name AND division = :division AND created_at > :accounting_date",
93
  #     { :id => 3, :name => "37signals", :division => "First", :accounting_date => '2005-01-01' }
94
  #   ).first
95
  #
96 97 98
  # Similarly, a simple hash without a statement will generate conditions based on equality with the SQL AND
  # operator. For instance:
  #
99 100
  #   Student.where(:first_name => "Harvey", :status => 1)
  #   Student.where(params[:student])
101
  #
102 103
  # A range may be used in the hash to use the SQL BETWEEN operator:
  #
104
  #   Student.where(:grade => 9..12)
105
  #
P
Pratik Naik 已提交
106 107
  # An array may be used in the hash to use the SQL IN operator:
  #
108
  #   Student.where(:grade => [9,11,12])
P
Pratik Naik 已提交
109
  #
110
  # When joining tables, nested hashes or keys written in the form 'table_name.column_name'
111
  # can be used to qualify the table name of a particular condition. For instance:
P
Pratik Naik 已提交
112
  #
113 114
  #   Student.joins(:schools).where(:schools => { :type => 'public' })
  #   Student.joins(:schools).where('schools.type' => 'public' )
P
Pratik Naik 已提交
115
  #
D
Initial  
David Heinemeier Hansson 已提交
116
  # == Overwriting default accessors
117
  #
118 119 120 121
  # 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
122
  # change things.
123
  #
D
Initial  
David Heinemeier Hansson 已提交
124 125
  #   class Song < ActiveRecord::Base
  #     # Uses an integer of seconds to hold the length of the song
126
  #
D
Initial  
David Heinemeier Hansson 已提交
127
  #     def length=(minutes)
128
  #       write_attribute(:length, minutes.to_i * 60)
D
Initial  
David Heinemeier Hansson 已提交
129
  #     end
130
  #
D
Initial  
David Heinemeier Hansson 已提交
131
  #     def length
132
  #       read_attribute(:length) / 60
D
Initial  
David Heinemeier Hansson 已提交
133 134
  #     end
  #   end
135
  #
136
  # You can alternatively use <tt>self[:attribute]=(value)</tt> and <tt>self[:attribute]</tt>
137
  # instead of <tt>write_attribute(:attribute, value)</tt> and <tt>read_attribute(:attribute)</tt>.
138
  #
139 140 141 142
  # == 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.
143
  #
144 145 146 147 148 149 150 151 152
  # 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
  #
153
  # == Accessing attributes before they have been typecasted
154
  #
155 156 157
  # 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,
158
  # you can call <tt>account.balance_before_type_cast</tt> or <tt>account.id_before_type_cast</tt>.
159
  #
160 161
  # 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
162
  # attribute normally would typecast the string to 0, which isn't what you want.
163
  #
164 165
  # == Dynamic attribute-based finders
  #
166 167 168 169 170
  # 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
171
  # <tt>Person.where(:user_name => user_name).first</tt>, you just do <tt>Person.find_by_user_name(user_name)</tt>.
172
  # And instead of writing <tt>Person.where(:last_name => last_name).all</tt>, you just do
173
  # <tt>Person.find_all_by_last_name(last_name)</tt>.
174
  #
175
  # It's also possible to use multiple attributes in the same find by separating them with "_and_".
176
  #
177 178
  #  Person.where(:user_name => user_name, :password => password).first
  #  Person.find_by_user_name_and_password #with dynamic finder
179
  #
180 181
  #  Person.where(:user_name => user_name, :password => password, :gender => 'male').first
  #  Payment.find_by_user_name_and_password_and_gender
182
  #
183
  # It's even possible to call these dynamic finder methods on relations and named scopes.
184 185 186
  #
  #   Payment.order("created_on").find_all_by_amount(50)
  #   Payment.pending.find_last_by_amount(100)
187
  #
188 189 190 191
  # 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.
192 193 194
  #
  #   # No 'Summer' tag exists
  #   Tag.find_or_create_by_name("Summer") # equal to Tag.create(:name => "Summer")
195
  #
196 197 198
  #   # Now the 'Summer' tag does exist
  #   Tag.find_or_create_by_name("Summer") # equal to Tag.find_by_name("Summer")
  #
199 200 201
  #   # Now 'Bob' exist and is an 'admin'
  #   User.find_or_create_by_name('Bob', :age => 40) { |u| u.admin = true }
  #
202
  # Use the <tt>find_or_initialize_by_</tt> finder if you want to return a new record without
203
  # saving it first. Protected attributes won't be set unless they are given in a block.
204 205 206
  #
  #   # No 'Winter' tag exists
  #   winter = Tag.find_or_initialize_by_name("Winter")
207
  #   winter.persisted? # false
208
  #
209
  # To find by a subset of the attributes to be used for instantiating a new object, pass a hash instead of
210
  # a list of parameters.
211 212 213
  #
  #   Tag.find_or_create_by_name(:name => "rails", :creator => current_user)
  #
214
  # That will either find an existing tag named "rails", or create a new one while setting the
215
  # user that created it.
216
  #
217 218 219 220
  # 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.
  #
221
  #  User.scoped_by_user_name('David')
222
  #
223
  # == Saving arrays, hashes, and other non-mappable objects in text columns
224
  #
225
  # Active Record can serialize any object in text columns using YAML. To do so, you must
226
  # specify this with a call to the class method +serialize+.
227 228
  # This makes it possible to store arrays, hashes, and other non-mappable objects without doing
  # any additional work.
229
  #
D
Initial  
David Heinemeier Hansson 已提交
230 231 232
  #   class User < ActiveRecord::Base
  #     serialize :preferences
  #   end
233
  #
234
  #   user = User.create(:preferences => { "background" => "black", "display" => large })
D
Initial  
David Heinemeier Hansson 已提交
235
  #   User.find(user.id).preferences # => { "background" => "black", "display" => large }
236
  #
237
  # You can also specify a class option as the second parameter that'll raise an exception
238
  # if a serialized object is retrieved as a descendant of a class not in the hierarchy.
239
  #
D
Initial  
David Heinemeier Hansson 已提交
240
  #   class User < ActiveRecord::Base
241
  #     serialize :preferences, Hash
D
Initial  
David Heinemeier Hansson 已提交
242
  #   end
243
  #
244
  #   user = User.create(:preferences => %w( one two three ))
D
Initial  
David Heinemeier Hansson 已提交
245
  #   User.find(user.id).preferences    # raises SerializationTypeMismatch
246
  #
D
Initial  
David Heinemeier Hansson 已提交
247 248
  # == Single table inheritance
  #
249 250
  # 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>).
251
  # This means that an inheritance looking like this:
D
Initial  
David Heinemeier Hansson 已提交
252 253 254 255 256 257
  #
  #   class Company < ActiveRecord::Base; end
  #   class Firm < Company; end
  #   class Client < Company; end
  #   class PriorityClient < Client; end
  #
258 259
  # 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
260
  # <tt>Company.where(:name => '37signals').first</tt> and it will return a Firm object.
D
Initial  
David Heinemeier Hansson 已提交
261
  #
262 263
  # 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
264
  # for differentiating between them or reloading the right type with find.
265
  #
D
Initial  
David Heinemeier Hansson 已提交
266 267
  # Note, all the attributes for all the cases are kept in the same table. Read more:
  # http://www.martinfowler.com/eaaCatalog/singleTableInheritance.html
268
  #
D
Initial  
David Heinemeier Hansson 已提交
269 270
  # == Connection to multiple databases in different models
  #
271 272 273
  # 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
274
  # ActiveRecord::Base, but resides in a different database, you can just say <tt>Course.establish_connection</tt>
P
Pratik Naik 已提交
275
  # and Course and all of its subclasses will use this connection instead.
D
Initial  
David Heinemeier Hansson 已提交
276
  #
277 278
  # 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
279
  # will go up the class-hierarchy until a connection is found in the connection pool.
D
Initial  
David Heinemeier Hansson 已提交
280 281
  #
  # == Exceptions
282
  #
P
Pratik Naik 已提交
283 284
  # * 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 已提交
285
  #   <tt>:adapter</tt> key.
286
  # * AdapterNotFound - The <tt>:adapter</tt> key used in <tt>establish_connection</tt> specified a
287
  #   non-existent adapter
288
  #   (or a bad spelling of an existing one).
289
  # * AssociationTypeMismatch - The object assigned to the association wasn't of the type
290
  #   specified in the association definition.
P
Pratik Naik 已提交
291
  # * SerializationTypeMismatch - The serialized object wasn't of the class specified as the second parameter.
292
  # * ConnectionNotEstablished+ - No connection has been established. Use <tt>establish_connection</tt>
293
  #   before querying.
P
Pratik Naik 已提交
294 295 296 297 298
  # * 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
299
  #   <tt>attributes=</tt> method. The +errors+ property of this exception contains an array of
300
  #   AttributeAssignmentError
301
  #   objects that should be inspected to determine which attributes triggered the errors.
302
  # * AttributeAssignmentError - An error occurred while doing a mass assignment through the
303
  #   <tt>attributes=</tt> method.
304
  #   You can inspect the +attribute+ property of the exception object to determine which attribute
305
  #   triggered the error.
306
  #
307
  # *Note*: The attributes listed are class-level attributes (accessible from both the class and instance level).
P
Pratik Naik 已提交
308
  # 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 已提交
309 310
  # instances in the current object space.
  class Base
E
Emilio Tagua 已提交
311
    ##
P
Pratik Naik 已提交
312
    # :singleton-method:
313 314
    # 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
315
    # a class and instance level by calling +logger+.
316
    cattr_accessor :logger, :instance_writer => false
J
Jeremy Kemper 已提交
317

P
Pratik Naik 已提交
318 319
    ##
    # :singleton-method:
P
Pratik Naik 已提交
320 321 322 323
    # Contains the database configuration - as is typically stored in config/database.yml -
    # as a Hash.
    #
    # For example, the following database.yml...
E
Emilio Tagua 已提交
324
    #
P
Pratik Naik 已提交
325 326 327
    #   development:
    #     adapter: sqlite3
    #     database: db/development.sqlite3
E
Emilio Tagua 已提交
328
    #
P
Pratik Naik 已提交
329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344
    #   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'
    #      }
    #   }
345
    cattr_accessor :configurations, :instance_writer => false
346 347
    @@configurations = {}

P
Pratik Naik 已提交
348 349
    ##
    # :singleton-method:
350 351 352
    # 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
353
    # latter is specified, the Product class will look for "product_id" instead of "id". Remember
354
    # that this is a global setting for all Active Records.
355
    cattr_accessor :primary_key_prefix_type, :instance_writer => false
D
Initial  
David Heinemeier Hansson 已提交
356 357
    @@primary_key_prefix_type = nil

P
Pratik Naik 已提交
358 359
    ##
    # :singleton-method:
360 361 362
    # 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.
363
    # By default, the prefix is the empty string.
364
    #
365 366
    # 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
367
    # returns your chosen prefix.
368 369
    class_attribute :table_name_prefix, :instance_writer => false
    self.table_name_prefix = ""
D
Initial  
David Heinemeier Hansson 已提交
370

P
Pratik Naik 已提交
371 372
    ##
    # :singleton-method:
D
Initial  
David Heinemeier Hansson 已提交
373 374
    # 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.
375 376
    class_attribute :table_name_suffix, :instance_writer => false
    self.table_name_suffix = ""
D
Initial  
David Heinemeier Hansson 已提交
377

P
Pratik Naik 已提交
378 379
    ##
    # :singleton-method:
380
    # Indicates whether table names should be the pluralized versions of the corresponding class names.
P
Pratik Naik 已提交
381
    # 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 已提交
382
    # See table_name for the full rules on table/class naming. This is true, by default.
383
    cattr_accessor :pluralize_table_names, :instance_writer => false
D
Initial  
David Heinemeier Hansson 已提交
384 385
    @@pluralize_table_names = true

P
Pratik Naik 已提交
386 387
    ##
    # :singleton-method:
388
    # Determines whether to use Time.local (using :local) or Time.utc (using :utc) when pulling
389
    # dates and times from the database. This is set to :local by default.
390
    cattr_accessor :default_timezone, :instance_writer => false
391
    @@default_timezone = :local
392

P
Pratik Naik 已提交
393 394
    ##
    # :singleton-method:
395 396
    # Specifies the format to use when dumping the database schema with Rails'
    # Rakefile.  If :sql, the schema is dumped as (potentially database-
397
    # specific) SQL statements.  If :ruby, the schema is dumped as an
398 399 400
    # 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.
401
    cattr_accessor :schema_format , :instance_writer => false
402
    @@schema_format = :ruby
403

P
Pratik Naik 已提交
404 405
    ##
    # :singleton-method:
406
    # Specify whether or not to use timestamps for migration versions
407 408 409
    cattr_accessor :timestamped_migrations , :instance_writer => false
    @@timestamped_migrations = true

410
    # Determine whether to store the full constant name including namespace when using STI
411
    class_attribute :store_full_sti_class
412
    self.store_full_sti_class = true
413

414
    # Stores the default scope for the class
415
    class_attribute :default_scoping, :instance_writer => false
416 417
    self.default_scoping = []

J
Jeremy Kemper 已提交
418 419 420 421 422
    # 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 = {}

423 424 425
    class_attribute :_attr_readonly, :instance_writer => false
    self._attr_readonly = []

D
Initial  
David Heinemeier Hansson 已提交
426
    class << self # Class methods
427
      delegate :find, :first, :last, :all, :destroy, :destroy_all, :exists?, :delete, :delete_all, :update, :update_all, :to => :scoped
P
Pratik Naik 已提交
428
      delegate :find_each, :find_in_batches, :to => :scoped
429
      delegate :select, :group, :order, :except, :limit, :offset, :joins, :where, :preload, :eager_load, :includes, :from, :lock, :readonly, :having, :create_with, :to => :scoped
430
      delegate :count, :average, :minimum, :maximum, :sum, :calculate, :to => :scoped
431

P
Pratik Naik 已提交
432
      # Executes a custom SQL query against your database and returns all the results.  The results will
433
      # be returned as an array with columns requested encapsulated as attributes of the model you call
P
Pratik Naik 已提交
434 435
      # 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.
436
      #
437 438
      # 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
439 440
      # table.
      #
P
Pratik Naik 已提交
441
      # The +sql+ parameter is a full SQL query as a string.  It will be called as is, there will be
442 443
      # 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 已提交
444
      # change your call if you switch engines.
445 446
      #
      # ==== Examples
P
Pratik Naik 已提交
447
      #   # A simple SQL query spanning multiple tables
448 449 450 451 452 453
      #   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]
      #   > [#<Post:0x36bff9c @attributes={"first_name"=>"The Cheap Man Buys Twice"}>, ...]
454 455
      def find_by_sql(sql, binds = [])
        connection.select_all(sanitize_sql(sql), "#{name} Load", binds).collect! { |record| instantiate(record) }
D
Initial  
David Heinemeier Hansson 已提交
456
      end
457

458
      # Creates an object (or multiple objects) and saves it to the database, if validations pass.
459 460 461 462 463 464 465 466
      # 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')
467
      #
468
      #   # Create an Array of new objects
P
Pratik Naik 已提交
469
      #   User.create([{ :first_name => 'Jamie' }, { :first_name => 'Jeremy' }])
470 471 472 473 474 475 476
      #
      #   # 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 已提交
477
      #   User.create([{ :first_name => 'Jamie' }, { :first_name => 'Jeremy' }]) do |u|
478
      #     u.is_admin = false
479
      #   end
480
      def create(attributes = nil, &block)
481
        if attributes.is_a?(Array)
482
          attributes.collect { |attr| create(attr, &block) }
483 484
        else
          object = new(attributes)
485
          yield(object) if block_given?
486 487 488
          object.save
          object
        end
D
Initial  
David Heinemeier Hansson 已提交
489 490 491
      end

      # Returns the result of an SQL statement that should only include a COUNT(*) in the SELECT part.
492
      # The use of this method should be restricted to complicated SQL queries that can't be executed
493 494
      # using the ActiveRecord::Calculations class methods.  Look into those before using this.
      #
P
Pratik Naik 已提交
495
      # ==== Parameters
496
      #
P
Pratik Naik 已提交
497
      # * +sql+ - An SQL statement which should return a count query from the database, see the example below.
498 499 500
      #
      # ==== Examples
      #
501
      #   Product.count_by_sql "SELECT COUNT(*) FROM sales s, customers c WHERE s.customer_id = c.id"
D
Initial  
David Heinemeier Hansson 已提交
502
      def count_by_sql(sql)
503
        sql = sanitize_conditions(sql)
504
        connection.select_value(sql, "#{name} Count").to_i
D
Initial  
David Heinemeier Hansson 已提交
505
      end
506

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

513 514
      # Returns an array of all the attributes that have been specified as readonly.
      def readonly_attributes
515
        self._attr_readonly
D
Initial  
David Heinemeier Hansson 已提交
516 517
      end

518 519 520
      # 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 已提交
521
      # class on retrieval or SerializationTypeMismatch will be raised.
D
David Heinemeier Hansson 已提交
522
      #
P
Pratik Naik 已提交
523
      # ==== Parameters
D
David Heinemeier Hansson 已提交
524
      #
P
Pratik Naik 已提交
525 526
      # * +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 已提交
527 528 529 530 531 532
      #
      # ==== Example
      #   # Serialize a preferences attribute
      #   class User
      #     serialize :preferences
      #   end
D
Initial  
David Heinemeier Hansson 已提交
533
      def serialize(attr_name, class_name = Object)
534
        serialized_attributes[attr_name.to_s] = class_name
D
Initial  
David Heinemeier Hansson 已提交
535
      end
536

537 538
      # 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
539
      # looks like: Reply < Message < ActiveRecord::Base, then Message is used
540 541
      # 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
542
      # English inflections. You can add new inflections in config/initializers/inflections.rb.
D
Initial  
David Heinemeier Hansson 已提交
543
      #
544
      # Nested classes are given table names prefixed by the singular form of
P
Pratik Naik 已提交
545 546 547
      # the parent's table name. Enclosing modules are not considered.
      #
      # ==== Examples
548 549
      #
      #   class Invoice < ActiveRecord::Base; end;
550 551
      #   file                  class               table_name
      #   invoice.rb            Invoice             invoices
552 553 554 555 556 557 558 559
      #
      #   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 已提交
560
      #
P
Pratik Naik 已提交
561 562
      # Additionally, the class-level +table_name_prefix+ is prepended and the
      # +table_name_suffix+ is appended.  So if you have "myapp_" as a prefix,
563 564 565 566 567
      # 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 已提交
568 569
      #
      #   class Mouse < ActiveRecord::Base
570
      #     set_table_name "mice"
D
Initial  
David Heinemeier Hansson 已提交
571
      #   end
572
      def table_name
573 574 575
        reset_table_name
      end

576
      # Returns a quoted version of the table name, used to construct SQL statements.
P
Pratik Naik 已提交
577 578 579 580
      def quoted_table_name
        @quoted_table_name ||= connection.quote_table_name(table_name)
      end

581
      # Computes the table name, (re)sets it internally, and returns it.
D
David Heinemeier Hansson 已提交
582
      def reset_table_name #:nodoc:
583
        self.table_name = compute_table_name
D
Initial  
David Heinemeier Hansson 已提交
584 585
      end

586
      def full_table_name_prefix #:nodoc:
587 588 589
        (parents.detect{ |p| p.respond_to?(:table_name_prefix) } || self).table_name_prefix
      end

590
      # Defines the column name for use with single table inheritance. Use
N
Neeraj Singh 已提交
591
      # <tt>set_inheritance_column</tt> to set a different value.
D
Initial  
David Heinemeier Hansson 已提交
592
      def inheritance_column
A
Aaron Patterson 已提交
593
        @inheritance_column ||= "type"
D
Initial  
David Heinemeier Hansson 已提交
594 595
      end

596 597
      # 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 已提交
598
      def sequence_name #:nodoc:
599 600 601
        reset_sequence_name
      end

D
David Heinemeier Hansson 已提交
602
      def reset_sequence_name #:nodoc:
603 604 605
        default = connection.default_sequence_name(table_name, primary_key)
        set_sequence_name(default)
        default
606 607
      end

608
      # Sets the table name. If the value is nil or false  then the value returned by the given
N
Neeraj Singh 已提交
609
      # block is used.
610 611 612 613
      #
      #   class Project < ActiveRecord::Base
      #     set_table_name "project"
      #   end
614
      def set_table_name(value = nil, &block)
615
        @quoted_table_name = nil
616 617 618 619 620 621
        define_attr_method :table_name, value, &block
      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
622
      # given block.
623 624 625 626 627 628
      #
      #   class Project < ActiveRecord::Base
      #     set_inheritance_column do
      #       original_inheritance_column + "_id"
      #     end
      #   end
629
      def set_inheritance_column(value = nil, &block)
630 631 632 633
        define_attr_method :inheritance_column, value, &block
      end
      alias :inheritance_column= :set_inheritance_column

634 635
      # 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
636 637
      # given block. This is required for Oracle and is useful for any
      # database which relies on sequences for primary key generation.
638
      #
639 640 641 642 643
      # 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.
644 645 646 647
      #
      #   class Project < ActiveRecord::Base
      #     set_sequence_name "projectseq"   # default would have been "project_seq"
      #   end
648
      def set_sequence_name(value = nil, &block)
649 650 651 652
        define_attr_method :sequence_name, value, &block
      end
      alias :sequence_name= :set_sequence_name

653 654
      # Indicates whether the table associated with this class exists
      def table_exists?
655
        connection.table_exists?(table_name)
656 657
      end

D
Initial  
David Heinemeier Hansson 已提交
658 659
      # Returns an array of column objects for the table associated with this class.
      def columns
660
        unless defined?(@columns) && @columns
661
          @columns = connection.columns(table_name, "#{name} Columns")
662
          @columns.each { |column| column.primary = column.name == primary_key }
663 664
        end
        @columns
D
Initial  
David Heinemeier Hansson 已提交
665
      end
666

667
      # Returns a hash of column objects for the table associated with this class.
D
Initial  
David Heinemeier Hansson 已提交
668
      def columns_hash
669
        @columns_hash ||= Hash[columns.map { |column| [column.name, column] }]
D
Initial  
David Heinemeier Hansson 已提交
670
      end
671

D
David Heinemeier Hansson 已提交
672
      # Returns an array of column names as strings.
673
      def column_names
674
        @column_names ||= columns.map { |column| column.name }
675
      end
D
Initial  
David Heinemeier Hansson 已提交
676

677 678
      # 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 已提交
679
      def content_columns
680
        @content_columns ||= columns.reject { |c| c.primary || c.name =~ /(_id|_count)$/ || c.name == inheritance_column }
D
Initial  
David Heinemeier Hansson 已提交
681 682 683 684
      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
685
      # is available.
D
David Heinemeier Hansson 已提交
686
      def column_methods_hash #:nodoc:
687
        @dynamic_methods_hash ||= column_names.inject(Hash.new(false)) do |methods, attr|
688 689 690 691 692
          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 已提交
693 694 695
          methods
        end
      end
696

P
Pratik Naik 已提交
697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722
      # 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
723
      def reset_column_information
A
Aaron Patterson 已提交
724
        connection.clear_cache!
725
        undefine_attribute_methods
726
        @column_names = @columns = @columns_hash = @content_columns = @dynamic_methods_hash = @inheritance_column = nil
727
        @arel_engine = @relation = @arel_table = nil
728 729
      end

730
      def attribute_method?(attribute)
J
José Valim 已提交
731
        super || (table_exists? && column_names.include?(attribute.to_s.sub(/=$/, '')))
732 733
      end

734 735
      # Set the lookup ancestors for ActiveModel.
      def lookup_ancestors #:nodoc:
736 737
        klass = self
        classes = [klass]
738 739
        return classes if klass == ActiveRecord::Base

E
Emilio Tagua 已提交
740
        while klass != klass.base_class
741 742 743 744 745
          classes << klass = klass.superclass
        end
        classes
      end

746 747 748
      # Set the i18n scope to overwrite ActiveModel.
      def i18n_scope #:nodoc:
        :activerecord
D
Initial  
David Heinemeier Hansson 已提交
749
      end
750

751 752 753 754 755 756 757
      # 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 已提交
758 759
      end

760 761 762 763 764
      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

765
      # Returns a string like 'Post id:integer, title:string, body:text'
766
      def inspect
767 768 769 770
        if self == Base
          super
        elsif abstract_class?
          "#{super}(abstract)"
771
        elsif table_exists?
772 773
          attr_list = columns.map { |c| "#{c.name}: #{c.type}" } * ', '
          "#{super}(#{attr_list})"
774 775
        else
          "#{super}(Table doesn't exist)"
776
        end
777 778
      end

779
      def quote_value(value, column = nil) #:nodoc:
780
        connection.quote(value,column)
781 782
      end

783
      # Used to sanitize objects before they're used in an SQL SELECT statement. Delegates to <tt>connection.quote</tt>.
784
      def sanitize(object) #:nodoc:
785
        connection.quote(object)
D
Initial  
David Heinemeier Hansson 已提交
786 787
      end

788 789 790
      # Overwrite the default class equality method to provide support for association proxies.
      def ===(object)
        object.is_a?(self)
791
      end
792

793 794 795
      # 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.
796 797 798
      #
      # 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.
799 800 801 802
      def base_class
        class_of_active_record_descendant(self)
      end

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

806
      # Returns whether this class is an abstract class or not.
807
      def abstract_class?
808
        defined?(@abstract_class) && @abstract_class == true
809 810
      end

811
      def respond_to?(method_id, include_private = false)
812 813
        if match = DynamicFinderMatch.match(method_id)
          return true if all_attributes_exists?(match.attribute_names)
814 815
        elsif match = DynamicScopeMatch.match(method_id)
          return true if all_attributes_exists?(match.attribute_names)
816
        end
817

818 819 820
        super
      end

821 822 823 824
      def sti_name
        store_full_sti_class ? name : name.demodulize
      end

P
Pratik Naik 已提交
825
      def arel_table
826
        @arel_table ||= Arel::Table.new(table_name, arel_engine)
827 828
      end

829 830
      def arel_engine
        @arel_engine ||= begin
831 832 833
          if self == ActiveRecord::Base
            Arel::Table.engine
          else
834
            connection_handler.connection_pools[name] ? self : superclass.arel_engine
835 836
          end
        end
837 838
      end

839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854
      # 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"
      #   }
      #
855 856
      # 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.
857
      #
858 859
      # Post.unscoped.published
      # Post.published
860
      def unscoped #:nodoc:
861 862 863 864 865 866 867 868
        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

869 870 871 872
      def before_remove_const #:nodoc:
        reset_scoped_methods
      end

J
Jon Leighton 已提交
873 874 875 876 877 878 879 880 881 882
      # 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

D
Initial  
David Heinemeier Hansson 已提交
883
      private
J
José Valim 已提交
884 885 886

        def relation #:nodoc:
          @relation ||= Relation.new(self, arel_table)
887 888 889 890 891 892

          if finder_needs_type_condition?
            @relation.where(type_condition).create_with(inheritance_column.to_sym => sti_name)
          else
            @relation
          end
J
José Valim 已提交
893 894
        end

895 896 897
        # 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.
D
Initial  
David Heinemeier Hansson 已提交
898
        def instantiate(record)
899 900 901
          model = find_sti_class(record[inheritance_column]).allocate
          model.init_with('attributes' => record)
          model
D
Initial  
David Heinemeier Hansson 已提交
902
        end
903

904
        def find_sti_class(type_name)
905
          if type_name.blank? || !columns_hash.include?(inheritance_column)
906 907 908
            self
          else
            begin
909 910 911 912 913
              if store_full_sti_class
                ActiveSupport::Dependencies.constantize(type_name)
              else
                compute_type(type_name)
              end
914 915 916 917 918 919 920 921 922 923
            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

924
        def construct_finder_arel(options = {}, scope = nil)
925
          relation = options.is_a?(Hash) ? unscoped.apply_finder_options(options) : options
926
          relation = scope.merge(relation) if scope
927
          relation
928 929
        end

930
        def type_condition
931
          sti_column = arel_table[inheritance_column.to_sym]
932
          sti_names  = ([self] + descendants).map { |model| model.sti_name }
933

934
          sti_column.in(sti_names)
D
Initial  
David Heinemeier Hansson 已提交
935 936 937
        end

        # Guesses the table name, but does not decorate it with prefix and suffix information.
938
        def undecorated_table_name(class_name = base_class.name)
939 940
          table_name = class_name.to_s.demodulize.underscore
          table_name = table_name.pluralize if pluralize_table_names
941
          table_name
D
Initial  
David Heinemeier Hansson 已提交
942 943
        end

944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960
        # 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
              contained << '_'
            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

961
        # Enables dynamic finders like <tt>User.find_by_user_name(user_name)</tt> and
962 963
        # <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.
964
        #
965
        # It's even possible to use all the additional parameters to +find+. For example, the
966
        # full interface for +find_all_by_amount+ is actually <tt>find_all_by_amount(amount, options)</tt>.
967
        #
968
        # Each dynamic finder using <tt>scoped_by_*</tt> is also defined in the class after it
969
        # is first invoked, so that future attempts to use it do not run through method_missing.
970
        def method_missing(method_id, *arguments, &block)
971 972
          if match = DynamicFinderMatch.match(method_id)
            attribute_names = match.attribute_names
973
            super unless all_attributes_exists?(attribute_names)
974
            if match.finder?
975
              options = arguments.extract_options!
976
              relation = options.any? ? construct_finder_arel(options, current_scoped_methods) : scoped
977
              relation.send :find_by_attributes, match, attribute_names, *arguments
978
            elsif match.instantiator?
979
              scoped.send :find_or_instantiator_by_attributes, match, attribute_names, *arguments, &block
980
            end
981 982 983 984
          elsif match = DynamicScopeMatch.match(method_id)
            attribute_names = match.attribute_names
            super unless all_attributes_exists?(attribute_names)
            if match.scope?
985
              self.class_eval <<-METHOD, __FILE__, __LINE__ + 1
986 987 988 989 990
                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
991
              METHOD
992 993
              send(method_id, *arguments)
            end
994 995 996 997
          else
            super
          end
        end
D
Initial  
David Heinemeier Hansson 已提交
998

999 1000
        # Similar in purpose to +expand_hash_conditions_for_aggregates+.
        def expand_attribute_names_for_aggregates(attribute_names)
1001
          attribute_names.map { |attribute_name|
1002
            unless (aggregation = reflect_on_aggregation(attribute_name.to_sym)).nil?
1003 1004
              aggregate_mapping(aggregation).map do |field_attr, _|
                field_attr.to_sym
1005 1006
              end
            else
1007
              attribute_name.to_sym
1008
            end
1009
          }.flatten
1010 1011
        end

1012
        def all_attributes_exists?(attribute_names)
1013 1014
          (expand_attribute_names_for_aggregates(attribute_names) -
           column_methods_hash.keys).empty?
1015
        end
1016

D
Initial  
David Heinemeier Hansson 已提交
1017
      protected
N
Neeraj Singh 已提交
1018 1019
        # 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
1020
        # <tt>:create</tt> parameters are an attributes hash.
1021 1022 1023
        #
        #   class Article < ActiveRecord::Base
        #     def self.create_with_scope
1024
        #       with_scope(:find => where(:blog_id => 1), :create => { :blog_id => 1 }) do
1025 1026 1027 1028 1029 1030 1031
        #         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 已提交
1032
        # In nested scopings, all previous parameters are overwritten by the innermost rule, with the exception of
1033
        # <tt>where</tt>, <tt>includes</tt>, and <tt>joins</tt> operations in <tt>Relation</tt>, which are merged.
P
Pratik Naik 已提交
1034
        #
1035
        # <tt>joins</tt> operations are uniqued so multiple scopes can join in the same table without table aliasing
P
Pratik Naik 已提交
1036 1037
        # 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.
1038 1039 1040
        #
        #   class Article < ActiveRecord::Base
        #     def self.find_with_scope
1041 1042 1043
        #       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
1044
        #         end
1045 1046
        #         with_scope(:find => where(:author_id => 3)) do
        #           all # => SELECT * from articles WHERE blog_id = 1 AND author_id = 3 LIMIT 1
1047 1048 1049 1050 1051
        #         end
        #       end
        #     end
        #   end
        #
1052
        # You can ignore any previous scopings by using the <tt>with_exclusive_scope</tt> method.
1053 1054 1055
        #
        #   class Article < ActiveRecord::Base
        #     def self.find_with_exclusive_scope
1056
        #       with_scope(:find => where(:blog_id => 1).limit(1)) do
1057
        #         with_exclusive_scope(:find => limit(10)) do
1058
        #           all # => SELECT * from articles LIMIT 10
1059 1060 1061 1062
        #         end
        #       end
        #     end
        #   end
P
Pratik Naik 已提交
1063
        #
N
Neeraj Singh 已提交
1064
        # *Note*: the +:find+ scope also has effect on update and deletion methods, like +update_all+ and +delete_all+.
1065 1066 1067
        def with_scope(method_scoping = {}, action = :merge, &block)
          method_scoping = method_scoping.method_scoping if method_scoping.respond_to?(:method_scoping)

1068 1069
          if method_scoping.is_a?(Hash)
            # Dup first and second level of hash (method and params).
1070 1071 1072
            method_scoping = method_scoping.dup
            method_scoping.each do |method, params|
              method_scoping[method] = params.dup unless params == true
1073
            end
1074

1075 1076 1077 1078
            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]
1079
              scope_for_create = if action == :merge
1080 1081 1082
                current_scoped_methods.create_with_value.merge(method_scoping[:create])
              else
                method_scoping[:create]
1083
              end
1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094

              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

1095
          method_scoping = current_scoped_methods.merge(method_scoping) if current_scoped_methods && action ==  :merge
1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106

          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)
1107
          if method_scoping.values.any? { |e| e.is_a?(ActiveRecord::Relation) }
1108 1109
            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:
1110

1111 1112 1113 1114 1115 1116 1117 1118 1119
  User.unscoped.where(:active => true)

Or call unscoped with a block:

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

MSG
1120
          end
1121 1122 1123
          with_scope(method_scoping, :overwrite, &block)
        end

1124
        # Sets the default options for the model. The format of the
1125
        # <tt>options</tt> argument is the same as in find.
1126 1127
        #
        #   class Person < ActiveRecord::Base
1128
        #     default_scope order('last_name, first_name')
1129
        #   end
1130 1131 1132 1133 1134 1135 1136 1137
        #
        # <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
        #
1138 1139
        #   Article.new.published    # => true
        #   Article.create.published # => true
1140
        def default_scope(options = {})
1141
          reset_scoped_methods
1142 1143
          default_scoping = self.default_scoping.dup
          self.default_scoping = default_scoping << construct_finder_arel(options, default_scoping.pop)
1144 1145
        end

D
David Heinemeier Hansson 已提交
1146
        def current_scoped_methods #:nodoc:
T
Tim Morgan 已提交
1147 1148
          method = scoped_methods.last
          if method.respond_to?(:call)
1149
            relation.scoping { method.call }
T
Tim Morgan 已提交
1150 1151 1152
          else
            method
          end
1153
        end
1154

1155 1156 1157 1158
        def reset_scoped_methods #:nodoc:
          Thread.current[:"#{self}_scoped_methods"] = nil
        end

P
Pratik Naik 已提交
1159
        # Returns the class type of the record using the current module as a prefix. So descendants of
1160
        # MyApp::Business::Account would appear as MyApp::Business::AccountSubclass.
D
Initial  
David Heinemeier Hansson 已提交
1161
        def compute_type(type_name)
1162 1163 1164
          if type_name.match(/^::/)
            # If the type is prefixed with a scope operator then we assume that
            # the type_name is an absolute reference.
1165
            ActiveSupport::Dependencies.constantize(type_name)
1166 1167 1168 1169 1170 1171 1172 1173
          else
            # Build a list of candidates to search for
            candidates = []
            name.scan(/::|$/) { candidates.unshift "#{$`}::#{type_name}" }
            candidates << type_name

            candidates.each do |candidate|
              begin
1174
                constant = ActiveSupport::Dependencies.constantize(candidate)
1175
                return constant if candidate == constant.to_s
1176 1177 1178
              rescue NameError => e
                # We don't want to swallow NoMethodError < NameError errors
                raise e unless e.instance_of?(NameError)
1179 1180
              rescue ArgumentError
              end
1181
            end
1182 1183

            raise NameError, "uninitialized constant #{candidates.first}"
D
Initial  
David Heinemeier Hansson 已提交
1184 1185 1186
          end
        end

P
Pratik Naik 已提交
1187 1188
        # Returns the class descending directly from ActiveRecord::Base or an
        # abstract class, if any, in the inheritance hierarchy.
1189
        def class_of_active_record_descendant(klass)
1190
          if klass.superclass == Base || klass.superclass.abstract_class?
1191
            klass
D
Initial  
David Heinemeier Hansson 已提交
1192 1193 1194
          elsif klass.superclass.nil?
            raise ActiveRecordError, "#{name} doesn't belong in a hierarchy descending from ActiveRecord"
          else
1195
            class_of_active_record_descendant(klass.superclass)
D
Initial  
David Heinemeier Hansson 已提交
1196 1197 1198
          end
        end

P
Pratik Naik 已提交
1199
        # Accepts an array, hash, or string of SQL conditions and sanitizes
1200
        # them into a valid SQL fragment for a WHERE clause.
1201 1202 1203
        #   ["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'"
1204
        def sanitize_sql_for_conditions(condition, table_name = self.table_name)
1205 1206
          return nil if condition.blank?

1207 1208
          case condition
            when Array; sanitize_sql_array(condition)
1209
            when Hash;  sanitize_sql_hash_for_conditions(condition, table_name)
1210 1211
            else        condition
          end
1212
        end
1213
        alias_method :sanitize_sql, :sanitize_sql_for_conditions
1214

P
Pratik Naik 已提交
1215
        # Accepts an array, hash, or string of SQL conditions and sanitizes
1216 1217 1218 1219 1220 1221 1222 1223 1224 1225
        # 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

1226 1227 1228 1229 1230
        def aggregate_mapping(reflection)
          mapping = reflection.options[:mapping] || [reflection.name, reflection.name]
          mapping.first.is_a?(Array) ? mapping : [mapping]
        end

P
Pratik Naik 已提交
1231
        # Accepts a hash of SQL conditions and replaces those attributes
1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260
        # 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

1261
        # Sanitizes a hash of attribute/value pairs into SQL conditions for a WHERE clause.
1262 1263 1264 1265
        #   { :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)"
1266 1267
        #   { :age => 13..18 }
        #     # => "age BETWEEN 13 AND 18"
1268 1269
        #   { 'other_records.id' => 7 }
        #     # => "`other_records`.`id` = 7"
1270 1271
        #   { :other_records => { :id => 7 } }
        #     # => "`other_records`.`id` = 7"
1272 1273 1274
        # 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'"
1275
        def sanitize_sql_hash_for_conditions(attrs, default_table_name = self.table_name)
1276 1277
          attrs = expand_hash_conditions_for_aggregates(attrs)

1278
          table = Arel::Table.new(self.table_name, :engine => arel_engine, :as => default_table_name)
1279 1280 1281 1282
          viz = Arel::Visitors.for(arel_engine)
          PredicateBuilder.build_from_hash(arel_engine, attrs, table).map { |b|
            viz.accept b
          }.join(' AND ')
1283
        end
1284 1285 1286 1287 1288 1289
        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)
1290
          attrs.map do |attr, value|
1291 1292 1293
            "#{connection.quote_column_name(attr)} = #{quote_bound_value(value)}"
          end.join(', ')
        end
1294

1295
        # Accepts an array of conditions.  The array has each value
P
Pratik Naik 已提交
1296
        # sanitized and interpolated into the SQL statement.
1297
        #   ["name='%s' and group_id='%s'", "foo'bar", 4]  returns  "name='foo''bar' and group_id='4'"
1298
        def sanitize_sql_array(ary)
1299
          statement, *values = ary
1300
          if values.first.is_a?(Hash) && statement =~ /:\w+/
1301 1302
            replace_named_bind_variables(statement, values.first)
          elsif statement.include?('?')
1303
            replace_bind_variables(statement, values)
1304 1305
          elsif statement.blank?
            statement
1306
          else
1307
            statement % values.collect { |value| connection.quote_string(value.to_s) }
1308
          end
1309 1310
        end

1311 1312
        alias_method :sanitize_conditions, :sanitize_sql

D
David Heinemeier Hansson 已提交
1313
        def replace_bind_variables(statement, values) #:nodoc:
1314
          raise_if_bind_arity_mismatch(statement, statement.count('?'), values.size)
1315
          bound = values.dup
C
Comron Sattari 已提交
1316 1317
          c = connection
          statement.gsub('?') { quote_bound_value(bound.shift, c) }
1318 1319
        end

D
David Heinemeier Hansson 已提交
1320
        def replace_named_bind_variables(statement, bind_vars) #:nodoc:
1321 1322 1323 1324
          statement.gsub(/(:?):([a-zA-Z]\w*)/) do
            if $1 == ':' # skip postgresql casts
              $& # return the whole match
            elsif bind_vars.include?(match = $2.to_sym)
1325
              quote_bound_value(bind_vars[match])
1326 1327
            else
              raise PreparedStatementInvalid, "missing value for :#{match} in #{statement}"
1328 1329
            end
          end
1330 1331
        end

1332
        def expand_range_bind_variables(bind_vars) #:nodoc:
1333 1334 1335
          expanded = []

          bind_vars.each do |var|
1336 1337
            next if var.is_a?(Hash)

1338
            if var.is_a?(Range)
1339 1340
              expanded << var.first
              expanded << var.last
1341
            else
1342
              expanded << var
1343
            end
1344
          end
1345 1346

          expanded
1347 1348
        end

C
Comron Sattari 已提交
1349
        def quote_bound_value(value, c = connection) #:nodoc:
1350
          if value.respond_to?(:map) && !value.acts_like?(:string)
1351
            if value.respond_to?(:empty?) && value.empty?
C
Comron Sattari 已提交
1352
              c.quote(nil)
1353
            else
C
Comron Sattari 已提交
1354
              value.map { |v| c.quote(v) }.join(',')
1355
            end
1356
          else
C
Comron Sattari 已提交
1357
            c.quote(value)
1358 1359 1360
          end
        end

D
David Heinemeier Hansson 已提交
1361
        def raise_if_bind_arity_mismatch(statement, expected, provided) #:nodoc:
1362 1363 1364
          unless expected == provided
            raise PreparedStatementInvalid, "wrong number of bind variables (#{provided} for #{expected}) in: #{statement}"
          end
1365
        end
1366

D
David Heinemeier Hansson 已提交
1367
        def encode_quoted_value(value) #:nodoc:
1368
          quoted_value = connection.quote(value)
1369 1370
          quoted_value = "'#{quoted_value[1..-2].gsub(/\'/, "\\\\'")}'" if quoted_value.include?("\\\'") # (for ruby mode) "
          quoted_value
1371
        end
D
Initial  
David Heinemeier Hansson 已提交
1372 1373 1374 1375 1376
    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).
1377
      # In both instances, valid attribute keys are determined by the column names of the associated table --
D
Initial  
David Heinemeier Hansson 已提交
1378 1379
      # hence you can't have attributes that aren't part of the table columns.
      def initialize(attributes = nil)
1380
        @attributes = attributes_from_column_definition
1381
        @association_cache = {}
1382
        @aggregation_cache = {}
1383
        @attributes_cache = {}
1384
        @new_record = true
1385
        @readonly = false
1386 1387
        @destroyed = false
        @marked_for_destruction = false
1388 1389
        @previously_changed = {}
        @changed_attributes = {}
1390

D
Initial  
David Heinemeier Hansson 已提交
1391
        ensure_proper_type
P
Pratik Naik 已提交
1392

1393
        populate_with_current_scope_attributes
1394
        self.attributes = attributes unless attributes.nil?
P
Pratik Naik 已提交
1395

1396
        result = yield self if block_given?
1397
        _run_initialize_callbacks
1398
        result
D
Initial  
David Heinemeier Hansson 已提交
1399
      end
1400

1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416
      # 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

1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429
      # 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']
        @attributes_cache, @previously_changed, @changed_attributes = {}, {}, {}
1430
        @association_cache = {}
1431
        @aggregation_cache = {}
1432
        @readonly = @destroyed = @marked_for_destruction = false
1433
        @new_record = false
1434 1435 1436 1437
        _run_find_callbacks
        _run_initialize_callbacks
      end

J
Jon Leighton 已提交
1438 1439 1440 1441 1442 1443 1444 1445 1446 1447
      # 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 已提交
1448 1449 1450 1451
      # 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 已提交
1452
      # For example, suppose that you have a User model, and that you have a
1453
      # <tt>resources :users</tt> route. Normally, +user_path+ will
P
Pratik Naik 已提交
1454
      # construct a path with the user object's 'id' in it:
P
Pratik Naik 已提交
1455 1456
      #
      #   user = User.find_by_name('Phusion')
1457
      #   user_path(user)  # => "/users/1"
P
Pratik Naik 已提交
1458
      #
P
Pratik Naik 已提交
1459 1460
      # 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 已提交
1461 1462 1463 1464 1465 1466
      #
      #   class User < ActiveRecord::Base
      #     def to_param  # overridden
      #       name
      #     end
      #   end
E
Emilio Tagua 已提交
1467
      #
P
Pratik Naik 已提交
1468
      #   user = User.find_by_name('Phusion')
1469
      #   user_path(user)  # => "/users/Phusion"
1470
      def to_param
1471
        # We can't use alias_method here, because method 'id' optimizes itself on the fly.
1472
        id && id.to_s # Be sure to stringify the id for routes
1473
      end
1474

P
Pratik Naik 已提交
1475 1476 1477
      # Returns a cache key that can be used to identify this record.
      #
      # ==== Examples
1478 1479 1480 1481 1482
      #
      #   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
1483
        case
1484
        when new_record?
J
Jeremy Kemper 已提交
1485 1486 1487
          "#{self.class.model_name.cache_key}/new"
        when timestamp = self[:updated_at]
          "#{self.class.model_name.cache_key}/#{id}-#{timestamp.to_s(:number)}"
1488
        else
J
Jeremy Kemper 已提交
1489
          "#{self.class.model_name.cache_key}/#{id}"
1490 1491
        end
      end
1492

1493
      def quoted_id #:nodoc:
1494
        quote_value(id, column_for_attribute(self.class.primary_key))
1495
      end
1496

1497 1498 1499 1500 1501
      # Returns true if the given attribute is in the attributes hash
      def has_attribute?(attr_name)
        @attributes.has_key?(attr_name.to_s)
      end

1502
      # Returns an array of names for the attributes available on this object.
1503
      def attribute_names
1504
        @attributes.keys
1505 1506
      end

D
Initial  
David Heinemeier Hansson 已提交
1507
      # Allows you to set all the attributes at once by passing in a hash with keys
P
Pratik Naik 已提交
1508 1509 1510 1511 1512 1513
      # 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
1514
      # attributes not included in that won't be allowed to be mass-assigned.
P
Pratik Naik 已提交
1515 1516 1517 1518
      #
      #   class User < ActiveRecord::Base
      #     attr_protected :is_admin
      #   end
E
Emilio Tagua 已提交
1519
      #
P
Pratik Naik 已提交
1520 1521 1522 1523
      #   user = User.new
      #   user.attributes = { :username => 'Phusion', :is_admin => true }
      #   user.username   # => "Phusion"
      #   user.is_admin?  # => false
E
Emilio Tagua 已提交
1524
      #
P
Pratik Naik 已提交
1525 1526
      #   user.send(:attributes=, { :username => 'Phusion', :is_admin => true }, false)
      #   user.is_admin?  # => true
1527
      def attributes=(new_attributes, guard_protected_attributes = true)
P
Pratik Naik 已提交
1528
        return unless new_attributes.is_a?(Hash)
1529
        attributes = new_attributes.stringify_keys
D
Initial  
David Heinemeier Hansson 已提交
1530 1531

        multi_parameter_attributes = []
1532
        attributes = sanitize_for_mass_assignment(attributes) if guard_protected_attributes
1533

1534
        attributes.each do |k, v|
1535 1536
          if k.include?("(")
            multi_parameter_attributes << [ k, v ]
J
José Valim 已提交
1537 1538
          elsif respond_to?("#{k}=")
            send("#{k}=", v)
1539
          else
J
José Valim 已提交
1540
            raise(UnknownAttributeError, "unknown attribute: #{k}")
1541
          end
D
Initial  
David Heinemeier Hansson 已提交
1542
        end
D
David Heinemeier Hansson 已提交
1543

D
Initial  
David Heinemeier Hansson 已提交
1544 1545 1546
        assign_multiparameter_attributes(multi_parameter_attributes)
      end

1547
      # Returns a hash of all the attributes with their names as keys and the values of the attributes as values.
1548
      def attributes
1549
        Hash[@attributes.map { |name, _| [name, read_attribute(name)] }]
1550 1551
      end

P
Pratik Naik 已提交
1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564
      # 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"'
1565 1566 1567
      def attribute_for_inspect(attr_name)
        value = read_attribute(attr_name)

1568
        if value.is_a?(String) && value.length > 50
1569
          "#{value[0..50]}...".inspect
1570 1571
        elsif value.is_a?(Date) || value.is_a?(Time)
          %("#{value.to_s(:db)}")
1572 1573 1574 1575 1576
        else
          value.inspect
        end
      end

D
Initial  
David Heinemeier Hansson 已提交
1577
      # Returns true if the specified +attribute+ has been set by the user or by a database load and is neither
1578
      # nil nor empty? (the latter only applies to objects that respond to empty?, most notably Strings).
D
Initial  
David Heinemeier Hansson 已提交
1579
      def attribute_present?(attribute)
E
Emilio Tagua 已提交
1580
        !read_attribute(attribute).blank?
D
Initial  
David Heinemeier Hansson 已提交
1581 1582 1583 1584
      end

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

1588
      # Returns true if +comparison_object+ is the same exact object, or +comparison_object+
X
Xavier Noria 已提交
1589 1590 1591 1592 1593 1594 1595 1596
      # 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 已提交
1597
      def ==(comparison_object)
1598
        comparison_object.equal?(self) ||
1599 1600 1601
          comparison_object.instance_of?(self.class) &&
          id.present? &&
          comparison_object.id == id
D
Initial  
David Heinemeier Hansson 已提交
1602 1603 1604 1605
      end

      # Delegates to ==
      def eql?(comparison_object)
E
Emilio Tagua 已提交
1606
        self == comparison_object
D
Initial  
David Heinemeier Hansson 已提交
1607
      end
1608

D
Initial  
David Heinemeier Hansson 已提交
1609 1610 1611
      # 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
1612
        id.hash
D
Initial  
David Heinemeier Hansson 已提交
1613 1614
      end

1615
      # Freeze the attributes hash such that associations are still accessible, even on destroyed records.
1616
      def freeze
1617
        @attributes.freeze; self
1618
      end
1619

1620
      # Returns +true+ if the attributes hash has been frozen.
1621 1622 1623
      def frozen?
        @attributes.frozen?
      end
1624

A
Aaron Patterson 已提交
1625 1626 1627 1628 1629 1630 1631 1632 1633
      # 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 已提交
1634
      # Duped objects have no id assigned and are treated as new records. Note
A
Aaron Patterson 已提交
1635 1636
      # 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 已提交
1637 1638
      # specific and is therefore left to the application to implement according
      # to its need.
1639
      # The dup method does not preserve the timestamps (created|updated)_(at|on).
A
Aaron Patterson 已提交
1640
      def initialize_dup(other)
A
Aaron Patterson 已提交
1641 1642 1643
        cloned_attributes = other.clone_attributes(:read_attribute_before_type_cast)
        cloned_attributes.delete(self.class.primary_key)

A
Aaron Patterson 已提交
1644 1645
        @attributes = cloned_attributes

A
Aaron Patterson 已提交
1646 1647
        _run_after_initialize_callbacks if respond_to?(:_run_after_initialize_callbacks)

A
Aaron Patterson 已提交
1648 1649 1650 1651
        @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 已提交
1652

1653
        @aggregation_cache = {}
1654 1655
        @association_cache = {}
        @attributes_cache = {}
1656
        @new_record  = true
A
Aaron Patterson 已提交
1657

A
Aaron Patterson 已提交
1658 1659
        ensure_proper_type
        populate_with_current_scope_attributes
1660
        clear_timestamp_attributes
A
Aaron Patterson 已提交
1661 1662
      end

1663 1664
      # 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.
1665
      def readonly?
1666
        @readonly
1667 1668
      end

1669 1670
      # Marks this record as read only.
      def readonly!
1671 1672
        @readonly = true
      end
1673

1674
      # Returns the contents of the record as a nicely formatted string.
1675
      def inspect
1676
        attributes_as_nice_string = self.class.column_names.collect { |name|
1677
          if has_attribute?(name)
1678 1679 1680
            "#{name}: #{attribute_for_inspect(name)}"
          end
        }.compact.join(", ")
1681
        "#<#{self.class} #{attributes_as_nice_string}>"
1682
      end
1683

1684 1685
    protected
      def clone_attributes(reader_method = :read_attribute, attributes = {})
1686 1687
        attribute_names.each do |name|
          attributes[name] = clone_attribute_value(reader_method, name)
1688
        end
1689
        attributes
1690 1691 1692 1693 1694 1695 1696 1697 1698
      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 已提交
1699 1700
    private

1701
      # Sets the attribute used for single table inheritance to this class name if this is not the
1702
      # ActiveRecord::Base descendant.
1703 1704
      # 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.
1705
      # No such attribute would be set for objects of the Message class in that example.
D
Initial  
David Heinemeier Hansson 已提交
1706 1707
      def ensure_proper_type
        unless self.class.descends_from_active_record?
1708
          write_attribute(self.class.inheritance_column, self.class.sti_name)
D
Initial  
David Heinemeier Hansson 已提交
1709 1710 1711
        end
      end

1712
      # The primary key and inheritance column can never be set by mass-assignment for security reasons.
1713 1714 1715
      def self.attributes_protected_by_default
        default = [ primary_key, inheritance_column ]
        default << 'id' unless primary_key.eql? 'id'
1716
        default
1717 1718
      end

1719 1720
      # Returns a copy of the attributes hash where all the values have been safely quoted for use in
      # an Arel insert/update method.
1721 1722 1723 1724 1725 1726
      def arel_attributes_values(include_primary_key = true, include_readonly_attributes = true, attribute_names = @attributes.keys)
        attrs = {}
        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))
1727 1728
              value = read_attribute(name)

1729
              if !value.nil? && self.class.serialized_attributes.key?(name)
1730
                value = YAML.dump value
1731
              end
1732
              attrs[self.class.arel_table[name]] = value
1733 1734 1735 1736 1737 1738
            end
          end
        end
        attrs
      end

D
Initial  
David Heinemeier Hansson 已提交
1739
      # Quote strings appropriately for SQL statements.
1740
      def quote_value(value, column = nil)
1741
        self.class.connection.quote(value, column)
D
Initial  
David Heinemeier Hansson 已提交
1742 1743
      end

P
Pratik Naik 已提交
1744
      # Interpolate custom SQL string in instance context.
D
Initial  
David Heinemeier Hansson 已提交
1745 1746
      # Optional record argument is meant for custom insert_sql.
      def interpolate_sql(sql, record = nil)
1747
        instance_eval("%@#{sql.gsub('@', '\@')}@", __FILE__, __LINE__)
D
Initial  
David Heinemeier Hansson 已提交
1748 1749 1750 1751 1752 1753
      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
1754 1755
      # 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
1756
      # attribute will be set to nil.
D
Initial  
David Heinemeier Hansson 已提交
1757 1758 1759 1760 1761
      def assign_multiparameter_attributes(pairs)
        execute_callstack_for_multiparameter_attributes(
          extract_callstack_for_multiparameter_attributes(pairs)
        )
      end
1762

1763
      def instantiate_time_object(name, values)
1764
        if self.class.send(:create_time_zone_conversion_attribute?, name, column_for_attribute(name))
1765
          Time.zone.local(*values)
1766
        else
1767
          Time.time_with_datetime_fallback(@@default_timezone, *values)
1768
        end
1769 1770
      end

D
Initial  
David Heinemeier Hansson 已提交
1771
      def execute_callstack_for_multiparameter_attributes(callstack)
1772
        errors = []
1773
        callstack.each do |name, values_with_empty_parameters|
1774 1775
          begin
            klass = (self.class.reflect_on_aggregation(name.to_sym) || column_for_attribute(name)).klass
1776 1777
            # 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.
1778
            values = values_with_empty_parameters.reject { |v| v.nil? }
1779

1780 1781 1782
            if values.empty?
              send(name + "=", nil)
            else
1783

1784
              value = if Time == klass
1785
                instantiate_time_object(name, values)
1786 1787
              elsif Date == klass
                begin
1788
                  values = values_with_empty_parameters.collect do |v| v.nil? ? 1 : v end
1789 1790
                  Date.new(*values)
                rescue ArgumentError => ex # if Date.new raises an exception on an invalid date
1791
                  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
1792 1793 1794 1795 1796 1797
                end
              else
                klass.new(*values)
              end

              send(name + "=", value)
1798
            end
1799 1800
          rescue => ex
            errors << AttributeAssignmentError.new("error on assignment #{values.inspect} to #{name}", ex, name)
D
Initial  
David Heinemeier Hansson 已提交
1801 1802
          end
        end
1803 1804 1805
        unless errors.empty?
          raise MultiparameterAssignmentErrors.new(errors), "#{errors.size} error(s) on assignment of multiparameter attributes"
        end
D
Initial  
David Heinemeier Hansson 已提交
1806
      end
1807

D
Initial  
David Heinemeier Hansson 已提交
1808 1809 1810 1811 1812 1813 1814 1815
      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)

1816 1817
          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 已提交
1818 1819 1820 1821
        end

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

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

D
Initial  
David Heinemeier Hansson 已提交
1827 1828 1829
      def find_parameter_position(multiparameter_name)
        multiparameter_name.scan(/\(([0-9]*).*\)/).first.first
      end
1830

D
Initial  
David Heinemeier Hansson 已提交
1831 1832
      # Returns a comma-separated pair list, like "key1 = val1, key2 = val2".
      def comma_pair_list(hash)
1833
        hash.map { |k,v| "#{k} = #{v}" }.join(", ")
D
Initial  
David Heinemeier Hansson 已提交
1834 1835
      end

1836
      def quote_columns(quoter, hash)
1837
        Hash[hash.map { |name, value| [quoter.quote_column_name(name), value] }]
D
Initial  
David Heinemeier Hansson 已提交
1838 1839
      end

1840 1841
      def quoted_comma_pair_list(quoter, hash)
        comma_pair_list(quote_columns(quoter, hash))
D
Initial  
David Heinemeier Hansson 已提交
1842 1843
      end

1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859
      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

      def object_from_yaml(string)
        return string unless string.is_a?(String) && string =~ /^---/
        YAML::load(string) rescue string
      end
1860 1861 1862 1863

      def populate_with_current_scope_attributes
        if scope = self.class.send(:current_scoped_methods)
          create_with = scope.scope_for_create
1864
          create_with.each { |att,value|
J
José Valim 已提交
1865
            respond_to?("#{att}=") && send("#{att}=", value)
1866
          }
1867 1868
        end
      end
1869

1870
      # Clear attributes and changed_attributes
1871 1872
      def clear_timestamp_attributes
        %w(created_at created_on updated_at updated_on).each do |attribute_name|
X
Xavier Noria 已提交
1873
          if has_attribute?(attribute_name)
1874
            self[attribute_name] = nil
X
Xavier Noria 已提交
1875
            changed_attributes.delete(attribute_name)
1876 1877 1878
          end
        end
      end
D
Initial  
David Heinemeier Hansson 已提交
1879
  end
J
Joshua Peek 已提交
1880 1881

  Base.class_eval do
1882
    include ActiveRecord::Persistence
J
Joshua Peek 已提交
1883
    extend ActiveModel::Naming
J
Joshua Peek 已提交
1884
    extend QueryCache::ClassMethods
J
José Valim 已提交
1885
    extend ActiveSupport::Benchmarkable
1886
    extend ActiveSupport::DescendantsTracker
J
José Valim 已提交
1887

1888
    include ActiveModel::Conversion
J
Joshua Peek 已提交
1889
    include Validations
1890
    extend CounterCache
J
Joshua Peek 已提交
1891 1892
    include Locking::Optimistic, Locking::Pessimistic
    include AttributeMethods
J
Joshua Peek 已提交
1893
    include AttributeMethods::Read, AttributeMethods::Write, AttributeMethods::BeforeTypeCast, AttributeMethods::Query
1894
    include AttributeMethods::PrimaryKey
J
Joshua Peek 已提交
1895 1896
    include AttributeMethods::TimeZoneConversion
    include AttributeMethods::Dirty
1897
    include ActiveModel::MassAssignmentSecurity
1898
    include Callbacks, ActiveModel::Observing, Timestamp
J
Joshua Peek 已提交
1899
    include Associations, AssociationPreload, NamedScope
1900
    include ActiveModel::SecurePassword
1901 1902 1903 1904

    # 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 已提交
1905
    include Aggregations, Transactions, Reflection, Serialization
1906

1907
    NilClass.add_whiner(self) if NilClass.respond_to?(:add_whiner)
1908 1909 1910 1911 1912 1913 1914 1915 1916 1917 1918

    # 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 已提交
1919
  end
1920
end
1921 1922 1923

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