[프로그래머스] 신규 아이디 추천 (python)

문제 링크 (Level 1)

https://programmers.co.kr/learn/courses/30/lessons/72410

코딩테스트 연습 - 신규 아이디 추천

 

나의 코드 (python)

def solution(new_id):
    answer = ''
    
    # 1단계
    new_id = new_id.lower()
    
    # 2단계
    for i in new_id:
    	if i.isalnum() or i == '-' or i == '_' or i == '.':
            answer += i
    
    # 3단계
    while ('..' in answer):
        answer = answer.replace('..', '.')
        
    # 4단계
    if answer[0] == '.':
        answer = answer[1:]
    
    if answer != '':
        if answer[-1] == '.':
            answer = answer[:-1]

    # 5단계
    if answer == '':
        answer += 'a'

    # 6단계
    if len(answer) >= 16:
        answer = answer[:15]
        if answer[-1] == '.':
            answer = answer[:-1]
    
    # 7단계
    if len(answer) <= 2:
        while len(answer) <= 2:
            answer += answer[-1]

    return answer