我在 python 中做了一个素数检查器。可以优化吗？

Question

我在 python 中做了一个简单的小质数检查器作为我学习时的第一个项目。如何改进？例如，这是我能做到的最紧凑的吗？

def isPrime(input):

    check = input - 1

    checkmod = input % check

    for i in range (input):

        checkmod = input % check

        check -=1

        if checkmod != 0 and input > 1 and check <= 1 or input == 2:

            return 'Prime'

        elif checkmod == 0 and check > 1:

            return 'Not prime'

print(isPrime(numberhere)) # where to put in number