2004-11-23 20:04:44 -05:00
require 'yaml'
2005-11-02 12:30:59 -05:00
require 'set'
2005-04-03 06:52:05 -04:00
require 'active_record/deprecated_finders'
2004-11-23 20:04:44 -05:00
module ActiveRecord #:nodoc:
class ActiveRecordError < StandardError #:nodoc:
end
2004-12-14 07:32:29 -05:00
class SubclassNotFound < ActiveRecordError #:nodoc:
end
2004-11-23 20:04:44 -05:00
class AssociationTypeMismatch < ActiveRecordError #:nodoc:
end
class SerializationTypeMismatch < ActiveRecordError #:nodoc:
end
class AdapterNotSpecified < ActiveRecordError # :nodoc:
end
class AdapterNotFound < ActiveRecordError # :nodoc:
end
class ConnectionNotEstablished < ActiveRecordError #:nodoc:
end
class ConnectionFailed < ActiveRecordError #:nodoc:
end
class RecordNotFound < ActiveRecordError #:nodoc:
end
class StatementInvalid < ActiveRecordError #:nodoc:
end
2004-12-08 05:38:10 -05:00
class PreparedStatementInvalid < ActiveRecordError #:nodoc:
end
2004-12-31 14:38:04 -05:00
class StaleObjectError < ActiveRecordError #:nodoc:
end
2005-07-11 02:09:08 -04:00
class ConfigurationError < StandardError #:nodoc:
end
2005-10-14 20:19:56 -04:00
class ReadOnlyRecord < StandardError #:nodoc:
end
2005-09-11 05:41:24 -04:00
2005-03-06 09:11:26 -05:00
class AttributeAssignmentError < ActiveRecordError #:nodoc:
attr_reader :exception , :attribute
def initialize ( message , exception , attribute )
@exception = exception
@attribute = attribute
@message = message
end
end
2005-07-03 04:32:43 -04:00
2005-03-06 09:11:26 -05:00
class MultiparameterAssignmentErrors < ActiveRecordError #:nodoc:
attr_reader :errors
def initialize ( errors )
@errors = errors
end
end
2005-07-03 04:32:43 -04:00
2005-10-10 23:55:49 -04:00
# Active Record objects don't specify their attributes directly, but rather infer them from the table definition with
2004-11-23 20:04:44 -05:00
# 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
2005-07-03 04:32:43 -04:00
# database table will happen automatically in most common cases, but can be overwritten for the uncommon ones.
#
2004-11-23 20:04:44 -05:00
# See the mapping rules in table_name and the full example in link:files/README.html for more insight.
2005-07-03 04:32:43 -04:00
#
2004-11-23 20:04:44 -05:00
# == Creation
2005-07-03 04:32:43 -04:00
#
2005-10-10 23:55:49 -04:00
# Active Records accept constructor parameters either in a hash or as a block. The hash method is especially useful when
2004-11-23 20:04:44 -05:00
# you're receiving the data from somewhere else, like a HTTP request. It works like this:
2005-07-03 04:32:43 -04:00
#
2005-04-17 13:16:24 -04:00
# user = User.new(:name => "David", :occupation => "Code Artist")
2004-11-23 20:04:44 -05:00
# user.name # => "David"
2005-07-03 04:32:43 -04:00
#
2004-11-23 20:04:44 -05:00
# You can also use block initialization:
2005-07-03 04:32:43 -04:00
#
2004-11-23 20:04:44 -05:00
# user = User.new do |u|
# u.name = "David"
# u.occupation = "Code Artist"
# end
2005-07-03 04:32:43 -04:00
#
2004-11-23 20:04:44 -05:00
# And of course you can just create a bare object and specify the attributes after the fact:
2005-07-03 04:32:43 -04:00
#
2004-11-23 20:04:44 -05:00
# user = User.new
# user.name = "David"
# user.occupation = "Code Artist"
2005-07-03 04:32:43 -04:00
#
2004-11-23 20:04:44 -05:00
# == Conditions
2005-07-03 04:32:43 -04:00
#
2004-11-23 20:04:44 -05:00
# Conditions can either be specified as a string or an array representing the WHERE-part of an SQL statement.
# The array form is to be used when the condition input is tainted and requires sanitization. The string form can
2005-10-10 23:55:49 -04:00
# be used for statements that don't involve tainted data. Examples:
2005-07-03 04:32:43 -04:00
#
2004-11-23 20:04:44 -05:00
# User < ActiveRecord::Base
# def self.authenticate_unsafely(user_name, password)
2005-06-26 07:25:32 -04:00
# find(:first, :conditions => "user_name = '#{user_name}' AND password = '#{password}'")
2004-11-23 20:04:44 -05:00
# end
2005-07-03 04:32:43 -04:00
#
2004-11-23 20:04:44 -05:00
# def self.authenticate_safely(user_name, password)
2005-06-26 07:25:32 -04:00
# find(:first, :conditions => [ "user_name = ? AND password = ?", user_name, password ])
2004-11-23 20:04:44 -05:00
# end
# end
2005-07-03 04:32:43 -04:00
#
2004-12-06 13:08:35 -05:00
# The <tt>authenticate_unsafely</tt> method inserts the parameters directly into the query and is thus susceptible to SQL-injection
2005-07-03 04:32:43 -04:00
# attacks if the <tt>user_name</tt> and +password+ parameters come directly from a HTTP request. The <tt>authenticate_safely</tt> method,
2004-12-06 13:08:35 -05:00
# on the other hand, will sanitize the <tt>user_name</tt> and +password+ before inserting them in the query, which will ensure that
2004-11-23 20:04:44 -05:00
# an attacker can't escape the query and fake the login (or worse).
2004-12-06 13:08:35 -05:00
#
2005-03-27 07:04:07 -05:00
# When using multiple parameters in the conditions, it can easily become hard to read exactly what the fourth or fifth
2005-07-03 04:32:43 -04:00
# question mark is supposed to represent. In those cases, you can resort to named bind variables instead. That's done by replacing
2005-03-27 07:04:07 -05:00
# the question marks with symbols and supplying a hash with values for the matching symbol keys:
#
2005-07-03 04:32:43 -04:00
# Company.find(:first, [
# "id = :id AND name = :name AND division = :division AND created_at > :accounting_date",
2005-03-27 07:04:07 -05:00
# { :id => 3, :name => "37signals", :division => "First", :accounting_date => '2005-01-01' }
# ])
#
2004-11-23 20:04:44 -05:00
# == Overwriting default accessors
2005-07-03 04:32:43 -04:00
#
2004-11-23 20:04:44 -05:00
# All column values are automatically available through basic accessors on the Active Record object, but some times you
# want to specialize this behavior. This can be done by either by overwriting the default accessors (using the same
# name as the attribute) calling read_attribute(attr_name) and write_attribute(attr_name, value) to actually change things.
# Example:
2005-07-03 04:32:43 -04:00
#
2004-11-23 20:04:44 -05:00
# class Song < ActiveRecord::Base
# # Uses an integer of seconds to hold the length of the song
2005-07-03 04:32:43 -04:00
#
2004-11-23 20:04:44 -05:00
# def length=(minutes)
2005-04-17 13:16:24 -04:00
# write_attribute(:length, minutes * 60)
2004-11-23 20:04:44 -05:00
# end
2005-07-03 04:32:43 -04:00
#
2004-11-23 20:04:44 -05:00
# def length
2005-04-17 13:16:24 -04:00
# read_attribute(:length) / 60
2004-11-23 20:04:44 -05:00
# end
# end
2005-07-03 04:32:43 -04:00
#
2005-04-17 13:16:24 -04:00
# You can alternatively use self[:attribute]=(value) and self[:attribute] instead of write_attribute(:attribute, vaule) and
# read_attribute(:attribute) as a shorter form.
#
2005-10-10 23:55:49 -04:00
# == Accessing attributes before they have been typecasted
2005-02-23 18:51:34 -05:00
#
2005-10-10 23:55:49 -04:00
# Sometimes you want to be able to read the raw attribute data without having the column-determined typecast run its course first.
2005-02-23 18:51:34 -05:00
# That can be done by using the <attribute>_before_type_cast accessors that all attributes have. For example, if your Account model
2005-07-03 04:32:43 -04:00
# has a balance attribute, you can call account.balance_before_type_cast or account.id_before_type_cast.
2005-02-23 18:51:34 -05:00
#
# This is especially useful in validation situations where the user might supply a string for an integer field and you want to display
2005-10-10 23:55:49 -04:00
# the original string back in an error message. Accessing the attribute normally would typecast the string to 0, which isn't what you
2005-02-23 18:51:34 -05:00
# want.
#
2005-01-02 08:31:00 -05:00
# == Dynamic attribute-based finders
#
2005-11-04 14:39:50 -05:00
# Dynamic attribute-based finders are a cleaner way of getting (and/or creating) objects by simple queries without turning to SQL. They work by
2005-06-25 07:47:37 -04:00
# appending the name of an attribute to <tt>find_by_</tt> or <tt>find_all_by_</tt>, so you get finders like Person.find_by_user_name,
2005-07-03 04:32:43 -04:00
# Person.find_all_by_last_name, Payment.find_by_transaction_id. So instead of writing
# <tt>Person.find(:first, ["user_name = ?", user_name])</tt>, you just do <tt>Person.find_by_user_name(user_name)</tt>.
2005-06-25 07:47:37 -04:00
# And instead of writing <tt>Person.find(:all, ["last_name = ?", last_name])</tt>, you just do <tt>Person.find_all_by_last_name(last_name)</tt>.
2005-07-03 04:32:43 -04:00
#
2005-01-02 08:31:00 -05:00
# 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
2005-07-03 04:32:43 -04:00
# <tt>Person.find(:first, ["user_name = ? AND password = ?", user_name, password])</tt>, you just do
2005-01-02 08:31:00 -05:00
# <tt>Person.find_by_user_name_and_password(user_name, password)</tt>.
2005-07-03 04:32:43 -04:00
#
2005-04-17 13:16:24 -04:00
# It's even possible to use all the additional parameters to find. For example, the full interface for Payment.find_all_by_amount
# is actually Payment.find_all_by_amount(amount, options). And the full interface to Person.find_by_user_name is
# actually Person.find_by_user_name(user_name, options). So you could call <tt>Payment.find_all_by_amount(50, :order => "created_on")</tt>.
2005-01-02 08:51:00 -05:00
#
2005-11-04 14:39:50 -05:00
# The same dynamic finder style can be used to create the object if it doesn't already exist. This dynamic finder is called with
# <tt>find_or_create_by_</tt> and will return the object if it already exists and otherwise creates it, then returns it. Example:
#
# # No 'Summer' tag exists
# Tag.find_or_create_by_name("Summer") # equal to Tag.create(:name => "Summer")
#
# # Now the 'Summer' tag does exist
# Tag.find_or_create_by_name("Summer") # equal to Tag.find_by_name("Summer")
#
2005-02-07 09:15:53 -05:00
# == Saving arrays, hashes, and other non-mappable objects in text columns
2005-07-03 04:32:43 -04:00
#
# 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+.
2005-10-10 23:55:49 -04:00
# This makes it possible to store arrays, hashes, and other non-mappable objects without doing any additional work. Example:
2005-07-03 04:32:43 -04:00
#
2004-11-23 20:04:44 -05:00
# class User < ActiveRecord::Base
# serialize :preferences
# end
2005-07-03 04:32:43 -04:00
#
2005-04-17 13:16:24 -04:00
# user = User.create(:preferences) => { "background" => "black", "display" => large })
2004-11-23 20:04:44 -05:00
# User.find(user.id).preferences # => { "background" => "black", "display" => large }
2005-07-03 04:32:43 -04:00
#
2005-10-10 23:55:49 -04:00
# You can also specify a class option as the second parameter that'll raise an exception if a serialized object is retrieved as a
2004-11-23 20:04:44 -05:00
# descendent of a class not in the hierarchy. Example:
2005-07-03 04:32:43 -04:00
#
2004-11-23 20:04:44 -05:00
# class User < ActiveRecord::Base
2005-01-25 15:00:20 -05:00
# serialize :preferences, Hash
2004-11-23 20:04:44 -05:00
# end
2005-07-03 04:32:43 -04:00
#
2005-04-17 13:16:24 -04:00
# user = User.create(:preferences => %w( one two three ))
2004-11-23 20:04:44 -05:00
# User.find(user.id).preferences # raises SerializationTypeMismatch
2005-07-03 04:32:43 -04:00
#
2004-11-23 20:04:44 -05:00
# == Single table inheritance
#
2005-07-03 04:32:43 -04:00
# Active Record allows inheritance by storing the name of the class in a column that by default is called "type" (can be changed
2004-11-23 20:04:44 -05:00
# 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
#
2005-04-17 13:16:24 -04:00
# When you do Firm.create(:name => "37signals"), this record will be saved in the companies table with type = "Firm". You can then
# fetch this row again using Company.find(:first, "name = '37signals'") and it will return a Firm object.
2004-11-23 20:04:44 -05:00
#
2004-12-16 11:56:30 -05:00
# 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.
#
2004-11-23 20:04:44 -05:00
# Note, all the attributes for all the cases are kept in the same table. Read more:
# http://www.martinfowler.com/eaaCatalog/singleTableInheritance.html
2005-07-03 04:32:43 -04:00
#
2004-11-23 20:04:44 -05:00
# == Connection to multiple databases in different models
#
# Connections are usually created through ActiveRecord::Base.establish_connection and retrieved by ActiveRecord::Base.connection.
2005-07-03 04:32:43 -04:00
# All classes inheriting from ActiveRecord::Base will use this connection. But you can also set a class-specific connection.
2004-11-23 20:04:44 -05:00
# For example, if Course is a ActiveRecord::Base, but resides in a different database you can just say Course.establish_connection
# and Course *and all its subclasses* will use this connection instead.
#
# 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
2005-07-03 04:32:43 -04:00
#
2004-11-23 20:04:44 -05:00
# * +ActiveRecordError+ -- generic error class and superclass of all other errors raised by Active Record
2005-07-03 04:32:43 -04:00
# * +AdapterNotSpecified+ -- the configuration hash used in <tt>establish_connection</tt> didn't include a
2004-11-23 20:04:44 -05:00
# <tt>:adapter</tt> key.
2005-07-22 02:27:07 -04:00
# * +AdapterNotFound+ -- the <tt>:adapter</tt> key used in <tt>establish_connection</tt> specified an non-existent adapter
2005-07-03 04:32:43 -04:00
# (or a bad spelling of an existing one).
# * +AssociationTypeMismatch+ -- the object assigned to the association wasn't of the type specified in the association definition.
# * +SerializationTypeMismatch+ -- the object serialized wasn't of the class specified as the second parameter.
2004-11-23 20:04:44 -05:00
# * +ConnectionNotEstablished+ -- no connection has been established. Use <tt>establish_connection</tt> before querying.
2005-07-03 04:32:43 -04:00
# * +RecordNotFound+ -- no record responded to the find* method.
2004-11-23 20:04:44 -05:00
# Either the row with the given ID doesn't exist or the row didn't meet the additional restrictions.
# * +StatementInvalid+ -- the database server rejected the SQL statement. The precise error is added in the message.
# Either the record with the given ID doesn't exist or the record didn't meet the additional restrictions.
2005-07-03 04:32:43 -04:00
# * +MultiparameterAssignmentErrors+ -- collection of errors that occurred during a mass assignment using the
# +attributes=+ method. The +errors+ property of this exception contains an array of +AttributeAssignmentError+
2005-03-06 09:11:26 -05:00
# objects that should be inspected to determine which attributes triggered the errors.
# * +AttributeAssignmentError+ -- an error occurred while doing a mass assignment through the +attributes=+ method.
# You can inspect the +attribute+ property of the exception object to determine which attribute triggered the error.
2005-09-11 05:41:24 -04:00
#
2005-07-03 04:32:43 -04:00
# *Note*: The attributes listed are class-level attributes (accessible from both the class and instance level).
2004-11-23 20:04:44 -05:00
# So it's possible to assign a logger to the class through Base.logger= which will then be used by all
# instances in the current object space.
class Base
# 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+.
cattr_accessor :logger
2006-02-02 00:41:00 -05:00
2006-02-04 01:11:42 -05:00
include Reloadable :: Subclasses
2006-02-02 00:41:00 -05:00
2004-11-23 20:04:44 -05:00
def self . inherited ( child ) #:nodoc:
@@subclasses [ self ] || = [ ]
@@subclasses [ self ] << child
super
end
2005-10-15 16:49:04 -04:00
2005-09-20 06:33:26 -04:00
def self . reset_subclasses
2005-10-15 16:49:04 -04:00
nonreloadables = [ ]
2005-10-15 14:21:27 -04:00
subclasses . each do | klass |
2005-10-15 16:49:04 -04:00
unless klass . reloadable?
nonreloadables << klass
next
end
2005-10-15 14:21:27 -04:00
klass . instance_variables . each { | var | klass . send ( :remove_instance_variable , var ) }
klass . instance_methods ( false ) . each { | m | klass . send :undef_method , m }
end
2005-10-15 16:49:04 -04:00
@@subclasses = { }
nonreloadables . each { | klass | ( @@subclasses [ klass . superclass ] || = [ ] ) << klass }
2005-09-20 06:33:26 -04:00
end
2004-11-23 20:04:44 -05:00
@@subclasses = { }
2005-06-12 09:32:06 -04:00
2004-11-23 20:04:44 -05:00
cattr_accessor :configurations
2005-06-12 09:32:06 -04:00
@@configurations = { }
2005-07-03 04:32:43 -04:00
# Accessor for the prefix type that will be prepended to every primary key column name. The options are :table_name and
2004-11-23 20:04:44 -05:00
# :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
2005-07-03 04:32:43 -04:00
# that this is a global setting for all Active Records.
2004-11-23 20:04:44 -05:00
cattr_accessor :primary_key_prefix_type
@@primary_key_prefix_type = nil
2005-07-03 04:32:43 -04:00
# Accessor for the name of the prefix string to prepend to every table name. So if set to "basecamp_", all
2005-02-07 09:15:53 -05:00
# table names will be named like "basecamp_projects", "basecamp_people", etc. This is a convenient way of creating a namespace
2004-11-23 20:04:44 -05:00
# for tables in a shared database. By default, the prefix is the empty string.
cattr_accessor :table_name_prefix
@@table_name_prefix = " "
# 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.
cattr_accessor :table_name_suffix
@@table_name_suffix = " "
2005-10-10 23:55:49 -04:00
# Indicates whether or not table names should be the pluralized versions of the corresponding class names.
# If true, the default table name for a +Product+ class will be +products+. If false, it would just be +product+.
2004-11-23 20:04:44 -05:00
# See table_name for the full rules on table/class naming. This is true, by default.
cattr_accessor :pluralize_table_names
@@pluralize_table_names = true
2005-03-06 12:18:34 -05:00
# Determines whether or not to use ANSI codes to colorize the logging statements committed by the connection adapter. These colors
2005-10-10 23:55:49 -04:00
# make it much easier to overview things during debugging (when used through a reader like +tail+ and on a black background), but
2005-03-06 12:18:34 -05:00
# may complicate matters if you use software like syslog. This is true, by default.
cattr_accessor :colorize_logging
@@colorize_logging = true
2004-12-28 12:30:17 -05:00
# 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.
cattr_accessor :default_timezone
@@default_timezone = :local
2005-07-10 00:54:34 -04:00
# Determines whether or not to use a connection for each thread, or a single shared connection for all threads.
# Defaults to true; Railties' WEBrick server sets this to false.
2005-10-12 18:37:28 -04:00
cattr_accessor :allow_concurrency
@@allow_concurrency = true
2005-10-06 20:53:05 -04:00
# Determines whether to speed up access by generating optimized reader
# methods to avoid expensive calls to method_missing when accessing
# attributes by name. You might want to set this to false in development
# mode, because the methods would be regenerated on each request.
cattr_accessor :generate_read_methods
@@generate_read_methods = true
2005-10-13 00:12:32 -04:00
# Specifies the format to use when dumping the database schema with Rails'
# Rakefile. If :sql, the schema is dumped as (potentially database-
# specific) SQL statements. If :ruby, the schema is dumped as an
# 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.
cattr_accessor :schema_format
@@schema_format = :sql
2005-10-06 20:53:05 -04:00
2004-11-23 20:04:44 -05:00
class << self # Class methods
2005-10-10 23:55:49 -04:00
# Find operates with three different retrieval approaches:
2005-04-18 01:03:56 -04:00
#
# * Find by id: This can either be a specific id (1), a list of ids (1, 5, 6), or an array of ids ([5, 6, 10]).
# If no record can be found for all of the listed ids, then RecordNotFound will be raised.
# * Find first: This will return the first record matched by the options used. These options can either be specific
# conditions or merely an order. If no record can matched, nil is returned.
2005-04-18 16:10:11 -04:00
# * Find all: This will return all the records matched by the options used. If no records are found, an empty array is returned.
2005-04-18 01:03:56 -04:00
#
2005-10-10 23:55:49 -04:00
# All approaches accept an option hash as their last parameter. The options are:
2005-04-18 01:03:56 -04:00
#
2005-04-18 10:39:36 -04:00
# * <tt>:conditions</tt>: An SQL fragment like "administrator = 1" or [ "user_name = ?", username ]. See conditions in the intro.
# * <tt>:order</tt>: An SQL fragment like "created_at DESC, name".
2005-11-10 11:36:01 -05:00
# * <tt>:group</tt>: An attribute name by which the result should be grouped. Uses the GROUP BY SQL-clause.
2005-04-18 10:39:36 -04:00
# * <tt>:limit</tt>: An integer determining the limit on the number of rows that should be returned.
# * <tt>:offset</tt>: An integer determining the offset from where the rows should be fetched. So at 5, it would skip the first 4 rows.
# * <tt>:joins</tt>: An SQL fragment for additional joins like "LEFT JOIN comments ON comments.post_id = id". (Rarely needed).
2005-10-14 20:19:56 -04:00
# The records will be returned read-only since they will have attributes that do not correspond to the table's columns.
2005-10-20 10:53:04 -04:00
# Pass :readonly => false to override.
2005-04-18 10:39:36 -04:00
# * <tt>:include</tt>: Names associations that should be loaded alongside using LEFT OUTER JOINs. The symbols named refer
# to already defined associations. See eager loading under Associations.
2005-07-14 03:18:26 -04:00
# * <tt>:select</tt>: By default, this is * as in SELECT * FROM, but can be changed if you for example want to do a join, but not
# include the joined columns.
2005-10-14 20:19:56 -04:00
# * <tt>:readonly</tt>: Mark the returned records read-only so they cannot be saved or updated.
2005-04-18 01:03:56 -04:00
#
2005-04-18 10:39:36 -04:00
# Examples for find by id:
2004-11-23 20:04:44 -05:00
# Person.find(1) # returns the object for ID = 1
# Person.find(1, 2, 6) # returns an array for objects with IDs in (1, 2, 6)
# Person.find([7, 17]) # returns an array for objects with IDs in (7, 17)
2004-12-15 06:55:43 -05:00
# Person.find([1]) # returns an array for objects the object with ID = 1
2005-04-18 10:39:36 -04:00
# Person.find(1, :conditions => "administrator = 1", :order => "created_on DESC")
#
# Examples for find first:
2005-04-18 16:10:11 -04:00
# Person.find(:first) # returns the first object fetched by SELECT * FROM people
2005-04-18 10:39:36 -04:00
# Person.find(:first, :conditions => [ "user_name = ?", user_name])
# Person.find(:first, :order => "created_on DESC", :offset => 5)
#
# Examples for find all:
2005-04-18 16:10:11 -04:00
# Person.find(:all) # returns an array of objects for all the rows fetched by SELECT * FROM people
2005-04-18 10:39:36 -04:00
# Person.find(:all, :conditions => [ "category IN (?)", categories], :limit => 50)
# Person.find(:all, :offset => 10, :limit => 10)
# Person.find(:all, :include => [ :account, :friends ])
2005-11-10 11:36:01 -05:00
# Person.find(:all, :group => "category")
2005-01-01 14:50:23 -05:00
def find ( * args )
options = extract_options_from_args! ( args )
2004-11-23 20:04:44 -05:00
2005-11-06 05:18:51 -05:00
# Inherit :readonly from finder scope if set. Otherwise,
2005-10-28 00:53:22 -04:00
# if :joins is not blank then :readonly defaults to true.
unless options . has_key? ( :readonly )
2005-11-06 05:18:51 -05:00
if scoped? ( :find , :readonly )
options [ :readonly ] = scope ( :find , :readonly )
2005-10-28 00:53:22 -04:00
elsif ! options [ :joins ] . blank?
options [ :readonly ] = true
end
2005-10-20 10:53:04 -04:00
end
2005-10-14 20:19:56 -04:00
2005-04-03 06:52:05 -04:00
case args . first
when :first
2005-07-11 02:09:08 -04:00
find ( :all , options . merge ( options [ :include ] ? { } : { :limit = > 1 } ) ) . first
2005-04-03 06:52:05 -04:00
when :all
2005-10-14 20:19:56 -04:00
records = options [ :include ] ? find_with_associations ( options ) : find_by_sql ( construct_finder_sql ( options ) )
records . each { | record | record . readonly! } if options [ :readonly ]
records
2004-11-23 20:04:44 -05:00
else
2005-06-21 12:36:18 -04:00
return args . first if args . first . kind_of? ( Array ) && args . first . empty?
2005-04-03 06:52:05 -04:00
expects_array = args . first . kind_of? ( Array )
2005-07-22 16:05:42 -04:00
2005-09-24 19:58:13 -04:00
conditions = " AND ( #{ sanitize_sql ( options [ :conditions ] ) } ) " if options [ :conditions ]
2005-04-03 06:52:05 -04:00
ids = args . flatten . compact . uniq
case ids . size
when 0
raise RecordNotFound , " Couldn't find #{ name } without an ID #{ conditions } "
when 1
2005-04-04 00:56:23 -04:00
if result = find ( :first , options . merge ( { :conditions = > " #{ table_name } . #{ primary_key } = #{ sanitize ( ids . first ) } #{ conditions } " } ) )
2005-04-03 06:52:05 -04:00
return expects_array ? [ result ] : result
else
raise RecordNotFound , " Couldn't find #{ name } with ID= #{ ids . first } #{ conditions } "
end
else
# Find multiple ids
ids_list = ids . map { | id | sanitize ( id ) } . join ( ',' )
2005-04-30 10:04:01 -04:00
result = find ( :all , options . merge ( { :conditions = > " #{ table_name } . #{ primary_key } IN ( #{ ids_list } ) #{ conditions } " } ) )
2005-04-03 06:52:05 -04:00
if result . size == ids . size
return result
else
raise RecordNotFound , " Couldn't find all #{ name . pluralize } with IDs ( #{ ids_list } ) #{ conditions } "
end
2005-01-01 14:50:23 -05:00
end
2004-11-23 20:04:44 -05:00
end
end
2005-06-26 07:25:32 -04:00
# Works like find(:all), but requires a complete SQL string. Examples:
2004-11-23 20:04:44 -05:00
# Post.find_by_sql "SELECT p.*, c.author FROM posts p, comments c WHERE p.id = c.post_id"
2004-12-07 16:14:20 -05:00
# Post.find_by_sql ["SELECT * FROM posts WHERE author = ? AND created > ?", author_id, start_date]
2004-11-23 20:04:44 -05:00
def find_by_sql ( sql )
2005-07-03 04:32:43 -04:00
connection . select_all ( sanitize_sql ( sql ) , " #{ name } Load " ) . collect! { | record | instantiate ( record ) }
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:43 -04:00
2005-04-03 06:52:05 -04:00
# Returns true if the given +id+ represents the primary key of a record in the database, false otherwise.
# Example:
# Person.exists?(5)
def exists? ( id )
2005-04-17 13:16:24 -04:00
! find ( :first , :conditions = > [ " #{ primary_key } = ? " , id ] ) . nil? rescue false
2004-11-23 20:04:44 -05:00
end
2005-04-03 06:52:05 -04:00
2004-11-23 20:04:44 -05:00
# Creates an object, instantly saves it as a record (if the validation permits it), and returns it. If the save
2005-10-10 23:55:49 -04:00
# fails under validations, the unsaved object is still returned.
2004-11-23 20:04:44 -05:00
def create ( attributes = nil )
2005-01-25 07:45:01 -05:00
if attributes . is_a? ( Array )
attributes . collect { | attr | create ( attr ) }
else
object = new ( attributes )
2006-02-04 01:11:42 -05:00
scope ( :create ) . each { | att , value | object . send ( " #{ att } = " , value ) } if scoped? ( :create )
2005-01-25 07:45:01 -05:00
object . save
object
end
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:43 -04:00
# Finds the record from the passed +id+, instantly saves it with the passed +attributes+ (if the validation permits it),
2005-10-10 23:55:49 -04:00
# and returns it. If the save fails under validations, the unsaved object is still returned.
2004-11-23 20:04:44 -05:00
def update ( id , attributes )
2005-01-25 07:45:01 -05:00
if id . is_a? ( Array )
idx = - 1
id . collect { | id | idx += 1 ; update ( id , attributes [ idx ] ) }
else
object = find ( id )
object . update_attributes ( attributes )
object
end
2004-11-23 20:04:44 -05:00
end
2005-01-25 07:45:01 -05:00
# Deletes the record with the given +id+ without instantiating an object first. If an array of ids is provided, all of them
# are deleted.
2004-12-17 16:36:13 -05:00
def delete ( id )
2005-01-25 07:45:01 -05:00
delete_all ( [ " #{ primary_key } IN (?) " , id ] )
2004-12-17 16:36:13 -05:00
end
2005-07-03 04:32:43 -04:00
2004-12-17 16:36:13 -05:00
# Destroys the record with the given +id+ by instantiating the object and calling #destroy (all the callbacks are the triggered).
2005-01-25 07:45:01 -05:00
# If an array of ids is provided, all of them are destroyed.
2004-12-17 16:36:13 -05:00
def destroy ( id )
2005-01-25 07:45:01 -05:00
id . is_a? ( Array ) ? id . each { | id | destroy ( id ) } : find ( id ) . destroy
2004-12-17 16:36:13 -05:00
end
2005-10-10 23:55:49 -04:00
# Updates all records with the SET-part of an SQL update statement in +updates+ and returns an integer with the number of rows updated.
2004-12-19 11:21:55 -05:00
# A subset of the records can be selected by specifying +conditions+. Example:
2004-11-23 20:04:44 -05:00
# Billing.update_all "category = 'authorized', approved = 1", "author = 'David'"
def update_all ( updates , conditions = nil )
2005-01-24 08:06:12 -05:00
sql = " UPDATE #{ table_name } SET #{ sanitize_sql ( updates ) } "
2004-11-23 20:04:44 -05:00
add_conditions! ( sql , conditions )
2005-07-03 04:32:43 -04:00
connection . update ( sql , " #{ name } Update " )
2004-11-23 20:04:44 -05:00
end
2005-01-24 09:13:10 -05:00
2005-10-10 23:55:49 -04:00
# Destroys the objects for all the records that match the +condition+ by instantiating each object and calling
2004-11-23 20:04:44 -05:00
# the destroy method. Example:
# Person.destroy_all "last_login < '2004-04-04'"
def destroy_all ( conditions = nil )
2005-06-26 07:25:32 -04:00
find ( :all , :conditions = > conditions ) . each { | object | object . destroy }
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:43 -04:00
2005-10-10 23:55:49 -04:00
# Deletes all the records that match the +condition+ without instantiating the objects first (and hence not
2004-11-23 20:04:44 -05:00
# calling the destroy method). Example:
2005-11-14 02:41:17 -05:00
# Post.delete_all "person_id = 5 AND (category = 'Something' OR category = 'Else')"
2004-11-23 20:04:44 -05:00
def delete_all ( conditions = nil )
sql = " DELETE FROM #{ table_name } "
add_conditions! ( sql , conditions )
connection . delete ( sql , " #{ name } Delete all " )
end
2005-07-03 04:32:43 -04:00
2006-02-09 04:17:40 -05:00
# Count operates using three different approaches.
#
# * Count all: By not passing any parameters to count, it will return a count of all the rows for the model.
# * Count by conditions or joins: For backwards compatibility, you can pass in +conditions+ and +joins+ as individual parameters.
# * Count using options will find the row count matched by the options used.
#
# The last approach, count using options, accepts an option hash as the only parameter. The options are:
#
# * <tt>:conditions</tt>: An SQL fragment like "administrator = 1" or [ "user_name = ?", username ]. See conditions in the intro.
# * <tt>:joins</tt>: An SQL fragment for additional joins like "LEFT JOIN comments ON comments.post_id = id". (Rarely needed).
# The records will be returned read-only since they will have attributes that do not correspond to the table's columns.
# * <tt>:include</tt>: Named associations that should be loaded alongside using LEFT OUTER JOINs. The symbols named refer
# to already defined associations. When using named associations count returns the number DISTINCT items for the model you're counting.
# See eager loading under Associations.
#
# Examples for counting all:
# Person.count # returns the total count of all people
#
# Examples for count by +conditions+ and +joins+ (for backwards compatibility):
# Person.count("age > 26") # returns the number of people older than 26
# Person.find("age > 26 AND job.salary > 60000", "LEFT JOIN jobs on jobs.person_id = person.id") # returns the total number of rows matching the conditions and joins fetched by SELECT COUNT(*).
#
# Examples for count with options:
# Person.count(:conditions => "age > 26")
# Person.count(:conditions => "age > 26 AND job.salary > 60000", :include => :job) # because of the named association, it finds the DISTINCT count using LEFT OUTER JOIN.
# Person.count(:conditions => "age > 26 AND job.salary > 60000", :joins => "LEFT JOIN jobs on jobs.person_id = person.id") # finds the number of rows matching the conditions and joins.
def count ( * args )
options = { }
#For backwards compatibility, we need to handle both count(conditions=nil, joins=nil) or count(options={}).
if args . size > = 0 and args . size < = 2
if args . first . is_a? ( Hash )
options = args . first
#should we verify the options hash???
else
#Handle legacy paramter options: def count(conditions=nil, joins=nil)
options . merge! ( :conditions = > args [ 0 ] ) if args . length > 0
options . merge! ( :joins = > args [ 1 ] ) if args . length > 1
end
else
raise ( ArgumentError , " Unexpected parameters passed to count(*args): expected either count(conditions=nil, joins=nil) or count(options={}) " )
end
options [ :include ] ? count_with_associations ( options ) : count_by_sql ( construct_counter_sql ( options ) )
end
def construct_counter_sql ( options )
sql = " SELECT COUNT( "
sql << " DISTINCT " if options [ :distinct ]
2006-02-19 22:15:22 -05:00
sql << " #{ options [ :select ] || " #{ table_name } . #{ primary_key } " } ) FROM #{ table_name } "
2006-02-09 04:17:40 -05:00
sql << " #{ options [ :joins ] } " if options [ :joins ]
add_conditions! ( sql , options [ :conditions ] )
sql
2004-11-23 20:04:44 -05:00
end
# Returns the result of an SQL statement that should only include a COUNT(*) in the SELECT part.
2005-08-14 04:20:51 -04:00
# Product.count_by_sql "SELECT COUNT(*) FROM sales s, customers c WHERE s.customer_id = c.id"
2004-11-23 20:04:44 -05:00
def count_by_sql ( sql )
2004-12-07 16:14:20 -05:00
sql = sanitize_conditions ( sql )
2005-09-24 15:50:57 -04:00
connection . select_value ( sql , " #{ name } Count " ) . to_i
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:43 -04:00
# Increments the specified counter by one. So <tt>DiscussionBoard.increment_counter("post_count",
2004-11-23 20:04:44 -05:00
# discussion_board_id)</tt> would increment the "post_count" counter on the board responding to discussion_board_id.
2005-10-10 23:55:49 -04:00
# This is used for caching aggregate values, so that they don't need to be computed every time. Especially important
2004-11-23 20:04:44 -05:00
# for looping over a collection where each element require a number of aggregate values. Like the DiscussionBoard
# that needs to list both the number of posts and comments.
def increment_counter ( counter_name , id )
2004-12-07 09:48:53 -05:00
update_all " #{ counter_name } = #{ counter_name } + 1 " , " #{ primary_key } = #{ quote ( id ) } "
2004-11-23 20:04:44 -05:00
end
# Works like increment_counter, but decrements instead.
def decrement_counter ( counter_name , id )
2004-12-07 09:48:53 -05:00
update_all " #{ counter_name } = #{ counter_name } - 1 " , " #{ primary_key } = #{ quote ( id ) } "
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:43 -04:00
# Attributes named in this macro are protected from mass-assignment, such as <tt>new(attributes)</tt> and
2004-11-23 20:04:44 -05:00
# <tt>attributes=(attributes)</tt>. Their assignment will simply be ignored. Instead, you can use the direct writer
2005-10-10 23:55:49 -04:00
# methods to do assignment. This is meant to protect sensitive attributes from being overwritten by URL/form hackers. Example:
2004-11-23 20:04:44 -05:00
#
# 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"
def attr_protected ( * attributes )
2005-10-12 15:59:13 -04:00
write_inheritable_array ( " attr_protected " , attributes - ( protected_attributes || [ ] ) )
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:43 -04:00
2005-02-07 09:15:53 -05:00
# Returns an array of all the attributes that have been protected from mass-assignment.
2004-11-23 20:04:44 -05:00
def protected_attributes # :nodoc:
read_inheritable_attribute ( " attr_protected " )
end
2005-10-10 23:55:49 -04:00
# If this macro is used, only those attributes named in it will be accessible for mass-assignment, such as
2004-11-23 20:04:44 -05:00
# <tt>new(attributes)</tt> and <tt>attributes=(attributes)</tt>. This is the more conservative choice for mass-assignment
# protection. If you'd rather start from an all-open default and restrict attributes as needed, have a look at
# attr_protected.
def attr_accessible ( * attributes )
2005-10-12 15:59:13 -04:00
write_inheritable_array ( " attr_accessible " , attributes - ( accessible_attributes || [ ] ) )
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:43 -04:00
2005-02-07 09:15:53 -05:00
# Returns an array of all the attributes that have been made accessible to mass-assignment.
2004-11-23 20:04:44 -05:00
def accessible_attributes # :nodoc:
read_inheritable_attribute ( " attr_accessible " )
end
# Specifies that the attribute by the name of +attr_name+ should be serialized before saving to the database and unserialized
# after loading from the database. The serialization is done through YAML. If +class_name+ is specified, the serialized
2005-02-07 09:15:53 -05:00
# object must be of that class on retrieval or +SerializationTypeMismatch+ will be raised.
2004-11-23 20:04:44 -05:00
def serialize ( attr_name , class_name = Object )
2005-07-03 04:32:43 -04:00
serialized_attributes [ attr_name . to_s ] = class_name
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:43 -04:00
2004-11-23 20:04:44 -05:00
# Returns a hash of all the attributes that have been specified for serialization as keys and their class restriction as values.
def serialized_attributes
2005-07-03 04:32:43 -04:00
read_inheritable_attribute ( " attr_serialized " ) or write_inheritable_attribute ( " attr_serialized " , { } )
2004-11-23 20:04:44 -05:00
end
# Guesses the table name (in forced lower-case) based on the name of the class in the inheritance hierarchy descending
# directly from ActiveRecord. So if the hierarchy looks like: Reply < Message < ActiveRecord, then Message is used
2005-02-23 18:51:34 -05:00
# to guess the table name from even when called on Reply. The rules used to do the guess are handled by the Inflector class
# in Active Support, which knows almost all common English inflections (report a bug if your inflection isn't covered).
2004-11-23 20:04:44 -05:00
#
# Additionally, the class-level table_name_prefix is prepended to the table_name and the table_name_suffix is appended.
# So if you have "myapp_" as a prefix, the table name guess for an Account class becomes "myapp_accounts".
#
# 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:
#
# class Mouse < ActiveRecord::Base
2005-03-13 19:23:34 -05:00
# set_table_name "mice"
2004-11-23 20:04:44 -05:00
# end
2004-12-22 08:00:15 -05:00
def table_name
2005-10-10 14:59:56 -04:00
reset_table_name
end
def reset_table_name
name = " #{ table_name_prefix } #{ undecorated_table_name ( class_name_of_active_record_descendant ( self ) ) } #{ table_name_suffix } "
set_table_name name
name
2004-11-23 20:04:44 -05:00
end
2005-02-07 09:15:53 -05:00
# Defines the primary key field -- can be overridden in subclasses. Overwriting will negate any effect of the
2004-11-23 20:04:44 -05:00
# primary_key_prefix_type setting, though.
def primary_key
2005-10-06 19:19:55 -04:00
reset_primary_key
end
def reset_primary_key
key = 'id'
2004-11-23 20:04:44 -05:00
case primary_key_prefix_type
when :table_name
2005-10-06 19:19:55 -04:00
key = Inflector . foreign_key ( class_name_of_active_record_descendant ( self ) , false )
2004-11-23 20:04:44 -05:00
when :table_name_with_underscore
2005-10-06 19:19:55 -04:00
key = Inflector . foreign_key ( class_name_of_active_record_descendant ( self ) )
2004-11-23 20:04:44 -05:00
end
2005-10-06 19:19:55 -04:00
set_primary_key ( key )
key
2004-11-23 20:04:44 -05:00
end
# Defines the column name for use with single table inheritance -- can be overridden in subclasses.
def inheritance_column
" type "
end
2005-11-12 06:59:54 -05:00
# Lazy-set the sequence name to the connection's default. This method
# is only ever called once since set_sequence_name overrides it.
2005-07-24 10:01:35 -04:00
def sequence_name
2005-11-12 06:59:54 -05:00
reset_sequence_name
end
def reset_sequence_name
default = connection . default_sequence_name ( table_name , primary_key )
set_sequence_name ( default )
default
2005-07-24 10:01:35 -04:00
end
2005-02-07 09:26:57 -05:00
# Sets the table name to use to the given value, or (if the value
2005-02-23 18:51:34 -05:00
# is nil or false) to the value returned by the given block.
2005-02-07 09:26:57 -05:00
#
# Example:
#
# class Project < ActiveRecord::Base
# set_table_name "project"
# end
def set_table_name ( value = nil , & block )
define_attr_method :table_name , value , & block
end
alias :table_name = :set_table_name
# Sets the name of the primary key column to use to the given value,
# or (if the value is nil or false) to the value returned by the given
2005-02-23 18:51:34 -05:00
# block.
2005-02-07 09:26:57 -05:00
#
# Example:
#
# class Project < ActiveRecord::Base
# set_primary_key "sysid"
# end
def set_primary_key ( value = nil , & block )
define_attr_method :primary_key , value , & block
end
alias :primary_key = :set_primary_key
# 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
2005-02-23 18:51:34 -05:00
# given block.
2005-02-07 09:26:57 -05:00
#
# Example:
#
# class Project < ActiveRecord::Base
# set_inheritance_column do
# original_inheritance_column + "_id"
# end
# end
def set_inheritance_column ( value = nil , & block )
define_attr_method :inheritance_column , value , & block
end
alias :inheritance_column = :set_inheritance_column
2005-07-24 10:01:35 -04:00
# 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
2005-10-15 23:45:39 -04:00
# given block. This is required for Oracle and is useful for any
# database which relies on sequences for primary key generation.
2005-07-24 10:01:35 -04:00
#
2005-11-16 03:16:54 -05:00
# 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.
2005-07-24 10:01:35 -04:00
#
# Example:
#
# class Project < ActiveRecord::Base
# set_sequence_name "projectseq" # default would have been "project_seq"
# end
def set_sequence_name ( value = nil , & block )
define_attr_method :sequence_name , value , & block
end
alias :sequence_name = :set_sequence_name
2004-11-23 20:04:44 -05:00
# 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
2005-07-03 04:32:39 -04:00
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
2004-11-23 20:04:44 -05:00
end
2005-10-28 03:40:28 -04:00
# Indicates whether the table associated with this class exists
def table_exists?
if connection . respond_to? ( :tables )
connection . tables . include? table_name
else
# if the connection adapter hasn't implemented tables, there are two crude tests that can be
# used - see if getting column info raises an error, or if the number of columns returned is zero
begin
reset_column_information
columns . size > 0
rescue ActiveRecord :: StatementInvalid
false
end
end
end
2004-11-23 20:04:44 -05:00
# Returns an array of column objects for the table associated with this class.
def columns
2005-10-06 19:19:55 -04:00
unless @columns
@columns = connection . columns ( table_name , " #{ name } Columns " )
@columns . each { | column | column . primary = column . name == primary_key }
end
@columns
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:43 -04:00
2004-11-23 20:04:44 -05:00
# Returns an array of column objects for the table associated with this class.
def columns_hash
@columns_hash || = columns . inject ( { } ) { | hash , column | hash [ column . name ] = column ; hash }
end
2005-06-12 02:56:51 -04:00
2005-04-18 14:49:34 -04:00
def column_names
2005-06-12 02:56:51 -04:00
@column_names || = columns . map { | column | column . name }
2005-04-18 14:49:34 -04:00
end
2004-11-23 20:04:44 -05:00
2005-10-10 23:55:49 -04:00
# 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.
2004-11-23 20:04:44 -05:00
def content_columns
2005-10-06 19:19:55 -04:00
@content_columns || = columns . reject { | c | c . primary || c . name =~ / (_id|_count)$ / || c . name == inheritance_column }
2004-11-23 20:04:44 -05:00
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
2005-07-03 04:32:43 -04:00
# is available.
2004-11-23 20:04:44 -05:00
def column_methods_hash
2005-06-12 02:56:51 -04:00
@dynamic_methods_hash || = column_names . inject ( Hash . new ( false ) ) do | methods , attr |
2005-10-22 12:43:39 -04:00
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
2004-11-23 20:04:44 -05:00
methods
end
end
2005-07-03 04:32:43 -04:00
2005-10-06 20:53:05 -04:00
# Contains the names of the generated reader methods.
def read_methods
2005-11-02 12:30:59 -05:00
@read_methods || = Set . new
2005-10-06 20:53:05 -04:00
end
2005-11-02 12:30:59 -05:00
2005-10-10 23:55:49 -04:00
# Resets all the cached information about columns, which will cause them to be reloaded on the next request.
2004-12-12 12:49:38 -05:00
def reset_column_information
2005-11-02 12:30:59 -05:00
read_methods . each { | name | undef_method ( name ) }
2005-10-06 20:53:05 -04:00
@column_names = @columns = @columns_hash = @content_columns = @dynamic_methods_hash = @read_methods = nil
2004-12-12 12:49:38 -05:00
end
2005-02-23 18:51:34 -05:00
def reset_column_information_and_inheritable_attributes_for_all_subclasses #:nodoc:
2004-12-12 12:49:38 -05:00
subclasses . each { | klass | klass . reset_inheritable_attributes ; klass . reset_column_information }
end
2004-11-23 20:04:44 -05:00
# Transforms attribute key names into a more humane format, such as "First name" instead of "first_name". Example:
# Person.human_attribute_name("first_name") # => "First name"
2005-02-23 18:51:34 -05:00
# Deprecated in favor of just calling "first_name".humanize
def human_attribute_name ( attribute_key_name ) #:nodoc:
2005-01-17 14:58:02 -05:00
attribute_key_name . humanize
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:43 -04:00
2004-11-23 20:04:44 -05:00
def descends_from_active_record? # :nodoc:
2005-07-03 04:32:39 -04:00
superclass == Base || ! columns_hash . include? ( inheritance_column )
2004-11-23 20:04:44 -05:00
end
2005-02-23 18:51:34 -05:00
def quote ( object ) #:nodoc:
2004-12-07 09:48:53 -05:00
connection . quote ( object )
end
# Used to sanitize objects before they're used in an SELECT SQL-statement. Delegates to <tt>connection.quote</tt>.
2005-02-23 18:51:34 -05:00
def sanitize ( object ) #:nodoc:
2004-12-07 09:48:53 -05:00
connection . quote ( object )
2004-11-23 20:04:44 -05:00
end
2005-09-06 13:47:31 -04:00
# Log and benchmark multiple statements in a single block. Example:
2004-11-23 20:04:44 -05:00
#
# Project.benchmark("Creating project") do
# project = Project.create("name" => "stuff")
# project.create_manager("name" => "David")
2005-06-26 07:25:32 -04:00
# project.milestones << Milestone.find(:all)
2004-11-23 20:04:44 -05:00
# end
2005-09-06 12:48:18 -04:00
#
2005-09-06 13:47:31 -04:00
# The benchmark is only recorded if the current level of the logger matches the <tt>log_level</tt>, which makes it
# easy to include benchmarking statements in production software that will remain inexpensive because the benchmark
# will only be conducted if the log level is low enough.
#
2005-10-10 23:55:49 -04:00
# The logging of the multiple statements is turned off unless <tt>use_silence</tt> is set to false.
2005-09-06 13:47:31 -04:00
def benchmark ( title , log_level = Logger :: DEBUG , use_silence = true )
if logger && logger . level == log_level
2005-09-06 12:48:18 -04:00
result = nil
seconds = Benchmark . realtime { result = use_silence ? silence { yield } : yield }
2005-09-06 13:47:31 -04:00
logger . add ( log_level , " #{ title } ( #{ '%.5f' % seconds } ) " )
2005-09-06 12:48:18 -04:00
result
else
yield
end
2005-02-16 20:23:41 -05:00
end
2005-07-03 04:32:39 -04:00
2005-02-16 20:23:41 -05:00
# Silences the logger for the duration of the block.
def silence
2005-07-03 04:32:39 -04:00
old_logger_level , logger . level = logger . level , Logger :: ERROR if logger
yield
ensure
2005-04-13 01:16:52 -04:00
logger . level = old_logger_level if logger
2004-11-23 20:04:44 -05:00
end
2005-11-06 05:18:51 -05:00
# Scope parameters to method calls within the block. Takes a hash of method_name => parameters hash.
# method_name may be :find or :create.
# :find parameters may include the <tt>:conditions</tt>, <tt>:joins</tt>,
# <tt>:offset</tt>, <tt>:limit</tt>, and <tt>:readonly</tt> options.
# :create parameters are an attributes hash.
2005-07-22 16:05:42 -04:00
#
2005-11-06 05:18:51 -05:00
# Article.with_scope(:find => { :conditions => "blog_id = 1" }, :create => { :blog_id => 1 }) do
2005-07-22 16:05:42 -04:00
# Article.find(1) # => SELECT * from articles WHERE blog_id = 1 AND id = 1
2005-11-06 05:18:51 -05:00
# a = Article.create(1)
# a.blog_id == 1
2005-07-22 16:05:42 -04:00
# end
2005-11-06 05:18:51 -05:00
def with_scope ( method_scoping = { } )
# Dup first and second level of hash (method and params).
method_scoping = method_scoping . inject ( { } ) do | hash , ( method , params ) |
hash [ method ] = params . dup
hash
end
2005-12-21 01:16:42 -05:00
method_scoping . assert_valid_keys ( [ :find , :create ] )
2005-11-06 05:18:51 -05:00
if f = method_scoping [ :find ]
2005-12-21 01:16:42 -05:00
f . assert_valid_keys ( [ :conditions , :joins , :from , :offset , :limit , :readonly ] )
2005-11-06 05:18:51 -05:00
f [ :readonly ] = true if ! f [ :joins ] . blank? && ! f . has_key? ( :readonly )
end
2005-11-04 14:39:50 -05:00
2005-11-06 05:18:51 -05:00
raise ArgumentError , " Nested scopes are not yet supported: #{ scoped_methods . inspect } " unless scoped_methods . nil?
2005-11-04 14:39:50 -05:00
2005-11-06 05:18:51 -05:00
self . scoped_methods = method_scoping
yield
2005-10-28 00:53:22 -04:00
ensure
2005-11-06 05:18:51 -05:00
self . scoped_methods = nil
2005-07-22 16:05:42 -04:00
end
2004-11-23 20:04:44 -05:00
2005-01-23 10:19:33 -05:00
# Overwrite the default class equality method to provide support for association proxies.
def === ( object )
object . is_a? ( self )
2005-07-22 16:05:42 -04:00
end
2005-10-12 18:37:28 -04:00
# Deprecated
def threaded_connections
allow_concurrency
end
# Deprecated
def threaded_connections = ( value )
self . allow_concurrency = value
end
2006-01-20 15:43:40 -05:00
# 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
2005-07-22 16:05:42 -04:00
2004-11-23 20:04:44 -05:00
private
# Finder methods must instantiate through this method to work with the single-table inheritance model
# that makes it possible to create objects of different types from the same table.
def instantiate ( record )
2005-10-09 18:26:54 -04:00
object =
if subclass_name = record [ inheritance_column ]
if subclass_name . empty?
allocate
else
require_association_class ( subclass_name )
begin
compute_type ( subclass_name ) . allocate
rescue NameError
raise SubclassNotFound ,
" The single-table inheritance mechanism failed to locate the subclass: ' #{ record [ inheritance_column ] } '. " +
" 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 #{ self . to_s } .inheritance_column to use another column for that information. "
end
end
else
allocate
2005-07-03 04:32:39 -04:00
end
2004-12-14 07:32:29 -05:00
2004-11-23 20:04:44 -05:00
object . instance_variable_set ( " @attributes " , record )
2005-07-03 04:32:39 -04:00
object
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:39 -04:00
2004-11-23 20:04:44 -05:00
# Returns the name of the type of the record using the current module as a prefix. So descendents of
2005-10-10 23:55:49 -04:00
# MyApp::Business::Account would appear as "MyApp::Business::AccountSubclass".
2004-11-23 20:04:44 -05:00
def type_name_with_module ( type_name )
self . name =~ / :: / ? self . name . scan ( / (.*):: / ) . first . first + " :: " + type_name : type_name
end
2005-04-03 06:52:05 -04:00
def construct_finder_sql ( options )
2005-12-02 23:29:55 -05:00
sql = " SELECT #{ options [ :select ] || '*' } "
2005-12-20 16:20:35 -05:00
sql << " FROM #{ scope ( :find , :from ) || options [ :from ] || table_name } "
2005-12-02 23:29:55 -05:00
2005-07-22 16:05:42 -04:00
add_joins! ( sql , options )
2005-04-03 06:52:05 -04:00
add_conditions! ( sql , options [ :conditions ] )
2005-12-02 23:29:55 -05:00
2005-11-10 11:36:01 -05:00
sql << " GROUP BY #{ options [ :group ] } " if options [ :group ]
2005-07-22 16:05:42 -04:00
sql << " ORDER BY #{ options [ :order ] } " if options [ :order ]
2005-12-02 23:29:55 -05:00
2005-04-03 08:06:23 -04:00
add_limit! ( sql , options )
2005-12-02 23:29:55 -05:00
2005-07-03 04:32:43 -04:00
sql
2005-04-03 08:06:23 -04:00
end
2005-04-03 06:52:05 -04:00
2005-04-03 08:06:23 -04:00
def add_limit! ( sql , options )
2005-11-06 05:18:51 -05:00
options [ :limit ] || = scope ( :find , :limit )
options [ :offset ] || = scope ( :find , :offset )
2005-07-01 13:20:04 -04:00
connection . add_limit_offset! ( sql , options )
2005-04-03 06:52:05 -04:00
end
2005-11-06 05:18:51 -05:00
2005-07-22 16:05:42 -04:00
def add_joins! ( sql , options )
2005-11-06 05:18:51 -05:00
join = scope ( :find , :joins ) || options [ :joins ]
2005-07-22 16:05:42 -04:00
sql << " #{ join } " if join
end
2005-11-06 05:18:51 -05:00
2005-10-10 23:55:49 -04:00
# Adds a sanitized version of +conditions+ to the +sql+ string. Note that the passed-in +sql+ string is changed.
2005-07-22 16:05:42 -04:00
def add_conditions! ( sql , conditions )
2006-01-03 22:43:28 -05:00
segments = [ sanitize_sql ( scope ( :find , :conditions ) ) ]
2005-11-06 05:18:51 -05:00
segments << sanitize_sql ( conditions ) unless conditions . nil?
segments << type_condition unless descends_from_active_record?
segments . compact!
sql << " WHERE ( #{ segments . join ( " ) AND ( " ) } ) " unless segments . empty?
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:43 -04:00
2004-11-23 20:04:44 -05:00
def type_condition
2005-11-16 03:16:54 -05:00
quoted_inheritance_column = connection . quote_column_name ( inheritance_column )
type_condition = subclasses . inject ( " #{ table_name } . #{ quoted_inheritance_column } = ' #{ name . demodulize } ' " ) do | condition , subclass |
condition << " OR #{ table_name } . #{ quoted_inheritance_column } = ' #{ subclass . name . demodulize } ' "
2005-04-10 11:49:49 -04:00
end
2005-07-03 04:32:43 -04:00
" ( #{ type_condition } ) "
2004-11-23 20:04:44 -05:00
end
# Guesses the table name, but does not decorate it with prefix and suffix information.
def undecorated_table_name ( class_name = class_name_of_active_record_descendant ( self ) )
table_name = Inflector . underscore ( Inflector . demodulize ( class_name ) )
table_name = Inflector . pluralize ( table_name ) if pluralize_table_names
2005-07-03 04:32:43 -04:00
table_name
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:43 -04:00
# Enables dynamic finders like find_by_user_name(user_name) and find_by_user_name_and_password(user_name, password) that are turned into
# find(:first, :conditions => ["user_name = ?", user_name]) and find(:first, :conditions => ["user_name = ? AND password = ?", user_name, password])
2005-06-26 07:25:32 -04:00
# respectively. Also works for find(:all), but using find_all_by_amount(50) that are turned into find(:all, :conditions => ["amount = ?", 50]).
2005-07-03 04:32:43 -04:00
#
2005-04-17 13:16:24 -04:00
# It's even possible to use all the additional parameters to find. For example, the full interface for find_all_by_amount
# is actually find_all_by_amount(amount, options).
2005-01-02 08:31:00 -05:00
def method_missing ( method_id , * arguments )
2005-11-04 14:39:50 -05:00
if match = / find_(all_by|by)_([_a-zA-Z] \ w*) / . match ( method_id . to_s )
finder = determine_finder ( match )
2005-01-02 08:31:00 -05:00
2005-11-04 14:39:50 -05:00
attribute_names = extract_attribute_names_from_match ( match )
super unless all_attributes_exists? ( attribute_names )
2005-01-24 08:02:50 -05:00
2005-11-04 14:39:50 -05:00
conditions = construct_conditions_from_arguments ( attribute_names , arguments )
2005-07-03 04:32:39 -04:00
2006-01-22 04:34:41 -05:00
if ( extra_options = arguments [ attribute_names . size ] ) . is_a? ( Hash )
finder_options = extra_options . merge ( :conditions = > conditions )
if extra_options [ :conditions ]
with_scope ( :find = > { :conditions = > extra_options [ :conditions ] } ) do
find ( finder , finder_options )
end
else
find ( finder , finder_options )
end
2005-04-17 13:16:24 -04:00
else
2005-11-04 14:39:50 -05:00
send ( " find_ #{ finder } " , conditions , * arguments [ attribute_names . length .. - 1 ] ) # deprecated API
2005-04-17 13:16:24 -04:00
end
2005-11-04 14:39:50 -05:00
elsif match = / find_or_create_by_([_a-zA-Z] \ w*) / . match ( method_id . to_s )
attribute_names = extract_attribute_names_from_match ( match )
super unless all_attributes_exists? ( attribute_names )
find ( :first , :conditions = > construct_conditions_from_arguments ( attribute_names , arguments ) ) ||
create ( construct_attributes_from_arguments ( attribute_names , arguments ) )
2005-01-02 08:31:00 -05:00
else
super
end
end
2004-11-23 20:04:44 -05:00
2005-11-04 14:39:50 -05:00
def determine_finder ( match )
match . captures . first == 'all_by' ? :all : :first
end
def extract_attribute_names_from_match ( match )
match . captures . last . split ( '_and_' )
end
def construct_conditions_from_arguments ( attribute_names , arguments )
conditions = [ ]
2005-11-16 03:16:54 -05:00
attribute_names . each_with_index { | name , idx | conditions << " #{ table_name } . #{ connection . quote_column_name ( name ) } #{ attribute_condition ( arguments [ idx ] ) } " }
2005-11-04 14:39:50 -05:00
[ conditions . join ( " AND " ) , * arguments [ 0 ... attribute_names . length ] ]
end
def construct_attributes_from_arguments ( attribute_names , arguments )
attributes = { }
attribute_names . each_with_index { | name , idx | attributes [ name ] = arguments [ idx ] }
attributes
end
def all_attributes_exists? ( attribute_names )
attribute_names . all? { | name | column_methods_hash . include? ( name . to_sym ) }
end
2005-05-02 01:34:27 -04:00
def attribute_condition ( argument )
case argument
when nil then " IS ? "
when Array then " IN (?) "
else " = ? "
end
end
2005-02-23 18:51:34 -05:00
# Defines an "attribute" method (like #inheritance_column or
# #table_name). A new (class) method will be created with the
# given name. If a value is specified, the new method will
# return that value (as a string). Otherwise, the given block
# will be used to compute the value of the method.
#
# The original method will be aliased, with the new name being
# prefixed with "original_". This allows the new method to
# access the original value.
#
# Example:
#
# class A < ActiveRecord::Base
# define_attr_method :primary_key, "sysid"
# define_attr_method( :inheritance_column ) do
# original_inheritance_column + "_id"
# end
# end
def define_attr_method ( name , value = nil , & block )
sing = class << self ; self ; end
2005-10-14 11:04:57 -04:00
sing . send :alias_method , " original_ #{ name } " , name
2005-11-12 06:59:54 -05:00
if block_given?
sing . send :define_method , name , & block
else
2005-10-14 11:04:57 -04:00
# use eval instead of a block to work around a memory leak in dev
# mode in fcgi
sing . class_eval " def #{ name } ; #{ value . to_s . inspect } ; end "
end
2005-02-23 18:51:34 -05:00
end
2004-11-23 20:04:44 -05:00
protected
def subclasses
@@subclasses [ self ] || = [ ]
@@subclasses [ self ] + extra = @@subclasses [ self ] . inject ( [ ] ) { | list , subclass | list + subclass . subclasses }
end
2005-11-06 05:18:51 -05:00
# Test whether the given method and optional key are scoped.
def scoped? ( method , key = nil )
scoped_methods and scoped_methods . has_key? ( method ) and ( key . nil? or scope ( method ) . has_key? ( key ) )
end
# Retrieve the scope for the given method and optional key.
def scope ( method , key = nil )
if scoped_methods and scope = scoped_methods [ method ]
key ? scope [ key ] : scope
end
end
def scoped_methods
2005-10-12 18:37:28 -04:00
if allow_concurrency
2005-11-06 05:18:51 -05:00
Thread . current [ :scoped_methods ] || = { }
Thread . current [ :scoped_methods ] [ self ] || = nil
2005-10-10 23:40:11 -04:00
else
2005-11-06 05:18:51 -05:00
@scoped_methods || = nil
2005-10-10 23:40:11 -04:00
end
2005-07-22 16:05:42 -04:00
end
2005-10-09 18:08:16 -04:00
2005-11-06 05:18:51 -05:00
def scoped_methods = ( value )
2005-10-12 18:37:28 -04:00
if allow_concurrency
2005-11-06 05:18:51 -05:00
Thread . current [ :scoped_methods ] || = { }
Thread . current [ :scoped_methods ] [ self ] = value
2005-10-10 23:40:11 -04:00
else
2005-11-06 05:18:51 -05:00
@scoped_methods = value
2005-10-10 23:40:11 -04:00
end
2005-07-22 16:05:42 -04:00
end
2005-11-06 05:18:51 -05:00
2005-10-10 23:55:49 -04:00
# Returns the class type of the record using the current module as a prefix. So descendents of
# MyApp::Business::Account would appear as MyApp::Business::AccountSubclass.
2004-11-23 20:04:44 -05:00
def compute_type ( type_name )
2005-07-03 04:32:43 -04:00
type_name_with_module ( type_name ) . split ( " :: " ) . inject ( Object ) do | final_type , part |
2005-07-10 00:22:08 -04:00
final_type . const_get ( part )
2004-11-23 20:04:44 -05:00
end
end
2006-01-20 15:43:40 -05:00
# Returns the class descending directly from ActiveRecord in the inheritance hierarchy.
def class_of_active_record_descendant ( klass )
2004-11-23 20:04:44 -05:00
if klass . superclass == Base
2006-01-20 15:43:40 -05:00
klass
2004-11-23 20:04:44 -05:00
elsif klass . superclass . nil?
raise ActiveRecordError , " #{ name } doesn't belong in a hierarchy descending from ActiveRecord "
else
2006-01-20 15:43:40 -05:00
class_of_active_record_descendant ( klass . superclass )
2004-11-23 20:04:44 -05:00
end
end
2006-01-20 15:43:40 -05:00
# Returns the name of the class descending directly from ActiveRecord in the inheritance hierarchy.
def class_name_of_active_record_descendant ( klass )
class_of_active_record_descendant ( klass ) . name
end
2005-01-01 14:50:23 -05:00
# Accepts an array or string. The string is returned untouched, but the array has each value
# sanitized and interpolated into the sql statement.
# ["name='%s' and group_id='%s'", "foo'bar", 4] returns "name='foo''bar' and group_id='4'"
def sanitize_sql ( ary )
return ary unless ary . is_a? ( Array )
2004-12-07 05:37:50 -05:00
2005-01-01 14:50:23 -05:00
statement , * values = ary
if values . first . is_a? ( Hash ) and statement =~ / : \ w+ /
replace_named_bind_variables ( statement , values . first )
elsif statement . include? ( '?' )
2004-12-08 05:38:10 -05:00
replace_bind_variables ( statement , values )
else
2004-12-07 09:48:53 -05:00
statement % values . collect { | value | connection . quote_string ( value . to_s ) }
2004-12-08 05:38:10 -05:00
end
2004-12-07 05:37:50 -05:00
end
2005-01-01 14:50:23 -05:00
alias_method :sanitize_conditions , :sanitize_sql
2004-12-07 05:37:50 -05:00
def replace_bind_variables ( statement , values )
2005-01-02 07:45:53 -05:00
raise_if_bind_arity_mismatch ( statement , statement . count ( '?' ) , values . size )
2005-01-01 14:50:23 -05:00
bound = values . dup
2005-01-24 06:57:22 -05:00
statement . gsub ( '?' ) { quote_bound_value ( bound . shift ) }
2004-12-08 05:38:10 -05:00
end
2005-01-01 14:50:23 -05:00
def replace_named_bind_variables ( statement , bind_vars )
statement . gsub ( / :( \ w+) / ) do
match = $1 . to_sym
2005-07-03 04:32:39 -04:00
if bind_vars . include? ( match )
2005-01-24 06:57:22 -05:00
quote_bound_value ( bind_vars [ match ] )
2005-01-01 14:50:23 -05:00
else
raise PreparedStatementInvalid , " missing value for : #{ match } in #{ statement } "
2004-12-08 05:38:10 -05:00
end
end
2005-01-02 07:45:53 -05:00
end
2005-01-24 06:57:22 -05:00
def quote_bound_value ( value )
2005-06-16 06:13:37 -04:00
if ( value . respond_to? ( :map ) && ! value . is_a? ( String ) )
value . map { | v | connection . quote ( v ) } . join ( ',' )
else
connection . quote ( value )
2005-01-24 06:57:22 -05:00
end
end
2005-01-02 07:45:53 -05:00
def raise_if_bind_arity_mismatch ( statement , expected , provided )
unless expected == provided
raise PreparedStatementInvalid , " wrong number of bind variables ( #{ provided } for #{ expected } ) in: #{ statement } "
end
2005-01-01 14:50:23 -05:00
end
2004-12-08 05:38:10 -05:00
2005-01-01 14:50:23 -05:00
def extract_options_from_args! ( args )
2005-10-06 18:21:10 -04:00
options = args . last . is_a? ( Hash ) ? args . pop : { }
validate_find_options ( options )
options
end
2005-10-14 20:19:56 -04:00
2005-10-06 18:21:10 -04:00
def validate_find_options ( options )
2005-12-02 23:29:55 -05:00
options . assert_valid_keys [ :conditions , :include , :joins , :limit , :offset , :order , :select , :readonly , :group , :from ]
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:43 -04:00
2004-12-10 11:02:11 -05:00
def encode_quoted_value ( value )
quoted_value = connection . quote ( value )
2005-10-22 12:43:39 -04:00
quoted_value = " ' #{ quoted_value [ 1 .. - 2 ] . gsub ( / \ ' / , " \\ \\ ' " ) } ' " if quoted_value . include? ( " \\ \' " ) # (for ruby mode) "
quoted_value
2004-12-10 11:02:11 -05:00
end
2004-11-23 20:04:44 -05:00
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).
2005-07-03 04:32:43 -04:00
# In both instances, valid attribute keys are determined by the column names of the associated table --
2004-11-23 20:04:44 -05:00
# hence you can't have attributes that aren't part of the table columns.
def initialize ( attributes = nil )
@attributes = attributes_from_column_definition
@new_record = true
ensure_proper_type
self . attributes = attributes unless attributes . nil?
yield self if block_given?
end
2005-07-03 04:32:43 -04:00
2005-11-02 12:30:59 -05:00
# A model instance's primary key is always available as model.id
# whether you name it the default 'id' or set it to something else.
2004-11-23 20:04:44 -05:00
def id
2005-10-06 20:53:05 -04:00
attr_name = self . class . primary_key
2005-11-02 12:30:59 -05:00
column = column_for_attribute ( attr_name )
2005-10-06 20:53:05 -04:00
define_read_method ( :id , attr_name , column ) if self . class . generate_read_methods
2005-11-02 12:30:59 -05:00
read_attribute ( attr_name )
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:43 -04:00
2005-03-20 10:45:36 -05:00
# Enables Active Record objects to be used as URL parameters in Action Pack automatically.
alias_method :to_param , :id
2005-07-03 04:32:43 -04:00
2005-02-23 18:51:34 -05:00
def id_before_type_cast #:nodoc:
2004-12-20 08:32:09 -05:00
read_attribute_before_type_cast ( self . class . primary_key )
end
2005-02-23 18:51:34 -05:00
def quoted_id #:nodoc:
2005-07-03 04:32:39 -04:00
quote ( id , column_for_attribute ( self . class . primary_key ) )
2004-12-07 09:48:53 -05:00
end
2005-07-03 04:32:39 -04:00
2004-11-23 20:04:44 -05:00
# Sets the primary ID.
def id = ( value )
write_attribute ( self . class . primary_key , value )
end
2005-07-03 04:32:43 -04:00
2004-11-23 20:04:44 -05:00
# Returns true if this object hasn't been saved yet -- that is, a record for the object doesn't exist yet.
def new_record?
@new_record
end
2005-07-03 04:32:43 -04:00
2004-11-23 20:04:44 -05:00
# * No record exists: Creates a new record with values matching those of the object attributes.
# * A record does exist: Updates the record with values matching those of the object attributes.
def save
2005-10-14 20:19:56 -04:00
raise ActiveRecord :: ReadOnlyRecord if readonly?
2004-11-23 20:04:44 -05:00
create_or_update
end
2005-07-03 04:32:43 -04:00
2004-11-23 20:04:44 -05:00
# Deletes the record in the database and freezes this instance to reflect that no changes should
# be made (since they can't be persisted).
def destroy
unless new_record?
2005-07-03 04:32:47 -04:00
connection . delete <<-end_sql, "#{self.class.name} Destroy"
DELETE FROM #{self.class.table_name}
WHERE #{self.class.primary_key} = #{quoted_id}
end_sql
2004-11-23 20:04:44 -05:00
end
freeze
end
2005-11-10 13:53:31 -05:00
# Returns a clone of the record that hasn't been assigned an id yet and
# is treated as a new record. Note that this is a "shallow" clone:
# 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.
2004-11-23 20:04:44 -05:00
def clone
2005-05-19 12:39:50 -04:00
attrs = self . attributes_before_type_cast
2005-02-14 19:03:18 -05:00
attrs . delete ( self . class . primary_key )
2005-05-19 12:39:50 -04:00
self . class . new do | record |
record . send :instance_variable_set , '@attributes' , attrs
end
2004-11-23 20:04:44 -05:00
end
2005-05-19 12:39:50 -04:00
2004-11-23 20:04:44 -05:00
# Updates a single attribute and saves the record. This is especially useful for boolean flags on existing records.
2004-12-18 06:17:22 -05:00
# Note: This method is overwritten by the Validation module that'll make sure that updates made with this method
# doesn't get subjected to validation checks. Hence, attributes can be updated even if the full object isn't valid.
2004-11-23 20:04:44 -05:00
def update_attribute ( name , value )
2005-09-26 17:55:37 -04:00
send ( name . to_s + '=' , value )
2005-01-05 21:31:35 -05:00
save
2004-12-17 16:51:14 -05:00
end
2005-10-10 23:55:49 -04:00
# Updates all the attributes from the passed-in Hash and saves the record. If the object is invalid, the saving will
2004-12-18 06:17:22 -05:00
# fail and false will be returned.
2004-12-17 16:51:14 -05:00
def update_attributes ( attributes )
2004-12-17 20:27:57 -05:00
self . attributes = attributes
2005-07-03 04:32:43 -04:00
save
2004-11-23 20:04:44 -05:00
end
2005-01-05 21:31:35 -05:00
# Initializes the +attribute+ to zero if nil and adds one. Only makes sense for number-based attributes. Returns self.
def increment ( attribute )
self [ attribute ] || = 0
self [ attribute ] += 1
self
end
2005-07-03 04:32:43 -04:00
2005-01-05 21:31:35 -05:00
# Increments the +attribute+ and saves the record.
def increment! ( attribute )
increment ( attribute ) . update_attribute ( attribute , self [ attribute ] )
end
# Initializes the +attribute+ to zero if nil and subtracts one. Only makes sense for number-based attributes. Returns self.
def decrement ( attribute )
self [ attribute ] || = 0
self [ attribute ] -= 1
self
end
# Decrements the +attribute+ and saves the record.
def decrement! ( attribute )
decrement ( attribute ) . update_attribute ( attribute , self [ attribute ] )
end
2005-07-03 04:32:43 -04:00
2005-01-05 21:31:35 -05:00
# Turns an +attribute+ that's currently true into false and vice versa. Returns self.
def toggle ( attribute )
2005-10-06 00:15:14 -04:00
self [ attribute ] = ! send ( " #{ attribute } ? " )
2005-01-05 21:31:35 -05:00
self
end
# Toggles the +attribute+ and saves the record.
def toggle! ( attribute )
toggle ( attribute ) . update_attribute ( attribute , self [ attribute ] )
end
2005-01-10 18:49:57 -05:00
# Reloads the attributes of this object from the database.
def reload
2005-12-07 23:46:57 -05:00
clear_aggregation_cache
2005-01-10 18:49:57 -05:00
clear_association_cache
@attributes . update ( self . class . find ( self . id ) . instance_variable_get ( '@attributes' ) )
2005-07-03 04:32:43 -04:00
self
2005-01-10 18:49:57 -05:00
end
2005-10-10 23:55:49 -04:00
# Returns the value of the attribute identified by <tt>attr_name</tt> after it has been typecast (for example,
2004-11-23 20:04:44 -05:00
# "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).
2005-07-03 04:32:43 -04:00
def [] ( attr_name )
2005-09-02 04:31:11 -04:00
read_attribute ( attr_name )
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:43 -04:00
2004-11-23 20:04:44 -05:00
# Updates the attribute identified by <tt>attr_name</tt> with the specified +value+.
# (Alias for the protected write_attribute method).
2005-07-03 04:32:43 -04:00
def []= ( attr_name , value )
2005-09-02 04:31:11 -04:00
write_attribute ( attr_name , value )
2004-11-23 20:04:44 -05:00
end
# Allows you to set all the attributes at once by passing in a hash with keys
# matching the attribute names (which again matches the column names). 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 in with the +attr_accessible+ macro. Then all the
# attributes not included in that won't be allowed to be mass-assigned.
2006-01-14 03:26:20 -05:00
def attributes = ( attrs )
return if attrs . nil?
attributes = attrs . dup
2005-03-10 09:44:01 -05:00
attributes . stringify_keys!
2004-11-23 20:04:44 -05:00
multi_parameter_attributes = [ ]
2005-07-03 04:32:43 -04:00
remove_attributes_protected_from_mass_assignment ( attributes ) . each do | k , v |
2004-11-23 20:04:44 -05:00
k . include? ( " ( " ) ? multi_parameter_attributes << [ k , v ] : send ( k + " = " , v )
end
assign_multiparameter_attributes ( multi_parameter_attributes )
end
2005-01-10 19:45:26 -05:00
# Returns a hash of all the attributes with their names as keys and clones of their objects as values.
def attributes
2005-05-19 12:39:50 -04:00
clone_attributes :read_attribute
end
# Returns a hash of cloned attributes before typecasting and deserialization.
def attributes_before_type_cast
clone_attributes :read_attribute_before_type_cast
2005-01-10 19:45:26 -05:00
end
2004-11-23 20:04:44 -05:00
# Returns true if the specified +attribute+ has been set by the user or by a database load and is neither
2005-10-10 23:55:49 -04:00
# nil nor empty? (the latter only applies to objects that respond to empty?, most notably Strings).
2004-11-23 20:04:44 -05:00
def attribute_present? ( attribute )
2005-07-03 04:33:03 -04:00
value = read_attribute ( attribute )
! value . blank? or value == 0
2004-11-23 20:04:44 -05:00
end
2005-10-22 12:43:39 -04:00
# Returns true if the given attribute is in the attributes hash
def has_attribute? ( attr_name )
@attributes . has_key? ( attr_name . to_s )
end
2004-11-23 20:04:44 -05:00
# Returns an array of names for the attributes available on this object sorted alphabetically.
def attribute_names
@attributes . keys . sort
end
# Returns the column object for the named attribute.
def column_for_attribute ( name )
2005-01-05 21:31:35 -05:00
self . class . columns_hash [ name . to_s ]
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:43 -04:00
2005-01-15 12:45:16 -05:00
# Returns true if the +comparison_object+ is the same object, or is of the same type and has the same id.
2004-11-23 20:04:44 -05:00
def == ( comparison_object )
2005-09-09 04:54:02 -04:00
comparison_object . equal? ( self ) ||
( comparison_object . instance_of? ( self . class ) &&
comparison_object . id == id &&
! comparison_object . new_record? )
2004-11-23 20:04:44 -05:00
end
# Delegates to ==
def eql? ( comparison_object )
self == ( comparison_object )
end
2005-07-03 04:32:43 -04:00
2004-11-23 20:04:44 -05:00
# 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
2005-03-30 09:25:29 -05:00
id . hash
2004-11-23 20:04:44 -05:00
end
# For checking respond_to? without searching the attributes (which is faster).
alias_method :respond_to_without_attributes? , :respond_to?
# A Person object with a name attribute can ask person.respond_to?("name"), person.respond_to?("name="), and
# person.respond_to?("name?") which will all return true.
2005-01-03 18:23:02 -05:00
def respond_to? ( method , include_priv = false )
2005-10-22 12:43:39 -04:00
if attr_name = self . class . column_methods_hash [ method . to_sym ]
return true if @attributes . include? ( attr_name ) || attr_name == self . class . primary_key
return false if self . class . read_methods . include? ( attr_name )
elsif @attributes . include? ( method_name = method . to_s )
return true
elsif md = / (=| \ ?|_before_type_cast)$ / . match ( method_name )
return true if @attributes . include? ( md . pre_match )
end
# super must be called at the end of the method, because the inherited respond_to?
# would return true for generated readers, even if the attribute wasn't present
super
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:43 -04:00
2005-05-02 12:46:30 -04:00
# Just freeze the attributes hash, such that associations are still accessible even on destroyed records.
def freeze
2005-08-14 04:58:53 -04:00
@attributes . freeze ; self
2005-05-02 12:46:30 -04:00
end
2005-07-03 04:32:43 -04:00
2005-05-02 12:46:30 -04:00
def frozen?
@attributes . frozen?
end
2005-07-03 04:32:43 -04:00
2005-10-14 20:19:56 -04:00
def readonly?
@readonly == true
end
def readonly!
@readonly = true
end
2004-11-23 20:04:44 -05:00
private
def create_or_update
if new_record? then create else update end
end
2005-10-10 23:55:49 -04:00
# Updates the associated record with values matching those of the instance attributes.
2004-11-23 20:04:44 -05:00
def update
connection . update (
" UPDATE #{ self . class . table_name } " +
2004-12-01 10:20:53 -05:00
" SET #{ quoted_comma_pair_list ( connection , attributes_with_quotes ( false ) ) } " +
2004-12-07 09:48:53 -05:00
" WHERE #{ self . class . primary_key } = #{ quote ( id ) } " ,
2004-11-23 20:04:44 -05:00
" #{ self . class . name } Update "
)
end
2005-10-10 23:55:49 -04:00
# Creates a new record with values matching those of the instance attributes.
2004-11-23 20:04:44 -05:00
def create
2005-12-21 01:16:42 -05:00
if self . id . nil? && connection . prefetch_primary_key? ( self . class . table_name )
2005-11-16 03:16:54 -05:00
self . id = connection . next_sequence_value ( self . class . sequence_name )
end
2004-11-23 20:04:44 -05:00
self . id = connection . insert (
" INSERT INTO #{ self . class . table_name } " +
" ( #{ quoted_column_names . join ( ', ' ) } ) " +
" VALUES( #{ attributes_with_quotes . values . join ( ', ' ) } ) " ,
" #{ self . class . name } Create " ,
2005-07-24 10:01:35 -04:00
self . class . primary_key , self . id , self . class . sequence_name
2004-11-23 20:04:44 -05:00
)
2005-07-03 04:32:43 -04:00
2004-11-23 20:04:44 -05:00
@new_record = false
end
2005-10-10 23:55:49 -04:00
# Sets the attribute used for single table inheritance to this class name if this is not the ActiveRecord descendent.
2005-07-03 04:32:43 -04:00
# Considering the hierarchy Reply < Message < ActiveRecord, this makes it possible to do Reply.new without having to
# set Reply[Reply.inheritance_column] = "Reply" yourself. No such attribute would be set for objects of the
2004-11-23 20:04:44 -05:00
# Message class in that example.
def ensure_proper_type
unless self . class . descends_from_active_record?
write_attribute ( self . class . inheritance_column , Inflector . demodulize ( self . class . name ) )
end
end
# Allows access to the object attributes, which are held in the @attributes hash, as were
# they first-class methods. So a Person class with a name attribute can use Person#name and
# Person#name= and never directly use the attributes hash -- except for multiple assigns with
# ActiveRecord#attributes=. A Milestone class can also ask Milestone#completed? to test that
2005-07-03 04:32:43 -04:00
# the completed attribute is not nil or 0.
2004-11-23 20:04:44 -05:00
#
# It's also possible to instantiate related objects, so a Client class belonging to the clients
# table with a master_id foreign key can instantiate master through Client#master.
2005-07-03 04:32:39 -04:00
def method_missing ( method_id , * args , & block )
method_name = method_id . to_s
if @attributes . include? ( method_name )
2005-10-06 20:53:05 -04:00
define_read_methods if self . class . read_methods . empty? && self . class . generate_read_methods
2005-07-03 04:32:39 -04:00
read_attribute ( method_name )
2005-10-13 16:44:39 -04:00
elsif self . class . primary_key . to_s == method_name
id
2005-07-03 04:32:39 -04:00
elsif md = / (=| \ ?|_before_type_cast)$ / . match ( method_name )
attribute_name , method_type = md . pre_match , md . to_s
if @attributes . include? ( attribute_name )
case method_type
when '='
write_attribute ( attribute_name , args . first )
when '?'
query_attribute ( attribute_name )
when '_before_type_cast'
read_attribute_before_type_cast ( attribute_name )
end
else
super
end
2004-11-23 20:04:44 -05:00
else
super
end
end
2005-10-10 23:55:49 -04:00
# Returns the value of the attribute identified by <tt>attr_name</tt> after it has been typecast (for example,
2004-11-23 20:04:44 -05:00
# "2004-12-12" in a data column is cast to a date object, like Date.new(2004, 12, 12)).
2005-02-23 18:51:34 -05:00
def read_attribute ( attr_name )
2005-09-02 04:31:11 -04:00
attr_name = attr_name . to_s
2005-07-05 08:38:06 -04:00
if ! ( value = @attributes [ attr_name ] ) . nil?
2004-11-23 20:04:44 -05:00
if column = column_for_attribute ( attr_name )
2005-07-03 04:32:17 -04:00
if unserializable_attribute? ( attr_name , column )
unserialize_attribute ( attr_name )
else
2005-07-03 04:32:39 -04:00
column . type_cast ( value )
2005-07-03 04:32:17 -04:00
end
2004-12-19 06:25:55 -05:00
else
2005-07-03 04:32:39 -04:00
value
2004-11-23 20:04:44 -05:00
end
else
nil
end
end
2004-12-19 06:25:55 -05:00
def read_attribute_before_type_cast ( attr_name )
@attributes [ attr_name ]
end
2005-10-06 20:53:05 -04:00
# Called on first read access to any given column and generates reader
# methods for all columns in the columns_hash if
# ActiveRecord::Base.generate_read_methods is set to true.
def define_read_methods
self . class . columns_hash . each do | name , column |
2005-10-22 12:43:39 -04:00
unless self . class . serialized_attributes [ name ] || respond_to_without_attributes? ( name )
2005-10-06 20:53:05 -04:00
define_read_method ( name . to_sym , name , column )
end
end
end
2005-11-02 12:30:59 -05:00
# Define an attribute reader method. Cope with nil column.
2005-10-06 20:53:05 -04:00
def define_read_method ( symbol , attr_name , column )
2005-11-02 12:30:59 -05:00
cast_code = column . type_cast_code ( 'v' ) if column
2005-10-06 20:53:05 -04:00
access_code = cast_code ? " (v=@attributes[' #{ attr_name } ']) && #{ cast_code } " : " @attributes[' #{ attr_name } '] "
2005-10-24 12:45:22 -04:00
unless attr_name . to_s == self . class . primary_key . to_s
2005-10-22 12:43:39 -04:00
access_code = access_code . insert ( 0 , " raise NoMethodError, 'missing attribute: #{ attr_name } ', caller unless @attributes.has_key?(' #{ attr_name } '); " )
2005-11-02 12:30:59 -05:00
self . class . read_methods << attr_name
2005-10-06 20:53:05 -04:00
end
2005-11-19 01:47:26 -05:00
begin
self . class . class_eval ( " def #{ symbol } ; #{ access_code } ; end " )
rescue SyntaxError = > err
self . class . read_methods . delete ( attr_name )
if logger
logger . warn " Exception occured during reader method compilation. "
logger . warn " Maybe #{ attr_name } is not a valid Ruby identifier? "
logger . warn " #{ err . message } "
end
end
2005-10-06 20:53:05 -04:00
end
2004-11-23 20:04:44 -05:00
# Returns true if the attribute is of a text column and marked for serialization.
def unserializable_attribute? ( attr_name , column )
2005-10-06 20:53:05 -04:00
column . text? && self . class . serialized_attributes [ attr_name ]
2004-11-23 20:04:44 -05:00
end
# Returns the unserialized object of the attribute.
def unserialize_attribute ( attr_name )
unserialized_object = object_from_yaml ( @attributes [ attr_name ] )
if unserialized_object . is_a? ( self . class . serialized_attributes [ attr_name ] )
@attributes [ attr_name ] = unserialized_object
else
2005-07-03 04:32:43 -04:00
raise SerializationTypeMismatch ,
2005-07-03 04:32:39 -04:00
" #{ attr_name } was supposed to be a #{ self . class . serialized_attributes [ attr_name ] } , but was a #{ unserialized_object . class . to_s } "
2004-11-23 20:04:44 -05:00
end
end
# Updates the attribute identified by <tt>attr_name</tt> with the specified +value+. Empty strings for fixnum and float
# columns are turned into nil.
2005-02-23 18:51:34 -05:00
def write_attribute ( attr_name , value )
2005-10-06 20:53:05 -04:00
attr_name = attr_name . to_s
if ( column = column_for_attribute ( attr_name ) ) && column . number?
@attributes [ attr_name ] = convert_number_column_value ( value )
else
@attributes [ attr_name ] = value
end
2004-11-23 20:04:44 -05:00
end
2005-10-06 20:53:05 -04:00
def convert_number_column_value ( value )
case value
when FalseClass : 0
when TrueClass : 1
when '' : nil
else value
end
2004-11-23 20:04:44 -05:00
end
def query_attribute ( attr_name )
attribute = @attributes [ attr_name ]
if attribute . kind_of? ( Fixnum ) && attribute == 0
false
elsif attribute . kind_of? ( String ) && attribute == " 0 "
false
elsif attribute . kind_of? ( String ) && attribute . empty?
false
elsif attribute . nil?
false
elsif attribute == false
false
elsif attribute == " f "
false
elsif attribute == " false "
false
else
true
end
end
def remove_attributes_protected_from_mass_assignment ( attributes )
if self . class . accessible_attributes . nil? && self . class . protected_attributes . nil?
2005-06-28 13:15:01 -04:00
attributes . reject { | key , value | attributes_protected_by_default . include? ( key . gsub ( / \ (.+ / , " " ) ) }
2004-11-23 20:04:44 -05:00
elsif self . class . protected_attributes . nil?
2005-06-28 13:15:01 -04:00
attributes . reject { | key , value | ! self . class . accessible_attributes . include? ( key . gsub ( / \ (.+ / , " " ) . intern ) || attributes_protected_by_default . include? ( key . gsub ( / \ (.+ / , " " ) ) }
2004-11-23 20:04:44 -05:00
elsif self . class . accessible_attributes . nil?
2005-06-28 13:15:01 -04:00
attributes . reject { | key , value | self . class . protected_attributes . include? ( key . gsub ( / \ (.+ / , " " ) . intern ) || attributes_protected_by_default . include? ( key . gsub ( / \ (.+ / , " " ) ) }
2004-11-23 20:04:44 -05:00
end
end
2005-01-23 12:24:54 -05:00
# The primary key and inheritance column can never be set by mass-assignment for security reasons.
def attributes_protected_by_default
2005-10-12 15:59:13 -04:00
default = [ self . class . primary_key , self . class . inheritance_column ]
default << 'id' unless self . class . primary_key . eql? 'id'
default
2005-01-23 12:24:54 -05:00
end
2004-11-23 20:04:44 -05:00
# Returns copy of the attributes hash where all the values have been safely quoted for use in
2005-07-03 04:32:43 -04:00
# an SQL statement.
2004-12-01 10:20:53 -05:00
def attributes_with_quotes ( include_primary_key = true )
2005-07-03 04:32:43 -04:00
attributes . inject ( { } ) do | quoted , ( name , value ) |
2005-06-12 02:56:51 -04:00
if column = column_for_attribute ( name )
2005-10-06 19:19:55 -04:00
quoted [ name ] = quote ( value , column ) unless ! include_primary_key && column . primary
2005-06-12 02:56:51 -04:00
end
quoted
2004-11-23 20:04:44 -05:00
end
end
2005-06-12 02:56:51 -04:00
2004-11-23 20:04:44 -05:00
# Quote strings appropriately for SQL statements.
def quote ( value , column = nil )
2005-07-03 04:32:17 -04:00
self . class . connection . quote ( value , column )
2004-11-23 20:04:44 -05:00
end
# Interpolate custom sql string in instance context.
# Optional record argument is meant for custom insert_sql.
def interpolate_sql ( sql , record = nil )
2005-10-14 19:28:35 -04:00
instance_eval ( " %@ #{ sql . gsub ( '@' , '\@' ) } @ " )
2004-11-23 20:04:44 -05:00
end
# Initializes the attributes array with keys matching the columns from the linked table and
# the values matching the corresponding default value of that column, so
# that a new instance, or one populated from a passed-in Hash, still has all the attributes
# that instances loaded from the database would.
def attributes_from_column_definition
2005-09-08 08:21:36 -04:00
self . class . columns . inject ( { } ) do | attributes , column |
2004-12-20 08:32:09 -05:00
attributes [ column . name ] = column . default unless column . name == self . class . primary_key
2004-11-23 20:04:44 -05:00
attributes
end
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
2005-02-07 09:15:53 -05:00
# parentheses to have the parameters typecasted before they're used in the constructor. Use i for Fixnum, f for Float,
2004-11-23 20:04:44 -05:00
# s for String, and a for Array. If all the values for a given attribute is empty, the attribute will be set to nil.
def assign_multiparameter_attributes ( pairs )
execute_callstack_for_multiparameter_attributes (
extract_callstack_for_multiparameter_attributes ( pairs )
)
end
2005-07-03 04:32:43 -04:00
2004-11-23 20:04:44 -05:00
# Includes an ugly hack for Time.local instead of Time.new because the latter is reserved by Time itself.
def execute_callstack_for_multiparameter_attributes ( callstack )
2005-03-06 09:11:26 -05:00
errors = [ ]
2004-11-23 20:04:44 -05:00
callstack . each do | name , values |
klass = ( self . class . reflect_on_aggregation ( name ) || column_for_attribute ( name ) ) . klass
if values . empty?
send ( name + " = " , nil )
else
2005-03-06 09:11:26 -05:00
begin
send ( name + " = " , Time == klass ? klass . local ( * values ) : klass . new ( * values ) )
rescue = > ex
errors << AttributeAssignmentError . new ( " error on assignment #{ values . inspect } to #{ name } " , ex , name )
end
2004-11-23 20:04:44 -05:00
end
end
2005-03-06 09:11:26 -05:00
unless errors . empty?
raise MultiparameterAssignmentErrors . new ( errors ) , " #{ errors . size } error(s) on assignment of multiparameter attributes "
end
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:43 -04:00
2004-11-23 20:04:44 -05:00
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 )
unless value . empty?
2005-07-03 04:32:43 -04:00
attributes [ attribute_name ] <<
2005-01-24 06:39:23 -05:00
[ find_parameter_position ( multiparameter_name ) , type_cast_attribute_value ( multiparameter_name , value ) ]
2004-11-23 20:04:44 -05:00
end
end
attributes . each { | name , values | attributes [ name ] = values . sort_by { | v | v . first } . collect { | v | v . last } }
end
2005-07-03 04:32:43 -04:00
2004-11-23 20:04:44 -05:00
def type_cast_attribute_value ( multiparameter_name , value )
multiparameter_name =~ / \ ([0-9]*([a-z]) \ ) / ? value . send ( " to_ " + $1 ) : value
end
2005-07-03 04:32:43 -04:00
2004-11-23 20:04:44 -05:00
def find_parameter_position ( multiparameter_name )
multiparameter_name . scan ( / \ (([0-9]*).* \ ) / ) . first . first
end
2005-07-03 04:32:43 -04:00
2004-11-23 20:04:44 -05:00
# Returns a comma-separated pair list, like "key1 = val1, key2 = val2".
def comma_pair_list ( hash )
hash . inject ( [ ] ) { | list , pair | list << " #{ pair . first } = #{ pair . last } " } . join ( " , " )
end
def quoted_column_names ( attributes = attributes_with_quotes )
2005-07-03 04:32:12 -04:00
attributes . keys . collect do | column_name |
self . class . connection . quote_column_name ( column_name )
end
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:12 -04:00
def quote_columns ( quoter , hash )
hash . inject ( { } ) do | quoted , ( name , value ) |
quoted [ quoter . quote_column_name ( name ) ] = value
quoted
2005-01-24 06:39:23 -05:00
end
2004-11-23 20:04:44 -05:00
end
2005-07-03 04:32:12 -04:00
def quoted_comma_pair_list ( quoter , hash )
comma_pair_list ( quote_columns ( quoter , hash ) )
2004-11-23 20:04:44 -05:00
end
def object_from_yaml ( string )
2005-01-20 09:40:11 -05:00
return string unless string . is_a? ( String )
2005-09-27 05:25:17 -04:00
YAML :: load ( string ) rescue string
2004-11-23 20:04:44 -05:00
end
2005-05-19 12:39:50 -04:00
def clone_attributes ( reader_method = :read_attribute , attributes = { } )
self . attribute_names . inject ( attributes ) do | attributes , name |
attributes [ name ] = clone_attribute_value ( reader_method , name )
attributes
end
end
def clone_attribute_value ( reader_method , attribute_name )
value = send ( reader_method , attribute_name )
value . clone
rescue TypeError , NoMethodError
value
end
2004-11-23 20:04:44 -05:00
end
2005-11-06 05:18:51 -05:00
end