Python模块·Pickle读写文件（不同类型）

一、Pickle简介：

• 把数据写入文件时，常规的文件方法只能把字符串对象写入。其他数据需先转换成字符串再写入文件
• Pickle可以将其他类型的数据写入文件，之后又可以把它完整无缺地取出来

补充：

• 常规方法写入数据，只能是字符串类型，其他类型无法写入，例如：int，字典，列表等类型；pickle模块可以在文件中存储任何类型的数据，也可以完整取出任何类型的数据；

二、Pickle模块常用函数：

三、使用示例

import pickle

# 将其他类型的数据写入文件
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)

四、练习：创建系统用户

需求：

• 编写一个程序(函数)，实现创建用户的功能
• 提示用户输入 用户名
• 随机生成 8位密码 (导入之前的模块文件)
• 创建用户并设置密码
• 将用户相关信息存储文件内，并打印

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}已存在，请重新输入！")
    else:
        # 随机生成8位密码
        password = rp.get_pass()
        print(f"您的用户名:{username},您的密码:{password}")
        # 创建用户
        subprocess.run(f"useradd {username}", shell=True)
        # 设置密码
        subprocess.run(f"echo {password} | passwd --stdin {username}", shell=True)

        # 将用户相关信息存储文件
        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生成随机密码类

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


# 随机生成密码
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())