제 목적은

어느 한 폴더에 각각 모델명이 붙은 파일을 넣으면

자동으로 해당 모델 폴더로 가서 압축파일 안에 이동 시키는건데요

 

모델이 많아서 스크립트점 만들어 볼라고 하는 중 입니다.

파이썬 이고 모고 잘 모르고 그냥 gpt 켜놓고 하는 중인데 

계속 막히네요

 

import os

import shutil

import zipfile

import time

 

source_folder = "c:\\0_자동\\"

destination_folder = "c:\\0_백업\\"

 

# 폴더가 존재하지 않으면 생성하는 함수

def create_folder_if_not_exists(path):

    if not os.path.exists(path):

        os.makedirs(path)

 

# source_folder 안의 파일들을 검색하여 이동시키는 함수

def move_files():

    files = os.listdir(source_folder)

    for file in files:

        file_path = os.path.join(source_folder, file)

        if os.path.isfile(file_path):

            # 파일명에서 "_"를 기준으로 모델명 추출

            if "_" in file:

                model_name = file.split("_")[0]

            else:

                continue  # "_"가 없는 파일은 이동하지 않음

 

            # 모델명을 기반으로 폴더 경로 생성

            model_folder = find_model_folder(model_name)

 

            # 모델 폴더가 없으면 생성

            create_folder_if_not_exists(model_folder)

 

            # plot 폴더 경로 생성

            plot_folder = os.path.join(model_folder, "plot")

 

            # plot 폴더가 없으면 생성

            create_folder_if_not_exists(plot_folder)

 

            # 파일 복사

            dest_path = os.path.join(plot_folder, file)

            if os.path.exists(dest_path):

                os.remove(dest_path)

            shutil.copy2(file_path, dest_path)

 

            # 원본 파일 삭제

            os.remove(file_path)

 

            # plot.zip 압축 파일 업데이트

            update_zip_archive(plot_folder)

 

# 모델명과 일치하는 폴더를 검색하는 함수

def find_model_folder(model_name):

    for root, _, _ in os.walk(destination_folder):

        for folder in os.listdir(root):

            if model_name in folder:

                return os.path.join(root, folder)

 

    # 일치하는 폴더가 없을 경우 모델명으로 폴더 생성

    return os.path.join(destination_folder, model_name)

 

# 압축 파일 내에 파일 존재 여부 확인하는 함수

def file_exists_in_zip(zip_file, file):

    try:

        zip_file.getinfo(file)

        return True

    except KeyError:

        return False

 

# 압축 파일 내의 파일 제거하는 함수

def remove_file_in_zip(zip_file, file):

    temp_zip_path = os.path.join(zip_file.filename + '.tmp')

    with zipfile.ZipFile(zip_file.filename, 'r') as source_zip, zipfile.ZipFile(temp_zip_path, 'w') as temp_zip:

        for item in source_zip.infolist():

            if item.filename != file:

                temp_zip.writestr(item, source_zip.read(item.filename))

 

    os.remove(zip_file.filename)

    os.rename(temp_zip_path, zip_file.filename)

 

# plot.zip 압축 파일 업데이트하는 함수

def update_zip_archive(plot_folder):

    zip_path = os.path.join(plot_folder, "plot.zip")

 

    if not os.path.exists(zip_path):

        # 압축 파일이 없는 경우 새로 생성

        with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED, allowZip64=True) as zip_file:

            pass

 

    # 중첩된 압축 파일 검사

    with zipfile.ZipFile(zip_path, "r") as zip_file:

        if "plot.zip" in zip_file.namelist():

            # 중첩된 압축 파일이 있는 경우 제거

            zip_file.extract("plot.zip", path=plot_folder)

            os.remove(os.path.join(plot_folder, "plot.zip"))

 

    # plot 폴더 내 파일들 추가

    files = os.listdir(plot_folder)

    with zipfile.ZipFile(zip_path, "a", zipfile.ZIP_DEFLATED, allowZip64=True) as zip_file:

        for file in files:

            file_path = os.path.join(plot_folder, file)

            if os.path.isfile(file_path) and file != "plot.zip":

                zip_file.write(file_path, file)

 

    # 임시 압축 파일 삭제

    temp_zip_path = os.path.join(plot_folder, "plot_tmp.zip")

    if os.path.exists(temp_zip_path):

        os.remove(temp_zip_path)

 

# 파일 이동 함수 실행

move_files()

 

일단 gpt로 여기까진 짰는데

압축파일 안에 계속 중복으로 파일을 집어 넣고 

압축파일 에 집어 넣고 남은 파일도 안지우고 냅두고 있네요

 

일단 폴더까진 잘 찾아가서 이동은 시킵니다.

 

해결해야 할게  압축 파일 내  중복된 파일을 처리해야 하는데

 

그런 코드를 적을떄 마다 프로세스가 엑세스 할 수 없다고 떠버림