#!/usr/bin/env python

import sys
import pexpect

# Read variables of first and second given argument
user = str(sys.argv[1])
password = str(sys.argv[2])

# Define function
def _vadduser(user, password):
    adduser = pexpect.spawn(
        command='vadduser',  # actual shell command which should be started
        args=[user]   # necessary argument
    )

    adduser.expect('password:')
    adduser.sendline(password)

    adduser.expect('verification:')
    adduser.sendline(password)

    adduser.expect(pexpect.EOF)
    adduser.close()
    
    # Error handling
    if adduser.exitstatus != 0:
        last_line = filter(None, adduser.before.split('\r\n'))[-1]
        problem = last_line.split(':')[-1]
        print "ERROR! Code: " + str(adduser.exitstatus)
        print "Last error: " + str(problem)
    else:
        print "SUCCESS!"
        print "New user " + str(user) + " has been created with the password: " + str(password)

# Start function
_vadduser(user, password)