每日python三分钟

41-创建用户,密码随机

2019-12-17  本文已影响0人  JSFnull

```

"""创建用户

这是一个用于创建用户的脚本,用到有4个函数

"""

import sys

import randpass

import subprocess

def add_user(user, passwd, fname):

    #如果用户已存在,则返回,不要继续执行函数

    result = subprocess.run(

        'id %s &> /dev/null' % user, shell=True

    )

    if result.returncode == 0 :

        print('用户已存在')

        #return默认返回None,类似于break,函数遇到return也会提前结束

        return

    # 创建用户, 设置密码

    subprocess.run(

        'useradd %s' % user, shell=True

    )

    subprocess.run(

        'echo %s | passwd --stdin %s' % (passwd,user),shell=True

    )

    #写入文件

    info = """用户信息:

    用户名: %s

    密码: %s

    """ % (user,passwd)

    with open(fname,'a') as fobj:

        fobj.write(info)

if __name__ == '__main__':

    user = sys.argv[1]

    passwd = randpass.mk_pass2()

    fname = sys.argv[2]

    add_user(user,passwd,fname)

```

上一篇 下一篇

猜你喜欢

热点阅读