Files
excel-merge-rpa/create_sample_data.py

44 lines
1.3 KiB
Python

"""
샘플 엑셀 데이터 생성 스크립트
테스트용 엑셀 파일들을 자동으로 생성합니다.
"""
import pandas as pd
import random
from datetime import datetime, timedelta
def create_sample_sales_data():
"""샘플 매출 데이터 생성"""
months = ['1월', '2월', '3월']
for month in months:
# 랜덤 행 수 결정
num_rows = random.randint(25, 30)
# 랜덤 데이터 생성
data = {
'날짜': [datetime(2025, months.index(month)+1, random.randint(1, 28))
for _ in range(num_rows)],
'제품명': [f'제품{random.randint(1,10)}'
for _ in range(num_rows)],
'수량': [random.randint(1, 100)
for _ in range(num_rows)],
'단가': [random.randint(1000, 50000)
for _ in range(num_rows)],
}
# 매출액 계산
df = pd.DataFrame(data)
df['매출액'] = df['수량'] * df['단가']
# 파일 저장
filename = f'input/{month}_매출.xlsx'
df.to_excel(filename, index=False)
print(f'{filename} 생성 완료 ({len(df)}행)')
if __name__ == '__main__':
print('📊 샘플 데이터 생성 중...\n')
create_sample_sales_data()
print('\n✅ 완료! input 폴더를 확인하세요.')