"""
샘플 엑셀 데이터 생성 스크립트
테스트용 엑셀 파일들을 자동으로 생성합니다.
"""

import pandas as pd
import random
from datetime import datetime, timedelta

def create_sample_sales_data():
    """샘플 매출 데이터 생성"""

    months = ['1월', '2월', '3월']

    for month in months:
        # 랜덤 데이터 생성
        data = {
            '날짜': [datetime(2025, months.index(month)+1, day)
                    for day in range(1, random.randint(25, 31))],
            '제품명': [f'제품{random.randint(1,10)}'
                      for _ in range(random.randint(25, 30))],
            '수량': [random.randint(1, 100)
                    for _ in range(random.randint(25, 30))],
            '단가': [random.randint(1000, 50000)
                    for _ in range(random.randint(25, 30))],
        }

        # 매출액 계산
        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 폴더를 확인하세요.')