base.rb 82.9 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'
D
Initial  
David Heinemeier Hansson 已提交
23 24

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

P
Pratik Naik 已提交
280 281
    ##
    # :singleton-method:
P
Pratik Naik 已提交
282 283 284 285
    # Contains the database configuration - as is typically stored in config/database.yml -
    # as a Hash.
    #
    # For example, the following database.yml...
E
Emilio Tagua 已提交
286
    #
P
Pratik Naik 已提交
287 288 289
    #   development:
    #     adapter: sqlite3
    #     database: db/development.sqlite3
E
Emilio Tagua 已提交
290
    #
P
Pratik Naik 已提交
291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306
    #   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'
    #      }
    #   }
307
    cattr_accessor :configurations, :instance_writer => false
308 309
    @@configurations = {}

P
Pratik Naik 已提交
310 311
    ##
    # :singleton-method:
312
    # Accessor for the prefix type that will be prepended to every primary key column name. The options are :table_name and
D
Initial  
David Heinemeier Hansson 已提交
313 314
    # :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 latter is specified, the Product class will look for "product_id" instead of "id". Remember
315
    # that this is a global setting for all Active Records.
316
    cattr_accessor :primary_key_prefix_type, :instance_writer => false
D
Initial  
David Heinemeier Hansson 已提交
317 318
    @@primary_key_prefix_type = nil

P
Pratik Naik 已提交
319 320
    ##
    # :singleton-method:
321
    # Accessor for the name of the prefix string to prepend to every table name. So if set to "basecamp_", all
322
    # table names will be named like "basecamp_projects", "basecamp_people", etc. This is a convenient way of creating a namespace
D
Initial  
David Heinemeier Hansson 已提交
323
    # for tables in a shared database. By default, the prefix is the empty string.
324 325 326
    #
    # 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 returns your chosen prefix.
327 328
    class_attribute :table_name_prefix, :instance_writer => false
    self.table_name_prefix = ""
D
Initial  
David Heinemeier Hansson 已提交
329

P
Pratik Naik 已提交
330 331
    ##
    # :singleton-method:
D
Initial  
David Heinemeier Hansson 已提交
332 333
    # 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.
334 335
    class_attribute :table_name_suffix, :instance_writer => false
    self.table_name_suffix = ""
D
Initial  
David Heinemeier Hansson 已提交
336

P
Pratik Naik 已提交
337 338
    ##
    # :singleton-method:
339
    # Indicates whether table names should be the pluralized versions of the corresponding class names.
P
Pratik Naik 已提交
340
    # 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 已提交
341
    # See table_name for the full rules on table/class naming. This is true, by default.
342
    cattr_accessor :pluralize_table_names, :instance_writer => false
D
Initial  
David Heinemeier Hansson 已提交
343 344
    @@pluralize_table_names = true

P
Pratik Naik 已提交
345 346
    ##
    # :singleton-method:
347 348
    # Determines whether to use Time.local (using :local) or Time.utc (using :utc) when pulling dates and times from the database.
    # This is set to :local by default.
349
    cattr_accessor :default_timezone, :instance_writer => false
350
    @@default_timezone = :local
351

P
Pratik Naik 已提交
352 353
    ##
    # :singleton-method:
354 355
    # Specifies the format to use when dumping the database schema with Rails'
    # Rakefile.  If :sql, the schema is dumped as (potentially database-
356
    # specific) SQL statements.  If :ruby, the schema is dumped as an
357 358 359
    # 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.
360
    cattr_accessor :schema_format , :instance_writer => false
361
    @@schema_format = :ruby
362

P
Pratik Naik 已提交
363 364
    ##
    # :singleton-method:
365
    # Specify whether or not to use timestamps for migration versions
366 367 368
    cattr_accessor :timestamped_migrations , :instance_writer => false
    @@timestamped_migrations = true

369 370
    # Determine whether to store the full constant name including namespace when using STI
    superclass_delegating_accessor :store_full_sti_class
371
    self.store_full_sti_class = true
372

373 374 375 376
    # Stores the default scope for the class
    class_inheritable_accessor :default_scoping, :instance_writer => false
    self.default_scoping = []

D
Initial  
David Heinemeier Hansson 已提交
377
    class << self # Class methods
378 379
      def colorize_logging(*args)
        ActiveSupport::Deprecation.warn "ActiveRecord::Base.colorize_logging and " <<
380
          "config.active_record.colorize_logging are deprecated. Please use " <<
381
          "Rails::LogSubscriber.colorize_logging or config.colorize_logging instead", caller
382 383 384
      end
      alias :colorize_logging= :colorize_logging

385
      delegate :find, :first, :last, :all, :destroy, :destroy_all, :exists?, :delete, :delete_all, :update, :update_all, :to => :scoped
P
Pratik Naik 已提交
386
      delegate :find_each, :find_in_batches, :to => :scoped
387
      delegate :select, :group, :order, :limit, :joins, :where, :preload, :eager_load, :includes, :from, :lock, :readonly, :having, :to => :scoped
388
      delegate :count, :average, :minimum, :maximum, :sum, :calculate, :to => :scoped
389

P
Pratik Naik 已提交
390
      # Executes a custom SQL query against your database and returns all the results.  The results will
391
      # be returned as an array with columns requested encapsulated as attributes of the model you call
P
Pratik Naik 已提交
392 393
      # 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.
394
      #
395 396
      # 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
397 398
      # table.
      #
P
Pratik Naik 已提交
399
      # The +sql+ parameter is a full SQL query as a string.  It will be called as is, there will be
400 401
      # 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 已提交
402
      # change your call if you switch engines.
403 404
      #
      # ==== Examples
P
Pratik Naik 已提交
405
      #   # A simple SQL query spanning multiple tables
406 407 408 409 410 411
      #   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 已提交
412
      def find_by_sql(sql)
413
        connection.select_all(sanitize_sql(sql), "#{name} Load").collect! { |record| instantiate(record) }
D
Initial  
David Heinemeier Hansson 已提交
414
      end
415

416
      # Creates an object (or multiple objects) and saves it to the database, if validations pass.
417 418 419 420 421 422 423 424
      # 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')
425
      #
426
      #   # Create an Array of new objects
P
Pratik Naik 已提交
427
      #   User.create([{ :first_name => 'Jamie' }, { :first_name => 'Jeremy' }])
428 429 430 431 432 433 434
      #
      #   # 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 已提交
435
      #   User.create([{ :first_name => 'Jamie' }, { :first_name => 'Jeremy' }]) do |u|
436
      #     u.is_admin = false
437
      #   end
438
      def create(attributes = nil, &block)
439
        if attributes.is_a?(Array)
440
          attributes.collect { |attr| create(attr, &block) }
441 442
        else
          object = new(attributes)
443
          yield(object) if block_given?
444 445 446
          object.save
          object
        end
D
Initial  
David Heinemeier Hansson 已提交
447 448 449
      end

      # Returns the result of an SQL statement that should only include a COUNT(*) in the SELECT part.
450
      # The use of this method should be restricted to complicated SQL queries that can't be executed
451 452
      # using the ActiveRecord::Calculations class methods.  Look into those before using this.
      #
P
Pratik Naik 已提交
453
      # ==== Parameters
454
      #
P
Pratik Naik 已提交
455
      # * +sql+ - An SQL statement which should return a count query from the database, see the example below.
456 457 458
      #
      # ==== Examples
      #
459
      #   Product.count_by_sql "SELECT COUNT(*) FROM sales s, customers c WHERE s.customer_id = c.id"
D
Initial  
David Heinemeier Hansson 已提交
460
      def count_by_sql(sql)
461
        sql = sanitize_conditions(sql)
462
        connection.select_value(sql, "#{name} Count").to_i
D
Initial  
David Heinemeier Hansson 已提交
463
      end
464

P
Pratik Naik 已提交
465 466 467 468 469 470 471 472 473
      # Attributes named in this macro are protected from mass-assignment,
      # such as <tt>new(attributes)</tt>,
      # <tt>update_attributes(attributes)</tt>, or
      # <tt>attributes=(attributes)</tt>.
      #
      # Mass-assignment to these attributes will simply be ignored, to assign
      # to them you can use direct writer methods. This is meant to protect
      # sensitive attributes from being overwritten by malicious users
      # tampering with URLs or forms.
D
Initial  
David Heinemeier Hansson 已提交
474 475 476 477 478 479 480 481 482 483 484 485
      #
      #   class Customer < ActiveRecord::Base
      #     attr_protected :credit_rating
      #   end
      #
      #   customer = Customer.new("name" => David, "credit_rating" => "Excellent")
      #   customer.credit_rating # => nil
      #   customer.attributes = { "description" => "Jolly fellow", "credit_rating" => "Superb" }
      #   customer.credit_rating # => nil
      #
      #   customer.credit_rating = "Average"
      #   customer.credit_rating # => "Average"
486
      #
P
Pratik Naik 已提交
487 488
      # To start from an all-closed default and enable attributes as needed,
      # have a look at +attr_accessible+.
P
Pratik Naik 已提交
489 490 491 492
      #
      # If the access logic of your application is richer you can use <tt>Hash#except</tt>
      # or <tt>Hash#slice</tt> to sanitize the hash of parameters before they are
      # passed to Active Record.
493
      #
P
Pratik Naik 已提交
494 495 496 497 498 499 500 501 502 503
      # For example, it could be the case that the list of protected attributes
      # for a given model depends on the role of the user:
      #
      #   # Assumes plan_id is not protected because it depends on the role.
      #   params[:account] = params[:account].except(:plan_id) unless admin?
      #   @account.update_attributes(params[:account])
      #
      # Note that +attr_protected+ is still applied to the received hash. Thus,
      # with this technique you can at most _extend_ the list of protected
      # attributes for a particular mass-assignment call.
D
Initial  
David Heinemeier Hansson 已提交
504
      def attr_protected(*attributes)
505
        write_inheritable_attribute(:attr_protected, Set.new(attributes.map {|a| a.to_s}) + (protected_attributes || []))
D
Initial  
David Heinemeier Hansson 已提交
506
      end
507

508
      # Returns an array of all the attributes that have been protected from mass-assignment.
D
Initial  
David Heinemeier Hansson 已提交
509
      def protected_attributes # :nodoc:
510
        read_inheritable_attribute(:attr_protected)
D
Initial  
David Heinemeier Hansson 已提交
511 512
      end

P
Pratik Naik 已提交
513 514 515 516
      # Specifies a white list of model attributes that can be set via
      # mass-assignment, such as <tt>new(attributes)</tt>,
      # <tt>update_attributes(attributes)</tt>, or
      # <tt>attributes=(attributes)</tt>
517
      #
P
Pratik Naik 已提交
518 519 520 521 522 523 524
      # This is the opposite of the +attr_protected+ macro: Mass-assignment
      # will only set attributes in this list, to assign to the rest of
      # attributes you can use direct writer methods. This is meant to protect
      # sensitive attributes from being overwritten by malicious users
      # tampering with URLs or forms. If you'd rather start from an all-open
      # default and restrict attributes as needed, have a look at
      # +attr_protected+.
525 526
      #
      #   class Customer < ActiveRecord::Base
527
      #     attr_accessible :name, :nickname
528 529
      #   end
      #
530 531 532 533
      #   customer = Customer.new(:name => "David", :nickname => "Dave", :credit_rating => "Excellent")
      #   customer.credit_rating # => nil
      #   customer.attributes = { :name => "Jolly fellow", :credit_rating => "Superb" }
      #   customer.credit_rating # => nil
534
      #
535 536
      #   customer.credit_rating = "Average"
      #   customer.credit_rating # => "Average"
P
Pratik Naik 已提交
537 538 539 540
      #
      # If the access logic of your application is richer you can use <tt>Hash#except</tt>
      # or <tt>Hash#slice</tt> to sanitize the hash of parameters before they are
      # passed to Active Record.
541
      #
P
Pratik Naik 已提交
542 543 544 545 546 547 548 549 550 551
      # For example, it could be the case that the list of accessible attributes
      # for a given model depends on the role of the user:
      #
      #   # Assumes plan_id is accessible because it depends on the role.
      #   params[:account] = params[:account].except(:plan_id) unless admin?
      #   @account.update_attributes(params[:account])
      #
      # Note that +attr_accessible+ is still applied to the received hash. Thus,
      # with this technique you can at most _narrow_ the list of accessible
      # attributes for a particular mass-assignment call.
D
Initial  
David Heinemeier Hansson 已提交
552
      def attr_accessible(*attributes)
553
        write_inheritable_attribute(:attr_accessible, Set.new(attributes.map(&:to_s)) + (accessible_attributes || []))
D
Initial  
David Heinemeier Hansson 已提交
554
      end
555

556
      # Returns an array of all the attributes that have been made accessible to mass-assignment.
D
Initial  
David Heinemeier Hansson 已提交
557
      def accessible_attributes # :nodoc:
558
        read_inheritable_attribute(:attr_accessible)
D
Initial  
David Heinemeier Hansson 已提交
559 560
      end

561 562
       # Attributes listed as readonly can be set for a new record, but will be ignored in database updates afterwards.
       def attr_readonly(*attributes)
563
         write_inheritable_attribute(:attr_readonly, Set.new(attributes.map(&:to_s)) + (readonly_attributes || []))
564 565 566 567
       end

       # Returns an array of all the attributes that have been specified as readonly.
       def readonly_attributes
568
         read_inheritable_attribute(:attr_readonly) || []
569
       end
D
David Heinemeier Hansson 已提交
570

571 572 573
      # 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 已提交
574
      # class on retrieval or SerializationTypeMismatch will be raised.
D
David Heinemeier Hansson 已提交
575
      #
P
Pratik Naik 已提交
576
      # ==== Parameters
D
David Heinemeier Hansson 已提交
577
      #
P
Pratik Naik 已提交
578 579
      # * +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 已提交
580 581 582 583 584 585
      #
      # ==== Example
      #   # Serialize a preferences attribute
      #   class User
      #     serialize :preferences
      #   end
D
Initial  
David Heinemeier Hansson 已提交
586
      def serialize(attr_name, class_name = Object)
587
        serialized_attributes[attr_name.to_s] = class_name
D
Initial  
David Heinemeier Hansson 已提交
588
      end
589

D
Initial  
David Heinemeier Hansson 已提交
590 591
      # Returns a hash of all the attributes that have been specified for serialization as keys and their class restriction as values.
      def serialized_attributes
592
        read_inheritable_attribute(:attr_serialized) or write_inheritable_attribute(:attr_serialized, {})
D
Initial  
David Heinemeier Hansson 已提交
593 594 595
      end

      # Guesses the table name (in forced lower-case) based on the name of the class in the inheritance hierarchy descending
P
Pratik Naik 已提交
596
      # directly from ActiveRecord::Base. So if the hierarchy looks like: Reply < Message < ActiveRecord::Base, then Message is used
597
      # to guess the table name even when called on Reply. The rules used to do the guess are handled by the Inflector class
598
      # in Active Support, which knows almost all common English inflections. You can add new inflections in config/initializers/inflections.rb.
D
Initial  
David Heinemeier Hansson 已提交
599
      #
600
      # Nested classes are given table names prefixed by the singular form of
P
Pratik Naik 已提交
601 602 603
      # the parent's table name. Enclosing modules are not considered.
      #
      # ==== Examples
604 605
      #
      #   class Invoice < ActiveRecord::Base; end;
606 607
      #   file                  class               table_name
      #   invoice.rb            Invoice             invoices
608 609 610 611 612 613 614 615
      #
      #   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 已提交
616
      #
P
Pratik Naik 已提交
617 618
      # Additionally, the class-level +table_name_prefix+ is prepended and the
      # +table_name_suffix+ is appended.  So if you have "myapp_" as a prefix,
619 620 621 622 623
      # 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 已提交
624 625
      #
      #   class Mouse < ActiveRecord::Base
626
      #     set_table_name "mice"
D
Initial  
David Heinemeier Hansson 已提交
627
      #   end
628
      def table_name
629 630 631
        reset_table_name
      end

632
      # Returns a quoted version of the table name, used to construct SQL statements.
P
Pratik Naik 已提交
633 634 635 636
      def quoted_table_name
        @quoted_table_name ||= connection.quote_table_name(table_name)
      end

637
      # Computes the table name, (re)sets it internally, and returns it.
D
David Heinemeier Hansson 已提交
638
      def reset_table_name #:nodoc:
639
        self.table_name = compute_table_name
D
Initial  
David Heinemeier Hansson 已提交
640 641
      end

642
      def full_table_name_prefix #:nodoc:
643 644 645
        (parents.detect{ |p| p.respond_to?(:table_name_prefix) } || self).table_name_prefix
      end

646 647
      # Defines the column name for use with single table inheritance
      # -- can be set in subclasses like so: self.inheritance_column = "type_id"
D
Initial  
David Heinemeier Hansson 已提交
648
      def inheritance_column
649
        @inheritance_column ||= "type".freeze
D
Initial  
David Heinemeier Hansson 已提交
650 651
      end

652 653
      # 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 已提交
654
      def sequence_name #:nodoc:
655 656 657
        reset_sequence_name
      end

D
David Heinemeier Hansson 已提交
658
      def reset_sequence_name #:nodoc:
659 660 661
        default = connection.default_sequence_name(table_name, primary_key)
        set_sequence_name(default)
        default
662 663
      end

664
      # Sets the table name to use to the given value, or (if the value
665
      # is nil or false) to the value returned by the given block.
666 667 668 669
      #
      #   class Project < ActiveRecord::Base
      #     set_table_name "project"
      #   end
670
      def set_table_name(value = nil, &block)
671
        @quoted_table_name = nil
672 673 674 675 676 677
        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
678
      # given block.
679 680 681 682 683 684
      #
      #   class Project < ActiveRecord::Base
      #     set_inheritance_column do
      #       original_inheritance_column + "_id"
      #     end
      #   end
685
      def set_inheritance_column(value = nil, &block)
686 687 688 689
        define_attr_method :inheritance_column, value, &block
      end
      alias :inheritance_column= :set_inheritance_column

690 691
      # 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
692 693
      # given block. This is required for Oracle and is useful for any
      # database which relies on sequences for primary key generation.
694
      #
695 696 697 698 699
      # 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.
700 701 702 703
      #
      #   class Project < ActiveRecord::Base
      #     set_sequence_name "projectseq"   # default would have been "project_seq"
      #   end
704
      def set_sequence_name(value = nil, &block)
705 706 707 708
        define_attr_method :sequence_name, value, &block
      end
      alias :sequence_name= :set_sequence_name

D
Initial  
David Heinemeier Hansson 已提交
709 710 711
      # Turns the +table_name+ back into a class name following the reverse rules of +table_name+.
      def class_name(table_name = table_name) # :nodoc:
        # remove any prefix and/or suffix from the table name
712 713 714
        class_name = table_name[table_name_prefix.length..-(table_name_suffix.length + 1)].camelize
        class_name = class_name.singularize if pluralize_table_names
        class_name
D
Initial  
David Heinemeier Hansson 已提交
715 716
      end

717 718
      # Indicates whether the table associated with this class exists
      def table_exists?
719
        connection.table_exists?(table_name)
720 721
      end

D
Initial  
David Heinemeier Hansson 已提交
722 723
      # Returns an array of column objects for the table associated with this class.
      def columns
724
        unless defined?(@columns) && @columns
725
          @columns = connection.columns(table_name, "#{name} Columns")
726
          @columns.each { |column| column.primary = column.name == primary_key }
727 728
        end
        @columns
D
Initial  
David Heinemeier Hansson 已提交
729
      end
730

731
      # Returns a hash of column objects for the table associated with this class.
D
Initial  
David Heinemeier Hansson 已提交
732 733 734
      def columns_hash
        @columns_hash ||= columns.inject({}) { |hash, column| hash[column.name] = column; hash }
      end
735

D
David Heinemeier Hansson 已提交
736
      # Returns an array of column names as strings.
737
      def column_names
738
        @column_names ||= columns.map { |column| column.name }
739
      end
D
Initial  
David Heinemeier Hansson 已提交
740

741 742
      # 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 已提交
743
      def content_columns
744
        @content_columns ||= columns.reject { |c| c.primary || c.name =~ /(_id|_count)$/ || c.name == inheritance_column }
D
Initial  
David Heinemeier Hansson 已提交
745 746 747 748
      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
749
      # is available.
D
David Heinemeier Hansson 已提交
750
      def column_methods_hash #:nodoc:
751
        @dynamic_methods_hash ||= column_names.inject(Hash.new(false)) do |methods, attr|
752 753 754 755 756
          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 已提交
757 758 759
          methods
        end
      end
760

P
Pratik Naik 已提交
761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786
      # 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
787
      def reset_column_information
788
        undefine_attribute_methods
789
        @column_names = @columns = @columns_hash = @content_columns = @dynamic_methods_hash = @inheritance_column = nil
P
Pratik Naik 已提交
790
        @arel_engine = @unscoped = @arel_table = nil
791 792
      end

793
      def reset_column_information_and_inheritable_attributes_for_all_subclasses#:nodoc:
794
        descendants.each { |klass| klass.reset_inheritable_attributes; klass.reset_column_information }
795
      end
D
Initial  
David Heinemeier Hansson 已提交
796

797
      def attribute_method?(attribute)
J
José Valim 已提交
798
        super || (table_exists? && column_names.include?(attribute.to_s.sub(/=$/, '')))
799 800
      end

801 802
      # Set the lookup ancestors for ActiveModel.
      def lookup_ancestors #:nodoc:
803 804
        klass = self
        classes = [klass]
E
Emilio Tagua 已提交
805
        while klass != klass.base_class
806 807 808 809 810
          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 已提交
811
        # Apparently the method base_class causes some trouble.
812 813 814 815
        # It now works for sure.
        [self]
      end

816 817 818
      # Set the i18n scope to overwrite ActiveModel.
      def i18n_scope #:nodoc:
        :activerecord
D
Initial  
David Heinemeier Hansson 已提交
819
      end
820

821 822 823 824 825 826 827
      # 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 已提交
828 829
      end

830 831 832 833 834
      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

835
      # Returns a string like 'Post id:integer, title:string, body:text'
836
      def inspect
837 838 839 840
        if self == Base
          super
        elsif abstract_class?
          "#{super}(abstract)"
841
        elsif table_exists?
842 843
          attr_list = columns.map { |c| "#{c.name}: #{c.type}" } * ', '
          "#{super}(#{attr_list})"
844 845
        else
          "#{super}(Table doesn't exist)"
846
        end
847 848
      end

849
      def quote_value(value, column = nil) #:nodoc:
850
        connection.quote(value,column)
851 852
      end

853
      # Used to sanitize objects before they're used in an SQL SELECT statement. Delegates to <tt>connection.quote</tt>.
854
      def sanitize(object) #:nodoc:
855
        connection.quote(object)
D
Initial  
David Heinemeier Hansson 已提交
856 857
      end

858 859 860
      # Overwrite the default class equality method to provide support for association proxies.
      def ===(object)
        object.is_a?(self)
861
      end
862

863 864 865 866 867 868 869
      # 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.
      def base_class
        class_of_active_record_descendant(self)
      end

P
Pratik Naik 已提交
870
      # Set this to true if this is an abstract class (see <tt>abstract_class?</tt>).
871 872 873 874 875
      attr_accessor :abstract_class

      # Returns whether this class is a base AR class.  If A is a base class and
      # B descends from A, then B.base_class will return B.
      def abstract_class?
876
        defined?(@abstract_class) && @abstract_class == true
877 878
      end

879
      def respond_to?(method_id, include_private = false)
880 881
        if match = DynamicFinderMatch.match(method_id)
          return true if all_attributes_exists?(match.attribute_names)
882 883
        elsif match = DynamicScopeMatch.match(method_id)
          return true if all_attributes_exists?(match.attribute_names)
884
        end
885

886 887 888
        super
      end

889 890 891 892
      def sti_name
        store_full_sti_class ? name : name.demodulize
      end

893 894 895
      def unscoped
        @unscoped ||= Relation.new(self, arel_table)
        finder_needs_type_condition? ? @unscoped.where(type_condition) : @unscoped
896 897
      end

P
Pratik Naik 已提交
898 899
      def arel_table
        @arel_table ||= Arel::Table.new(table_name, :engine => arel_engine)
900 901
      end

902 903
      def arel_engine
        @arel_engine ||= begin
904 905 906
          if self == ActiveRecord::Base
            Arel::Table.engine
          else
907
            connection_handler.connection_pools[name] ? Arel::Sql::Engine.new(self) : superclass.arel_engine
908 909
          end
        end
910 911
      end

D
Initial  
David Heinemeier Hansson 已提交
912
      private
913 914 915
        # 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 已提交
916
        def instantiate(record)
917
          object = find_sti_class(record[inheritance_column]).allocate
918

919
          object.instance_variable_set(:'@attributes', record)
920
          object.instance_variable_set(:'@attributes_cache', {})
A
Aaron Patterson 已提交
921
          object.instance_variable_set(:@new_record, false)
922 923 924
          object.instance_variable_set(:@readonly, false)
          object.instance_variable_set(:@destroyed, false)
          object.instance_variable_set(:@marked_for_destruction, false)
925 926
          object.instance_variable_set(:@previously_changed, {})
          object.instance_variable_set(:@changed_attributes, {})
927

928 929
          object.send(:_run_find_callbacks)
          object.send(:_run_initialize_callbacks)
930

931
          object
D
Initial  
David Heinemeier Hansson 已提交
932
        end
933

934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949
        def find_sti_class(type_name)
          if type_name.blank? || !columns_hash.include?(inheritance_column)
            self
          else
            begin
              compute_type(type_name)
            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

950
        def construct_finder_arel(options = {}, scope = nil)
951
          relation = options.is_a?(Hash) ? unscoped.apply_finder_options(options) : unscoped.merge(options)
952
          relation = scope.merge(relation) if scope
953
          relation
954 955
        end

956
        def type_condition
957
          sti_column = arel_table[inheritance_column]
958
          condition = sti_column.eq(sti_name)
959
          descendants.each { |subclass| condition = condition.or(sti_column.eq(subclass.sti_name)) }
960

961
          condition
D
Initial  
David Heinemeier Hansson 已提交
962 963 964
        end

        # Guesses the table name, but does not decorate it with prefix and suffix information.
965
        def undecorated_table_name(class_name = base_class.name)
966 967
          table_name = class_name.to_s.demodulize.underscore
          table_name = table_name.pluralize if pluralize_table_names
968
          table_name
D
Initial  
David Heinemeier Hansson 已提交
969 970
        end

971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987
        # 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

P
Pratik Naik 已提交
988
        # Enables dynamic finders like <tt>find_by_user_name(user_name)</tt> and <tt>find_by_user_name_and_password(user_name, password)</tt>
P
Pratik Naik 已提交
989 990
        # that are turned into <tt>where(:user_name => user_name).first</tt> and <tt>where(:user_name => user_name, :password => :password).first</tt>
        # respectively. Also works for <tt>all</tt> by using <tt>find_all_by_amount(50)</tt> that is turned into <tt>where(:amount => 50).all</tt>.
991
        #
P
Pratik Naik 已提交
992 993
        # It's even possible to use all the additional parameters to +find+. For example, the full interface for +find_all_by_amount+
        # is actually <tt>find_all_by_amount(amount, options)</tt>.
994
        #
995
        # Each dynamic finder, scope or initializer/creator is also defined in the class after it is first invoked, so that future
996
        # attempts to use it do not run through method_missing.
997
        def method_missing(method_id, *arguments, &block)
998 999
          if match = DynamicFinderMatch.match(method_id)
            attribute_names = match.attribute_names
1000
            super unless all_attributes_exists?(attribute_names)
1001
            if match.finder?
1002
              options = arguments.extract_options!
1003
              relation = options.any? ? construct_finder_arel(options, current_scoped_methods) : scoped
1004
              relation.send :find_by_attributes, match, attribute_names, *arguments
1005
            elsif match.instantiator?
1006
              scoped.send :find_or_instantiator_by_attributes, match, attribute_names, *arguments, &block
1007
            end
1008 1009 1010 1011
          elsif match = DynamicScopeMatch.match(method_id)
            attribute_names = match.attribute_names
            super unless all_attributes_exists?(attribute_names)
            if match.scope?
1012
              self.class_eval <<-METHOD, __FILE__, __LINE__ + 1
1013 1014 1015 1016 1017
                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 已提交
1018
                                                                    #
1019 1020
                  scoped(:conditions => attributes)                 #   scoped(:conditions => attributes)
                end                                                 # end
1021
              METHOD
1022 1023
              send(method_id, *arguments)
            end
1024 1025 1026 1027
          else
            super
          end
        end
D
Initial  
David Heinemeier Hansson 已提交
1028

1029 1030 1031 1032 1033 1034
        def construct_attributes_from_arguments(attribute_names, arguments)
          attributes = {}
          attribute_names.each_with_index { |name, idx| attributes[name] = arguments[idx] }
          attributes
        end

1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049
        # 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

1050
        def all_attributes_exists?(attribute_names)
1051
          attribute_names = expand_attribute_names_for_aggregates(attribute_names)
1052
          attribute_names.all? { |name| column_methods_hash.include?(name.to_sym) }
1053
        end
1054

D
Initial  
David Heinemeier Hansson 已提交
1055
      protected
1056
        # Scope parameters to method calls within the block.  Takes a hash of method_name => parameters hash.
1057 1058
        # method_name may be <tt>:find</tt> or <tt>:create</tt>. <tt>:find</tt> parameter is <tt>Relation</tt> while
        # <tt>:create</tt> parameters are an attributes hash.
1059 1060 1061
        #
        #   class Article < ActiveRecord::Base
        #     def self.create_with_scope
1062
        #       with_scope(:find => where(:blog_id => 1), :create => { :blog_id => 1 }) do
1063 1064 1065 1066 1067 1068 1069
        #         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 已提交
1070
        # In nested scopings, all previous parameters are overwritten by the innermost rule, with the exception of
1071
        # <tt>where</tt>, <tt>includes</tt>, and <tt>joins</tt> operations in <tt>Relation</tt>, which are merged.
P
Pratik Naik 已提交
1072
        #
1073
        # <tt>joins</tt> operations are uniqued so multiple scopes can join in the same table without table aliasing
P
Pratik Naik 已提交
1074 1075
        # 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.
1076 1077 1078
        #
        #   class Article < ActiveRecord::Base
        #     def self.find_with_scope
1079 1080 1081
        #       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
1082
        #         end
1083 1084
        #         with_scope(:find => where(:author_id => 3)) do
        #           all # => SELECT * from articles WHERE blog_id = 1 AND author_id = 3 LIMIT 1
1085 1086 1087 1088 1089
        #         end
        #       end
        #     end
        #   end
        #
1090
        # You can ignore any previous scopings by using the <tt>with_exclusive_scope</tt> method.
1091 1092 1093
        #
        #   class Article < ActiveRecord::Base
        #     def self.find_with_exclusive_scope
1094 1095 1096
        #       with_scope(:find => where(:blog_id => 1).limit(1)) do
        #         with_exclusive_scope(:find => limit(10))
        #           all # => SELECT * from articles LIMIT 10
1097 1098 1099 1100
        #         end
        #       end
        #     end
        #   end
P
Pratik Naik 已提交
1101 1102 1103
        #
        # *Note*: the +:find+ scope also has effect on update and deletion methods,
        # like +update_all+ and +delete_all+.
1104 1105 1106
        def with_scope(method_scoping = {}, action = :merge, &block)
          method_scoping = method_scoping.method_scoping if method_scoping.respond_to?(:method_scoping)

1107 1108 1109 1110 1111 1112
          if method_scoping.is_a?(Hash)
            # Dup first and second level of hash (method and params).
            method_scoping = method_scoping.inject({}) do |hash, (method, params)|
              hash[method] = (params == true) ? params : params.dup
              hash
            end
1113

1114 1115 1116 1117
            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]
1118
              scope_for_create = if action == :merge
1119 1120 1121
                current_scoped_methods.create_with_value.merge(method_scoping[:create])
              else
                method_scoping[:create]
1122
              end
1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133

              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

1134
          method_scoping = current_scoped_methods.merge(method_scoping) if current_scoped_methods && action ==  :merge
1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148

          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)
          with_scope(method_scoping, :overwrite, &block)
        end

1149
        # Sets the default options for the model. The format of the
1150
        # <tt>options</tt> argument is the same as in find.
1151 1152
        #
        #   class Person < ActiveRecord::Base
1153
        #     default_scope order('last_name, first_name')
1154 1155
        #   end
        def default_scope(options = {})
1156 1157 1158
          self.default_scoping << construct_finder_arel(options, default_scoping.pop)
        end

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

D
David Heinemeier Hansson 已提交
1164
        def current_scoped_methods #:nodoc:
1165
          scoped_methods.last
1166
        end
1167

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

            candidates.each do |candidate|
              begin
                constant = candidate.constantize
                return constant if candidate == constant.to_s
1185 1186 1187
              rescue NameError => e
                # We don't want to swallow NoMethodError < NameError errors
                raise e unless e.instance_of?(NameError)
1188 1189
              rescue ArgumentError
              end
1190
            end
1191 1192

            raise NameError, "uninitialized constant #{candidates.first}"
D
Initial  
David Heinemeier Hansson 已提交
1193 1194 1195
          end
        end

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

P
Pratik Naik 已提交
1208
        # Returns the name of the class descending directly from Active Record in the inheritance hierarchy.
D
David Heinemeier Hansson 已提交
1209
        def class_name_of_active_record_descendant(klass) #:nodoc:
1210
          klass.base_class.name
1211 1212
        end

P
Pratik Naik 已提交
1213
        # Accepts an array, hash, or string of SQL conditions and sanitizes
1214
        # them into a valid SQL fragment for a WHERE clause.
1215 1216 1217
        #   ["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'"
1218
        def sanitize_sql_for_conditions(condition, table_name = self.table_name)
1219 1220
          return nil if condition.blank?

1221 1222
          case condition
            when Array; sanitize_sql_array(condition)
1223
            when Hash;  sanitize_sql_hash_for_conditions(condition, table_name)
1224 1225
            else        condition
          end
1226
        end
1227
        alias_method :sanitize_sql, :sanitize_sql_for_conditions
1228

P
Pratik Naik 已提交
1229
        # Accepts an array, hash, or string of SQL conditions and sanitizes
1230 1231 1232 1233 1234 1235 1236 1237 1238 1239
        # 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

1240 1241 1242 1243 1244
        def aggregate_mapping(reflection)
          mapping = reflection.options[:mapping] || [reflection.name, reflection.name]
          mapping.first.is_a?(Array) ? mapping : [mapping]
        end

P
Pratik Naik 已提交
1245
        # Accepts a hash of SQL conditions and replaces those attributes
1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274
        # 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

1275
        # Sanitizes a hash of attribute/value pairs into SQL conditions for a WHERE clause.
1276 1277 1278 1279
        #   { :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)"
1280 1281
        #   { :age => 13..18 }
        #     # => "age BETWEEN 13 AND 18"
1282 1283
        #   { 'other_records.id' => 7 }
        #     # => "`other_records`.`id` = 7"
1284 1285
        #   { :other_records => { :id => 7 } }
        #     # => "`other_records`.`id` = 7"
1286 1287 1288
        # 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'"
1289
        def sanitize_sql_hash_for_conditions(attrs, default_table_name = self.table_name)
1290 1291
          attrs = expand_hash_conditions_for_aggregates(attrs)

1292 1293
          table = Arel::Table.new(self.table_name, :engine => arel_engine, :as => default_table_name)
          builder = PredicateBuilder.new(arel_engine)
1294
          builder.build_from_hash(attrs, table).map(&:to_sql).join(' AND ')
1295
        end
1296 1297 1298 1299 1300 1301
        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)
1302
          attrs.map do |attr, value|
1303 1304 1305
            "#{connection.quote_column_name(attr)} = #{quote_bound_value(value)}"
          end.join(', ')
        end
1306

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

1321 1322
        alias_method :sanitize_conditions, :sanitize_sql

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

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

1342
        def expand_range_bind_variables(bind_vars) #:nodoc:
1343 1344 1345
          expanded = []

          bind_vars.each do |var|
1346 1347
            next if var.is_a?(Hash)

1348
            if var.is_a?(Range)
1349 1350
              expanded << var.first
              expanded << var.last
1351
            else
1352
              expanded << var
1353
            end
1354
          end
1355 1356

          expanded
1357 1358
        end

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

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

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

D
Initial  
David Heinemeier Hansson 已提交
1399
        ensure_proper_type
P
Pratik Naik 已提交
1400 1401 1402 1403 1404

        if scope = self.class.send(:current_scoped_methods)
          create_with = scope.scope_for_create
          create_with.each { |att,value| self.send("#{att}=", value) } if create_with
        end
1405
        self.attributes = attributes unless attributes.nil?
P
Pratik Naik 已提交
1406

1407
        result = yield self if block_given?
1408
        _run_initialize_callbacks
1409
        result
D
Initial  
David Heinemeier Hansson 已提交
1410
      end
1411

1412 1413 1414 1415 1416 1417 1418
      # 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)
        callback(:after_initialize) if respond_to_without_attributes?(:after_initialize)
        cloned_attributes = other.clone_attributes(:read_attribute_before_type_cast)
        cloned_attributes.delete(self.class.primary_key)
1419

1420
        @attributes = cloned_attributes
1421 1422 1423 1424 1425 1426

        @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

1427
        clear_aggregation_cache
1428
        clear_association_cache
1429 1430 1431
        @attributes_cache = {}
        @new_record = true
        ensure_proper_type
1432 1433 1434 1435 1436

        if scope = self.class.send(:current_scoped_methods)
          create_with = scope.scope_for_create
          create_with.each { |att,value| self.send("#{att}=", value) } if create_with
        end
1437 1438
      end

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

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

1484
      def quoted_id #:nodoc:
1485
        quote_value(id, column_for_attribute(self.class.primary_key))
1486
      end
1487

1488 1489 1490 1491 1492 1493 1494 1495 1496 1497
      # 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

1498
      # Returns the value of the attribute identified by <tt>attr_name</tt> after it has been typecast (for example,
D
Initial  
David Heinemeier Hansson 已提交
1499 1500
      # "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).
1501
      def [](attr_name)
1502
        read_attribute(attr_name)
D
Initial  
David Heinemeier Hansson 已提交
1503
      end
1504

D
Initial  
David Heinemeier Hansson 已提交
1505 1506
      # Updates the attribute identified by <tt>attr_name</tt> with the specified +value+.
      # (Alias for the protected write_attribute method).
1507
      def []=(attr_name, value)
1508
        write_attribute(attr_name, value)
D
Initial  
David Heinemeier Hansson 已提交
1509 1510 1511
      end

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

        multi_parameter_attributes = []
1536
        attributes = remove_attributes_protected_from_mass_assignment(attributes) if guard_protected_attributes
1537

1538
        attributes.each do |k, v|
1539 1540 1541 1542 1543
          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 已提交
1544
        end
D
David Heinemeier Hansson 已提交
1545

D
Initial  
David Heinemeier Hansson 已提交
1546 1547 1548
        assign_multiparameter_attributes(multi_parameter_attributes)
      end

1549
      # Returns a hash of all the attributes with their names as keys and the values of the attributes as values.
1550
      def attributes
1551 1552 1553
        attrs = {}
        attribute_names.each { |name| attrs[name] = read_attribute(name) }
        attrs
1554 1555
      end

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

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

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

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

1593
      # 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 已提交
1594
      def ==(comparison_object)
1595
        comparison_object.equal?(self) ||
1596
          (comparison_object.instance_of?(self.class) &&
1597
            comparison_object.id == id && !comparison_object.new_record?)
D
Initial  
David Heinemeier Hansson 已提交
1598 1599 1600 1601 1602 1603
      end

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

D
Initial  
David Heinemeier Hansson 已提交
1605 1606 1607
      # 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
1608
        id.hash
D
Initial  
David Heinemeier Hansson 已提交
1609 1610
      end

1611
      # Freeze the attributes hash such that associations are still accessible, even on destroyed records.
1612
      def freeze
1613
        @attributes.freeze; self
1614
      end
1615

1616
      # Returns +true+ if the attributes hash has been frozen.
1617 1618 1619
      def frozen?
        @attributes.frozen?
      end
1620

1621 1622 1623
      # Returns duplicated record with unfreezed attributes.
      def dup
        obj = super
1624
        obj.instance_variable_set('@attributes', @attributes.dup)
1625 1626 1627
        obj
      end

1628 1629
      # 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.
1630
      def readonly?
1631
        @readonly
1632 1633
      end

1634 1635
      # Marks this record as read only.
      def readonly!
1636 1637
        @readonly = true
      end
1638

1639
      # Returns the contents of the record as a nicely formatted string.
1640
      def inspect
1641
        attributes_as_nice_string = self.class.column_names.collect { |name|
1642 1643 1644 1645
          if has_attribute?(name) || new_record?
            "#{name}: #{attribute_for_inspect(name)}"
          end
        }.compact.join(", ")
1646
        "#<#{self.class} #{attributes_as_nice_string}>"
1647
      end
1648

1649 1650
    protected
      def clone_attributes(reader_method = :read_attribute, attributes = {})
1651 1652
        attribute_names.each do |name|
          attributes[name] = clone_attribute_value(reader_method, name)
1653
        end
1654
        attributes
1655 1656 1657 1658 1659 1660 1661 1662 1663
      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 已提交
1664 1665
    private

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

      def remove_attributes_protected_from_mass_assignment(attributes)
1677 1678 1679 1680
        safe_attributes =
          if self.class.accessible_attributes.nil? && self.class.protected_attributes.nil?
            attributes.reject { |key, value| attributes_protected_by_default.include?(key.gsub(/\(.+/, "")) }
          elsif self.class.protected_attributes.nil?
1681
            attributes.reject { |key, value| !self.class.accessible_attributes.include?(key.gsub(/\(.+/, "")) || attributes_protected_by_default.include?(key.gsub(/\(.+/, "")) }
1682
          elsif self.class.accessible_attributes.nil?
1683
            attributes.reject { |key, value| self.class.protected_attributes.include?(key.gsub(/\(.+/,"")) || attributes_protected_by_default.include?(key.gsub(/\(.+/, "")) }
1684 1685 1686 1687 1688 1689 1690
          else
            raise "Declare either attr_protected or attr_accessible for #{self.class}, but not both."
          end

        removed_attributes = attributes.keys - safe_attributes.keys

        if removed_attributes.any?
1691
          log_protected_attribute_removal(removed_attributes)
D
Initial  
David Heinemeier Hansson 已提交
1692
        end
1693 1694

        safe_attributes
D
Initial  
David Heinemeier Hansson 已提交
1695
      end
1696

1697 1698 1699
      # Removes attributes which have been marked as readonly.
      def remove_readonly_attributes(attributes)
        unless self.class.readonly_attributes.nil?
1700
          attributes.delete_if { |key, value| self.class.readonly_attributes.include?(key.gsub(/\(.+/,"")) }
1701 1702 1703 1704
        else
          attributes
        end
      end
D
Initial  
David Heinemeier Hansson 已提交
1705

1706
      def log_protected_attribute_removal(*attributes)
1707 1708 1709
        if logger
          logger.debug "WARNING: Can't mass-assign these protected attributes: #{attributes.join(', ')}"
        end
1710 1711
      end

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

1719 1720
      # Returns a copy of the attributes hash where all the values have been safely quoted for use in
      # an Arel insert/update method.
1721 1722 1723 1724 1725 1726
      def arel_attributes_values(include_primary_key = true, include_readonly_attributes = true, attribute_names = @attributes.keys)
        attrs = {}
        attribute_names.each do |name|
          if (column = column_for_attribute(name)) && (include_primary_key || !column.primary)

            if include_readonly_attributes || (!include_readonly_attributes && !self.class.readonly_attributes.include?(name))
1727 1728 1729
              value = read_attribute(name)

              if value && ((self.class.serialized_attributes.has_key?(name) && (value.acts_like?(:date) || value.acts_like?(:time))) || value.is_a?(Hash) || value.is_a?(Array))
1730 1731
                value = value.to_yaml
              end
1732
              attrs[self.class.arel_table[name]] = value
1733 1734 1735 1736 1737 1738
            end
          end
        end
        attrs
      end

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

P
Pratik Naik 已提交
1744
      # Interpolate custom SQL string in instance context.
D
Initial  
David Heinemeier Hansson 已提交
1745 1746
      # Optional record argument is meant for custom insert_sql.
      def interpolate_sql(sql, record = nil)
1747
        instance_eval("%@#{sql.gsub('@', '\@')}@", __FILE__, __LINE__)
D
Initial  
David Heinemeier Hansson 已提交
1748 1749 1750 1751 1752 1753
      end

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

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

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

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

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

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

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

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

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

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

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

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

1835 1836 1837 1838
      def quote_columns(quoter, hash)
        hash.inject({}) do |quoted, (name, value)|
          quoted[quoter.quote_column_name(name)] = value
          quoted
1839
        end
D
Initial  
David Heinemeier Hansson 已提交
1840 1841
      end

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

1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861
      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
D
Initial  
David Heinemeier Hansson 已提交
1862
  end
J
Joshua Peek 已提交
1863 1864

  Base.class_eval do
1865
    include ActiveRecord::Persistence
J
Joshua Peek 已提交
1866
    extend ActiveModel::Naming
J
Joshua Peek 已提交
1867
    extend QueryCache::ClassMethods
J
José Valim 已提交
1868
    extend ActiveSupport::Benchmarkable
1869
    extend ActiveSupport::DescendantsTracker
J
José Valim 已提交
1870

1871
    include ActiveModel::Conversion
J
Joshua Peek 已提交
1872
    include Validations
1873
    extend CounterCache
J
Joshua Peek 已提交
1874 1875
    include Locking::Optimistic, Locking::Pessimistic
    include AttributeMethods
J
Joshua Peek 已提交
1876
    include AttributeMethods::Read, AttributeMethods::Write, AttributeMethods::BeforeTypeCast, AttributeMethods::Query
1877
    include AttributeMethods::PrimaryKey
J
Joshua Peek 已提交
1878 1879
    include AttributeMethods::TimeZoneConversion
    include AttributeMethods::Dirty
1880
    include Callbacks, ActiveModel::Observing, Timestamp
J
Joshua Peek 已提交
1881
    include Associations, AssociationPreload, NamedScope
1882 1883 1884 1885

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

1888
    NilClass.add_whiner(self) if NilClass.respond_to?(:add_whiner)
J
Joshua Peek 已提交
1889
  end
1890
end
1891 1892 1893

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