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

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

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

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

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

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

P
Pratik Naik 已提交
376 377
    ##
    # :singleton-method:
D
Initial  
David Heinemeier Hansson 已提交
378 379
    # 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.
380 381
    class_attribute :table_name_suffix, :instance_writer => false
    self.table_name_suffix = ""
D
Initial  
David Heinemeier Hansson 已提交
382

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

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

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

P
Pratik Naik 已提交
409 410
    ##
    # :singleton-method:
411
    # Specify whether or not to use timestamps for migration versions
412 413 414
    cattr_accessor :timestamped_migrations , :instance_writer => false
    @@timestamped_migrations = true

415
    # Determine whether to store the full constant name including namespace when using STI
416
    class_attribute :store_full_sti_class
417
    self.store_full_sti_class = true
418

419
    # Stores the default scope for the class
420
    class_attribute :default_scoping, :instance_writer => false
421 422
    self.default_scoping = []

J
Jeremy Kemper 已提交
423 424 425 426 427
    # 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 = {}

428 429 430
    class_attribute :_attr_readonly, :instance_writer => false
    self._attr_readonly = []

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

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

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

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

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

518 519
      # Returns an array of all the attributes that have been specified as readonly.
      def readonly_attributes
520
        self._attr_readonly
D
Initial  
David Heinemeier Hansson 已提交
521 522
      end

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

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

581
      # Returns a quoted version of the table name, used to construct SQL statements.
P
Pratik Naik 已提交
582 583 584 585
      def quoted_table_name
        @quoted_table_name ||= connection.quote_table_name(table_name)
      end

586
      # Computes the table name, (re)sets it internally, and returns it.
D
David Heinemeier Hansson 已提交
587
      def reset_table_name #:nodoc:
588
        self.table_name = compute_table_name
D
Initial  
David Heinemeier Hansson 已提交
589 590
      end

591
      def full_table_name_prefix #:nodoc:
592 593 594
        (parents.detect{ |p| p.respond_to?(:table_name_prefix) } || self).table_name_prefix
      end

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

601 602
      # 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 已提交
603
      def sequence_name #:nodoc:
604 605 606
        reset_sequence_name
      end

D
David Heinemeier Hansson 已提交
607
      def reset_sequence_name #:nodoc:
608 609 610
        default = connection.default_sequence_name(table_name, primary_key)
        set_sequence_name(default)
        default
611 612
      end

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

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

658 659
      # Indicates whether the table associated with this class exists
      def table_exists?
660
        connection.table_exists?(table_name)
661 662
      end

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

672
      # Returns a hash of column objects for the table associated with this class.
D
Initial  
David Heinemeier Hansson 已提交
673
      def columns_hash
674
        @columns_hash ||= Hash[columns.map { |column| [column.name, column] }]
D
Initial  
David Heinemeier Hansson 已提交
675
      end
676

D
David Heinemeier Hansson 已提交
677
      # Returns an array of column names as strings.
678
      def column_names
679
        @column_names ||= columns.map { |column| column.name }
680
      end
D
Initial  
David Heinemeier Hansson 已提交
681

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

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

735
      def attribute_method?(attribute)
J
José Valim 已提交
736
        super || (table_exists? && column_names.include?(attribute.to_s.sub(/=$/, '')))
737 738
      end

739 740
      # Set the lookup ancestors for ActiveModel.
      def lookup_ancestors #:nodoc:
741 742
        klass = self
        classes = [klass]
743 744
        return classes if klass == ActiveRecord::Base

E
Emilio Tagua 已提交
745
        while klass != klass.base_class
746 747 748 749 750
          classes << klass = klass.superclass
        end
        classes
      end

751 752 753
      # Set the i18n scope to overwrite ActiveModel.
      def i18n_scope #:nodoc:
        :activerecord
D
Initial  
David Heinemeier Hansson 已提交
754
      end
755

756 757 758 759 760 761 762
      # 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 已提交
763 764
      end

765 766 767 768 769
      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

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

784
      def quote_value(value, column = nil) #:nodoc:
785
        connection.quote(value,column)
786 787
      end

788
      # Used to sanitize objects before they're used in an SQL SELECT statement. Delegates to <tt>connection.quote</tt>.
789
      def sanitize(object) #:nodoc:
790
        connection.quote(object)
D
Initial  
David Heinemeier Hansson 已提交
791 792
      end

793 794 795
      # Overwrite the default class equality method to provide support for association proxies.
      def ===(object)
        object.is_a?(self)
796
      end
797

798 799 800
      # 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.
801 802 803
      #
      # 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.
804 805 806 807
      def base_class
        class_of_active_record_descendant(self)
      end

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

811
      # Returns whether this class is an abstract class or not.
812
      def abstract_class?
813
        defined?(@abstract_class) && @abstract_class == true
814 815
      end

816
      def respond_to?(method_id, include_private = false)
817 818
        if match = DynamicFinderMatch.match(method_id)
          return true if all_attributes_exists?(match.attribute_names)
819 820
        elsif match = DynamicScopeMatch.match(method_id)
          return true if all_attributes_exists?(match.attribute_names)
821
        end
822

823 824 825
        super
      end

826 827 828 829
      def sti_name
        store_full_sti_class ? name : name.demodulize
      end

P
Pratik Naik 已提交
830
      def arel_table
831
        @arel_table ||= Arel::Table.new(table_name, arel_engine)
832 833
      end

834 835
      def arel_engine
        @arel_engine ||= begin
836
          if self == ActiveRecord::Base
837
            ActiveRecord::Base
838
          else
839
            connection_handler.connection_pools[name] ? self : superclass.arel_engine
840 841
          end
        end
842 843
      end

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

874 875 876 877
      def before_remove_const #:nodoc:
        reset_scoped_methods
      end

J
Jon Leighton 已提交
878 879 880 881 882 883 884 885 886 887
      # 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

888 889 890 891 892 893 894 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.
      def instantiate(record) # :nodoc:
        model = find_sti_class(record[inheritance_column]).allocate
        model.init_with('attributes' => record)
        model
      end

D
Initial  
David Heinemeier Hansson 已提交
898
      private
J
José Valim 已提交
899 900 901

        def relation #:nodoc:
          @relation ||= Relation.new(self, arel_table)
902 903 904 905 906 907

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

910
        def find_sti_class(type_name)
911
          if type_name.blank? || !columns_hash.include?(inheritance_column)
912 913 914
            self
          else
            begin
915 916 917 918 919
              if store_full_sti_class
                ActiveSupport::Dependencies.constantize(type_name)
              else
                compute_type(type_name)
              end
920 921 922 923 924 925 926 927 928 929
            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

930
        def construct_finder_arel(options = {}, scope = nil)
931
          relation = options.is_a?(Hash) ? unscoped.apply_finder_options(options) : options
932
          relation = scope.merge(relation) if scope
933
          relation
934 935
        end

936
        def type_condition
937
          sti_column = arel_table[inheritance_column.to_sym]
938
          sti_names  = ([self] + descendants).map { |model| model.sti_name }
939

940
          sti_column.in(sti_names)
D
Initial  
David Heinemeier Hansson 已提交
941 942 943
        end

        # Guesses the table name, but does not decorate it with prefix and suffix information.
944
        def undecorated_table_name(class_name = base_class.name)
945 946
          table_name = class_name.to_s.demodulize.underscore
          table_name = table_name.pluralize if pluralize_table_names
947
          table_name
D
Initial  
David Heinemeier Hansson 已提交
948 949
        end

950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966
        # 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

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

1005 1006
        # Similar in purpose to +expand_hash_conditions_for_aggregates+.
        def expand_attribute_names_for_aggregates(attribute_names)
1007
          attribute_names.map { |attribute_name|
1008
            unless (aggregation = reflect_on_aggregation(attribute_name.to_sym)).nil?
1009 1010
              aggregate_mapping(aggregation).map do |field_attr, _|
                field_attr.to_sym
1011 1012
              end
            else
1013
              attribute_name.to_sym
1014
            end
1015
          }.flatten
1016 1017
        end

1018
        def all_attributes_exists?(attribute_names)
1019 1020
          (expand_attribute_names_for_aggregates(attribute_names) -
           column_methods_hash.keys).empty?
1021
        end
1022

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

1074 1075
          if method_scoping.is_a?(Hash)
            # Dup first and second level of hash (method and params).
1076 1077 1078
            method_scoping = method_scoping.dup
            method_scoping.each do |method, params|
              method_scoping[method] = params.dup unless params == true
1079
            end
1080

1081 1082 1083 1084
            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]
1085
              scope_for_create = if action == :merge
1086 1087 1088
                current_scoped_methods.create_with_value.merge(method_scoping[:create])
              else
                method_scoping[:create]
1089
              end
1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100

              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

1101
          method_scoping = current_scoped_methods.merge(method_scoping) if current_scoped_methods && action ==  :merge
1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112

          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)
1113
          if method_scoping.values.any? { |e| e.is_a?(ActiveRecord::Relation) }
1114 1115
            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:
1116

1117 1118 1119 1120 1121 1122 1123 1124 1125
  User.unscoped.where(:active => true)

Or call unscoped with a block:

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

MSG
1126
          end
1127 1128 1129
          with_scope(method_scoping, :overwrite, &block)
        end

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

D
David Heinemeier Hansson 已提交
1152
        def current_scoped_methods #:nodoc:
T
Tim Morgan 已提交
1153 1154
          method = scoped_methods.last
          if method.respond_to?(:call)
1155
            relation.scoping { method.call }
T
Tim Morgan 已提交
1156 1157 1158
          else
            method
          end
1159
        end
1160

1161 1162 1163 1164
        def reset_scoped_methods #:nodoc:
          Thread.current[:"#{self}_scoped_methods"] = nil
        end

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

            candidates.each do |candidate|
              begin
1180
                constant = ActiveSupport::Dependencies.constantize(candidate)
1181
                return constant if candidate == constant.to_s
1182 1183 1184
              rescue NameError => e
                # We don't want to swallow NoMethodError < NameError errors
                raise e unless e.instance_of?(NameError)
1185 1186
              rescue ArgumentError
              end
1187
            end
1188 1189

            raise NameError, "uninitialized constant #{candidates.first}"
D
Initial  
David Heinemeier Hansson 已提交
1190 1191 1192
          end
        end

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

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

1213 1214
          case condition
            when Array; sanitize_sql_array(condition)
1215
            when Hash;  sanitize_sql_hash_for_conditions(condition, table_name)
1216 1217
            else        condition
          end
1218
        end
1219
        alias_method :sanitize_sql, :sanitize_sql_for_conditions
1220

P
Pratik Naik 已提交
1221
        # Accepts an array, hash, or string of SQL conditions and sanitizes
1222 1223 1224 1225 1226 1227 1228 1229 1230 1231
        # 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

1232 1233 1234 1235 1236
        def aggregate_mapping(reflection)
          mapping = reflection.options[:mapping] || [reflection.name, reflection.name]
          mapping.first.is_a?(Array) ? mapping : [mapping]
        end

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

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

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

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

1317 1318
        alias_method :sanitize_conditions, :sanitize_sql

D
David Heinemeier Hansson 已提交
1319
        def replace_bind_variables(statement, values) #:nodoc:
1320
          raise_if_bind_arity_mismatch(statement, statement.count('?'), values.size)
1321
          bound = values.dup
C
Comron Sattari 已提交
1322 1323
          c = connection
          statement.gsub('?') { quote_bound_value(bound.shift, c) }
1324 1325
        end

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

1338
        def expand_range_bind_variables(bind_vars) #:nodoc:
1339 1340 1341
          expanded = []

          bind_vars.each do |var|
1342 1343
            next if var.is_a?(Hash)

1344
            if var.is_a?(Range)
1345 1346
              expanded << var.first
              expanded << var.last
1347
            else
1348
              expanded << var
1349
            end
1350
          end
1351 1352

          expanded
1353 1354
        end

C
Comron Sattari 已提交
1355
        def quote_bound_value(value, c = connection) #:nodoc:
1356
          if value.respond_to?(:map) && !value.acts_like?(:string)
1357
            if value.respond_to?(:empty?) && value.empty?
C
Comron Sattari 已提交
1358
              c.quote(nil)
1359
            else
C
Comron Sattari 已提交
1360
              value.map { |v| c.quote(v) }.join(',')
1361
            end
1362
          else
C
Comron Sattari 已提交
1363
            c.quote(value)
1364 1365 1366
          end
        end

D
David Heinemeier Hansson 已提交
1367
        def raise_if_bind_arity_mismatch(statement, expected, provided) #:nodoc:
1368 1369 1370
          unless expected == provided
            raise PreparedStatementInvalid, "wrong number of bind variables (#{provided} for #{expected}) in: #{statement}"
          end
1371
        end
1372

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

D
Initial  
David Heinemeier Hansson 已提交
1397
        ensure_proper_type
P
Pratik Naik 已提交
1398

1399
        populate_with_current_scope_attributes
1400
        self.attributes = attributes unless attributes.nil?
P
Pratik Naik 已提交
1401

1402
        result = yield self if block_given?
1403
        _run_initialize_callbacks
1404
        result
D
Initial  
David Heinemeier Hansson 已提交
1405
      end
1406

1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422
      # 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

1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435
      # 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 = {}, {}, {}
1436
        @association_cache = {}
1437
        @aggregation_cache = {}
1438
        @readonly = @destroyed = @marked_for_destruction = false
1439
        @new_record = false
1440 1441 1442 1443
        _run_find_callbacks
        _run_initialize_callbacks
      end

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

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

1499
      def quoted_id #:nodoc:
1500
        quote_value(id, column_for_attribute(self.class.primary_key))
1501
      end
1502

1503 1504 1505 1506 1507
      # Returns true if the given attribute is in the attributes hash
      def has_attribute?(attr_name)
        @attributes.has_key?(attr_name.to_s)
      end

1508
      # Returns an array of names for the attributes available on this object.
1509
      def attribute_names
1510
        @attributes.keys
1511 1512
      end

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

        multi_parameter_attributes = []
1538
        attributes = sanitize_for_mass_assignment(attributes) if guard_protected_attributes
1539

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

D
Initial  
David Heinemeier Hansson 已提交
1550 1551 1552
        assign_multiparameter_attributes(multi_parameter_attributes)
      end

1553
      # Returns a hash of all the attributes with their names as keys and the values of the attributes as values.
1554
      def attributes
1555
        Hash[@attributes.map { |name, _| [name, read_attribute(name)] }]
1556 1557
      end

P
Pratik Naik 已提交
1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570
      # 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"'
1571 1572 1573
      def attribute_for_inspect(attr_name)
        value = read_attribute(attr_name)

1574
        if value.is_a?(String) && value.length > 50
1575
          "#{value[0..50]}...".inspect
1576 1577
        elsif value.is_a?(Date) || value.is_a?(Time)
          %("#{value.to_s(:db)}")
1578 1579 1580 1581 1582
        else
          value.inspect
        end
      end

D
Initial  
David Heinemeier Hansson 已提交
1583
      # Returns true if the specified +attribute+ has been set by the user or by a database load and is neither
1584
      # nil nor empty? (the latter only applies to objects that respond to empty?, most notably Strings).
D
Initial  
David Heinemeier Hansson 已提交
1585
      def attribute_present?(attribute)
1586
        !_read_attribute(attribute).blank?
D
Initial  
David Heinemeier Hansson 已提交
1587 1588 1589 1590
      end

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

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

      # Delegates to ==
      def eql?(comparison_object)
E
Emilio Tagua 已提交
1612
        self == comparison_object
D
Initial  
David Heinemeier Hansson 已提交
1613
      end
1614

D
Initial  
David Heinemeier Hansson 已提交
1615 1616 1617
      # 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
1618
        id.hash
D
Initial  
David Heinemeier Hansson 已提交
1619 1620
      end

1621
      # Freeze the attributes hash such that associations are still accessible, even on destroyed records.
1622
      def freeze
1623
        @attributes.freeze; self
1624
      end
1625

1626
      # Returns +true+ if the attributes hash has been frozen.
1627 1628 1629
      def frozen?
        @attributes.frozen?
      end
1630

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

A
Aaron Patterson 已提交
1650 1651
        @attributes = cloned_attributes

A
Aaron Patterson 已提交
1652 1653
        _run_after_initialize_callbacks if respond_to?(:_run_after_initialize_callbacks)

A
Aaron Patterson 已提交
1654 1655 1656 1657
        @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 已提交
1658

1659
        @aggregation_cache = {}
1660 1661
        @association_cache = {}
        @attributes_cache = {}
1662
        @new_record  = true
A
Aaron Patterson 已提交
1663

A
Aaron Patterson 已提交
1664 1665
        ensure_proper_type
        populate_with_current_scope_attributes
1666
        clear_timestamp_attributes
A
Aaron Patterson 已提交
1667 1668
      end

1669 1670
      # 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.
1671
      def readonly?
1672
        @readonly
1673 1674
      end

1675 1676
      # Marks this record as read only.
      def readonly!
1677 1678
        @readonly = true
      end
1679

1680
      # Returns the contents of the record as a nicely formatted string.
1681
      def inspect
1682
        attributes_as_nice_string = self.class.column_names.collect { |name|
1683
          if has_attribute?(name)
1684 1685 1686
            "#{name}: #{attribute_for_inspect(name)}"
          end
        }.compact.join(", ")
1687
        "#<#{self.class} #{attributes_as_nice_string}>"
1688
      end
1689

1690 1691
    protected
      def clone_attributes(reader_method = :read_attribute, attributes = {})
1692 1693
        attribute_names.each do |name|
          attributes[name] = clone_attribute_value(reader_method, name)
1694
        end
1695
        attributes
1696 1697 1698 1699 1700 1701 1702 1703 1704
      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 已提交
1705 1706
    private

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

1718
      # The primary key and inheritance column can never be set by mass-assignment for security reasons.
1719 1720 1721
      def self.attributes_protected_by_default
        default = [ primary_key, inheritance_column ]
        default << 'id' unless primary_key.eql? 'id'
1722
        default
1723 1724
      end

1725 1726
      # Returns a copy of the attributes hash where all the values have been safely quoted for use in
      # an Arel insert/update method.
1727 1728 1729 1730 1731 1732
      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))
1733 1734
              value = read_attribute(name)

1735
              if !value.nil? && self.class.serialized_attributes.key?(name)
1736
                value = YAML.dump value
1737
              end
1738
              attrs[self.class.arel_table[name]] = value
1739 1740 1741 1742 1743 1744
            end
          end
        end
        attrs
      end

D
Initial  
David Heinemeier Hansson 已提交
1745
      # Quote strings appropriately for SQL statements.
1746
      def quote_value(value, column = nil)
1747
        self.class.connection.quote(value, column)
D
Initial  
David Heinemeier Hansson 已提交
1748 1749
      end

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

1769
      def instantiate_time_object(name, values)
1770
        if self.class.send(:create_time_zone_conversion_attribute?, name, column_for_attribute(name))
1771
          Time.zone.local(*values)
1772
        else
1773
          Time.time_with_datetime_fallback(@@default_timezone, *values)
1774
        end
1775 1776
      end

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

1786 1787 1788
            if values.empty?
              send(name + "=", nil)
            else
1789

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

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

D
Initial  
David Heinemeier Hansson 已提交
1814 1815 1816 1817 1818 1819 1820 1821
      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)

1822 1823
          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 已提交
1824 1825 1826 1827
        end

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

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

D
Initial  
David Heinemeier Hansson 已提交
1833 1834 1835
      def find_parameter_position(multiparameter_name)
        multiparameter_name.scan(/\(([0-9]*).*\)/).first.first
      end
1836

D
Initial  
David Heinemeier Hansson 已提交
1837 1838
      # Returns a comma-separated pair list, like "key1 = val1, key2 = val2".
      def comma_pair_list(hash)
1839
        hash.map { |k,v| "#{k} = #{v}" }.join(", ")
D
Initial  
David Heinemeier Hansson 已提交
1840 1841
      end

1842
      def quote_columns(quoter, hash)
1843
        Hash[hash.map { |name, value| [quoter.quote_column_name(name), value] }]
D
Initial  
David Heinemeier Hansson 已提交
1844 1845
      end

1846 1847
      def quoted_comma_pair_list(quoter, hash)
        comma_pair_list(quote_columns(quoter, hash))
D
Initial  
David Heinemeier Hansson 已提交
1848 1849
      end

1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865
      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
1866 1867 1868 1869

      def populate_with_current_scope_attributes
        if scope = self.class.send(:current_scoped_methods)
          create_with = scope.scope_for_create
1870
          create_with.each { |att,value|
J
José Valim 已提交
1871
            respond_to?("#{att}=") && send("#{att}=", value)
1872
          }
1873 1874
        end
      end
1875

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

  Base.class_eval do
1888
    include ActiveRecord::Persistence
J
Joshua Peek 已提交
1889
    extend ActiveModel::Naming
J
Joshua Peek 已提交
1890
    extend QueryCache::ClassMethods
J
José Valim 已提交
1891
    extend ActiveSupport::Benchmarkable
1892
    extend ActiveSupport::DescendantsTracker
J
José Valim 已提交
1893

1894
    include ActiveModel::Conversion
J
Joshua Peek 已提交
1895
    include Validations
1896
    extend CounterCache
J
Joshua Peek 已提交
1897 1898
    include Locking::Optimistic, Locking::Pessimistic
    include AttributeMethods
J
Joshua Peek 已提交
1899
    include AttributeMethods::Read, AttributeMethods::Write, AttributeMethods::BeforeTypeCast, AttributeMethods::Query
1900
    include AttributeMethods::PrimaryKey
J
Joshua Peek 已提交
1901 1902
    include AttributeMethods::TimeZoneConversion
    include AttributeMethods::Dirty
1903
    include ActiveModel::MassAssignmentSecurity
1904
    include Callbacks, ActiveModel::Observing, Timestamp
J
Joshua Peek 已提交
1905
    include Associations, AssociationPreload, NamedScope
1906
    include ActiveModel::SecurePassword
1907 1908 1909 1910

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

1913
    NilClass.add_whiner(self) if NilClass.respond_to?(:add_whiner)
1914 1915 1916 1917 1918 1919 1920 1921 1922 1923 1924

    # 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 已提交
1925
  end
1926
end
1927 1928 1929

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