PIA-SPACE_LeaderBoard / utils /except_dir.py
jisujang's picture
first
a005c19
raw
history blame contribute delete
532 Bytes
import os
from typing import List
import enviroments.config as config
def cust_listdir(directory: str) -> List[str]:
"""
os.listdir์™€ ์œ ์‚ฌํ•˜๊ฒŒ ์ž‘๋™ํ•˜์ง€๋งŒ config์— ์ •์˜๋œ ํด๋”/ํŒŒ์ผ๋“ค์„ ์ œ์™ธํ•˜๊ณ  ๋ชฉ๋ก์„ ๋ฐ˜ํ™˜ํ•ฉ๋‹ˆ๋‹ค.
Args:
directory (str): ํƒ์ƒ‰ํ•  ๋””๋ ‰ํ† ๋ฆฌ ๊ฒฝ๋กœ
Returns:
List[str]: config์˜ EXCLUDE_DIRS์— ์ •์˜๋œ ํด๋”/ํŒŒ์ผ๋“ค์„ ์ œ์™ธํ•œ ๋ชฉ๋ก
"""
return [item for item in os.listdir(directory) if item not in config.EXCLUDE_DIRS]