-
Notifications
You must be signed in to change notification settings - Fork 0
/
calculator.py
42 lines (32 loc) · 1 KB
/
calculator.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
from typing import Tuple, Optional
from typing_extensions import Self
Num = float | int
class Calculator:
def __init__(self, init_num: Optional[Num] = None) -> None:
self.result = init_num if init_num else 0
def add(self, *nums: Tuple[Num, ...]) -> Self:
for num in nums:
self.result += num
return self
def subtract(self, *nums: Tuple[Num, ...]) -> Self:
for num in nums:
self.result -= num
return self
def multiply(self, *nums: Tuple[Num, ...]) -> Self:
for num in nums:
self.result *= num
return self
def divide(self, num: Num) -> Self:
try:
self.result /= num
except ZeroDivisionError:
print("Divisor can't be 0")
return self
def reset(self) -> Self:
self.set_result(0)
return self
def get_result(self) -> Num:
return self.result
def set_result(self, num: Num) -> Self:
self.result = num
return self