Here's what I found after a quick search in google. Tested and seems to work. The check for if the number is higher than 1 can be changed to num > 0 if you consider 1 a prime number.
# Program to check if a number is prime or not
num = 407
# prime numbers are greater than 1
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")
# if input number is less than
# or equal to 1, it is not prime
else:
print(num,"is not a prime number")
Credit: https://www.programiz.com/python-programming/examples/prime-number