一、Pickle簡(jiǎn)介：

• 把數(shù)據(jù)寫入文件時(shí)，常規(guī)的文件方法只能把字符串對(duì)象寫入。其他數(shù)據(jù)需先轉(zhuǎn)換成字符串再寫入文件
• Pickle可以將其他類型的數(shù)據(jù)寫入文件，之后又可以把它完整無(wú)缺地取出來(lái)

補(bǔ)充：

• 常規(guī)方法寫入數(shù)據(jù)，只能是字符串類型，其他類型無(wú)法寫入，例如：int，字典，列表等類型；pickle模塊可以在文件中存儲(chǔ)任何類型的數(shù)據(jù)，也可以完整取出任何類型的數(shù)據(jù)；

二、Pickle模塊常用函數(shù)：

三、使用示例

import pickle

# 將其他類型的數(shù)據(jù)寫入文件
list01 = [1, 2, "213", "he", {213: "23", "sad": 23}]
with open("/opt/list.data", mode="wb") as fw:
    pickle.dump(list01, fw)

with open("/opt/list.data", mode="rb") as fr:
    list_data = pickle.load(fr)

print(list_data)

四、練習(xí)：創(chuàng)建系統(tǒng)用戶

需求：

• 編寫一個(gè)程序(函數(shù))，實(shí)現(xiàn)創(chuàng)建用戶的功能
• 提示用戶輸入 用戶名
• 隨機(jī)生成 8位密碼 (導(dǎo)入之前的模塊文件)
• 創(chuàng)建用戶并設(shè)置密碼
• 將用戶相關(guān)信息存儲(chǔ)文件內(nèi)，并打印

import pickle
import subprocess
import randpass as rp

def create_user(username, file_name):
    result = subprocess.run(f'id {username} &> /dev/null', shell=True)
    if result.returncode == 0:
        print(f"用戶{username}已存在，請(qǐng)重新輸入！")
    else:
        # 隨機(jī)生成8位密碼
        password = rp.get_pass()
        print(f"您的用戶名:{username},您的密碼:{password}")
        # 創(chuàng)建用戶
        subprocess.run(f"useradd {username}", shell=True)
        # 設(shè)置密碼
        subprocess.run(f"echo {password} | passwd --stdin {username}", shell=True)

        # 將用戶相關(guān)信息存儲(chǔ)文件
        with open(file_name, mode="rb") as fr:
            userdb = pickle.load(fr)
        userdb[username] = password
        with open(file_name, mode="wb") as fw:
            pickle.dump(userdb, fw)
            print("userdb:", userdb)


if __name__ == '__main__':
    # 提示用戶輸入用戶名
    user = input("輸入用戶名：\n")
    create_user(user, "/opt/user_db")

• randpass生成隨機(jī)密碼類

import random as rd
from string import digits as dg, ascii_letters as al


# 隨機(jī)生成密碼
def get_pass(pass_number=8):
    passwd = ""  # 最終密碼
    for item in range(pass_number):
        # 密碼拼接
        passwd += rd.choice(dg + al)
    return passwd


if __name__ == '__main__':
    print(get_pass())