Attempting exercises from Practice Python( http://www.practicepython.org/ )

16.Password_generator.py

userin = input("Would you like a strong or weak password? ")
userin = userin.lower()

import string
import random

def generatePW():
    """
    Evaluate if user wants a weak or strong password, before passing it respective password generator.
    """

    if userin == "weak":
        print(weakPW())
    elif userin == "strong":
        print(strongPW())
    else:
        print("Invalid input! Type 'Weak' or 'Strong' to generate a password.")

        
def weakPW(size = 6, chars = string.ascii_lowercase + string.digits):
    """
    Will generate a weak password if user input matches.
    """

    weak_pass = "".join(random.SystemRandom().choice(chars) for _ in range(size))
    return "Your weak password is {}, please keep it with you at all times.".format(weak_pass)

  
def strongPW(size = 12, chars = string.ascii_lowercase + string.ascii_uppercase + string.digits + string.punctuation):
    """
    Will generate a strong password if user input matches.
    """

    strong_pass = "".join(random.SystemRandom().choice(chars) for _ in range(size))
    return "Your strong password is {}, please keep it with you at all times.".format(strong_pass)

  
generatePW()