prime.py

#!usr/bin/env python3

"""
prime or composite
"""

from random import randint

num = randint(0, 100)
div = [2, 3, 5, 7]

def pri(init):
    """divider"""
    for i in div:
        if init % i == 0:
            return 1
            break

if pri(num) is None:
    print("%s is prime number" % num)
else:
    print("%s is composite number" % num)