import os
from random import choice
import string
import subprocess #subprocess可以直接运行linux命令
def ran_passwd():
ran_range = string.ascii_letters + string.digits #大小写字母和数字
password = []
for i in range(6):
p = choice(ran_range)
password.append(p)
return ''.join(password) #把列表中的元素拼接到一起,形成字符串密码
def add_user():
while True:
name = input('用户名:').strip()
rc = subprocess.run('id %s &> /dev/null' % name, shell=True)
if rc.returncode == 0: #用户存在返回0,不存在返回非0值
print('用户名已存在!')
continue
else:
subprocess.run('useradd %s' % name, shell=True)
break
password = ran_passwd()
subprocess.run('echo %s | passwd --stdin %s' % (password, name), shell=True)
if __name__ == '__main__':
add_user()