TextRearrange/arrange.py

import os
from os import path

def rename_file(old_path, new_path):
    """
    파일 이름을 수정하는 함수

    Parameters:
        old_path (str): 파일 경로 + 기존 파일 이름
        new_path (str): 파일 경로 + 수정할 파일 이름

    Changes:
        old_path > new_path
    """
    if path.exists(new_path)==False:
        if path.exists(old_path):
	        # 파일이 있는 디렉토리 경로 얻기
            src = path.realpath(old_path)
	        # 원본파일 파일명 변경
            os.rename(old_path,new_path)
        else:
            print("{old_path} doesn't exist")

def extract_text_from_filename(filename):
    """
    파일 이름에서 'x'와 'y' 사이의 텍스트와 'y'와 'w' 사이의 텍스트를 추출하는 함수
    ex) x10y148w157h184.jpg
    
    Parameters:
        filename (str): 파일 이름

    Returns:
        tuple: 'x'와 'y' 사이의 텍스트, 'y'와 'w' 사이의 텍스트
    """
    start_x = filename.index('x')
    end_x = filename.index('y')

    start_y = end_x + 1
    end_y = filename.index('w')

    x_text = filename[start_x+1:end_x]
    y_text = filename[start_y:end_y]

    return x_text, y_text

def get_folder_contents_with_text(folder_path, file_extension=".jpg"):
    """
    폴더 내부의 파일들의 이름에서 'x'와 'y' 사이의 텍스트를 리스트 x에,
    'y'와 'w' 사이의 텍스트를 리스트 y에 각각 저장하는 함수

    Parameters:
        folder_path (str): 폴더 경로
        file_extension (str, optional): 가져올 파일의 확장자 (기본값: ".jpg")

    Returns:
        list, list: 'x'와 'y' 사이의 텍스트를 저장한 리스트 x,
                    'y'와 'w' 사이의 텍스트를 저장한 리스트 y
    """
    x_list = []
    y_list = []

    folder_contents = os.listdir(folder_path)
    print(folder_contents)
    jpg_files = [filename for filename in folder_contents if filename.endswith('jpg')]

    for jpg_file in jpg_files:

        x_text, y_text = extract_text_from_filename(jpg_file)
        x_list.append(x_text)
        y_list.append(y_text)

    return x_list, y_list, jpg_files