diff options
Diffstat (limited to 'app')
| -rw-r--r-- | app/models/company.rb | 3 | ||||
| -rw-r--r-- | app/models/license_status.rb | 4 | ||||
| -rw-r--r-- | app/models/license_status/confidential.rb | 5 |
3 files changed, 8 insertions, 4 deletions
diff --git a/app/models/company.rb b/app/models/company.rb index 7233020..4b6e798 100644 --- a/app/models/company.rb +++ b/app/models/company.rb @@ -2,9 +2,6 @@ class Company < ActiveRecord::Base has_many :licenses def status(status) - #today = DateTime.now - #licenses.where('issued_at < ? AND expired_at > ?', today, today) - status.filter(licenses) end end diff --git a/app/models/license_status.rb b/app/models/license_status.rb index 1c5833f..82af2fd 100644 --- a/app/models/license_status.rb +++ b/app/models/license_status.rb @@ -1,7 +1,9 @@ class LicenseStatus ACTIVE=ActiveStatus.new("active") EXPIRED=ExpiredStatus.new("expired") - ALL=[ACTIVE, EXPIRED] + CONFIDENTIAL=Confidential.new + + ALL=[ACTIVE, EXPIRED, CONFIDENTIAL] def self.find_match(status) ACTIVE diff --git a/app/models/license_status/confidential.rb b/app/models/license_status/confidential.rb new file mode 100644 index 0000000..f3e980a --- /dev/null +++ b/app/models/license_status/confidential.rb @@ -0,0 +1,5 @@ +class LicenseStatus::Confidential + def filter(licenses) + licenses.where(confidential: true) + end +end |
