[programmers/py] 신규 아이디 추천

신규아이디 추천

https://school.programmers.co.kr/learn/courses/30/lessons/72410?language=python3

문제 설명

1. new_id의 모든 대문자를 대응되는 소문자로 치환합니다.
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이 될 때까지 반복해서 끝에 붙입니다.

제한사항

• new_id에 나타날 수 있는 특수문자는 -_.~!@#$%^&*()=+[{]}:?,<>/로 한정됩니다
• 아이디의 길이는 3자 이상 15자 이하여야 합니다.
• 아이디는 알파벳 소문자, 숫자, 빼기(-), 밑줄(_), 마침표(.) 문자만 사용할 수 있습니다.
• 단, 마침표(.)는 처음과 끝에 사용할 수 없으며 또한 연속으로 사용할 수 없습니다

풀이

• 정규표현식을 사용해서 해결

import re
def solution(new_id):
    
    s = new_id
    s = s.lower() # 1
    s = re.sub('[^a-z0-9\-_.]',"",s) # 2
    s = re.sub('\.+', ".", s) # 3
    s = re.sub('^[.]|[.]$', "", s) # 4
    s = 'a' if len(s)==0 else s[:15] # 5
    s = s[:14] if s[-1] == "." else s # 6
    s = s if len(s) > 2 else s + "".join(s[-1] for i in range(3-len(s)))
      
    return s