diff options
Diffstat (limited to 'app/models')
| -rw-r--r-- | app/models/license.rb | 1 | ||||
| -rw-r--r-- | app/models/user.rb | 10 |
2 files changed, 4 insertions, 7 deletions
diff --git a/app/models/license.rb b/app/models/license.rb index ccf9b23..6170683 100644 --- a/app/models/license.rb +++ b/app/models/license.rb @@ -2,6 +2,7 @@ class License < ActiveRecord::Base belongs_to :company belongs_to :well_type has_one :location + belongs_to :applicant, class_name: 'User', foreign_key: 'user_id' def self.most_recent(page: 1, per_page: 10) offset = (page - 1) * per_page diff --git a/app/models/user.rb b/app/models/user.rb index f2197dd..446d3e2 100644 --- a/app/models/user.rb +++ b/app/models/user.rb @@ -1,16 +1,12 @@ -class User - attr_reader :company - - def initialize(company: nil) - @company = company - end +class User < ActiveRecord::Base + belongs_to :company def apply_for(well_type, location) license = License.new license.company = company license.well_type = well_type license.location = location - #license.applicant = self + license.applicant = self license.save! license end |