Check if the number is the prime in Python

I am learning programming from scratch first time in my life. I am learning the Python language. My first difficult task is to write algorythm which checks if the number is the prime number.

The script should work in very simple way. You enter:

is_prime(29)

and you should get output like this:

The number 29 is the prime number.

or

The number 29 is NOT the prime number.

I did not check any solution on the internet. I did it by myself. My assumptions was as follows:

1. from the school I remember that the prime number is divided only by 1 and by itself
2. '0' and '1' are not the prime number

So I wrote the code which checks if the given number is divided by all numbers from 2 to (number-1). For example if the given value is '6', the script checks first if the 6 is divided by 2. If this is true that means the number is NOT a prime number. If 6 would be not divided by 2, the script checks if 6 is divided by 3. If so, that means the number is NOT a prime number. In case of '7'number the script checks 7/2, then 7/3, then 7/4 then 7/5, then 7/6.

The code is like this:

def is_prime(number):    if number == 0 or number == 1:        print(f"The number {number} is NOT the prime number.")    elif number == 2:        print(f"The number {number} is the prime number.")    else:        for i in range(2, number):            if number % i == 0:                check = "is NOT"                break            else:                check = "is"        print(f"The number {number} {check} the prime number.")

But then, I actually realized three things:

1. If the number is divided by 2, definitely it is not the prime number
2. If the number is not divided by 2, it can be divided by 3 or by 5.
3. If the number is not divided by 2, is not divided by 3 and is not divided by 5 that means that this number is the prime number. The only exceptions from this rules are these three numbers 2,3 and 5.

And that's it. So I wrote the code as follows

def is_prime(number):    if number > 1:        if (number %2 == 0 and number != 2) or (number %3 == 0 and number != 3 ) or(number %5 == 0 and number != 5):            print(f"The number {number} is NOT the prime number. ")        else:            print(f"The number {number} is the prime number. ")    else:        print(f"The number {number} is NOT the prime number. ")

I think that both solutions are ok. Please correct me if I am wrongBut I would like to ask you which solution is better from the programming point of view?