import json
import random
from faker import Faker
import time

한글 더미 데이터 생성을 위해 ko_KR 로케일 설정

fake = Faker('ko_KR')

--- 1. 데이터 베이스 (현실감 부여) ---

CITIES = [
{"name": "오사카", "country": "일본", "currency": "JPY"},
{"name": "도쿄", "country": "일본", "currency": "JPY"},
{"name": "다낭", "country": "베트남", "currency": "VND"},
{"name": "방콕", "country": "태국", "currency": "THB"},
{"name": "파리", "country": "프랑스", "currency": "EUR"},
{"name": "뉴욕", "country": "미국", "currency": "USD"},
{"name": "제주", "country": "한국", "currency": "KRW"},
{"name": "서울", "country": "한국", "currency": "KRW"},
{"name": "바르셀로나", "country": "스페인", "currency": "EUR"}
]

AIRLINES = ["대한항공", "아시아나", "진에어", "제주항공", "티웨이", "에어부산", "에어서울"]
HOTEL_BRANDS = ["신라호텔", "힐튼", "하얏트", "메리어트", "도미인", "토요코인", "에어비앤비", "감성 숙소"]
CONCERT_NAMES = ["싸이 흠뻑쇼", "아이유 콘서트", "임영웅 리사이틀", "콜드플레이 내한", "뮤지컬 시카고", "지킬앤하이드"]

마케팅 수식어 (검색어 매칭 테스트용)

ADJECTIVES = [
"초특가", "마감임박", "가성비 갑", "럭셔리", "5성급",
"아이와 함께", "커플 강추", "부모님 효도", "혼행 추천",
"오션뷰", "시티뷰", "조식포함", "무료취소 가능"
]

--- 2. 카테고리별 생성 로직 ---

def generate_flight(id):
city = random.choice(CITIES)
airline = random.choice(AIRLINES)
price_base = 200000 if city['country'] in ['일본', '중국'] else 500000 if city['country'] in ['베트남', '태국'] else 1200000
price = int(price_base random.uniform(0.8, 1.5) / 100) 100 # 100원 단위 절삭

title = f"[{airline}] {city['name']} 왕복 항공권 ({random.choice(ADJECTIVES)})"
desc = f"{city['name']}로 떠나는 가장 합리적인 선택. {airline} 직항으로 편안하게 모십니다. {fake.sentence()}"

return {
    "id": id,
    "type": "FLIGHT",
    "title": title,
    "description": desc,
    "location": city['name'],
    "country": city['country'],
    "price": price,
    "airline": airline,
    "is_direct": random.choice([True, True, False]), # 직항 확률 높음
    "departure_date": fake.date_between(start_date='+1d', end_date='+30d').isoformat()
}

def generate_hotel(id):
city = random.choice(CITIES)
brand = random.choice(HOTEL_BRANDS)
price = random.randint(5, 50) * 10000

title = f"{city['name']} {brand} - {random.choice(ADJECTIVES)}"
desc = f"{city['name']} 중심가에 위치한 최고의 숙소. {fake.catch_phrase()}. {fake.bs()}."

return {
    "id": id,
    "type": "HOTEL",
    "title": title,
    "description": desc,
    "location": city['name'],
    "country": city['country'],
    "price": price,
    "rating": round(random.uniform(3.0, 5.0), 1), # 3.0 ~ 5.0 평점
    "facilities": random.sample(["수영장", "와이파이", "조식", "피트니스", "주차장", "스파"], k=random.randint(2, 5))
}

def generate_ticket(id):

# 티켓은 주로 서울/한국 위주
item = random.choice(CONCERT_NAMES)
location = "서울"
price = random.randint(3, 20) * 10000

title = f"{item} 2026 - {random.choice(['R석', 'VIP석', 'S석'])} 티켓 오픈"
desc = f"당신의 심장을 뛰게 할 최고의 공연. {item} 놓치지 마세요. {fake.text(max_nb_chars=50)}"

return {
    "id": id,
    "type": "TICKET",
    "title": title,
    "description": desc,
    "location": location,
    "country": "한국",
    "price": price,
    "date": fake.date_between(start_date='+10d', end_date='+60d').isoformat()
}

--- 3. 실행 및 저장 ---

TOTAL_COUNT = 3000 # 원하는 데이터 개수 설정
data_list = []

print(f"🚀 여행 상품 데이터 {TOTAL_COUNT}개 생성 시작...")

for i in range(1, TOTAL_COUNT + 1):
category = random.choices(["FLIGHT", "HOTEL", "TICKET"], weights=[4, 4, 2], k=1)[0]

item = {}
if category == "FLIGHT":
    item = generate_flight(i)
elif category == "HOTEL":
    item = generate_hotel(i)
else:
    item = generate_ticket(i)

# 공통 필드 (랭킹 모델링용)
item['view_count'] = random.randint(0, 10000)   # 조회수 (인기순 정렬용)
item['review_count'] = random.randint(0, 500)   # 리뷰수 (신뢰도 정렬용)
item['created_at'] = fake.date_time_between(start_date='-1y', end_date='now').isoformat() # 최신순 정렬용

data_list.append(item)

JSON 파일 저장

file_name = 'real_travel_data.json'
with open(file_name, 'w', encoding='utf-8') as f:
json.dump(data_list, f, ensure_ascii=False, indent=2)

print(f"✅ 생성 완료! 파일명: {file_name}")