add term to inquiry

42floors · Oct 9, 2017 · bcb0bde · bcb0bde
1 parent 96e4f7d
commit bcb0bde
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/lib/mls/inquiry.rb b/lib/mls/inquiry.rb
@@ -1,4 +1,6 @@
 class Inquiry < MLS::Model
+
+  TERMS = %w(<1 1-2 3-5 5+ flexible)
 
   has_many :emails
   belongs_to :subject, polymorphic: true
@@ -9,6 +11,20 @@ class Inquiry < MLS::Model
   def property
     subject.is_a? MLS::Model::Listing ? subject.property : subject
   end
+
+  def term_units(value=nil)
+    value ||= self.term
+    case value
+    when "<1"
+      "year"
+    when "flexible"
+      ""
+    when nil
+      ""
+    else
+      "years"
+    end
+  end
 
   def account_attributes=(account_attrs)
     account_attrs = account_attrs&.with_indifferent_access