Class: Sequel::Model
- Inherits:
-
Object
- Object
- Sequel::Model
- Extended by:
- Enumerable, Associations
- Defined in:
- lib/sequel_model.rb,
lib/sequel_model/base.rb,
lib/sequel_model/hooks.rb,
lib/sequel_model/record.rb,
lib/sequel_model/schema.rb,
lib/sequel_model/caching.rb,
lib/sequel_model/plugins.rb,
lib/sequel_model/validations.rb,
lib/sequel_model/association_reflection.rb
Overview
Model has some methods that are added via metaprogramming:
-
All of the methods in DATASET_METHODS have class methods created that call the Model’s dataset with the method of the same name with the given arguments.
-
All of the methods in HOOKS have class methods created that accept either a method name symbol or an optional tag and a block. These methods run the code as a callback at the specified time. For example:
Model.before_save :do_something Model.before_save(:do_something_else){ self.something_else = 42} object = Model.new object.save
Would run the object’s :do_something method following by the code block related to :do_something_else. Note that if you specify a block, a tag is optional. If the tag is not nil, it will overwrite a previous block with the same tag. This allows hooks to work with systems that reload code.
-
All of the methods in HOOKS also create instance methods, but you should not override these instance methods.
-
The following instance_methods all call the class method of the same name: columns, dataset, db, primary_key, str_columns.
-
The following class level attr_readers are created: allowed_columns, cache_store, cache_ttl, dataset_methods, primary_key, restricted_columns, sti_dataset, and sti_key. You should not usually need to access these directly.
-
The following class level attr_accessors are created: strict_param_setting and typecast_on_assignment:
# Don't raise errors in new/set/update/etc. if an attempt to # access a missing/restricted method occurs (just silently # skip it) Model.strict_param_setting = false Model.new(:id=>1) # No Error # Don't typecast attribute values on assignment Model.typecast_on_assignment = false m = Model.new m.number = '10' m.number # => '10' instead of 10
-
The following class level method aliases are defined:
-
Model.dataset= => set_dataset
-
Model.is_a => is
-
Defined Under Namespace
Modules: Associations, DatasetMethods, Validation
Constant Summary collapse
- DATASET_METHODS =
Dataset methods to proxy via metaprogramming
%w'<< all avg count delete distinct eager eager_graph each each_page empty? except exclude filter first from_self full_outer_join get graph group group_and_count group_by having import inner_join insert insert_multiple intersect interval invert_order join join_table last left_outer_join limit map multi_insert naked order order_by order_more paginate print query range reverse_order right_outer_join select select_all select_more set set_graph_aliases single_value size to_csv transform union uniq unordered update where'.map{|x| x.to_sym}
- INHERITED_INSTANCE_VARIABLES =
Instance variables that are inherited in subclasses
{:@allowed_columns=>:dup, :@dataset_methods=>:dup, :@primary_key=>nil, :@restricted_columns=>:dup, :@restrict_primary_key=>nil, :@sti_dataset=>nil, :@sti_key=>nil, :@strict_param_setting=>nil, :@typecast_on_assignment=>nil}
- HOOKS =
Hooks that are safe for public use
[:after_initialize, :before_create, :after_create, :before_update, :after_update, :before_save, :after_save, :before_destroy, :after_destroy, :before_validation, :after_validation]
- PRIVATE_HOOKS =
Hooks that are only for internal use
[:before_update_values, :before_delete]
- RESTRICTED_SETTER_METHODS =
The setter methods (methods ending with =) that are never allowed to be called automatically via set.
%w"== === []= taguri= typecast_on_assignment="
- @@lazy_load_schema =
Whether to lazily load the schema for future subclasses. Unless turned off, checks the database for the table schema whenever a subclass is created
false
Instance Attribute Summary collapse
-
#associations ⇒ Object
readonly
The current cached associations.
-
#changed_columns ⇒ Object
readonly
The columns that have been updated.
-
#strict_param_setting ⇒ Object
writeonly
Whether this model instance should raise an error if attempting to call a method through set/update and their variants that either doesn’t exist or access to it is denied.
-
#typecast_on_assignment ⇒ Object
writeonly
Whether this model instance should typecast on attribute assignment.
-
#values ⇒ Object
readonly
The hash of attribute values.
Class Method Summary collapse
-
.[](*args) ⇒ Object
Returns the first record from the database matching the conditions.
-
.columns ⇒ Object
Returns the columns in the result set in their original order.
-
.create(values = {}, &block) ⇒ Object
Creates new instance with values set to passed-in Hash, saves it (running any callbacks), and returns the instance if the object was saved correctly.
-
.create_table ⇒ Object
Creates table.
-
.create_table! ⇒ Object
Drops the table if it exists and then runs create_table.
-
.dataset ⇒ Object
Returns the dataset associated with the Model class.
-
.db ⇒ Object
Returns the database associated with the Model class.
-
.db=(db) ⇒ Object
Sets the database associated with the Model class.
-
.db_schema ⇒ Object
Returns the cached schema information if available or gets it from the database.
-
.def_dataset_method(*args, &block) ⇒ Object
If a block is given, define a method on the dataset with the given argument name using the given block as well as a method on the model that calls the dataset method.
-
.delete_all ⇒ Object
Deletes all records in the model’s table.
-
.destroy_all ⇒ Object
Like delete_all, but invokes before_destroy and after_destroy hooks if used.
-
.drop_table ⇒ Object
Drops table.
-
.fetch(*args) ⇒ Object
Returns a dataset with custom SQL that yields model objects.
-
.find(*args, &block) ⇒ Object
Finds a single record according to the supplied filter, e.g.:.
-
.find_or_create(cond) ⇒ Object
Like find but invokes create with given conditions when record does not exists.
-
.has_hooks?(key) ⇒ Boolean
Returns true if the model class or any of its ancestors have defined hooks for the given hook key.
-
.has_validations? ⇒ Boolean
Returns true if validations are defined.
-
.implicit_table_name ⇒ Object
Returns the implicit table name for the model class.
-
.inherited(subclass) ⇒ Object
If possible, set the dataset for the model subclass as soon as it is created.
-
.is(plugin, *args) ⇒ Object
Loads a plugin for use with the model class, passing optional arguments to the plugin.
-
.lazy_load_schema=(value) ⇒ Object
Set whether to lazily load the schema for future model classes.
-
.load(values) ⇒ Object
Initializes a model instance as an existing record.
-
.no_primary_key ⇒ Object
Mark the model as not having a primary key.
-
.primary_key_hash(value) ⇒ Object
Returns primary key attribute hash.
-
.restrict_primary_key ⇒ Object
Restrict the setting of the primary key(s) inside new/set/update.
-
.restrict_primary_key? ⇒ Boolean
Whether or not setting the primary key inside new/set/update is restricted, true by default.
-
.schema ⇒ Object
Returns table schema created with set_schema for direct descendant of Model.
-
.serialize(*columns) ⇒ Object
Serializes column with YAML or through marshalling.
-
.set_allowed_columns(*cols) ⇒ Object
Set the columns to allow in new/set/update.
-
.set_cache(store, opts = {}) ⇒ Object
Set the cache store for the model, as well as the caching before_* hooks.
-
.set_cache_ttl(ttl) ⇒ Object
Set the time to live for the cache store, in seconds (default is 3600, so 1 hour).
-
.set_dataset(ds) ⇒ Object
Sets the dataset associated with the Model class.
-
.set_primary_key(*key) ⇒ Object
Sets primary key, regular and composite are possible.
-
.set_restricted_columns(*cols) ⇒ Object
Set the columns to restrict in new/set/update.
-
.set_schema(name = nil, &block) ⇒ Object
Defines a table schema (see Schema::Generator for more information).
-
.set_sti_key(key) ⇒ Object
Makes this model a polymorphic model with the given key being a string field in the database holding the name of the class to use.
-
.skip_superclass_validations ⇒ Object
Instructs the model to skip validations defined in superclasses.
-
.str_columns ⇒ Object
Returns the columns as a list of frozen strings instead of a list of symbols.
-
.subset(name, *args, &block) ⇒ Object
Defines a method that returns a filtered dataset.
-
.table_exists? ⇒ Boolean
Returns true if table exists, false otherwise.
-
.table_name ⇒ Object
Returns name of primary table for the dataset.
-
.unrestrict_primary_key ⇒ Object
Allow the setting of the primary key(s) inside new/set/update.
-
.validate(o) ⇒ Object
Validates the given instance.
-
.validates(&block) ⇒ Object
Defines validations by converting a longhand block into a series of shorthand definitions.
-
.validates_acceptance_of(*atts) ⇒ Object
Validates acceptance of an attribute.
-
.validates_confirmation_of(*atts) ⇒ Object
Validates confirmation of an attribute.
-
.validates_each(*atts, &block) ⇒ Object
Adds a validation for each of the given attributes using the supplied block.
-
.validates_format_of(*atts) ⇒ Object
Validates the format of an attribute.
-
.validates_length_of(*atts) ⇒ Object
Validates the length of an attribute.
-
.validates_numericality_of(*atts) ⇒ Object
Validates whether an attribute is a number.
-
.validates_presence_of(*atts) ⇒ Object
Validates the presence of an attribute.
-
.validates_uniqueness_of(*atts) ⇒ Object
Validates only if the fields in the model (specified by atts) are unique in the database.
-
.validations ⇒ Object
Returns the validations hash for the class.
Instance Method Summary collapse
-
#==(obj) ⇒ Object
(also: #eql?)
Compares model instances by values.
-
#===(obj) ⇒ Object
If pk is not nil, true only if the objects have the same class and pk.
-
#[](column) ⇒ Object
Returns value of the column’s attribute.
-
#[]=(column, value) ⇒ Object
Sets value of the column’s attribute and marks the column as changed.
-
#cache_key ⇒ Object
Return a key unique to the underlying record for caching, based on the primary key value(s) for the object.
-
#delete ⇒ Object
Deletes and returns self.
-
#destroy ⇒ Object
Like delete but runs hooks before and after delete.
-
#each(&block) ⇒ Object
Enumerates through all attributes.
-
#errors ⇒ Object
Returns the validation errors associated with the object.
-
#exists? ⇒ Boolean
Returns true when current instance exists, false otherwise.
-
#hash ⇒ Object
Unique for objects with the same class and pk (if pk is not nil), or the same class and values (if pk is nil).
-
#id ⇒ Object
Returns value for the :id attribute, even if the primary key is not id.
-
#initialize(values = nil, from_db = false) {|_self| ... } ⇒ Model
constructor
Creates new instance with values set to passed-in Hash.
-
#inspect ⇒ Object
Returns a string representation of the model instance including the class name and values.
-
#keys ⇒ Object
Returns attribute names as an array of symbols.
-
#new? ⇒ Boolean
Returns true if the current instance represents a new record.
-
#pk ⇒ Object
Returns the primary key value identifying the model instance.
-
#pk_hash ⇒ Object
Returns a hash identifying the model instance.
-
#refresh ⇒ Object
(also: #reload)
Reloads attributes from database and returns self.
-
#save(*columns) ⇒ Object
Creates or updates the record, after making sure the record is valid.
-
#save!(*columns) ⇒ Object
Creates or updates the record, without attempting to validate it first.
-
#save_changes ⇒ Object
Saves only changed columns or does nothing if no columns are marked as chanaged.
-
#set(hash) ⇒ Object
(also: #set_with_params)
Updates the instance with the supplied values with support for virtual attributes, ignoring any values for which no setter method is available.
-
#set_all(hash) ⇒ Object
Set all values using the entries in the hash, ignoring any setting of allowed_columns or restricted columns in the model.
-
#set_except(hash, *except) ⇒ Object
Set all values using the entries in the hash, except for the keys given in except.
-
#set_only(hash, *only) ⇒ Object
Set the values using the entries in the hash, only if the key is included in only.
-
#set_values(values) ⇒ Object
Sets the value attributes without saving the record.
-
#this ⇒ Object
Returns (naked) dataset that should return only this instance.
-
#update(hash) ⇒ Object
(also: #update_with_params)
Runs set with the passed hash and runs save_changes (which runs any callback methods).
-
#update_all(hash) ⇒ Object
Update all values using the entries in the hash, ignoring any setting of allowed_columns or restricted columns in the model.
-
#update_except(hash, *except) ⇒ Object
Update all values using the entries in the hash, except for the keys given in except.
-
#update_only(hash, *only) ⇒ Object
Update the values using the entries in the hash, only if the key is included in only.
-
#update_values(values) ⇒ Object
Sets the values attributes with set_values and then updates the record in the database using those values.
-
#valid? ⇒ Boolean
Validates the object and returns true if no errors are reported.
-
#validate ⇒ Object
Validates the object.
Methods included from Associations
all_association_reflections, associate, association_reflection, many_to_many, many_to_one, one_to_many
Constructor Details
#initialize(values = nil, from_db = false) {|_self| ... } ⇒ Model
Creates new instance with values set to passed-in Hash. If a block is given, yield the instance to the block. This method runs the after_initialize hook after it has optionally yielded itself to the block.
Arguments:
-
values - should be a hash with symbol keys, though string keys will work if from_db is false.
-
from_db - should only be set by Model.load, forget it exists.
40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
# File 'lib/sequel_model/record.rb', line 40 def initialize(values = nil, from_db = false, &block) values ||= {} @associations = {} @db_schema = model.db_schema @changed_columns = [] @strict_param_setting = model.strict_param_setting @typecast_on_assignment = model.typecast_on_assignment if from_db @new = false @values = values else @values = {} @new = true set(values) end @changed_columns.clear yield self if block after_initialize end |
Instance Attribute Details
#associations ⇒ Object (readonly)
The current cached associations. A hash with the keys being the association name symbols and the values being the associated object or nil (many_to_one), or the array of associated objects (*_to_many).
10 11 12 |
# File 'lib/sequel_model/record.rb', line 10 def associations @associations end |
#changed_columns ⇒ Object (readonly)
The columns that have been updated. This isn’t completely accurate, see Model#[]=.
14 15 16 |
# File 'lib/sequel_model/record.rb', line 14 def changed_columns @changed_columns end |
#strict_param_setting=(value) ⇒ Object (writeonly)
Whether this model instance should raise an error if attempting to call a method through set/update and their variants that either doesn’t exist or access to it is denied.
19 20 21 |
# File 'lib/sequel_model/record.rb', line 19 def strict_param_setting=(value) @strict_param_setting = value end |
#typecast_on_assignment=(value) ⇒ Object (writeonly)
Whether this model instance should typecast on attribute assignment
22 23 24 |
# File 'lib/sequel_model/record.rb', line 22 def typecast_on_assignment=(value) @typecast_on_assignment = value end |
#values ⇒ Object (readonly)
The hash of attribute values. Keys are symbols with the names of the underlying database columns.
26 27 28 |
# File 'lib/sequel_model/record.rb', line 26 def values @values end |
Class Method Details
.[](*args) ⇒ Object
Returns the first record from the database matching the conditions. If a hash is given, it is used as the conditions. If another object is given, it finds the first record whose primary key(s) match the given argument(s). If caching is used, the cache is checked first before a dataset lookup is attempted unless a hash is supplied.
69 70 71 72 73 74 75 76 77 78 |
# File 'lib/sequel_model/base.rb', line 69 def self.[](*args) args = args.first if (args.size == 1) raise(Error::InvalidFilter, "Did you mean to supply a hash?") if args === true || args === false if Hash === args dataset[args] else @cache_store ? cache_lookup(args) : dataset[primary_key_hash(args)] end end |
.columns ⇒ Object
Returns the columns in the result set in their original order. Generally, this will used the columns determined via the database schema, but in certain cases (e.g. models that are based on a joined dataset) it will use Dataset#columns to find the columns, which may be empty if the Dataset has no records.
85 86 87 |
# File 'lib/sequel_model/base.rb', line 85 def self.columns @columns || set_columns(dataset.naked.columns || raise(Error, "Could not fetch columns for #{self}")) end |
.create(values = {}, &block) ⇒ Object
Creates new instance with values set to passed-in Hash, saves it (running any callbacks), and returns the instance if the object was saved correctly. If there was an error saving the object, returns false.
93 94 95 96 97 |
# File 'lib/sequel_model/base.rb', line 93 def self.create(values = {}, &block) obj = new(values, &block) return false if obj.save == false obj end |
.create_table ⇒ Object
Creates table.
4 5 6 7 8 |
# File 'lib/sequel_model/schema.rb', line 4 def self.create_table db.create_table_sql_list(table_name, *schema.create_info).each {|s| db << s} @db_schema = get_db_schema(true) unless @@lazy_load_schema columns end |
.create_table! ⇒ Object
Drops the table if it exists and then runs create_table.
11 12 13 14 |
# File 'lib/sequel_model/schema.rb', line 11 def self.create_table! drop_table if table_exists? create_table end |
.dataset ⇒ Object
Returns the dataset associated with the Model class.
100 101 102 |
# File 'lib/sequel_model/base.rb', line 100 def self.dataset @dataset || raise(Error, "No dataset associated with #{self}") end |
.db ⇒ Object
Returns the database associated with the Model class.
105 106 107 108 109 110 |
# File 'lib/sequel_model/base.rb', line 105 def self.db return @db if @db @db = self == Model ? DATABASES.first : superclass.db raise(Error, "No database associated with #{self}") unless @db @db end |
.db=(db) ⇒ Object
Sets the database associated with the Model class.
113 114 115 116 117 118 |
# File 'lib/sequel_model/base.rb', line 113 def self.db=(db) @db = db if @dataset set_dataset(db[table_name]) end end |
.db_schema ⇒ Object
Returns the cached schema information if available or gets it from the database.
122 123 124 |
# File 'lib/sequel_model/base.rb', line 122 def self.db_schema @db_schema ||= get_db_schema end |
.def_dataset_method(*args, &block) ⇒ Object
If a block is given, define a method on the dataset with the given argument name using the given block as well as a method on the model that calls the dataset method.
If a block is not given, define a method on the model for each argument that calls the dataset method of the same argument name.
132 133 134 135 136 137 138 139 140 141 |
# File 'lib/sequel_model/base.rb', line 132 def self.def_dataset_method(*args, &block) raise(Error, "No arguments given") if args.empty? if block_given? raise(Error, "Defining a dataset method using a block requires only one argument") if args.length > 1 meth = args.first @dataset_methods[meth] = block dataset.(meth, &block) end args.each{|arg| instance_eval("def #{arg}(*args, &block); dataset.#{arg}(*args, &block) end", __FILE__, __LINE__)} end |
.delete_all ⇒ Object
Deletes all records in the model’s table.
144 145 146 |
# File 'lib/sequel_model/base.rb', line 144 def self.delete_all dataset.delete end |
.destroy_all ⇒ Object
Like delete_all, but invokes before_destroy and after_destroy hooks if used.
149 150 151 |
# File 'lib/sequel_model/base.rb', line 149 def self.destroy_all dataset.destroy end |
.drop_table ⇒ Object
Drops table.
17 18 19 |
# File 'lib/sequel_model/schema.rb', line 17 def self.drop_table db.execute db.drop_table_sql(table_name) end |
.fetch(*args) ⇒ Object
Returns a dataset with custom SQL that yields model objects.
154 155 156 |
# File 'lib/sequel_model/base.rb', line 154 def self.fetch(*args) db.fetch(*args).set_model(self) end |
.find(*args, &block) ⇒ Object
Finds a single record according to the supplied filter, e.g.:
Ticket.find :author => 'Sharon' # => record
161 162 163 |
# File 'lib/sequel_model/base.rb', line 161 def self.find(*args, &block) dataset.filter(*args, &block).first end |
.find_or_create(cond) ⇒ Object
Like find but invokes create with given conditions when record does not exists.
167 168 169 |
# File 'lib/sequel_model/base.rb', line 167 def self.find_or_create(cond) find(cond) || create(cond) end |
.has_hooks?(key) ⇒ Boolean
Returns true if the model class or any of its ancestors have defined hooks for the given hook key. Notice that this method cannot detect hooks defined using overridden methods.
14 15 16 17 |
# File 'lib/sequel_model/hooks.rb', line 14 def self.has_hooks?(key) has = hooks[key] && !hooks[key].empty? has || ((self != Model) && superclass.has_hooks?(key)) end |
.has_validations? ⇒ Boolean
Returns true if validations are defined.
51 52 53 |
# File 'lib/sequel_model/validations.rb', line 51 def self.has_validations? !validations.empty? end |
.implicit_table_name ⇒ Object
Returns the implicit table name for the model class.
196 197 198 |
# File 'lib/sequel_model/base.rb', line 196 def self.implicit_table_name name.demodulize.underscore.pluralize.to_sym end |
.inherited(subclass) ⇒ Object
If possible, set the dataset for the model subclass as soon as it is created. Also, inherit the INHERITED_INSTANCE_VARIABLES from the parent class.
174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 |
# File 'lib/sequel_model/base.rb', line 174 def self.inherited(subclass) sup_class = subclass.superclass ivs = subclass.instance_variables INHERITED_INSTANCE_VARIABLES.each do |iv, dup| next if ivs.include?(iv.to_s) sup_class_value = sup_class.instance_variable_get(iv) sup_class_value = sup_class_value.dup if dup == :dup && sup_class_value subclass.instance_variable_set(iv, sup_class_value) end unless ivs.include?("@dataset") begin if sup_class == Model subclass.set_dataset(Model.db[subclass.implicit_table_name]) unless subclass.name.empty? elsif ds = sup_class.instance_variable_get(:@dataset) subclass.set_dataset(sup_class.sti_key ? sup_class.sti_dataset.filter(sup_class.sti_key=>subclass.name) : ds.clone) end rescue end end end |
.is(plugin, *args) ⇒ Object
Loads a plugin for use with the model class, passing optional arguments to the plugin. If the plugin has a DatasetMethods module and the model doesn’t have a dataset, raise an Error.
21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
# File 'lib/sequel_model/plugins.rb', line 21 def self.is(plugin, *args) m = plugin_module(plugin) raise(Error, "Plugin cannot be applied because the model class has no dataset") if m.const_defined?("DatasetMethods") && !@dataset if m.respond_to?(:apply) m.apply(self, *args) end if m.const_defined?("InstanceMethods") class_def(:"#{plugin}_opts") {args.first} include(m::InstanceMethods) end if m.const_defined?("ClassMethods") (:"#{plugin}_opts") {args.first} extend(m::ClassMethods) end if m.const_defined?("DatasetMethods") dataset.(:"#{plugin}_opts") {args.first} dataset..send(:include, m::DatasetMethods) def_dataset_method(*m::DatasetMethods.instance_methods) end end |
.lazy_load_schema=(value) ⇒ Object
Set whether to lazily load the schema for future model classes. When the schema is lazy loaded, the schema information is grabbed during the first instantiation of the class instead of when the class is created.
204 205 206 |
# File 'lib/sequel_model/base.rb', line 204 def self.lazy_load_schema=(value) @@lazy_load_schema = value end |
.load(values) ⇒ Object
Initializes a model instance as an existing record. This constructor is used by Sequel to initialize model instances when fetching records. #load requires that values be a hash where all keys are symbols. It probably should not be used by external code.
212 213 214 |
# File 'lib/sequel_model/base.rb', line 212 def self.load(values) new(values, true) end |
.no_primary_key ⇒ Object
Mark the model as not having a primary key. Not having a primary key can cause issues, among which is that you won’t be able to update records.
218 219 220 |
# File 'lib/sequel_model/base.rb', line 218 def self.no_primary_key @primary_key = nil end |
.primary_key_hash(value) ⇒ Object
Returns primary key attribute hash. If using a composite primary key value such be an array with values for each primary key in the correct order. For a standard primary key, value should be an object with a compatible type for the key. If the model does not have a primary key, raises an Error.
227 228 229 230 231 232 233 234 235 236 237 |
# File 'lib/sequel_model/base.rb', line 227 def self.primary_key_hash(value) raise(Error, "#{self} does not have a primary key") unless key = @primary_key case key when Array hash = {} key.each_with_index{|k,i| hash[k] = value[i]} hash else {key => value} end end |
.restrict_primary_key ⇒ Object
Restrict the setting of the primary key(s) inside new/set/update. Because this is the default, this only make sense to use in a subclass where the parent class has used unrestrict_primary_key.
242 243 244 |
# File 'lib/sequel_model/base.rb', line 242 def self.restrict_primary_key @restrict_primary_key = true end |
.restrict_primary_key? ⇒ Boolean
Whether or not setting the primary key inside new/set/update is restricted, true by default.
248 249 250 |
# File 'lib/sequel_model/base.rb', line 248 def self.restrict_primary_key? @restrict_primary_key end |
.schema ⇒ Object
Returns table schema created with set_schema for direct descendant of Model. Does not retreive schema information from the database, see db_schema if you want that.
24 25 26 |
# File 'lib/sequel_model/schema.rb', line 24 def self.schema @schema || ((superclass != Model) && (superclass.schema)) end |
.serialize(*columns) ⇒ Object
Serializes column with YAML or through marshalling. Arguments should be column symbols, with an optional trailing hash with a :format key set to :yaml or :marshal (:yaml is the default). Setting this adds a transform to the model and dataset so that columns values will be serialized when saved and deserialized when returned from the database.
257 258 259 260 261 262 263 264 |
# File 'lib/sequel_model/base.rb', line 257 def self.serialize(*columns) format = columns.[:format] || :yaml @transform = columns.inject({}) do |m, c| m[c] = format m end @dataset.transform(@transform) if @dataset end |
.set_allowed_columns(*cols) ⇒ Object
Set the columns to allow in new/set/update. Using this means that any columns not listed here will not be modified. If you have any virtual setter methods (methods that end in =) that you want to be used in new/set/update, they need to be listed here as well (without the =).
It may be better to use (set|update)_only instead of this in places where only certain columns may be allowed.
273 274 275 |
# File 'lib/sequel_model/base.rb', line 273 def self.set_allowed_columns(*cols) @allowed_columns = cols end |
.set_cache(store, opts = {}) ⇒ Object
Set the cache store for the model, as well as the caching before_* hooks.
The cache store should implement the following API:
cache_store.set(key, obj, time) # Associate the obj with the given key
# in the cache for the time (specified
# in seconds)
cache_store.get(key) => obj # Returns object set with same key
cache_store.get(key2) => nil # nil returned if there isn't an object
# currently in the cache with that key
17 18 19 20 21 22 23 |
# File 'lib/sequel_model/caching.rb', line 17 def self.set_cache(store, opts = {}) @cache_store = store @cache_ttl = opts[:ttl] || 3600 before_save :cache_delete_unless_new before_update_values :cache_delete before_delete :cache_delete end |
.set_cache_ttl(ttl) ⇒ Object
Set the time to live for the cache store, in seconds (default is 3600, so 1 hour).
27 28 29 |
# File 'lib/sequel_model/caching.rb', line 27 def self.set_cache_ttl(ttl) @cache_ttl = ttl end |
.set_dataset(ds) ⇒ Object
Sets the dataset associated with the Model class. ds can be a Symbol (specifying a table name in the current database), or a Dataset. If a dataset is used, the model’s database is changed to the given dataset. If a symbol is used, a dataset is created from the current database with the table name given. Other arguments raise an Error.
This sets the model of the the given/created dataset to the current model and adds a destroy method to it. It also extends the dataset with the Associations::EagerLoading methods, and assigns a transform to it if there is one associated with the model. Finally, it attempts to determine the database schema based on the given/created dataset unless lazy_load_schema is set.
289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 |
# File 'lib/sequel_model/base.rb', line 289 def self.set_dataset(ds) @dataset = case ds when Symbol db[ds] when Dataset @db = ds.db ds else raise(Error, "Model.set_dataset takes a Symbol or a Sequel::Dataset") end @dataset.set_model(self) @dataset.extend(DatasetMethods) @dataset.extend(Associations::EagerLoading) @dataset.transform(@transform) if @transform @dataset_methods.each{|meth, block| @dataset.(meth, &block)} if @dataset_methods begin (@db_schema = get_db_schema) unless @@lazy_load_schema rescue end self end |
.set_primary_key(*key) ⇒ Object
Sets primary key, regular and composite are possible.
Example:
class Tagging < Sequel::Model
# composite key
set_primary_key :taggable_id, :tag_id
end
class Person < Sequel::Model
# regular key
set_primary_key :person_id
end
You can set it to nil to not have a primary key, but that cause certain things not to work, see #no_primary_key.
327 328 329 |
# File 'lib/sequel_model/base.rb', line 327 def self.set_primary_key(*key) @primary_key = (key.length == 1) ? key[0] : key.flatten end |
.set_restricted_columns(*cols) ⇒ Object
Set the columns to restrict in new/set/update. Using this means that any columns listed here will not be modified. If you have any virtual setter methods (methods that end in =) that you want not to be used in new/set/update, they need to be listed here as well (without the =).
It may be better to use (set|update)_except instead of this in places where only certain columns may be allowed.
338 339 340 |
# File 'lib/sequel_model/base.rb', line 338 def self.set_restricted_columns(*cols) @restricted_columns = cols end |
.set_schema(name = nil, &block) ⇒ Object
Defines a table schema (see Schema::Generator for more information).
This is only needed if you want to use the create_table or drop_table methods.
32 33 34 35 36 |
# File 'lib/sequel_model/schema.rb', line 32 def self.set_schema(name = nil, &block) set_dataset(db[name]) if name @schema = Schema::Generator.new(db, &block) set_primary_key(@schema.primary_key_name) if @schema.primary_key_name end |
.set_sti_key(key) ⇒ Object
Makes this model a polymorphic model with the given key being a string field in the database holding the name of the class to use. If the key given has a NULL value or there are any problems looking up the class, uses the current class.
This should be used to set up single table inheritance for the model, and it only makes sense to use this in the parent class.
You should call sti_key after any calls to set_dataset in the model, otherwise subclasses might not have the filters set up correctly.
The filters that sti_key sets up in subclasses will not work if those subclasses have further subclasses. For those middle subclasses, you will need to call set_dataset manually with the correct filter set.
356 357 358 359 360 361 362 |
# File 'lib/sequel_model/base.rb', line 356 def self.set_sti_key(key) m = self @sti_key = key @sti_dataset = dataset dataset.set_model(key, Hash.new{|h,k| h[k] = (k.constantize rescue m)}) before_create(:set_sti_key){send("#{key}=", model.name)} end |
.skip_superclass_validations ⇒ Object
Instructs the model to skip validations defined in superclasses
56 57 58 |
# File 'lib/sequel_model/validations.rb', line 56 def self.skip_superclass_validations @skip_superclass_validations = true end |
.str_columns ⇒ Object
Returns the columns as a list of frozen strings instead of a list of symbols. This makes it possible to check whether a column exists without creating a symbol, which would be a memory leak if called with user input.
368 369 370 |
# File 'lib/sequel_model/base.rb', line 368 def self.str_columns @str_columns ||= columns.map{|c| c.to_s.freeze} end |
.subset(name, *args, &block) ⇒ Object
Defines a method that returns a filtered dataset. Subsets create dataset methods, so they can be chained for scoping. For example:
Topic.subset(:popular, :num_posts > 100)
Topic.subset(:recent, :created_on > Date.today - 7)
Allows you to do:
Topic.filter(:username.like('%joe%')).popular.recent
to get topics with a username that includes joe that have more than 100 posts and were created less than 7 days ago.
386 387 388 |
# File 'lib/sequel_model/base.rb', line 386 def self.subset(name, *args, &block) def_dataset_method(name){filter(*args, &block)} end |
.table_exists? ⇒ Boolean
Returns true if table exists, false otherwise.
39 40 41 |
# File 'lib/sequel_model/schema.rb', line 39 def self.table_exists? db.table_exists?(table_name) end |
.table_name ⇒ Object
Returns name of primary table for the dataset.
391 392 393 |
# File 'lib/sequel_model/base.rb', line 391 def self.table_name dataset.opts[:from].first end |
.unrestrict_primary_key ⇒ Object
Allow the setting of the primary key(s) inside new/set/update.
396 397 398 |
# File 'lib/sequel_model/base.rb', line 396 def self.unrestrict_primary_key @restrict_primary_key = false end |
.validate(o) ⇒ Object
Validates the given instance.
82 83 84 85 86 87 88 89 90 |
# File 'lib/sequel_model/validations.rb', line 82 def self.validate(o) if superclass.respond_to?(:validate) && !@skip_superclass_validations superclass.validate(o) end validations.each do |att, procs| v = o.send(att) procs.each {|p| p[o, att, v]} end end |
.validates(&block) ⇒ Object
Defines validations by converting a longhand block into a series of shorthand definitions. For example:
class MyClass
include Validation
validates do
length_of :name, :minimum => 6
length_of :password, :minimum => 8
end
end
is equivalent to:
class MyClass
include Validation
validates_length_of :name, :minimum => 6
validates_length_of :password, :minimum => 8
end
77 78 79 |
# File 'lib/sequel_model/validations.rb', line 77 def self.validates(&block) Validation::Generator.new(self, &block) end |
.validates_acceptance_of(*atts) ⇒ Object
Validates acceptance of an attribute.
93 94 95 96 97 98 99 100 101 102 103 104 105 |
# File 'lib/sequel_model/validations.rb', line 93 def self.validates_acceptance_of(*atts) opts = { :message => 'is not accepted', :allow_nil => true, :accept => '1' }.merge!(atts.) validates_each(*atts) do |o, a, v| next unless o.instance_eval(&if_proc(opts)) next if (v.nil? && opts[:allow_nil]) || (v.blank? && opts[:allow_blank]) o.errors[a] << opts[:message] unless v == opts[:accept] end end |
.validates_confirmation_of(*atts) ⇒ Object
Validates confirmation of an attribute.
108 109 110 111 112 113 114 115 116 117 118 119 |
# File 'lib/sequel_model/validations.rb', line 108 def self.validates_confirmation_of(*atts) opts = { :message => 'is not confirmed', }.merge!(atts.) validates_each(*atts) do |o, a, v| next unless o.instance_eval(&if_proc(opts)) next if (v.nil? && opts[:allow_nil]) || (v.blank? && opts[:allow_blank]) c = o.send(:"#{a}_confirmation") o.errors[a] << opts[:message] unless v == c end end |
.validates_each(*atts, &block) ⇒ Object
Adds a validation for each of the given attributes using the supplied block. The block must accept three arguments: instance, attribute and value, e.g.:
validates_each :name, :password do |object, attribute, value|
object.errors[attribute] << 'is not nice' unless value.nice?
end
128 129 130 |
# File 'lib/sequel_model/validations.rb', line 128 def self.validates_each(*atts, &block) atts.each{|a| validations[a] << block} end |
.validates_format_of(*atts) ⇒ Object
Validates the format of an attribute.
133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 |
# File 'lib/sequel_model/validations.rb', line 133 def self.validates_format_of(*atts) opts = { :message => 'is invalid', }.merge!(atts.) unless opts[:with].is_a?(Regexp) raise ArgumentError, "A regular expression must be supplied as the :with option of the options hash" end validates_each(*atts) do |o, a, v| next unless o.instance_eval(&if_proc(opts)) next if (v.nil? && opts[:allow_nil]) || (v.blank? && opts[:allow_blank]) o.errors[a] << opts[:message] unless v.to_s =~ opts[:with] end end |
.validates_length_of(*atts) ⇒ Object
Validates the length of an attribute.
150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 |
# File 'lib/sequel_model/validations.rb', line 150 def self.validates_length_of(*atts) opts = { :too_long => 'is too long', :too_short => 'is too short', :wrong_length => 'is the wrong length' }.merge!(atts.) validates_each(*atts) do |o, a, v| next unless o.instance_eval(&if_proc(opts)) next if (v.nil? && opts[:allow_nil]) || (v.blank? && opts[:allow_blank]) if m = opts[:maximum] o.errors[a] << (opts[:message] || opts[:too_long]) unless v && v.size <= m end if m = opts[:minimum] o.errors[a] << (opts[:message] || opts[:too_short]) unless v && v.size >= m end if i = opts[:is] o.errors[a] << (opts[:message] || opts[:wrong_length]) unless v && v.size == i end if w = opts[:within] o.errors[a] << (opts[:message] || opts[:wrong_length]) unless v && w.include?(v.size) end end end |
.validates_numericality_of(*atts) ⇒ Object
Validates whether an attribute is a number.
176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 |
# File 'lib/sequel_model/validations.rb', line 176 def self.validates_numericality_of(*atts) opts = { :message => 'is not a number', }.merge!(atts.) validates_each(*atts) do |o, a, v| next unless o.instance_eval(&if_proc(opts)) next if (v.nil? && opts[:allow_nil]) || (v.blank? && opts[:allow_blank]) begin if opts[:only_integer] Kernel.Integer(v.to_s) else Kernel.Float(v.to_s) end rescue o.errors[a] << opts[:message] end end end |
.validates_presence_of(*atts) ⇒ Object
Validates the presence of an attribute.
197 198 199 200 201 202 203 204 205 206 |
# File 'lib/sequel_model/validations.rb', line 197 def self.validates_presence_of(*atts) opts = { :message => 'is not present', }.merge!(atts.) validates_each(*atts) do |o, a, v| next unless o.instance_eval(&if_proc(opts)) o.errors[a] << opts[:message] unless v && !v.blank? end end |
.validates_uniqueness_of(*atts) ⇒ Object
Validates only if the fields in the model (specified by atts) are unique in the database. You should also add a unique index in the database, as this suffers from a fairly obvious race condition.
211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 |
# File 'lib/sequel_model/validations.rb', line 211 def self.validates_uniqueness_of(*atts) opts = { :message => 'is already taken', }.merge!(atts.) validates_each(*atts) do |o, a, v| next unless o.instance_eval(&if_proc(opts)) next if v.blank? num_dups = o.class.filter(a => v).count allow = if num_dups == 0 # No unique value in the database true elsif num_dups > 1 # Multiple "unique" values in the database!! # Someone didn't add a unique index false elsif o.new? # New record, but unique value already exists in the database false elsif o.class[a => v].pk == o.pk # Unique value exists in database, but for the same record, so the update won't cause a duplicate record true else false end o.errors[a] << opts[:message] unless allow end end |
.validations ⇒ Object
Returns the validations hash for the class.
241 242 243 |
# File 'lib/sequel_model/validations.rb', line 241 def self.validations @validations ||= Hash.new {|h, k| h[k] = []} end |
Instance Method Details
#==(obj) ⇒ Object Also known as: eql?
Compares model instances by values.
80 81 82 |
# File 'lib/sequel_model/record.rb', line 80 def ==(obj) (obj.class == model) && (obj.values == @values) end |
#===(obj) ⇒ Object
If pk is not nil, true only if the objects have the same class and pk. If pk is nil, false.
87 88 89 |
# File 'lib/sequel_model/record.rb', line 87 def ===(obj) pk.nil? ? false : (obj.class == model) && (obj.pk == pk) end |
#[](column) ⇒ Object
Returns value of the column’s attribute.
62 63 64 |
# File 'lib/sequel_model/record.rb', line 62 def [](column) @values[column] end |
#[]=(column, value) ⇒ Object
Sets value of the column’s attribute and marks the column as changed. If the column already has the same value, this is a no-op.
68 69 70 71 72 73 74 75 76 77 |
# File 'lib/sequel_model/record.rb', line 68 def []=(column, value) # If it is new, it doesn't have a value yet, so we should # definitely set the new value. # If the column isn't in @values, we can't assume it is # NULL in the database, so assume it has changed. if new? || !@values.include?(column) || value != @values[column] @changed_columns << column unless @changed_columns.include?(column) @values[column] = typecast_value(column, value) end end |
#cache_key ⇒ Object
Return a key unique to the underlying record for caching, based on the primary key value(s) for the object. If the model does not have a primary key, raise an Error.
64 65 66 67 68 69 70 71 72 73 |
# File 'lib/sequel_model/caching.rb', line 64 def cache_key raise(Error, "No primary key is associated with this model") unless key = primary_key pk = case key when Array key.collect{|k| @values[k]} else @values[key] || (raise Error, 'no primary key for this record') end model.send(:cache_key, pk) end |
#delete ⇒ Object
Deletes and returns self. Does not run destroy hooks. Look into using destroy instead.
99 100 101 102 103 |
# File 'lib/sequel_model/record.rb', line 99 def delete before_delete this.delete self end |
#destroy ⇒ Object
Like delete but runs hooks before and after delete. If before_destroy returns false, returns false without deleting the object the the database. Otherwise, deletes the item from the database and returns self.
109 110 111 112 113 114 115 116 |
# File 'lib/sequel_model/record.rb', line 109 def destroy db.transaction do return false if before_destroy == false delete after_destroy end self end |
#each(&block) ⇒ Object
Enumerates through all attributes.
Example:
Ticket.find(7).each { |k, v| puts "#{k} => #{v}" }
122 123 124 |
# File 'lib/sequel_model/record.rb', line 122 def each(&block) @values.each(&block) end |
#errors ⇒ Object
Returns the validation errors associated with the object.
261 262 263 |
# File 'lib/sequel_model/validations.rb', line 261 def errors @errors ||= Validation::Errors.new end |
#exists? ⇒ Boolean
Returns true when current instance exists, false otherwise.
127 128 129 |
# File 'lib/sequel_model/record.rb', line 127 def exists? this.count > 0 end |
#hash ⇒ Object
Unique for objects with the same class and pk (if pk is not nil), or the same class and values (if pk is nil).
133 134 135 |
# File 'lib/sequel_model/record.rb', line 133 def hash [model, pk.nil? ? @values.sort_by{|k,v| k.to_s} : pk].hash end |
#id ⇒ Object
Returns value for the :id attribute, even if the primary key is not id. To get the primary key value, use #pk.
139 140 141 |
# File 'lib/sequel_model/record.rb', line 139 def id @values[:id] end |
#inspect ⇒ Object
Returns a string representation of the model instance including the class name and values.
145 146 147 |
# File 'lib/sequel_model/record.rb', line 145 def inspect "#<#{model.name} @values=#{@values.inspect}>" end |
#keys ⇒ Object
Returns attribute names as an array of symbols.
150 151 152 |
# File 'lib/sequel_model/record.rb', line 150 def keys @values.keys end |
#new? ⇒ Boolean
Returns true if the current instance represents a new record.
155 156 157 |
# File 'lib/sequel_model/record.rb', line 155 def new? @new end |
#pk ⇒ Object
Returns the primary key value identifying the model instance. Raises an error if this model does not have a primary key. If the model has a composite primary key, returns an array of values.
162 163 164 165 166 167 168 169 170 |
# File 'lib/sequel_model/record.rb', line 162 def pk raise(Error, "No primary key is associated with this model") unless key = primary_key case key when Array key.collect{|k| @values[k]} else @values[key] end end |
#pk_hash ⇒ Object
Returns a hash identifying the model instance. It should be true that:
Model[model_instance.pk_hash] === model_instance
175 176 177 |
# File 'lib/sequel_model/record.rb', line 175 def pk_hash model.primary_key_hash(pk) end |
#refresh ⇒ Object Also known as: reload
Reloads attributes from database and returns self. Also clears all cached association information. Raises an Error if the record no longer exists in the database.
182 183 184 185 186 |
# File 'lib/sequel_model/record.rb', line 182 def refresh @values = this.first || raise(Error, "Record not found") @associations.clear self end |
#save(*columns) ⇒ Object
Creates or updates the record, after making sure the record is valid. If the record is not valid, returns false. If before_save, before_create (if new?), or before_update (if !new?) return false, returns false. Otherwise, returns self.
194 195 196 197 |
# File 'lib/sequel_model/record.rb', line 194 def save(*columns) return false unless valid? save!(*columns) end |
#save!(*columns) ⇒ Object
Creates or updates the record, without attempting to validate it first. You can provide an optional list of columns to update, in which case it only updates those columns. If before_save, before_create (if new?), or before_update (if !new?) return false, returns false. Otherwise, returns self.
205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 |
# File 'lib/sequel_model/record.rb', line 205 def save!(*columns) return false if before_save == false if @new return false if before_create == false iid = model.dataset.insert(@values) # if we have a regular primary key and it's not set in @values, # we assume it's the last inserted id if (pk = primary_key) && !(Array === pk) && !@values[pk] @values[pk] = iid end if pk @this = nil # remove memoized this dataset refresh end @new = false after_create else return false if before_update == false if columns.empty? this.update(@values) @changed_columns = [] else # update only the specified columns this.update(@values.reject {|k, v| !columns.include?(k)}) @changed_columns.reject! {|c| columns.include?(c)} end after_update end after_save self end |
#save_changes ⇒ Object
Saves only changed columns or does nothing if no columns are marked as chanaged.
238 239 240 |
# File 'lib/sequel_model/record.rb', line 238 def save_changes save(*@changed_columns) unless @changed_columns.empty? end |
#set(hash) ⇒ Object Also known as: set_with_params
Updates the instance with the supplied values with support for virtual attributes, ignoring any values for which no setter method is available. Does not save the record.
If no columns have been set for this model (very unlikely), assume symbol keys are valid column names, and assign the column value based on that.
248 249 250 |
# File 'lib/sequel_model/record.rb', line 248 def set(hash) set_restricted(hash, nil, nil) end |
#set_all(hash) ⇒ Object
Set all values using the entries in the hash, ignoring any setting of allowed_columns or restricted columns in the model.
255 256 257 |
# File 'lib/sequel_model/record.rb', line 255 def set_all(hash) set_restricted(hash, false, false) end |
#set_except(hash, *except) ⇒ Object
Set all values using the entries in the hash, except for the keys given in except.
261 262 263 |
# File 'lib/sequel_model/record.rb', line 261 def set_except(hash, *except) set_restricted(hash, false, except.flatten) end |
#set_only(hash, *only) ⇒ Object
Set the values using the entries in the hash, only if the key is included in only.
267 268 269 |
# File 'lib/sequel_model/record.rb', line 267 def set_only(hash, *only) set_restricted(hash, only.flatten, false) end |
#set_values(values) ⇒ Object
Sets the value attributes without saving the record. Returns the values changed. Raises an error if the keys are not symbols or strings or a string key was passed that was not a valid column. This is a low level method that does not respect virtual attributes. It should probably be avoided. Look into using set instead.
276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 |
# File 'lib/sequel_model/record.rb', line 276 def set_values(values) s = str_columns vals = values.inject({}) do |m, kv| k, v = kv k = case k when Symbol k when String # Prevent denial of service via memory exhaustion by only # calling to_sym if the symbol already exists. raise(Error, "all string keys must be a valid columns") unless s.include?(k) k.to_sym else raise(Error, "Only symbols and strings allows as keys") end m[k] = v m end vals.each {|k, v| @values[k] = v} vals end |
#this ⇒ Object
Returns (naked) dataset that should return only this instance.
299 300 301 |
# File 'lib/sequel_model/record.rb', line 299 def this @this ||= dataset.filter(pk_hash).limit(1).naked end |
#update(hash) ⇒ Object Also known as: update_with_params
Runs set with the passed hash and runs save_changes (which runs any callback methods).
304 305 306 |
# File 'lib/sequel_model/record.rb', line 304 def update(hash) update_restricted(hash, nil, nil) end |
#update_all(hash) ⇒ Object
Update all values using the entries in the hash, ignoring any setting of allowed_columns or restricted columns in the model.
311 312 313 |
# File 'lib/sequel_model/record.rb', line 311 def update_all(hash) update_restricted(hash, false, false) end |
#update_except(hash, *except) ⇒ Object
Update all values using the entries in the hash, except for the keys given in except.
317 318 319 |
# File 'lib/sequel_model/record.rb', line 317 def update_except(hash, *except) update_restricted(hash, false, except.flatten) end |
#update_only(hash, *only) ⇒ Object
Update the values using the entries in the hash, only if the key is included in only.
323 324 325 |
# File 'lib/sequel_model/record.rb', line 323 def update_only(hash, *only) update_restricted(hash, only.flatten, false) end |
#update_values(values) ⇒ Object
Sets the values attributes with set_values and then updates the record in the database using those values. This is a low level method that does not run the usual save callbacks. It should probably be avoided. Look into using update_with_params instead.
331 332 333 334 |
# File 'lib/sequel_model/record.rb', line 331 def update_values(values) before_update_values this.update(set_values(values)) end |
#valid? ⇒ Boolean
Validates the object and returns true if no errors are reported.
275 276 277 278 |
# File 'lib/sequel_model/validations.rb', line 275 def valid? return false if validate == false errors.empty? end |
#validate ⇒ Object
Validates the object.
266 267 268 269 270 271 272 |
# File 'lib/sequel_model/validations.rb', line 266 def validate errors.clear return false if before_validation == false self.class.validate(self) after_validation nil end |