python 银行系统 user.py 部分

python
import os
import pickle


# 用户的类
class User:
    def __init__(self, uid, name, card):
        self.uid = uid      # 身份证
        self.name = name    # 用户名
        self.card = card    # 银行卡

    def __str__(self):
        return '姓名:{},身份证:{},卡号:{}'.format(self.name, self.uid, self.card.cid)

    # 将用户对象保存到文件中
    @staticmethod
    def save_user(userinfo):
        pathname = os.path.join(os.getcwd(), 'userinfo.db')
        with open(pathname, 'wb') as fp:
            pickle.dump(userinfo, fp)

    # 从文件中加载数据重新转换为用户对象
    @staticmethod
    def load_user():
        pathname = os.path.join(os.getcwd(), 'userinfo.db')
        if os.path.exists(pathname):
            with open(pathname, 'rb') as fp:
                userinfo = pickle.load(fp)
            return userinfo
        else:
            return {}

猜你喜欢

转载自blog.csdn.net/qq_42795565/article/details/81461761