company.rb 5.3 KB
Newer Older
1 2 3 4 5
class AbstractCompany < ActiveRecord::Base
  self.abstract_class = true
end

class Company < AbstractCompany
D
Initial  
David Heinemeier Hansson 已提交
6
  attr_protected :rating
7
  set_sequence_name :companies_nonstd_seq
8 9

  validates_presence_of :name
10 11

  has_one :dummy_account, :foreign_key => "firm_id", :class_name => "Account"
J
Jeremy Kemper 已提交
12

13 14 15
  def arbitrary_method
    "I am Jack's profound disappointment"
  end
D
Initial  
David Heinemeier Hansson 已提交
16 17
end

18 19 20 21
module Namespaced
  class Company < ::Company
  end
end
D
Initial  
David Heinemeier Hansson 已提交
22 23

class Firm < Company
24
  has_many :clients, :order => "id", :dependent => :destroy, :counter_sql =>
25 26
      "SELECT COUNT(*) FROM companies WHERE firm_id = 1 " +
      "AND (#{QUOTED_TYPE} = 'Client' OR #{QUOTED_TYPE} = 'SpecialClient' OR #{QUOTED_TYPE} = 'VerySpecialClient' )"
D
Initial  
David Heinemeier Hansson 已提交
27 28
  has_many :clients_sorted_desc, :class_name => "Client", :order => "id DESC"
  has_many :clients_of_firm, :foreign_key => "client_of", :class_name => "Client", :order => "id"
29
  has_many :unvalidated_clients_of_firm, :foreign_key => "client_of", :class_name => "Client", :validate => false
30 31
  has_many :dependent_clients_of_firm, :foreign_key => "client_of", :class_name => "Client", :order => "id", :dependent => :destroy
  has_many :exclusively_dependent_clients_of_firm, :foreign_key => "client_of", :class_name => "Client", :order => "id", :dependent => :delete_all
32
  has_many :limited_clients, :class_name => "Client", :order => "id", :limit => 1
D
Initial  
David Heinemeier Hansson 已提交
33
  has_many :clients_like_ms, :conditions => "name = 'Microsoft'", :class_name => "Client", :order => "id"
34
  has_many :clients_with_interpolated_conditions, :class_name => "Client", :conditions => 'rating > #{rating}'
J
Jeremy Kemper 已提交
35
  has_many :clients_like_ms_with_hash_conditions, :conditions => { :name => 'Microsoft' }, :class_name => "Client", :order => "id"
D
Initial  
David Heinemeier Hansson 已提交
36
  has_many :clients_using_sql, :class_name => "Client", :finder_sql => 'SELECT * FROM companies WHERE client_of = #{id}'
37
  has_many :clients_using_counter_sql, :class_name => "Client",
38
           :finder_sql  => 'SELECT * FROM companies WHERE client_of = #{id}',
39 40
           :counter_sql => 'SELECT COUNT(*) FROM companies WHERE client_of = #{id}'
  has_many :clients_using_zero_counter_sql, :class_name => "Client",
41
           :finder_sql  => 'SELECT * FROM companies WHERE client_of = #{id}',
42
           :counter_sql => 'SELECT 0 FROM companies WHERE client_of = #{id}'
43 44 45
  has_many :no_clients_using_counter_sql, :class_name => "Client",
           :finder_sql  => 'SELECT * FROM companies WHERE client_of = 1000',
           :counter_sql => 'SELECT COUNT(*) FROM companies WHERE client_of = 1000'
46
  has_many :clients_using_finder_sql, :class_name => "Client", :finder_sql => 'SELECT * FROM companies WHERE 1=1'
47
  has_many :plain_clients, :class_name => 'Client'
48
  has_many :readonly_clients, :class_name => 'Client', :readonly => true
D
Initial  
David Heinemeier Hansson 已提交
49

50 51
  has_one :account, :foreign_key => "firm_id", :dependent => :destroy, :validate => true
  has_one :unvalidated_account, :foreign_key => "firm_id", :class_name => 'Account', :validate => false
52
  has_one :account_with_select, :foreign_key => "firm_id", :select => "id, firm_id", :class_name=>'Account'
53
  has_one :readonly_account, :foreign_key => "firm_id", :class_name => "Account", :readonly => true
D
Initial  
David Heinemeier Hansson 已提交
54 55
end

56 57 58 59 60
class DependentFirm < Company
  has_one :account, :foreign_key => "firm_id", :dependent => :nullify
  has_many :companies, :foreign_key => 'client_of', :order => "id", :dependent => :nullify
end

61 62
class ExclusivelyDependentFirm < Company
  has_one :account, :foreign_key => "firm_id", :dependent => :delete
63 64
  has_many :dependent_sanitized_conditional_clients_of_firm, :foreign_key => "client_of", :class_name => "Client", :order => "id", :dependent => :delete_all, :conditions => "name = 'BigShot Inc.'"
  has_many :dependent_conditional_clients_of_firm, :foreign_key => "client_of", :class_name => "Client", :order => "id", :dependent => :delete_all, :conditions => ["name = ?", 'BigShot Inc.']
65
end
66

D
Initial  
David Heinemeier Hansson 已提交
67 68 69
class Client < Company
  belongs_to :firm, :foreign_key => "client_of"
  belongs_to :firm_with_basic_id, :class_name => "Firm", :foreign_key => "firm_id"
70
  belongs_to :firm_with_select, :class_name => "Firm", :foreign_key => "firm_id", :select => "id"
D
Initial  
David Heinemeier Hansson 已提交
71
  belongs_to :firm_with_other_name, :class_name => "Firm", :foreign_key => "client_of"
72
  belongs_to :firm_with_condition, :class_name => "Firm", :foreign_key => "client_of", :conditions => ["1 = ?", 1]
73
  belongs_to :readonly_firm, :class_name => "Firm", :foreign_key => "firm_id", :readonly => true
74 75 76 77 78 79 80 81 82 83 84 85 86 87

  # Record destruction so we can test whether firm.clients.clear has
  # is calling client.destroy, deleting from the database, or setting
  # foreign keys to NULL.
  def self.destroyed_client_ids
    @destroyed_client_ids ||= Hash.new { |h,k| h[k] = [] }
  end

  before_destroy do |client|
    if client.firm
      Client.destroyed_client_ids[client.firm.id] << client.id
    end
    true
  end
J
Jeremy Kemper 已提交
88

89 90 91 92
  # Used to test that read and question methods are not generated for these attributes
  def ruby_type
    read_attribute :ruby_type
  end
J
Jeremy Kemper 已提交
93

94 95 96
  def rating?
    query_attribute :rating
  end
D
Initial  
David Heinemeier Hansson 已提交
97 98 99 100 101 102 103 104 105 106 107
end


class SpecialClient < Client
end

class VerySpecialClient < SpecialClient
end

class Account < ActiveRecord::Base
  belongs_to :firm
J
Jeremy Kemper 已提交
108

109 110 111 112 113 114 115 116 117 118
  def self.destroyed_account_ids
    @destroyed_account_ids ||= Hash.new { |h,k| h[k] = [] }
  end

  before_destroy do |account|
    if account.firm
      Account.destroyed_account_ids[account.firm.id] << account.id
    end
    true
  end
J
Jeremy Kemper 已提交
119 120


D
Initial  
David Heinemeier Hansson 已提交
121 122 123 124
  protected
    def validate
      errors.add_on_empty "credit_limit"
    end
J
Jeremy Kemper 已提交
125
end