需求:
批量新建10个测试用户,并且让其密码随机,把用户名和密码写入一个文件,并有创建时间和创建者
#!/usr/bin/python # -*- coding: utf-8 -*- import os import random import time import io def deluser(nums): '''删除用户测试用户''' for i in range(nums): username = 'stu' + str(i) linux_cmd = 'userdel -r {username}'.format(username=username) cmd_stat = os.system(linux_cmd) if cmd_stat == 0: print(username + " userdel: OK") else: print(username + " userdel: FAIL") def create_user(nums, record_adduser): '''添加用户测试用户''' if isinstance(nums, str): nums = int(nums) symbol = '1234567890' if not os.path.exists(os.path.dirname(record_adduser)): # 没有这个目录创建 os.mkdir(os.path.dirname(record_adduser)) # 这个文件直接打开,没有就创建 f = io.open(record_adduser, 'a+', encoding='utf-8') for i in range(nums): passwd = ''.join(random.sample(symbol, 6)) username = 'stu' + str(i) linux_cmd = 'useradd {username} && echo "{passwd}" | passwd {username} --stdin{linesep}'.format(username=username, passwd=passwd, linesep=os.linesep) current_user = os.popen('whoami').read().strip() cmd_stat = os.system(linux_cmd) tmp = str(time.strftime('%Y-%m-%d %H:%M ', time.localtime()) + ' ' + current_user + ' ' + linux_cmd + ' {stat}') if cmd_stat == 0: print(username + " useradd: OK") tmp = tmp.format(stat='OK').decode('utf-8') # 创建成功写一句 f.write(tmp) else: print(username + " useradd: FAIL") tmp = tmp.format(stat='FAIL').decode('utf-8') f.write(tmp) # f.flush() f.close() if __name__ == '__main__': record_adduser = '/root/adduser{sep}useradd.log'.format(sep=os.path.sep) create_user(10, record_adduser) # deluser(10) # python 2 下,str 是 bytes类型,文件通过 io模块打开