[ 프로그래머스 / PYTHON ] 신규 아이디 추천

문제 설명

1단계 newid의 모든 대문자를 대응되는 소문자로 치환합니다.
2단계 new_id에서 알파벳 소문자, 숫자, 빼기(-), 밑줄(), 마침표(.)를 제외한 모든 문자를 제거합니다.
3단계 new_id에서 마침표(.)가 2번 이상 연속된 부분을 하나의 마침표(.)로 치환합니다.
4단계 new_id에서 마침표(.)가 처음이나 끝에 위치한다면 제거합니다.
5단계 new_id가 빈 문자열이라면, new_id에 "a"를 대입합니다.
6단계 new_id의 길이가 16자 이상이면, new_id의 첫 15개의 문자를 제외한 나머지 문자들을 모두 제거합니다.
만약 제거 후 마침표(.)가 new_id의 끝에 위치한다면 끝에 위치한 마침표(.) 문자를 제거합니다.
7단계 new_id의 길이가 2자 이하라면, new_id의 마지막 문자를 new_id의 길이가 3이 될 때까지 반복해서 끝에 붙입니다.

정규식 간단히 공부하고 정규식으로 풀어보겠다고 나름 노력함.
코테 준비? 알고리즘 연습?을 한다면 모듈 사용(re) 즉, 정규식 사용하지 않고도 연습 겸 작성해보는 것도 좋다고 생각했지만 귀찮아서 나쁠 듯

[TIL] 정규 표현식 기본 간단 정리

내 코드

import re

def solution(new_id):
    #step 1
    new_id = new_id.lower()

    #step 2
    new_id = re.sub('[^\w\-_.]','',new_id)

    #step 3
    new_id = re.sub('\.{2,}','.',new_id)

    #step 4
    new_id = new_id.rstrip(".").lstrip(".")

    #step 5
    if new_id == "":    new_id = "a"

    #step 6
    if len(new_id) > 15:    new_id = new_id[:15]
    new_id = new_id.rstrip(".")

    #step 7
    while len(new_id) < 3:    new_id += new_id[-1]
        
    return new_id