X-Git-Url: https://irasnyder.com/gitweb/?a=blobdiff_plain;f=app%2Fmodels%2Fcoitem.rb;h=5d0a2129398d611722fb126fcb145eef2b756dc6;hb=50e2ce05003eb4a56d1daa8e030a2f477aeb59bc;hp=513625a985a539bc37dbe951f95bcdc8d026234a;hpb=d95e8f6a8205bf08d7de6d4f069acba9933cdba0;p=cs356-p2-videostore.git

diff --git a/app/models/coitem.rb b/app/models/coitem.rb
index 513625a..5d0a212 100644
--- a/app/models/coitem.rb
+++ b/app/models/coitem.rb
@@ -21,6 +21,15 @@ class Coitem < ActiveRecord::Base
     return due_date < Time.now.to_date
   end
 
+  def late_fee
+    days_late = Time.now.to_date - (due_date)
+    if rentable.class == Video
+      return VideoPolicy.find_by_day(7).fee * days_late.to_i
+    else
+      return GamePolicy.find_by_day(7).fee * days_late.to_i
+    end
+  end
+
   protected
   def validate
     errors.add(:customer_id, "does not exist is the database") if customer.nil?