[Python] 계좌관리 프로그램

[개요]

파이썬으로 간단하게 계좌관리 프로그램을 만들어보았습니다.

계좌번호를 랜덤으로 주되, 겹쳐지지 않도록 만들었습니다.

입금 및 출금 처리시 계좌번호를 먼저 묻고 해당 계좌의 존재유무를 묻습니다.

둘 이상의 고객에 대한 정보를 담아서 관리할 수 있습니다.

[실행창]

 

[코드]

import random
user = []
def start():
    while True:
        print("[Bank Menu]")
        print("1. 계좌개설\n2. 입금처리\n3. 출금처리\n4. 전체 잔액현황\n5. 프로그램 종료")
        a = int(input("선택 = "))
        if a == 1:
            print("================================================")
            print("[계좌개설]")
            user.append(Account())
            print("계좌가 계설되었습니다.")
            print("================================================")
        elif a == 2:
            print("================================================")
            print("[입금처리]")
            _id = input("입금할 계좌번호를 입력하세요")
            errorcheck(_id)
            mini_db(_id)
            ad = int(input("입금할 금액을 입력하세요."))
            deposit(_id,ad)
            print("입금이 완료되었습니다.")
            mini_db(_id)
            print("================================================")
        elif a == 3:
            print("================================================")
            print("[출금처리]")
            _id = input("출금할 계좌번호를 입력하세요")
            print("계좌를 조회합니다")
            errorcheck(_id)
            mini_db(_id)
            su = int(input("출금할 금액을 입력하세요."))
            withdraw(_id,su)
            mini_db(_id)
            print("================================================")
        elif a == 4:
            print("================================================")
            print("[전체고객 잔액현황]")
            if len(user) != 0 :
                mini_db("all")
            else :
                print("결과가 없습니다.")
            print("================================================")
        elif a == 5:
            print("================================================")
            print("종료하시겠습니까?")
            print("1.예 2.아니오")
            endnum = int(input())
            if endnum == 1:
                break
            elif endnum == 2:
                print("================================================")
            else:
                print("입력형식이 잘못되었습니다.")
                print("================================================")
            
        else:
            print("다시 입력하세요")
            print("================================================")

def errorcheck(_id):
    try:
        co = -1
        for i in range(0, len(user)+1):
            co +=1
            if user[i]._id == _id :
                break
        if user[co]._id == _id :
            print("계좌를 조회합니다")
    except:
        print("없는 계좌번호입니다.")
        start()
        

def deposit(_id,ad):
    co = -1
    for i in range(0, len(user)+1):
        co +=1
        if user[i]._id == _id :
            break
    if user[co]._id == _id :
            user[i].balance += ad

def withdraw(_id,su):
    co = -1
    for i in range(0, len(user)+1):
        co +=1
        if user[i]._id == _id :
            break
    if user[co]._id == _id :
            if user[i].balance >= su:
                user[i].balance -= su
                print("출금이 완료되었습니다.")
            else:
                print("금액이 모자랍니다")

def mini_db(_id):
    if _id == "all":
        for i in range(0, len(user)):
            print(i+1,end=" ")
            user[i].disp()
        return
    else:
        for i in range(0, len(user)):
            if user[i]._id == _id :
                print(i+1,end=" ")
                user[i].disp()
        return True
    return False
    

class Account:
    def __init__(self):
        ran_num = ["0", "0", "0","0","0","0","0","0","0","0","0","0","0","0"]
        for i in range(14):
            ran_num[i] = random.randrange(1,9,1)
        ran_num[0]='1'
        ran_num[1]='1'
        ran_num[2]='0'
        ran_num[3]='-'
        ran_num[7]='-'
        self._id=""
        for i in range(14):
            self._id += str(ran_num[i])
        for i in range(0, len(user)):
            if user[i]._id == self._id :
                user.append(Account())
        print(self._id, sep="")
        self.name= input("이름 입력 : ")
        self.balance = int(input("입금 금액 입력 : "))
    def set_id(self, _id):
        self._id = _id
    def setname(self, name):
        self.name = name
    def setbalance(self, balance):
        self.balance = balance
    def disp(self):
        print(self._id," ",self.name," ",self.balance)
        


if __name__ == "__main__":
    start()