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

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

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

P
Pratik Naik 已提交
348 349
    ##
    # :singleton-method:
350 351 352
    # Accessor for the prefix type that will be prepended to every primary key column name.
    # The options are :table_name and :table_name_with_underscore. If the first is specified,
    # the Product class will look for "productid" instead of "id" as the primary column. If the
353
    # latter is specified, the Product class will look for "product_id" instead of "id". Remember
354
    # that this is a global setting for all Active Records.
355
    cattr_accessor :primary_key_prefix_type, :instance_writer => false
D
Initial  
David Heinemeier Hansson 已提交
356 357
    @@primary_key_prefix_type = nil

P
Pratik Naik 已提交
358 359
    ##
    # :singleton-method:
360 361 362
    # Accessor for the name of the prefix string to prepend to every table name. So if set
    # to "basecamp_", all table names will be named like "basecamp_projects", "basecamp_people",
    # etc. This is a convenient way of creating a namespace for tables in a shared database.
363
    # By default, the prefix is the empty string.
364
    #
365 366
    # If you are organising your models within modules you can add a prefix to the models within
    # a namespace by defining a singleton method in the parent module called table_name_prefix which
367
    # returns your chosen prefix.
368 369
    class_attribute :table_name_prefix, :instance_writer => false
    self.table_name_prefix = ""
D
Initial  
David Heinemeier Hansson 已提交
370

P
Pratik Naik 已提交
371 372
    ##
    # :singleton-method:
D
Initial  
David Heinemeier Hansson 已提交
373 374
    # Works like +table_name_prefix+, but appends instead of prepends (set to "_basecamp" gives "projects_basecamp",
    # "people_basecamp"). By default, the suffix is the empty string.
375 376
    class_attribute :table_name_suffix, :instance_writer => false
    self.table_name_suffix = ""
D
Initial  
David Heinemeier Hansson 已提交
377

P
Pratik Naik 已提交
378 379
    ##
    # :singleton-method:
380
    # Indicates whether table names should be the pluralized versions of the corresponding class names.
P
Pratik Naik 已提交
381
    # If true, the default table name for a Product class will be +products+. If false, it would just be +product+.
D
Initial  
David Heinemeier Hansson 已提交
382
    # See table_name for the full rules on table/class naming. This is true, by default.
383
    cattr_accessor :pluralize_table_names, :instance_writer => false
D
Initial  
David Heinemeier Hansson 已提交
384 385
    @@pluralize_table_names = true

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

P
Pratik Naik 已提交
393 394
    ##
    # :singleton-method:
395 396
    # Specifies the format to use when dumping the database schema with Rails'
    # Rakefile.  If :sql, the schema is dumped as (potentially database-
397
    # specific) SQL statements.  If :ruby, the schema is dumped as an
398 399 400
    # ActiveRecord::Schema file which can be loaded into any database that
    # supports migrations.  Use :ruby if you want to have different database
    # adapters for, e.g., your development and test environments.
401
    cattr_accessor :schema_format , :instance_writer => false
402
    @@schema_format = :ruby
403

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

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

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

J
Jeremy Kemper 已提交
418 419 420 421 422
    # Returns a hash of all the attributes that have been specified for serialization as
    # keys and their class restriction as values.
    class_attribute :serialized_attributes
    self.serialized_attributes = {}

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

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

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

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

N
Neeraj Singh 已提交
504 505
      # Attributes listed as readonly will be used to create a new record but update operations will
      # ignore these fields.
506
      def attr_readonly(*attributes)
507
        write_inheritable_attribute(:attr_readonly, Set.new(attributes.map { |a| a.to_s }) + (readonly_attributes || []))
D
Initial  
David Heinemeier Hansson 已提交
508
      end
509

510 511 512
      # Returns an array of all the attributes that have been specified as readonly.
      def readonly_attributes
        read_inheritable_attribute(:attr_readonly) || []
D
Initial  
David Heinemeier Hansson 已提交
513 514
      end

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

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

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

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

583
      def full_table_name_prefix #:nodoc:
584 585 586
        (parents.detect{ |p| p.respond_to?(:table_name_prefix) } || self).table_name_prefix
      end

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

593 594
      # 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 已提交
595
      def sequence_name #:nodoc:
596 597 598
        reset_sequence_name
      end

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

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

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

650 651
      # Indicates whether the table associated with this class exists
      def table_exists?
652
        connection.table_exists?(table_name)
653 654
      end

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

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

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

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

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

726
      def reset_column_information_and_inheritable_attributes_for_all_subclasses#:nodoc:
727
        descendants.each { |klass| klass.reset_inheritable_attributes; klass.reset_column_information }
728
      end
D
Initial  
David Heinemeier Hansson 已提交
729

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

734 735
      # Set the lookup ancestors for ActiveModel.
      def lookup_ancestors #:nodoc:
736 737
        klass = self
        classes = [klass]
E
Emilio Tagua 已提交
738
        while klass != klass.base_class
739 740 741 742 743
          classes << klass = klass.superclass
        end
        classes
      rescue
        # OPTIMIZE this rescue is to fix this test: ./test/cases/reflection_test.rb:56:in `test_human_name_for_column'
P
Pratik Naik 已提交
744
        # Apparently the method base_class causes some trouble.
745 746 747 748
        # It now works for sure.
        [self]
      end

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

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

763 764 765 766 767
      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

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

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

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

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

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

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

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

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

821 822 823
        super
      end

824 825 826 827
      def sti_name
        store_full_sti_class ? name : name.demodulize
      end

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

832 833
      def arel_engine
        @arel_engine ||= begin
834 835 836
          if self == ActiveRecord::Base
            Arel::Table.engine
          else
837
            connection_handler.connection_pools[name] ? Arel::Sql::Engine.new(self) : superclass.arel_engine
838 839
          end
        end
840 841
      end

842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857
      # 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"
      #   }
      #
858 859 860
      # It is recommended to use block form of unscoped because chaining unscoped with <tt>named_scope</tt>
      # does not work. Assuming that <tt>published</tt> is a <tt>named_scope</tt> following two statements are same.
      #
861 862
      # Post.unscoped.published
      # Post.published
863
      def unscoped #:nodoc:
864 865 866 867 868 869 870 871
        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

872 873 874 875
      def before_remove_const #:nodoc:
        reset_scoped_methods
      end

D
Initial  
David Heinemeier Hansson 已提交
876
      private
J
José Valim 已提交
877 878 879 880 881 882

        def relation #:nodoc:
          @relation ||= Relation.new(self, arel_table)
          finder_needs_type_condition? ? @relation.where(type_condition) : @relation
        end

883 884 885
        # Finder methods must instantiate through this method to work with the
        # single-table inheritance model that makes it possible to create
        # objects of different types from the same table.
D
Initial  
David Heinemeier Hansson 已提交
886
        def instantiate(record)
887 888 889
          model = find_sti_class(record[inheritance_column]).allocate
          model.init_with('attributes' => record)
          model
D
Initial  
David Heinemeier Hansson 已提交
890
        end
891

892
        def find_sti_class(type_name)
893
          if type_name.nil? || !columns_hash.include?(inheritance_column)
894 895 896
            self
          else
            begin
897 898 899 900 901
              if store_full_sti_class
                ActiveSupport::Dependencies.constantize(type_name)
              else
                compute_type(type_name)
              end
902 903 904 905 906 907 908 909 910 911
            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

912
        def construct_finder_arel(options = {}, scope = nil)
913
          relation = options.is_a?(Hash) ? unscoped.apply_finder_options(options) : options
914
          relation = scope.merge(relation) if scope
915
          relation
916 917
        end

918
        def type_condition
919
          sti_column = arel_table[inheritance_column]
920
          condition = sti_column.eq(sti_name)
921
          descendants.each { |subclass| condition = condition.or(sti_column.eq(subclass.sti_name)) }
922

923
          condition
D
Initial  
David Heinemeier Hansson 已提交
924 925 926
        end

        # Guesses the table name, but does not decorate it with prefix and suffix information.
927
        def undecorated_table_name(class_name = base_class.name)
928 929
          table_name = class_name.to_s.demodulize.underscore
          table_name = table_name.pluralize if pluralize_table_names
930
          table_name
D
Initial  
David Heinemeier Hansson 已提交
931 932
        end

933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949
        # 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

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

991 992 993 994 995 996
        def construct_attributes_from_arguments(attribute_names, arguments)
          attributes = {}
          attribute_names.each_with_index { |name, idx| attributes[name] = arguments[idx] }
          attributes
        end

997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011
        # Similar in purpose to +expand_hash_conditions_for_aggregates+.
        def expand_attribute_names_for_aggregates(attribute_names)
          expanded_attribute_names = []
          attribute_names.each do |attribute_name|
            unless (aggregation = reflect_on_aggregation(attribute_name.to_sym)).nil?
              aggregate_mapping(aggregation).each do |field_attr, aggregate_attr|
                expanded_attribute_names << field_attr
              end
            else
              expanded_attribute_names << attribute_name
            end
          end
          expanded_attribute_names
        end

1012
        def all_attributes_exists?(attribute_names)
1013 1014 1015
          expand_attribute_names_for_aggregates(attribute_names).all? { |name|
            column_methods_hash.include?(name.to_sym)
          }
1016
        end
1017

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

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

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

              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

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

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

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

Or call unscoped with a block:

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

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

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

D
David Heinemeier Hansson 已提交
1145
        def current_scoped_methods #:nodoc:
1146
          scoped_methods.last
1147
        end
1148

1149 1150 1151 1152
        def reset_scoped_methods #:nodoc:
          Thread.current[:"#{self}_scoped_methods"] = nil
        end

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

            candidates.each do |candidate|
              begin
1168
                constant = ActiveSupport::Dependencies.constantize(candidate)
1169
                return constant if candidate == constant.to_s
1170 1171 1172
              rescue NameError => e
                # We don't want to swallow NoMethodError < NameError errors
                raise e unless e.instance_of?(NameError)
1173 1174
              rescue ArgumentError
              end
1175
            end
1176 1177

            raise NameError, "uninitialized constant #{candidates.first}"
D
Initial  
David Heinemeier Hansson 已提交
1178 1179 1180
          end
        end

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

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

1201 1202
          case condition
            when Array; sanitize_sql_array(condition)
1203
            when Hash;  sanitize_sql_hash_for_conditions(condition, table_name)
1204 1205
            else        condition
          end
1206
        end
1207
        alias_method :sanitize_sql, :sanitize_sql_for_conditions
1208

P
Pratik Naik 已提交
1209
        # Accepts an array, hash, or string of SQL conditions and sanitizes
1210 1211 1212 1213 1214 1215 1216 1217 1218 1219
        # 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

1220 1221 1222 1223 1224
        def aggregate_mapping(reflection)
          mapping = reflection.options[:mapping] || [reflection.name, reflection.name]
          mapping.first.is_a?(Array) ? mapping : [mapping]
        end

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

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

1272 1273
          table = Arel::Table.new(self.table_name, :engine => arel_engine, :as => default_table_name)
          builder = PredicateBuilder.new(arel_engine)
1274
          builder.build_from_hash(attrs, table).map{ |b| b.to_sql }.join(' AND ')
1275
        end
1276 1277 1278 1279 1280 1281
        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)
1282
          attrs.map do |attr, value|
1283 1284 1285
            "#{connection.quote_column_name(attr)} = #{quote_bound_value(value)}"
          end.join(', ')
        end
1286

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

1303 1304
        alias_method :sanitize_conditions, :sanitize_sql

D
David Heinemeier Hansson 已提交
1305
        def replace_bind_variables(statement, values) #:nodoc:
1306
          raise_if_bind_arity_mismatch(statement, statement.count('?'), values.size)
1307
          bound = values.dup
C
Comron Sattari 已提交
1308 1309
          c = connection
          statement.gsub('?') { quote_bound_value(bound.shift, c) }
1310 1311
        end

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

1324
        def expand_range_bind_variables(bind_vars) #:nodoc:
1325 1326 1327
          expanded = []

          bind_vars.each do |var|
1328 1329
            next if var.is_a?(Hash)

1330
            if var.is_a?(Range)
1331 1332
              expanded << var.first
              expanded << var.last
1333
            else
1334
              expanded << var
1335
            end
1336
          end
1337 1338

          expanded
1339 1340
        end

C
Comron Sattari 已提交
1341
        def quote_bound_value(value, c = connection) #:nodoc:
1342
          if value.respond_to?(:map) && !value.acts_like?(:string)
1343
            if value.respond_to?(:empty?) && value.empty?
C
Comron Sattari 已提交
1344
              c.quote(nil)
1345
            else
C
Comron Sattari 已提交
1346
              value.map { |v| c.quote(v) }.join(',')
1347
            end
1348
          else
C
Comron Sattari 已提交
1349
            c.quote(value)
1350 1351 1352
          end
        end

D
David Heinemeier Hansson 已提交
1353
        def raise_if_bind_arity_mismatch(statement, expected, provided) #:nodoc:
1354 1355 1356
          unless expected == provided
            raise PreparedStatementInvalid, "wrong number of bind variables (#{provided} for #{expected}) in: #{statement}"
          end
1357
        end
1358

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

D
Initial  
David Heinemeier Hansson 已提交
1381
        ensure_proper_type
P
Pratik Naik 已提交
1382

1383
        populate_with_current_scope_attributes
1384
        self.attributes = attributes unless attributes.nil?
P
Pratik Naik 已提交
1385

1386
        result = yield self if block_given?
1387
        _run_initialize_callbacks
1388
        result
D
Initial  
David Heinemeier Hansson 已提交
1389
      end
1390

1391 1392 1393 1394
      # Cloned objects have no id assigned and are treated as new records. Note that this is a "shallow" clone
      # as it copies the object's attributes only, not its associations. The extent of a "deep" clone is
      # application specific and is therefore left to the application to implement according to its need.
      def initialize_copy(other)
1395
        _run_after_initialize_callbacks if respond_to?(:_run_after_initialize_callbacks)
1396 1397
        cloned_attributes = other.clone_attributes(:read_attribute_before_type_cast)
        cloned_attributes.delete(self.class.primary_key)
1398

1399
        @attributes = cloned_attributes
1400 1401 1402 1403 1404 1405

        @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

1406
        clear_aggregation_cache
1407
        clear_association_cache
1408 1409 1410
        @attributes_cache = {}
        @new_record = true
        ensure_proper_type
1411

1412
        populate_with_current_scope_attributes
1413 1414
      end

1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432
      # 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 = {}, {}, {}
        @new_record = @readonly = @destroyed = @marked_for_destruction = false
        _run_find_callbacks
        _run_initialize_callbacks
      end

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

P
Pratik Naik 已提交
1460 1461 1462
      # Returns a cache key that can be used to identify this record.
      #
      # ==== Examples
1463 1464 1465 1466 1467
      #
      #   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
1468
        case
1469
        when new_record?
J
Jeremy Kemper 已提交
1470 1471 1472
          "#{self.class.model_name.cache_key}/new"
        when timestamp = self[:updated_at]
          "#{self.class.model_name.cache_key}/#{id}-#{timestamp.to_s(:number)}"
1473
        else
J
Jeremy Kemper 已提交
1474
          "#{self.class.model_name.cache_key}/#{id}"
1475 1476
        end
      end
1477

1478
      def quoted_id #:nodoc:
1479
        quote_value(id, column_for_attribute(self.class.primary_key))
1480
      end
1481

1482 1483 1484 1485 1486 1487 1488 1489 1490 1491
      # Returns true if the given attribute is in the attributes hash
      def has_attribute?(attr_name)
        @attributes.has_key?(attr_name.to_s)
      end

      # Returns an array of names for the attributes available on this object sorted alphabetically.
      def attribute_names
        @attributes.keys.sort
      end

1492
      # Returns the value of the attribute identified by <tt>attr_name</tt> after it has been typecast (for example,
D
Initial  
David Heinemeier Hansson 已提交
1493 1494
      # "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).
1495
      def [](attr_name)
1496
        read_attribute(attr_name)
D
Initial  
David Heinemeier Hansson 已提交
1497
      end
1498

D
Initial  
David Heinemeier Hansson 已提交
1499 1500
      # Updates the attribute identified by <tt>attr_name</tt> with the specified +value+.
      # (Alias for the protected write_attribute method).
1501
      def []=(attr_name, value)
1502
        write_attribute(attr_name, value)
D
Initial  
David Heinemeier Hansson 已提交
1503 1504 1505
      end

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

        multi_parameter_attributes = []
1530
        attributes = sanitize_for_mass_assignment(attributes) if guard_protected_attributes
1531

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

D
Initial  
David Heinemeier Hansson 已提交
1540 1541 1542
        assign_multiparameter_attributes(multi_parameter_attributes)
      end

1543
      # Returns a hash of all the attributes with their names as keys and the values of the attributes as values.
1544
      def attributes
1545 1546 1547
        attrs = {}
        attribute_names.each { |name| attrs[name] = read_attribute(name) }
        attrs
1548 1549
      end

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

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

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

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

1587
      # Returns true if the +comparison_object+ is the same object, or is of the same type and has the same id.
D
Initial  
David Heinemeier Hansson 已提交
1588
      def ==(comparison_object)
1589
        comparison_object.equal?(self) ||
1590
          (comparison_object.instance_of?(self.class) &&
1591
            comparison_object.id == id && !comparison_object.new_record?)
D
Initial  
David Heinemeier Hansson 已提交
1592 1593 1594 1595 1596 1597
      end

      # Delegates to ==
      def eql?(comparison_object)
        self == (comparison_object)
      end
1598

D
Initial  
David Heinemeier Hansson 已提交
1599 1600 1601
      # 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
1602
        id.hash
D
Initial  
David Heinemeier Hansson 已提交
1603 1604
      end

1605
      # Freeze the attributes hash such that associations are still accessible, even on destroyed records.
1606
      def freeze
1607
        @attributes.freeze; self
1608
      end
1609

1610
      # Returns +true+ if the attributes hash has been frozen.
1611 1612 1613
      def frozen?
        @attributes.frozen?
      end
1614

1615 1616 1617
      # Returns duplicated record with unfreezed attributes.
      def dup
        obj = super
1618
        obj.instance_variable_set('@attributes', @attributes.dup)
1619 1620 1621
        obj
      end

1622 1623
      # 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.
1624
      def readonly?
1625
        @readonly
1626 1627
      end

1628 1629
      # Marks this record as read only.
      def readonly!
1630 1631
        @readonly = true
      end
1632

1633
      # Returns the contents of the record as a nicely formatted string.
1634
      def inspect
1635
        attributes_as_nice_string = self.class.column_names.collect { |name|
1636 1637 1638 1639
          if has_attribute?(name) || new_record?
            "#{name}: #{attribute_for_inspect(name)}"
          end
        }.compact.join(", ")
1640
        "#<#{self.class} #{attributes_as_nice_string}>"
1641
      end
1642

1643 1644
    protected
      def clone_attributes(reader_method = :read_attribute, attributes = {})
1645 1646
        attribute_names.each do |name|
          attributes[name] = clone_attribute_value(reader_method, name)
1647
        end
1648
        attributes
1649 1650 1651 1652 1653 1654 1655 1656 1657
      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 已提交
1658 1659
    private

1660
      # Sets the attribute used for single table inheritance to this class name if this is not the
1661
      # ActiveRecord::Base descendant.
1662 1663
      # 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.
1664
      # No such attribute would be set for objects of the Message class in that example.
D
Initial  
David Heinemeier Hansson 已提交
1665 1666
      def ensure_proper_type
        unless self.class.descends_from_active_record?
1667
          write_attribute(self.class.inheritance_column, self.class.sti_name)
D
Initial  
David Heinemeier Hansson 已提交
1668 1669 1670
        end
      end

1671
      # The primary key and inheritance column can never be set by mass-assignment for security reasons.
1672 1673 1674
      def self.attributes_protected_by_default
        default = [ primary_key, inheritance_column ]
        default << 'id' unless primary_key.eql? 'id'
1675
        default
1676 1677
      end

1678 1679
      # Returns a copy of the attributes hash where all the values have been safely quoted for use in
      # an Arel insert/update method.
1680 1681 1682 1683 1684 1685
      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))
1686 1687
              value = read_attribute(name)

1688 1689
              if value && self.class.serialized_attributes.key?(name)
                value = YAML.dump value
1690
              end
1691
              attrs[self.class.arel_table[name]] = value
1692 1693 1694 1695 1696 1697
            end
          end
        end
        attrs
      end

D
Initial  
David Heinemeier Hansson 已提交
1698
      # Quote strings appropriately for SQL statements.
1699
      def quote_value(value, column = nil)
1700
        self.class.connection.quote(value, column)
D
Initial  
David Heinemeier Hansson 已提交
1701 1702
      end

P
Pratik Naik 已提交
1703
      # Interpolate custom SQL string in instance context.
D
Initial  
David Heinemeier Hansson 已提交
1704 1705
      # Optional record argument is meant for custom insert_sql.
      def interpolate_sql(sql, record = nil)
1706
        instance_eval("%@#{sql.gsub('@', '\@')}@", __FILE__, __LINE__)
D
Initial  
David Heinemeier Hansson 已提交
1707 1708 1709 1710 1711 1712
      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
1713 1714
      # 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
1715
      # attribute will be set to nil.
D
Initial  
David Heinemeier Hansson 已提交
1716 1717 1718 1719 1720
      def assign_multiparameter_attributes(pairs)
        execute_callstack_for_multiparameter_attributes(
          extract_callstack_for_multiparameter_attributes(pairs)
        )
      end
1721

1722
      def instantiate_time_object(name, values)
1723
        if self.class.send(:create_time_zone_conversion_attribute?, name, column_for_attribute(name))
1724
          Time.zone.local(*values)
1725
        else
1726
          Time.time_with_datetime_fallback(@@default_timezone, *values)
1727
        end
1728 1729
      end

D
Initial  
David Heinemeier Hansson 已提交
1730
      def execute_callstack_for_multiparameter_attributes(callstack)
1731
        errors = []
1732
        callstack.each do |name, values_with_empty_parameters|
1733 1734
          begin
            klass = (self.class.reflect_on_aggregation(name.to_sym) || column_for_attribute(name)).klass
1735 1736
            # 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.
1737
            values = values_with_empty_parameters.reject { |v| v.nil? }
1738

1739 1740 1741
            if values.empty?
              send(name + "=", nil)
            else
1742

1743
              value = if Time == klass
1744
                instantiate_time_object(name, values)
1745 1746
              elsif Date == klass
                begin
1747
                  values = values_with_empty_parameters.collect do |v| v.nil? ? 1 : v end
1748 1749
                  Date.new(*values)
                rescue ArgumentError => ex # if Date.new raises an exception on an invalid date
1750
                  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
1751 1752 1753 1754 1755 1756
                end
              else
                klass.new(*values)
              end

              send(name + "=", value)
1757
            end
1758 1759
          rescue => ex
            errors << AttributeAssignmentError.new("error on assignment #{values.inspect} to #{name}", ex, name)
D
Initial  
David Heinemeier Hansson 已提交
1760 1761
          end
        end
1762 1763 1764
        unless errors.empty?
          raise MultiparameterAssignmentErrors.new(errors), "#{errors.size} error(s) on assignment of multiparameter attributes"
        end
D
Initial  
David Heinemeier Hansson 已提交
1765
      end
1766

D
Initial  
David Heinemeier Hansson 已提交
1767 1768 1769 1770 1771 1772 1773 1774
      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)

1775 1776
          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 已提交
1777 1778 1779 1780
        end

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

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

D
Initial  
David Heinemeier Hansson 已提交
1786 1787 1788
      def find_parameter_position(multiparameter_name)
        multiparameter_name.scan(/\(([0-9]*).*\)/).first.first
      end
1789

D
Initial  
David Heinemeier Hansson 已提交
1790 1791
      # Returns a comma-separated pair list, like "key1 = val1, key2 = val2".
      def comma_pair_list(hash)
1792
        hash.map { |k,v| "#{k} = #{v}" }.join(", ")
D
Initial  
David Heinemeier Hansson 已提交
1793 1794
      end

1795
      def quote_columns(quoter, hash)
1796
        Hash[hash.map { |name, value| [quoter.quote_column_name(name), value] }]
D
Initial  
David Heinemeier Hansson 已提交
1797 1798
      end

1799 1800
      def quoted_comma_pair_list(quoter, hash)
        comma_pair_list(quote_columns(quoter, hash))
D
Initial  
David Heinemeier Hansson 已提交
1801 1802
      end

1803 1804 1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818
      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
1819 1820 1821 1822 1823 1824 1825

      def populate_with_current_scope_attributes
        if scope = self.class.send(:current_scoped_methods)
          create_with = scope.scope_for_create
          create_with.each { |att,value| self.respond_to?(:"#{att}=") && self.send("#{att}=", value) } if create_with
        end
      end
D
Initial  
David Heinemeier Hansson 已提交
1826
  end
J
Joshua Peek 已提交
1827 1828

  Base.class_eval do
1829
    include ActiveRecord::Persistence
J
Joshua Peek 已提交
1830
    extend ActiveModel::Naming
J
Joshua Peek 已提交
1831
    extend QueryCache::ClassMethods
J
José Valim 已提交
1832
    extend ActiveSupport::Benchmarkable
1833
    extend ActiveSupport::DescendantsTracker
J
José Valim 已提交
1834

1835
    include ActiveModel::Conversion
J
Joshua Peek 已提交
1836
    include Validations
1837
    extend CounterCache
J
Joshua Peek 已提交
1838 1839
    include Locking::Optimistic, Locking::Pessimistic
    include AttributeMethods
J
Joshua Peek 已提交
1840
    include AttributeMethods::Read, AttributeMethods::Write, AttributeMethods::BeforeTypeCast, AttributeMethods::Query
1841
    include AttributeMethods::PrimaryKey
J
Joshua Peek 已提交
1842 1843
    include AttributeMethods::TimeZoneConversion
    include AttributeMethods::Dirty
1844
    include ActiveModel::MassAssignmentSecurity
1845
    include Callbacks, ActiveModel::Observing, Timestamp
J
Joshua Peek 已提交
1846
    include Associations, AssociationPreload, NamedScope
1847 1848 1849 1850

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

1853
    NilClass.add_whiner(self) if NilClass.respond_to?(:add_whiner)
J
Joshua Peek 已提交
1854
  end
1855
end
1856 1857 1858

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