[프로그래머스] 로그인 성공? - 파이썬

문제

https://school.programmers.co.kr/learn/courses/30/lessons/120883

프로그래머스

설계

def solution(id_pw, db):
    answer = 'fail'
    for i in range(len(db)):
        if id_pw[0] == db[i][0] and id_pw[1] == db[i][1]:
            answer = 'login'
        elif id_pw[0] == db[i][0] and id_pw[1] != db[i][1]:
            if answer == 'fail':
                answer = 'wrong pw'
    return answer