add error handler

koshkar · Oct 6, 2023 · 30127f9 · 30127f9
1 parent 6c829e3
commit 30127f9
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/src/lab1/calculator.py b/src/lab1/calculator.py
@@ -94,13 +94,22 @@ def button_equal(self):
             self.number_entry.insert(0, self.f_num * int(second_number))
 
         if self.math == "division":
-            self.number_entry.insert(0, self.f_num / int(second_number))
+            if second_number == '0':
+                self.number_entry.insert(0, 'На 0 делить нельзя! (Здесь)')
+            else:    
+                self.number_entry.insert(0, self.f_num / int(second_number))
 
         if self.math == "floor_div":
-            self.number_entry.insert(0, self.f_num // int(second_number))
+            if second_number == '0':
+                self.number_entry.insert(0, 'На 0 делить нельзя! (Здесь)')
+            else:    
+                self.number_entry.insert(0, self.f_num // int(second_number))
 
         if self.math == "modulus":
-            self.number_entry.insert(0, self.f_num % int(second_number))
+            if second_number == '0':
+                self.number_entry.insert(0, 'На 0 делить нельзя! (Здесь)')
+            else:    
+                self.number_entry.insert(0, self.f_num % int(second_number))
 
     def button_subtraction(self):
         '''Обработка команды разность'''