refactor it using OOP

Author: sangitabera

math/Armstrong-Number/Armstrong-Number.py

@@ -1,52 +1,63 @@
-print("🔢 Armstrong Number Checker 🔢")
-print("An Armstrong number equals the sum of its digits raised to the power of number of digits")
-print("Example: 153 = 1³ + 5³ + 3³ = 1 + 125 + 27 = 153\n")
+class ArmstrongChecker:
+
+    def __init__(self, num):
+        self.num = num
+        self.original_num = num
+        self.num_digits = len(str(num))
+        self.total = 0
+
+    def calculate(self):
+        temp = self.num
 
-try:
-    num = int(input("➡️  Enter a number to check: "))
-    
-    if num < 0:
-        print("❌ Please enter a positive number!")
-    else:
-        original_num = num
-        
-        num_digits = len(str(num))
-        
-        total = 0
-        temp = num
-        
         while temp > 0:
             digit = temp % 10
-            total += digit ** num_digits
+            self.total += digit ** self.num_digits
             temp //= 10
-        
-        print(f"\n📊 Number: {original_num}")
-        print(f"📐 Number of digits: {num_digits}")
-        
-        print(f"\n🔍 Calculation:")
-        temp = original_num
+
+    def show_calculation(self):
+        temp = self.original_num
         digits = []
+
         while temp > 0:
             digits.append(temp % 10)
             temp //= 10
+
         digits.reverse()
-        
-        calculation_parts = [f"{d}^{num_digits}" for d in digits]
+
+        calculation_parts = [f"{d}^{self.num_digits}" for d in digits]
         print(f"   {' + '.join(calculation_parts)}")
-        
-        values = [f"{d**num_digits}" for d in digits]
+
+        values = [f"{d**self.num_digits}" for d in digits]
         print(f"   = {' + '.join(values)}")
-        print(f"   = {total}")
-        
-        if total == original_num:
-            print(f"\n✅ {original_num} is an Armstrong Number! 🎉")
+        print(f"   = {self.total}")
+
+    def check(self):
+        self.calculate()
+
+        print(f"\n📊 Number: {self.original_num}")
+        print(f"📐 Number of digits: {self.num_digits}")
+
+        print(f"\n🔍 Calculation:")
+        self.show_calculation()
+
+        if self.total == self.original_num:
+            print(f"\n✅ {self.original_num} is an Armstrong Number! 🎉")
         else:
-            print(f"\n❌ {original_num} is NOT an Armstrong Number.")
-        
-        print("\n💡 Some Armstrong Numbers:")
-        print("   1-digit: 0, 1, 2, 3, 4, 5, 6, 7, 8, 9")
-        print("   3-digit: 153, 370, 371, 407")
-        print("   4-digit: 1634, 8208, 9474")
+            print(f"\n❌ {self.original_num} is NOT an Armstrong Number.")
+
+
+print("🔢 Armstrong Number Checker 🔢")
+print("An Armstrong number equals the sum of its digits raised to the power of number of digits")
+print("Example: 153 = 1³ + 5³ + 3³ = 1 + 125 + 27 = 153\n")
+
+try:
+    num = int(input("➡️  Enter a number to check: "))
+
+    if num < 0:
+        print("❌ Please enter a positive number!")
+    else:
+        checker = ArmstrongChecker(num)
+        checker.check()
 
 except ValueError:
     print("❌ Invalid input! Please enter a valid number.")