[프로그래머스] 신규 아이디 추천 - Python

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

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

import re

def solution(new_id):
    # 1단계
    answer = new_id.lower()
    # 2단계
    answer = re.sub('[^0-9a-z\.\-\_]', '', answer)
    # 3단계
    answer = re.sub('[.]+', '.', answer)
    # 4단계
    if answer.startswith('.'):
        answer = answer[1:]
    if answer.endswith('.'):
        answer =  answer[:-1]
    # 5단계
    if len(answer) == 0:
        answer = 'a'
    # 6단계
    if len(answer) >= 16:
        answer = answer[:15]
        if answer.endswith('.'):
            answer = answer[:-1]
    # 7단계
    if len(answer) <= 2:
        answer += answer[-1] * (3-len(answer))

    return answer