Prevent divide by zero error when self.ratings is empty.

Slushnas · Jul 11, 2018 · 17ba1b9 · 17ba1b9
1 parent af213eb
commit 17ba1b9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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