형식이 다른 여러 엑셀파일들을 하나의 통합 엑셀 양식 자동화 by 파이썬 (2/2)

import os

import msoffcrypto

import openpyxl

import io

# 파일이 위치한 폴더 경로

folder_path = os.path.dirname(os.path.abspath(__file__))

# 각 파일의 필수 포함 문자열

a_file_must_contain = "스마트스토어"

b_file_must_contain = "DeliveryList"

c_file_must_contain = "모카팩토리_주문 내역_"

# A 파일 찾기

a_file = None

for file in os.listdir(folder_path):

    if a_file_must_contain in file and file.endswith(".xlsx"):

        a_file = os.path.join(folder_path, file)

        break

# B 파일 찾기

b_file = None

for file in os.listdir(folder_path):

    if b_file_must_contain in file and file.endswith(".xlsx"):

        b_file = os.path.join(folder_path, file)

        break

# C 파일 찾기

c_file = None

for file in os.listdir(folder_path):

    if c_file_must_contain in file and file.endswith(".xlsx"):

        c_file = os.path.join(folder_path, file)

        break

# 결과물 파일 경로

result_file = os.path.join(folder_path, "result.xlsx")

# 결과물 파일 생성

wb_result = openpyxl.Workbook()

ws_result = wb_result.active

# 헤더 행 추가

ws_result.append(['스토어명', '보내시는분 연락처', '발송지 주소', '받는분성명', '받는분전화번호', 

                  '받는분기타연락처', '받는분우편번호', '받는분 주소(전체)', '받는분주소(세부주소)',

                  '스토어명', '내품명', '내품수량', '박스타입', '배송메세지'])

def process_file(file_path, columns, store_name, password=None):

    try:

        if password:

            # 암호화된 파일 해독

            encrypted_file = msoffcrypto.OfficeFile(open(file_path, "rb"))

            decrypted_stream = io.BytesIO()

            encrypted_file.load_key(password=password)

            encrypted_file.decrypt(decrypted_stream)

            decrypted_stream.seek(0)

            wb = openpyxl.load_workbook(decrypted_stream)

        else:

            wb = openpyxl.load_workbook(file_path)

        ws = wb.active

        

        for row in ws.iter_rows(min_row=2, values_only=True):

            data = [''] * 14

            

            # 고정값 채우기

            data[0] = store_name

            data[1] = '0507-1315-7424'

            data[2] = '경기도 고양시 덕양구 마상로 125 주교대리점'

            data[9] = '모카팩토리'

            

            # 지정된 열 데이터 채우기

            for i, col in enumerate(columns):

                if col is not None:

                    data[i+3] = row[col]

            

            ws_result.append(data)

        

    except Exception as e:

        print(f"Error processing file {file_path}: {str(e)}")

# A 파일 처리

if a_file:

    columns_a = [3, 30, 31, 32, 33, 34, 35, 36, 38]

    process_file(a_file, columns_a, '2H9M(모카팩토리)', password='1111')

else:

    print("A file not found.")

# B 파일 처리

if b_file:

    columns_b = [26, 27, 25, 28, 29, 30, 16, 22]

    process_file(b_file, columns_b, None)

else:

    print("B file not found.")

    

# C 파일 처리

if c_file:

    columns_c = [18, 19, 15, 14, 16, 17, 3, 22]

    process_file(c_file, columns_c, None)

else:

    print("C file not found.")

# 완성된 엑셀 파일 저장

wb_result.save(result_file)

print("작업이 완료되었습니다.")