Python(ATM机low版)

import os
class ATM:
    @staticmethod
    def regst():
        while 1:
            nm = input('请输入你的名字：')
            mm = input('请输入你的密码：')
            mm1 = input('请再次输入你的密码：')
            if mm == mm1:
                print('注册成功')
                money=input('输入你存的钱：')
                user=User(nm,mm,money)
                f1=open('zhanghu.txt',mode='a',encoding='utf-8')
                f1.write(nm+' '+mm+' '+money)
                f1.write('
')
                f1.flush()
                f1.close()
                break
            else:
                print('两次密码不一样,重新注册')

    def quqian(self, status): # status  0 , 1
        money2 = int(input('请输入你想取的钱：'))
        f4 = open('zhanghu.txt', mode='r', encoding='utf-8')
        f5 = open('zhanghu1.txt', mode='w', encoding='utf-8')
        for line in f4:
            if line.split()[0]==self.name and line.split()[1]==self.mima:
                if status == 0:
                    self.qian =int(line.split()[2])+int(money2)
                else:
                    self.qian = int(line.split()[2]) - int(money2)
                print('你的账户还有%s元'%self.qian)
                f5.write(self.name + ' ' + self.mima + ' ' + str(self.qian))
                f5.write('
')
            else:
                f5.write(self.name + ' ' + self.mima + ' ' + self.qian)
                f5.write('
')
        f4.flush()
        f5.flush()
        f4.close()
        f5.close()
        os.remove('zhanghu.txt')
        os.rename('zhanghu1.txt', 'zhanghu.txt')


class User(ATM):
    def __init__(self,name,mima,qian):
        self.name=name
        self.mima=mima
        self.qian=qian
while 1:
    n=input('注册输1，登录输2，退出输其他')
    if n=='1':
        fn=getattr(ATM,'regst')
        fn()
    elif n=='2':
        na1=input('输入你的名字：')
        mi1=input('输入你密码：')
        f6 = open('zhanghu.txt', mode='r', encoding='utf-8')
        for line in f6:
            if line.split()[0] == na1 and line.split()[1] == mi1:
                user=User(line.split()[0],line.split()[1],line.split()[2])
                f6.flush()
                f6.close()
                print('登录成功')
                while 1:
                    n1=input('存钱输1，取钱输2，查询输3，退出输其他')
                    if n1=='1':
                        user.quqian(0)
                    elif n1=='2':
                        user.quqian(1)
                    elif n1=='3':
                        user.chaxun()
                    else:
                        break
                break
        else:
            print('登录不成功')