From 17ba1b9a9847e7d5deaa2b8ddce45abc50deef10 Mon Sep 17 00:00:00 2001 From: Slushnas Date: Tue, 10 Jul 2018 20:25:57 -0700 Subject: [PATCH] Prevent divide by zero error when self.ratings is empty. --- TomeRater.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/TomeRater.py b/TomeRater.py index c3fc1cb..1fc0586 100644 --- a/TomeRater.py +++ b/TomeRater.py @@ -65,6 +65,8 @@ def add_rating(self, rating): print("Invalid Rating") def get_average_rating(self): + if len(self.ratings) == 0: + return None sum = 0 for rating in self.ratings: sum += rating