#!/usr/bin/env python3
"""
1. create new Linux user with root privileges
2. remove a Linux user
"""
import crypt
import subprocess
import getpass


def useradd(username, user_password):
    """
    create user
    -m create the user's home directory
    """
    encPass = crypt.crypt(user_password, "22")

    sudo_password = getpass.getpass("Enter your password: ")
    command = ("useradd -m -p " + encPass + " " + username + " -G sudo").split()

    p = subprocess.Popen(['sudo', '-S'] + command, stdin=subprocess.PIPE, stderr=subprocess.PIPE, universal_newlines=True)
    sudo_prompt = p.communicate(sudo_password + '\n')[1]
    print(sudo_prompt)
    print('user created')


def userdel(username):
    """
    remove user
    -r option to remove the users's home directory and mail spool
    """
    #sudo_password = input("Please enter your Password: ")
    sudo_password = getpass.getpass()
    command = ("userdel -r " + username).split()

    p = subprocess.Popen(['sudo', '-S'] + command, stdin=subprocess.PIPE, stderr=subprocess.PIPE, universal_newlines=True)
    sudo_prompt = p.communicate(sudo_password + '\n')[1]
    print(sudo_prompt)
    print('user removed')


def main():
    print('Enter 1 to create new user.')
    print('Enter 2 to remove a user.')
    option = input('Option: ')

    if option is "1":
        username = input('Enter a new username: ')
        user_password = getpass.getpass('Enter a password for the new user: ')
        useradd(username, user_password)
    elif option is "2":
        username = input('Enter a username: ')
        userdel(username)
    else: main()


if __name__ == "__main__":
    main()