Update checkprime.py

VishwakarmaRishi · web-flow · commit 0e353275a5c8 · 2020-10-12T14:43:14.000+05:30
diff --git a/checkprime.py b/checkprime.py
@@ -1,16 +1,34 @@
 # https://www.facebook.com/jasmeet.singh.77920/posts/3180825812045692
 # Subscribed by Code House
 
-num = 407
-
-#num = int(input("Enter a number: "))
-
-if num > 1:
-   # check for factors
-   for i in range(2,num):
-       if (num % i) == 0:
-           print(num,"is not a prime number")
-           print(i,"times",num//i,"is",num)
-           break
-   else:
-       print(num,"is a prime number")
+# num = 407
+
+# #num = int(input("Enter a number: "))
+
+# if num > 1:
+#    # check for factors
+#    for i in range(2,num):
+#        if (num % i) == 0:
+#            print(num,"is not a prime number")
+#            print(i,"times",num//i,"is",num)
+#            break
+#    else:
+#        print(num,"is a prime number")
+
+
+
+# define the checking function
+def is_prime(number):
+    for i in range(2, int(number**(.5))):
+        if not(number % i):
+            return False
+    return True
+
+# driver code
+num = int(input("Enter a number: ")) or 69
+
+if is_prime(num) and num > 1:
+    print(num, 'is a prime number')
+else:
+    print(num, 'is not a prime number')
+
