import random
import string


def password_generator(number):
    uppalph = random.sample(string.ascii_uppercase,number)
    lowalph = random.sample(string.ascii_lowercase,number)
    dig = random.sample(string.digits,number)
    punc = random.sample(string.punctuation,number)
    password = uppalph + lowalph + dig + punc
    for _ in range(number):
        random.shuffle(password)
    print(''.join(password))


if __name__ == '__main__':
    flag = 1
    sec_num = 0
    while flag:
        security = input("Yo Password generator here!! Please enter how secure your generated password should be [ HIGH | MEDIUM | LOW ]:\t")
        if security == 'high':
            sec_num = 3
            flag = 0
        elif security == 'medium':
            sec_num = 2
            flag = 0
        elif security == 'low':
            sec_num = 1
            flag = 0
        else:
            print("\nHey come on dude!!! Just look what input is allowed... try again\n")
    password_generator(sec_num)