利用Python創建用戶

2.1 問題
創建adduser.py文件,實現以下目標:

編寫一個程序,實現創建用戶的功能
提示用戶輸入用戶名
隨機生成8位密碼
創建用戶並設置密碼
將用戶相關信息寫入指定文件
2.2 方案
創建add_user()函數,讓函數具有創建用戶、創建密碼、將用戶密碼寫入到指定文件三種方法,因此爲函數設置3個參數,分別是用戶名、密碼及用戶名密碼存放文件,最終通過函數調用上傳實參的方式,完成用戶創建

1.利用subprocess.call函數運行用戶創建命令

2.subprocess.call函數運行密碼設置命令

3.用with語句將指定的文件以追加模式打開,這樣打開代碼塊結束後文件會自動關閉,將用戶密碼用指定格式寫入指定文件

4.調用add_user()函數時上傳的用戶名實參,是利用sys.argv[]參數,在命令行調用的時候由系統傳遞給程序,這個變量其實是一個List列表,用於保存命令行上的參數,argv[0] 一般是“被調用的腳本文件名或全路徑”,argv[1]和以後就是傳入的系統命令參數

2.3 步驟
實現此案例需要按照如下步驟進行。

步驟一:編寫腳本

將randpass文件的代碼以模塊形式導入以下代碼中,直接調用gen_pass()函數獲取返回值(即獲取隨機生成的密碼):

[root@localhost day04] # vim adduser.py
#!/usr/bin/env python3
import sys
import subprocess
from randpass import gen_pass 
def add_user(username, password, fname):
    info = """user information:
username: %s
password: %s
"""
    subprocess.run('useradd %s' % username, shell=True)
    subprocess.run(
        'echo %s | passwd --stdin %s' % (password, username),
        shell=True
    )                                                                                                           
    with open(fname, 'a') as fobj:
        fobj.write(info % (username, password))
if __name__ == '__main__':
    username = sys.argv[1]
    password = gen_pass()
    fname = '/tmp/users.txt'
    add_user(username, password, fname) 

步驟二:測試腳本執行

[root@localhost day04]# python3 adduser.py b c d 
更改用戶 b 的密碼 。
passwd:所有的身份驗證令牌已經成功更新。
[root@localhost day04]# python3 adduser.py a c d 
useradd:用戶“a”已存在
更改用戶 a 的密碼 。
passwd:所有的身份驗證令牌已經成功更新。
[root@localhost day04]# cat /tmp/users.txt
user information:
username: a
password: hD31SmTS
user information:
username: b
password: DztS7ycn
user information:
username: a
password: f2iH0Znt
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章