Python的生成用户练习案例

#!/usr/local/bin/python3
'Python的生成用户练习案例'

import string
import subprocess
import random
import os

# 交互获取用户名，返回可用用户名
def get_un():
    while 1 :
        u_name = input('用户名')
        result = subprocess.run('id %s >/dev/null'%u_name,shell=True)
        if result.returncode == 0 :
            print('存在')
            continue
        print('创建 %s 成功'%u_name)
        return u_name

# 交互获取文件名，返回可用文件名
def get_fn() :
    while 1 :
        f_name = input('文件名：')
        if os.path.exists(f_name) :
          print('存在')
        return  f_name

# 生成随机密码，默认8位，返回8位随机字符串
def ran_pass(n=8):
    result = ''
    ran_list = string.digits + string.ascii_letters
    for i in range(n) :
        char = random.choice(ran_list)
        result += char
    return result

# 创建用户并设置密码，返回None
def cu_sp(u_name,passwd) :
    subprocess.run('useradd %s'%u_name,shell=True)
    subprocess.run('echo %s | passwd --stdin %s'%(passwd,u_name),shell=True)
    return

# 将用户信息写入文件中
def w_file(u_name,passwd,f_name) :
    f_obj = open(f_name,'a')
    f_obj.write('''用户：%s 
密码：%s
'''%(u_name,passwd))
    f_obj.close()

if __name__ == '__main__':
    u_name = get_un()
    f_name = get_fn()
    passwd = ran_pass()
    cu_sp(u_name,passwd)
    w_file(u_name,passwd,f_name)
    ```