쇼핑몰[장바구니]

# 쇼핑몰[장바구니]
#  1. 로그인 후 쇼핑 메뉴를 선택하면, 다음과 같이 상품목록을 보여준다.
#  1) 사과
#  2) 바나나
#  3) 딸기
#
#  2. 번호를 선택해 상품을 장바구니에 담을 수 있다.
#  3. 로그인 회원의 인덱스 번호는 각 행의 첫번째 열에 저장한다.
#  4. 해당 회원이 구매한 상품의 인덱스 번호는 각 행의 두번째 열에 저장한다.
#  예)
#  [
#  		[0, 1],				qwer회원 			> 사과구매
#  		[1, 2],				javaking회원 		> 바나나구매
#  		[2, 1],				abcd회원			> 사과구매
#  		[0, 3],				qwer회원			> 딸기구매
#  		...
#  ]
ids = ["qwer", "pythonking", "abcd"]
pws = ["1111", "2222", "3333"]
items = ["사과", "바나나", "딸기"]
jang = [[0] * 2 for i in range(100)]
count = 0
log = -1
while True:
    print("[MEGA MART]")
    print("[1]로 그 인")
    print("[2]로그아웃")
    print("[3]쇼    핑")
    print("[4]장바구니")
    print("[0]종    료")
    sel = int(input("메뉴 선택 : "))
    if sel == 1:
        if log != -1:
            print("이미 로그인되어있습니다.")
            continue
        my_id = input("input your ID: ")
        my_pw = input("input your PW: ")
        count = 0
        for i in range(len(ids)):
            if my_id == ids[i] and my_pw == pws[i]:
                log = i
                print(ids[i],"님, 안녕하세요! 로그인되었습니다.")
                count = 1
                break
        if count == 0:
            print("아이디 혹은 패스워드 오류입니다.")
    elif sel == 2:
        if log == -1:
            print("이미 로그아웃 되어있습니다.")
            continue
        print(ids[log],"님, 정상적으로 로그아웃 되었습니다.")
        log = -1
    elif sel == 3:
        if log == -1:
            print("로그인 후 이용해주세요.")
            continue
        while True:
            print("---Items---")
            i = 0
            while i < len(items):
                print("[%d] %s"%(i+1,items[i]))
                i += 1
            print("[4] 뒤로가기")
            cho = int(input("input item index: "))
            if cho == 4:
                break
            jang[count][0] = log
            jang[count][1] = cho
            count += 1
    elif sel == 4:
        if log == -1:
            print("로그인 후 이용해주세요.")
            continue
        apple = 0
        banana = 0
        strawberry = 0
        i = 0
        while i < count:
            if jang[i][0] == log:
                if jang[i][1] == 1:
                    apple += 1
                elif jang[i][1] == 2:
                    banana += 1
                elif jang[i][1] == 3:
                    strawberry += 1
            i += 1
        print("apple: ",apple)
        print("banana: ",banana)
        print("strawberry: ",strawberry)
    elif sel == 0:
        print("프로그램 종료")
        break