# test_template_insertion.py
from src.database_module import DatabaseManager, Template
from datetime import datetime

def insert_sample_templates():
    db_manager = DatabaseManager("sqlite:///orders.db")
    
    sample_templates = [
        # [단계 1] 주문 접수
        {
            "stage": 1,
            "name": "주문 접수 기본",
            "content": (
                "[발송안내]안녕하세요. 혜리수샵입니다.\n"
                "롯데온에서 구매하신 접이식 전동차 상품은\n"
                "중국 현지에서 발송하여 국제물류센터로 이동중입니다.\n"
                "물류센터에서 국내로 발송하게 되면 다시 안내해 드리겠습니다.\n"
                "중국현지에서 발송 후 수령까지 약 10일~14일 소요될 예정입니다.\n"
                "고객님의 소중한 물품을 안전하게 배송해 드리겠습니다.\n"
                "좋은하루 되세요."
            ),
            "is_default": True
        },
        {
            "stage": 1,
            "name": "주문 접수 변형",
            "content": (
                "고객님, 주문이 정상적으로 접수되었습니다.\n"
                "(타오바오 트래킹: {taobao_tracking}, 국내 주문번호: {domestic_tracking})"
            ),
            "is_default": False
        },
        {
            "stage": 1,
            "name": "주문 접수 상세 안내",
            "content": (
                "[주문접수안내]안녕하세요, 혜리수샵입니다.\n"
                "고객님 주문이 접수되었으며, 곧 중국에서 상품 준비가 시작됩니다.\n"
                "타오바오 트래킹 번호 {taobao_tracking} 및 국내 주문번호 {domestic_tracking}이 발행되었습니다.\n"
                "추후 배송 진행 상황을 안내해드리겠습니다.\n"
                "감사합니다."
            ),
            "is_default": False
        },
        
        # [단계 2] 배송대행지 도착
        {
            "stage": 2,
            "name": "배송대행지 도착 기본",
            "content": (
                "[센터도착안내]안녕하세요. 혜리수샵입니다.\n"
                "롯데온에서 구매하신 접이식 전동차상품은\n"
                "국제물류센터에 도착하여 금일 국내로 발송예정입니다.\n"
                "고객님의 소중한 물품을 안전하게 배송해 드리겠습니다.\n"
                "좋은하루 되세요."
            ),
            "is_default": True
        },
        {
            "stage": 2,
            "name": "배송대행지 도착 변형",
            "content": (
                "안녕하세요, 주문하신 상품이 국제물류센터에 도착하였습니다.\n"
                "오늘 중으로 국내 배송이 시작될 예정입니다."
            ),
            "is_default": False
        },
        {
            "stage": 2,
            "name": "배송대행지 도착 상세 안내",
            "content": (
                "[배송안내]고객님, 상품이 국제물류센터에 도착하여\n"
                "검수 중입니다. 검수가 완료되면 국내로 발송되오니\n"
                "배송 상황을 지속적으로 확인해 주세요."
            ),
            "is_default": False
        },
        
        # [단계 3] 통관 진행
        {
            "stage": 3,
            "name": "통관 기본",
            "content": (
                "[통관안내]안녕하세요. 리앤수인터입니다.\n"
                "롯데온에서 구매하신 접이식 전동차상품은\n"
                "국내에 도착하여 통관 진행 중입니다.\n"
                "통관이 완료되는 대로 국내택배가 진행될 예정이며,\n"
                "물동량 증가로 인해 통관 지연이 발생할 수 있는 점 양해 부탁드립니다.\n"
                "고객님의 소중한 물품을 안전하게 배송해 드리겠습니다.\n"
                "좋은하루 되세요."
            ),
            "is_default": True
        },
        {
            "stage": 3,
            "name": "통관 변형",
            "content": (
                "고객님, 상품이 국내에 도착하였습니다.\n"
                "현재 통관 절차가 진행 중이며, 통관 완료 시\n"
                "국내 배송이 시작됩니다."
            ),
            "is_default": False
        },
        {
            "stage": 3,
            "name": "통관 상세 안내",
            "content": (
                "[통관진행]주문하신 상품이 국내 도착 후\n"
                "관세 당국의 통관 심사 중입니다.\n"
                "심사 완료 후, 빠른 시일 내에 배송될 예정이오니\n"
                "진행 상황은 추후 안내드리겠습니다."
            ),
            "is_default": False
        },
        
        # [단계 4] 통관 완료 및 국내 배송 시작
        {
            "stage": 4,
            "name": "통관 완료 기본",
            "content": (
                "[통관완료안내]안녕하세요 고객님,\n"
                "주문하신 접이식 전동차 상품은 통관이 완료되어\n"
                "현재 국내배송 준비 중입니다.\n"
                "주문번호: {domestic_tracking}을 통해 배송 상태를 확인하실 수 있습니다.\n"
                "빠른 시일 내에 고객님께 도착할 예정입니다.\n"
                "감사합니다."
            ),
            "is_default": True
        },
        {
            "stage": 4,
            "name": "통관 완료 변형",
            "content": (
                "[배송시작안내]안녕하세요 고객님.\n"
                "통관이 완료되어 주문하신 상품의 국내 배송이 시작되었습니다.\n"
                "국내배송 주문번호: {domestic_tracking}을 확인해 주세요.\n"
                "배송 진행 상황은 실시간으로 안내드리겠습니다."
            ),
            "is_default": False
        },
        {
            "stage": 4,
            "name": "국내배송 상세 안내",
            "content": (
                "[국내배송안내]고객님, 통관이 마무리되었습니다.\n"
                "현재 국내택배사에 인계되어 배송이 진행 중입니다.\n"
                "주문번호 {domestic_tracking}로 조회 가능하오니 참고 바랍니다.\n"
                "최선을 다해 안전하게 배송하겠습니다."
            ),
            "is_default": False
        },
        
        # [단계 5] 화물 전환
        {
            "stage": 5,
            "name": "화물 전환 기본",
            "content": (
                "[화물택배안내]안녕하세요, 혜리수샵입니다.\n"
                "고객님 주문 상품이 중량 문제로 인해 일반 택배에서\n"
                "화물택배로 전환되었습니다.\n"
                "화물택배 트래킹번호: {freight_tracking}를 통해 확인해 주세요.\n"
                "배송 일정은 변동될 수 있으니 양해 부탁드립니다."
            ),
            "is_default": True
        },
        {
            "stage": 5,
            "name": "화물 전환 변형",
            "content": (
                "[화물전환안내]고객님, 주문하신 상품이 무게 문제로 인해\n"
                "화물택배로 전환되었습니다.\n"
                "트래킹번호 {freight_tracking}를 이용하여 배송 상태를 확인하세요.\n"
                "빠른 시일 내에 안전하게 배송해 드리겠습니다."
            ),
            "is_default": False
        },
        {
            "stage": 5,
            "name": "화물 배송 상세 안내",
            "content": (
                "[화물배송안내]안녕하세요. 혜리수샵입니다.\n"
                "상품이 중량 기준을 초과하여 화물택배로 처리되었습니다.\n"
                "화물택배 트래킹번호: {freight_tracking}로 배송 진행 상황을 확인해 주세요.\n"
                "불편을 드려 죄송하며, 최선을 다해 배송하겠습니다."
            ),
            "is_default": False
        }
    ]
    
    for tpl in sample_templates:
        inserted = db_manager.insert_template(**tpl)
        print(f"삽입된 템플릿 - ID: {inserted.id}, 단계: {inserted.stage}, 이름: {inserted.name}")
    
    # db_manager.close()

if __name__ == '__main__':
    insert_sample_templates()