ATM流程的实现

ATM
1. 管理员可添加账户、指定用户额度、冻结用户等
2. 指定最大透支额度
3. 可存取款
4. 定期还款（每月指定日期还款，如15号）
5. 定期出账单
6. 支持多用户登陆，用户间转帐

# -*- coding: utf-8 -*-
'''
Author:SL
'''
import os,time

#开始登录界面
def login():
    welcome ='''
----欢迎使用自动取款机系统-----
操作指南：
    【1】用户登录
    【2】管理员登录
    【3】退出系统 
    '''
    while True:
        print(welcome)
        choice_num =input('请输入指令序号：').strip()
        if choice_num.isdigit() and choice_num in ['1','2','3']:
            return choice_num
        else:
            print('您的输入指令有误，请重新选择！')
            continue

#用户登录
def user_login():
    while 1:
        user_name = input('请输入用户名：\n>>>').strip()
        user_pwd = input('请输入密码：\n>>>').strip()
        with open('user_locked.txt','r',encoding='utf8') as f1:
            for line in f1:
                if user_name in line:
                    print('该账户处于冻结状态，请联系管理员！')
                    return 0
        with open('user_info.txt','r',encoding='utf8') as f2:
            for line in f2:
                try:
                    line_list =line.split(':')
                    if line_list[1].startswith(user_name) and line_list[2].startswith(user_pwd):#验证用户名和密码
                        print('用户登录成功！')
                        return user_name
                    else:
                        print('账户名或密码错误，请重新输入！')
                        continue
                except Exception:
                    print('账户名或密码错误，请重新输入！')

#管理员登录
def root_login():
    '''
    系统默认管理员名字：root  密码：123
    '''
    while True:
        user_name = input('请输入管理员用户名，默认root【Q退出】:\n>>>').strip()
        if user_name.lower() == 'q':
            return 0                     #退出返回0
        password = input('请输入管理员密码，默认123:\n>>>').strip()
        if user_name.lower() == 'root' and password == '123':
            print('您已获得管理者权限！')
            return 1                     #成功返回1
        else:
            print('请重新输入！')
            continue

#用户登录操作说明
def user_welcome():
    user_welcome ='''
    操作说明:
    【取款】：1  【余额查询】：4
    【存款】：2  【定期还款】：5
    【转账】：3  【显示账单】：6
    【返回】：R
    '''
    print(user_welcome)

#管理员操作说明
def root_welcome():
    root_welcome ='''
    操作说明：
    【添加用户】：1
    【冻结用户】：2
    【设定额度】：3
    【返回】：R
    '''
    print(root_welcome)

#取款操作
def withdrawal(user_name):
    while True:
        get_money =input("请输入取款数额【返回R】：\n>>>").strip()
        if get_money.lower() =='r':
            break
        elif get_money.isdigit():
                f1 = open('user_info.txt', 'r', encoding='utf8')
                f2 = open('new_user_info.txt', 'w', encoding='utf8')
                f1.seek(0)
                for line in f1:
                    if user_name in line:
                        if int(line.split(':')[3][:-5]) >int(get_money):
                            balance =int(line.split(':')[3][:-5])-int(get_money)
                            line =line.replace(str(line.split(':')[3][:-5]),str(balance))
                            print('取款成功！余额为%s'%balance)
                        else:
                            print('余额不足！')
                    f2.write(line)
                f1.close()
                f2.close()
                os.remove('user_info.txt')
                os.rename('new_user_info.txt','user_info.txt')
        else:
            print('请输入正确的金额！')

#存款操作
def deposit(user_name):
    while True:
        give_money =input("请输入存款数额【返回R】：").strip()
        if give_money.lower() =='r':
            break
        elif give_money.isdigit():
                f1 = open('user_info.txt', 'r', encoding='utf8')
                f2 = open('new_user_info.txt', 'w', encoding='utf8')
                f1.seek(0)
                for line in f1:
                    if user_name in line:
                        balance =int(line.split(':')[3][:-5])+int(give_money)
                        line =line.replace(str(line.split(':')[3][:-5]),str(balance))
                        print('存款成功！余额为%s'%balance)
                    f2.write(line)
                f1.close()
                f2.close()
                os.remove('user_info.txt')
                os.rename('new_user_info.txt','user_info.txt')
        else:
            print('请输入正确的金额！')

#转账操作
def trans_account(user_name):
    while True:
        flag = False
        trans_name =input('请输入收款人账户名【返回R】：\n>>>').strip()
        if trans_name.lower() =='r':
            break
        with open('user_info.txt', 'r', encoding='utf8') as f1:
            for line in f1:
                if trans_name in line and trans_name != user_name:
                    flag =True
        if flag:
            while True:
                trans_money = input("请输入转账数额【重选收款人R】：\n>>>").strip()
                if trans_money.lower() == 'r':
                    break
                elif trans_money.isdigit():
                    msg_insure =input('【确定Y】【取消N】>>>')
                    if msg_insure.lower() =='y':
                        f1 = open('user_info.txt', 'r', encoding='utf8')
                        f2 = open('new_user_info.txt', 'w', encoding='utf8')
                        for line in f1:
                            if user_name in line:
                                balance = int(line.split(':')[3][:-5]) - int(trans_money)
                                line = line.replace(str(line.split(':')[3][:-5]), str(balance))
                            if trans_name in line:
                                balance = int(line.split(':')[3][:-5]) + int(trans_money)
                                line = line.replace(str(line.split(':')[3][:-5]), str(balance))
                            f2.write(line)
                        f1.close()
                        f2.close()
                        os.remove('user_info.txt')
                        os.rename('new_user_info.txt', 'user_info.txt')
                        print('转账成功！')
                    elif msg_insure.lower() =='n':
                        continue
                    else:
                        print('您的操作有误，请重新选择收款人！')
                        break
                else:
                    print('请输入正确的金额！')
        else:
            print('收款人不存在，请重新输入！')

#余额查询
def balance_query(user_name):
    with open('user_info.txt', 'r', encoding='utf8') as f:
        for line in f:
            if user_name in line:
                balance = int(line.split(':')[3][:-5])
                print('您的余额为%s!'%balance)

#定期还款
def reg_payment(user_name):
    while True:
        with open('user_info.txt', 'r', encoding='utf8') as f:
            for line in f:
                if user_name in line:
                    debt = int(line.split(':')[-1])
                    print('您的欠款为%s!' % debt)
        set_money =input('请输入您的每次分期还款金额【返回R】：\n>>>').strip()
        if set_money.lower() =='r':
            break
        set_freq =input('请输入您的分期还款次数（分期金额*次数应小于欠款数额，否则后果自负！）：\n>>>').strip()
        count = 0
        if set_money.isdigit() and set_freq.isdigit():
            for i in range(int(set_freq)):
                f1 = open('user_info.txt', 'r', encoding='utf8')
                f2 = open('new_user_info.txt', 'w', encoding='utf8')
                for line in f1:
                    if user_name in line:
                        debt = int(line.split(':')[-1]) - int(set_money)
                        line = line.replace(str(line.split(':')[-1]), str(debt)+'\n')
                        balance = int(line.split(':')[3][:-5]) - int(set_money)
                        line = line.replace(str(line.split(':')[3][:-5]), str(balance))
                    f2.write(line)
                f1.close()
                f2.close()
                os.remove('user_info.txt')
                os.rename('new_user_info.txt', 'user_info.txt')
                count +=1
                time.sleep(5)  # 设定分期还款间隔时间，这里默认每隔5秒还款一次
                print('已经还款%s期！'%count)
            print('分期还款完成！')
            with open('user_info.txt', 'r', encoding='utf8') as f:
                for line in f:
                    if user_name in line:
                        debt = int(line.split(':')[-1])
                        print('您剩余欠款为%s!' % debt)
                        return
        else:
            print('请重新输入！')

#显示账单
def debt_query(user_name):
    with open('user_info.txt', 'r', encoding='utf8') as f:
        for line in f:
            if user_name in line:
                balance = int(line.split(':')[3][:-5])
                debt = int(line.split(':')[-1])
                print('您的账户信息为余额：%s，欠款：%s!'%(balance,debt))

#添加用户
def add_user():
    while True:
        add_flag =True
        new_user_name =input('请输入注册账户名【返回R】:\n>>>').strip()
        if new_user_name.lower() == 'r':
            break
        new_user_pwd =input('请输入密码【返回R】:').strip()
        if new_user_pwd.lower() =='r':
            break
        else:
            f1 =open('user_info.txt', 'r+', encoding='utf8')
            f2 =open('user_locked.txt', 'r', encoding='utf8')
            for line in f2:
                if new_user_name in line:
                    print('该账户已经被冻结，请重新输入账户名！')
                    add_flag =False
            if add_flag:
                for line1 in f1:
                    if new_user_name in line1:
                        print('该账户已经存在，请重新输入账户名！')
                        break
                else:
                    f1.write('\nname:'+new_user_name+',password:'+new_user_pwd+',balance:10000,debt:0')
                    print('新用户注册成功！')
            f1.close()
            f2.close()

#冻结用户
def frozen_user():
    while 1:
        frozen_flag = False
        lock_flag = True
        set_frozen_user = input('请输入要冻结的账户名【返回R】:').strip()
        if set_frozen_user.lower() == 'r':
            break
        with open('user_info.txt', 'r', encoding='utf8') as f:
            for line in f:
                if set_frozen_user in line:
                    frozen_flag = True
        if frozen_flag:
            with open('user_locked.txt', 'r+', encoding='utf8') as f1:
                for line in f1:
                    if set_frozen_user in line:
                        print('该账户已经处于冻结状态，无需再次冻结！')
                        lock_flag = False
                if lock_flag:
                    f1.write(set_frozen_user + '\n')
                    print('成功冻结该账户！')
        else:
            print('对不起，该用户不存在！')
            
#设定用户额度
def set_account():
    pass

#用户操作选择
def user_switch(user_name):
    while 1:
        user_welcome()
        case = input('请输入要进行操作的代号：').strip()
        if case == '1':
            withdrawal(user_name)
        elif case == '2':
            deposit(user_name)
        elif case == '3':
            trans_account(user_name)
        elif case == '4':
            balance_query(user_name)
        elif case == '5':
            reg_payment(user_name)
        elif case == '6':
            debt_query(user_name)
        elif case.lower() == 'r':
            break
        else:
            print('您的输入有误，请重新选择！')

#管理员操作选择
def root_switch():
    while 1:
        root_welcome()
        case = input('请输入要进行操作的代号：').strip()
        if case == '1':
            add_user()
        elif case == '2':
            frozen_user()
        elif case == '3':
            set_account()
        elif case.lower() == 'r':
            break
        else:
            print('您的操作有误，请重新选择！')

#主程序入口
if __name__ == '__main__':
    while True:
        choice_num = login()
        if choice_num == '3':
            print('退出成功，感谢您的使用！')
            break
        if choice_num == '1':
            res = user_login()
            if res == 0:
                continue
            else:
                user_switch(res)
        if choice_num == '2':
            res = root_login()
            if res == 0:
                continue
            if res == 1:
                root_switch()