"""直接用 SQL 创建必要的表"""
import pymysql

SQL_STATEMENTS = """
CREATE TABLE IF NOT EXISTS `nf_lovers` (
  `id` bigint NOT NULL AUTO_INCREMENT,
  `user_id` bigint NOT NULL,
  `name` varchar(64) DEFAULT NULL,
  `gender` enum('male','female') NOT NULL,
  `intro` text,
  `story_background` text,
  `personality_tag` int DEFAULT NULL,
  `interest_tags` json DEFAULT NULL,
  `opening_line` text,
  `personality_prompt` text,
  `appearance_prompt` text,
  `appearance_params` json DEFAULT NULL,
  `hair_style_id` int DEFAULT NULL,
  `eye_color_id` int DEFAULT NULL,
  `outfit_desc` varchar(50) DEFAULT NULL,
  `outfit_top_id` bigint DEFAULT NULL,
  `outfit_bottom_id` bigint DEFAULT NULL,
  `outfit_dress_id` bigint DEFAULT NULL,
  `voice_id` bigint DEFAULT NULL,
  `image_url` varchar(255) DEFAULT NULL,
  `last_image_task_id` bigint DEFAULT NULL,
  `image_gen_used` int DEFAULT '0',
  `image_gen_limit` int DEFAULT '10',
  `image_gen_reset_date` date DEFAULT NULL,
  `init_model` varchar(64) DEFAULT NULL,
  `init_at` datetime DEFAULT NULL,
  `created_at` datetime DEFAULT CURRENT_TIMESTAMP,
  `updated_at` datetime DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
  PRIMARY KEY (`id`),
  UNIQUE KEY `user_id` (`user_id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;

CREATE TABLE IF NOT EXISTS `nf_voice_library` (
  `id` bigint NOT NULL AUTO_INCREMENT,
  `name` varchar(64) NOT NULL,
  `gender` enum('male','female') NOT NULL,
  `style_tag` varchar(32) DEFAULT NULL,
  `avatar_url` varchar(255) DEFAULT NULL,
  `sample_audio_url` varchar(255) DEFAULT NULL,
  `tts_model_id` varchar(64) DEFAULT NULL,
  `is_default` tinyint(1) DEFAULT '0',
  `voice_code` varchar(64) NOT NULL,
  `is_owned` tinyint(1) DEFAULT '1',
  `price_gold` int DEFAULT '0',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;

CREATE TABLE IF NOT EXISTS `nf_girlfriend_mould` (
  `id` int NOT NULL AUTO_INCREMENT,
  `name` varchar(64) NOT NULL,
  `gender` enum('male','female') DEFAULT NULL,
  `weigh` int DEFAULT '0',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;

CREATE TABLE IF NOT EXISTS `nf_girlfriend_hobbies` (
  `id` int NOT NULL AUTO_INCREMENT,
  `name` varchar(64) NOT NULL,
  `weigh` int DEFAULT '0',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;

CREATE TABLE IF NOT EXISTS `nf_girlfriend_hairstyles` (
  `id` int NOT NULL AUTO_INCREMENT,
  `name` varchar(64) NOT NULL,
  `weigh` int DEFAULT '0',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;

CREATE TABLE IF NOT EXISTS `nf_girlfriend_eyecolor` (
  `id` int NOT NULL AUTO_INCREMENT,
  `name` varchar(64) NOT NULL,
  `weigh` int DEFAULT '0',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;
"""

try:
    conn = pymysql.connect(
        host='127.0.0.1',
        port=3306,
        user='root',
        password='root',
        database='lover',
        charset='utf8mb4'
    )
    print("✓ 连接数据库成功")
    
    cursor = conn.cursor()
    for statement in SQL_STATEMENTS.strip().split(';'):
        statement = statement.strip()
        if statement:
            cursor.execute(statement)
            print(f"✓ 执行成功")
    
    conn.commit()
    print("✓ 所有表创建完成！")
    cursor.close()
    conn.close()
except Exception as e:
    print(f"✗ 失败: {e}")