ATM[1단계]

# ATM[1단계]
# 1. 로그인
# . 로그인 후 재 로그인 불가
# . 로그아웃 상태에서만 이용 가능
# 2. 로그아웃
# . 로그인 후 이용가능

db_acc1 = 1111
db_pw1 = 1234

db_acc2 = 2222
db_pw2 = 2345

log = -1          # -1(로그아웃), 1(dbAcc1로그인), 2(dbAcc2로그인)

run = True
while run:
    print("1.로그인")
    print("2.로그아웃")
    print("0.종료")

    choice = int(input("메뉴 선택 : "))
    if choice == 1:
        ID = int(input("id를 입력하세요: "))
        PW = int(input("pw를 입력하세요: "))
        if ID==db_acc1 and PW==db_pw1:
            log=1
            print("db_acc1님, 환영합니다.")
        elif ID==db_acc2 and PW==db_pw2:
            log=2
            print("db_acc2님, 환영합니다.")
        else:
            print("id 혹은 pw 오류입니다.")
    elif choice == 2:
        if log == -1:
            print("이미 로그아웃 되어있습니다.")
        else:
            print("로그아웃 되었습니다.")
            log = -1
    elif choice == 0:
        run = False
        print("프로그램 종료")