python学习之ATM功能实现

Posted 2021-03-04 python小小人

# ：编写ATM程序实现下述功能，数据来源于文件db.txt
# 1、充值功能：用户输入充值钱数，db.txt中该账号钱数完成修改
# 2、转账功能：用户A向用户B转账1000元，db.txt中完成用户A账号减钱，用户B账号加钱
# 3、提现功能：用户输入提现金额，db.txt中该账号钱数减少
# 4、查询余额功能：输入账号查询余额

# ：登录功能
# 用户登录成功后，内存中记录下该状态，上述功能以当前登录状态为准，必须先登录才能操作
import os

user_staus = {‘username‘: None}


def ad_credit(username, amount):
    """
    充值功能
    :param username:
    :param amount:
    :return:
    """
    if user_staus[‘username‘] is None:
        login()
    with open(‘db.txt‘, ‘rt‘, encoding=‘utf-8‘)as f1, 
            open(‘db.txt.swap‘, ‘wt‘, encoding=‘utf-8‘)as f2:
        while True:
            cont = f1.readline()
            if len(cont) == 0:
                break
            name, remain = cont.strip().split(‘:‘)
            if username in name:
                remain = int(remain) + int(amount)
                f2.write(‘{}:{}
‘.format(name, remain))
                print(‘充值成功，{}的余额为{}‘.format(username, remain))
            else:
                f2.write(cont)
    os.remove(‘db.txt‘)
    os.rename(‘db.txt.swap‘, ‘db.txt‘)


def transfer(user_out, user_in, amount):
    ‘‘‘3
    转账功能
    :param user_out:
    :param user_in:
    :param amount:
    :return:
    ‘‘‘
    if user_staus[‘username‘] is None:
        login()
    with open(‘db.txt‘, ‘rt‘, encoding=‘utf-8‘)as f1, 
            open(‘db.txt.swap‘, ‘wt‘, encoding=‘utf-8‘)as f2:

        userinfo = {}
        for line in f1:
            name, remind = line.strip().split(‘:‘)
            userinfo[name] = int(remind)
        if user_out not in userinfo:
            print(‘用户不存在‘)
            return
        if user_in not in userinfo:
            print(‘收款方不存在‘)
            return
        if user_out in userinfo and user_in in userinfo:
            if userinfo[user_out] >= int(amount):
                userinfo[user_out] -= int(amount)
                userinfo[user_in] += int(amount)
                print(‘转账成功，已成功从{}向{}汇款{}‘.format(user_out, user_in, amount))
            elif userinfo[user_out] < amount:
                print(‘余额不足‘)
                return
        for name, remind in userinfo.items():
            f2.write(‘{}:{}
‘.format(name, remind))

    os.remove(‘db.txt‘)
    os.rename(‘db.txt.swap‘, ‘db.txt‘)


def cashon(username, amount):
    ‘‘‘
    提现功能
    :param username:
    :param amount:
    :return:
    ‘‘‘
    if user_staus[‘username‘] is None:
        login()
    with open(‘db.txt‘, ‘rt‘, encoding=‘utf-8‘)as f1, 
            open(‘db.txt.swap‘, ‘wt‘, encoding=‘utf-8‘)as f2:
        userinfo = {}
        for line in f1:
            name, remind = line.strip().split(‘:‘)
            userinfo[name] = int(remind)
        if username not in userinfo:
            print(‘用户不存在‘)
            return
        if username in userinfo and userinfo[username] >= int(amount):
            userinfo[username] -= int(amount)
            print(‘已从余额中取出{}，现余额为{}‘.format(amount, userinfo[username]))
        elif userinfo[username] < amount:
            print(‘余额不足，提现失败‘)
            return
        for name, remind in userinfo.items():
            f2.write(‘{}:{}
‘.format(name, remind))
    os.remove(‘db.txt‘)
    os.rename(‘db.txt.swap‘, ‘db.txt‘)


def check(username):
    ‘‘‘
    余额查询功能
    :param username:
    :return:
    ‘‘‘
    if user_staus[‘username‘] is None:
        login()
    with open(‘db.txt‘, ‘rt‘, encoding=‘utf-8‘)as f:
        userinfo = {}
        for line in f:
            name, remind = line.strip().split(‘:‘)
            userinfo[name] = remind
        if username not in userinfo:
            print(‘用户不存在‘)
            return
        if username in userinfo:
            print(‘当前余额为：{}‘.format(userinfo[username]))


def login():
    username = input(‘输入用户名‘)
    userpassword = input(‘输入密码‘)
    with open(‘login.txt‘, ‘rt‘, encoding=‘utf-8‘)as login_f:
        login = {}
        for line in login_f:
            name, psd = line.strip().split(‘:‘)
            login[name] = psd
            if username in login:
                if login[username] == userpassword:
                    print(‘登陆成功‘)
                    user_staus[‘username‘] = username
                    break
            elif username not in login:
                print(‘用户名不存在‘)
                return


def logout():
    user_staus[‘username‘] = None
    print(‘已成功登出‘)
    return


login()
tag = True
while tag:
    cmd = input(‘‘‘
    请输入你想使用的功能序号
    1：充值
    2：转账
    3：提现
    4：查询余额
    0：登出
    ‘‘‘)
    if cmd == ‘1‘:
        username = input(‘输入用户名：‘)
        amount = input(‘输入充值金额：‘)
        ad_credit(username, amount)
    elif cmd == ‘2‘:
        user_out = input(‘请输入转账方：‘)
        user_in = input(‘请输入接收方：‘)
        amount = input(‘输入转账金额：‘)
        transfer(user_out, user_in, amount)
    elif cmd == ‘3‘:
        username = input(‘输入用户名：‘)
        amount = input(‘输入提现金额‘)
        cashon(username, amount)
    elif cmd == ‘4‘:
        username = input(‘输入用户名：‘)
        check(username)
    elif cmd == ‘0‘:
        logout()
        tag = False
    else:
        print(‘请正确输入序号‘)

————————————————————————以下是模拟结果————————————————————————————————————————————
‘‘‘
/Users/chenfeng/PycharmProjects/ATM/venv/bin/python /Users/chenfeng/PycharmProjects/ATM/main.py
输入用户名xilou
输入密码666
登陆成功

    请输入你想使用的功能序号
    1：充值
    2：转账
    3：提现
    4：查询余额
    0：登出
    1
输入用户名：xilou
输入充值金额：200
充值成功，xilou的余额为700

    请输入你想使用的功能序号
    1：充值
    2：转账
    3：提现
    4：查询余额
    0：登出
    2
请输入转账方：xilou
请输入接收方：heiren
输入转账金额：200
转账成功，已成功从xilou向heiren汇款200

    请输入你想使用的功能序号
    1：充值
    2：转账
    3：提现
    4：查询余额
    0：登出
    3
输入用户名：xilou
输入提现金额100
已从余额中取出100，现余额为400

    请输入你想使用的功能序号
    1：充值
    2：转账
    3：提现
    4：查询余额
    0：登出
    4
输入用户名：xilou
当前余额为：400

    请输入你想使用的功能序号
    1：充值
    2：转账
    3：提现
    4：查询余额
    0：登出
    0
已成功登出

Process finished with exit code 0

‘‘‘