import pymysql

conn = pymysql.connect(
    host='officialsite.kr',
    port=23306,
    user='admin',
    password='dlgks~123',
    database='app_master',
    charset='utf8mb4'
)

cursor = conn.cursor(pymysql.cursors.DictCursor)

# 현재 카테고리 확인
print("=" * 60)
print("Current category data")
print("=" * 60)

cursor.execute("SELECT * FROM balancepick_categories ORDER BY order_index")
categories = cursor.fetchall()

for cat in categories:
    print(f"  {cat['id']:15} | {cat['name']:10} | {cat['description']}")

# 카테고리 수정
print("\n" + "=" * 60)
print("Updating categories...")
print("=" * 60)

updated_categories = [
    {'id': 'honbulho', 'name': '호불호', 'description': '좋아하거나 싫어하거나, 당신은 어느 쪽?'},
    {'id': 'couple', 'name': '커플', 'description': '직진파 vs 썸타기파, 당신의 연애 밸런스는?'},
    {'id': 'worker', 'name': '직장인', 'description': '칼퇴냐 야근이냐, 직장 밸런스 전쟁!'},
    {'id': 'tmi', 'name': 'TMI', 'description': '내 사소한 취향, 누가 더 잘 맞출까?'},
    {'id': 'value', 'name': '가치관', 'description': '이성보다 감성? 현실파 vs 꿈파?'},
    {'id': 'preference', 'name': '추구미', 'description': '소확행 vs 대확행, 당신의 인생 밸런스는?'},
    {'id': 'ideal_type', 'name': '이상형', 'description': '비주얼 vs 성격, 당신의 최종 선택은?'},
    {'id': 'random', 'name': '랜덤', 'description': '무슨 게임이 나올지 몰라요!'}
]

for cat in updated_categories:
    cursor.execute(
        """UPDATE balancepick_categories
           SET name = %s, description = %s
           WHERE id = %s""",
        (cat['name'], cat['description'], cat['id'])
    )
    print(f"  [OK] {cat['id']} updated: {cat['name']} - {cat['description']}")

conn.commit()

# 결과 확인
print("\n" + "=" * 60)
print("Updated category data")
print("=" * 60)

cursor.execute("SELECT * FROM balancepick_categories ORDER BY order_index")
categories = cursor.fetchall()

for cat in categories:
    print(f"  {cat['id']:15} | {cat['name']:10} | {cat['description']}")

cursor.close()
conn.close()

print("\n[DONE] Category update completed!")