python 增加用户

#!/usr/bin/python
import sys
import subprocess
import getopt
import pexpect
import string
import datetime

opts,args = getopt.getopt(sys.argv[1:], "ha:l:u:")
hostip = ""
hostip_list = ""
username = ""
useruid = ""
usergid = ""
user_NUG_list =[["pup",107,111],["zax",903,903],["hic",901,901],["apos",889,889],["admin",888,888],["dev",1231,1102],["sear",1233,1102],["hise",1102,1102],["kan",1107,1107],["tgo",1104,1104],["rcce",1234,1102],["pe_hid",1001,1001],["pe_hig",1002,1002],["nooc",890,890],["tte_1",1271,1106],["rocs",892,892],["t551",1272,100],["clus_hk",1115,1115],["mysql",60,112],["audde",1251,1251],["ttsic",1109,1109],["ttse",1255,1255],["bank",1110,1110],["vallmall",1111,1111],["semo",1116,1116],["imo",3112,1112],["aace",1114,1114],["ahd",1121,1121],["clus",1105,1105],["cc",1306,1306],["devbl",1235,1102],["fhpush",1213,1101],["hiud",1101,1101],["himg",1103,1103],["hiro",2103,1103],["iyoumax",1112,1112],["utep",1108,1108],["ute",1243,1108],["opw",1211,1101],["ocesec",891,891],["asdsus",1305,1305],["skkge",1261,1110],["sepue",1262,1110],["db_root",2110,1110],["acefe",1237,1237],["pklsim",1236,1236],["web_admin",1263,1110],["tk_edu",1239,1239],["bi_gov",1245,1245],["test",3333,3333],["nsp_dev",1999,1999]]

user_N_list = [ n for n,u,g in user_NUG_list ]
user_U_list = [ u for n,u,g in user_NUG_list ]
user_G_list = [ g for n,u,g in user_NUG_list ]

def usage():
    print "usage: %s -l hostip.list -u username\n or \nusage: %s -a hostip -u username" %(sys.argv[0],sys.argv[0])

def get_userinfo(Uid,Gid):
    Uid = input('Please input User UID(1000~60000): ')
    Gid = input('Please input User GID(1000~60000): ')
    if (isinstance(Uid,int) and isinstance(Gid,int)):
        return Uid,Gid
    else:
        print "UID or GID was no number!!"
        sys.exit()

def ssh_cmd(hostip):
    cmd = "groupadd -g %d %s;useradd -g %d -u %d -m %s" %(usergid,username,usergid,useruid,username)
    ssh = pexpect.spawn('ssh %s "%s"' %(hostip,cmd))
    ssh.sendline(cmd)
    r = ssh.read()
    ssh.close()
    print "*** %s information:\n%s***************" %(hostip,r)
    now = datetime.datetime.now().strftime('%Y-%m-%d %H:%M')
    file_log = open('User_add.log','a')
    file_log.write(str(hostip)+' '+str(now)+' add user:'+str(username)+' gid:'+str(usergid)+' uid:'+str(useruid)+'\n')
    file_log.close()


def add_user(hostip):
    if hostip_list != "":
        try:
            f_list = open(hostip_list,'r')
        except IOError,e:
            print '*** file open error ',e
            sys.exit()
        else:
            for hostip in open(hostip_list).readlines():
                ssh_cmd(hostip)
    else:
        ssh_cmd(hostip)


if "-h" in sys.argv:
        usage()
        sys.exit()
elif len(sys.argv) != 5 or "-u" not in sys.argv:
        print "Options error"
        usage()
        sys.exit()


for op,value in opts:
    if op == "-a":
        hostip = value
    elif op == "-u":
       username = value
    elif op == "-l":
        hostip_list = value

if username in user_N_list:
    useruid = user_U_list[user_N_list.index(username,)]
    usergid = user_G_list[user_N_list.index(username,)]
    add_user(hostip)
else:
    useruid,usergid = get_userinfo(useruid,usergid)
    add_user(hostip)