Programmers_LV0_로그인성공?

리스트 조회에 관한 문제, for문으로 돌려서 id와 pw가 같은 원소 있다면 바로 'login' 리턴, 해당 if문에 걸리지 않는다면, id는 같은 지 확인, id가 같다면, check_id = True로 만들고 for문이 모두 돌았다는건 첫번째 if문에 해당하는 원소 없다는 것으로 check_id boolean값 기준으로 wrong pw인지 fail인지 출력.

def solution(id_pw, db):
    check_id = False
    for i in db:
        if i[0] == id_pw[0] and i[1] == id_pw[1]:
            return 'login'
        if i[0] == id_pw[0]:
            check_id = True
                
            
    if check_id == True:
        return 'wrong pw'
    else:
        return 'fail'