2.1 : New

ETL/L3_Builder.py

@@ -18,6 +18,17 @@ logger = logging.getLogger(__name__)
 L3_DB_PATH = Config.DB_L3_PATH
 SCHEMA_PATH = os.path.join(Config.BASE_DIR, 'database', 'L3', 'schema.sql')
 
+def _get_existing_columns(conn, table_name):
+    cur = conn.execute(f"PRAGMA table_info({table_name})")
+    return {row[1] for row in cur.fetchall()}
+
+def _ensure_columns(conn, table_name, columns):
+    existing = _get_existing_columns(conn, table_name)
+    for col, col_type in columns.items():
+        if col in existing:
+            continue
+        conn.execute(f"ALTER TABLE {table_name} ADD COLUMN {col} {col_type}")
+
 def init_db():
     l3_dir = os.path.dirname(L3_DB_PATH)
     if not os.path.exists(l3_dir):
@@ -26,6 +37,40 @@ def init_db():
     conn = sqlite3.connect(L3_DB_PATH)
     with open(SCHEMA_PATH, 'r', encoding='utf-8') as f:
         conn.executescript(f.read())
+
+    _ensure_columns(
+        conn,
+        "dm_player_features",
+        {
+            "rd_phase_kill_early_share": "REAL",
+            "rd_phase_kill_mid_share": "REAL",
+            "rd_phase_kill_late_share": "REAL",
+            "rd_phase_death_early_share": "REAL",
+            "rd_phase_death_mid_share": "REAL",
+            "rd_phase_death_late_share": "REAL",
+            "rd_firstdeath_team_first_death_rounds": "INTEGER",
+            "rd_firstdeath_team_first_death_win_rate": "REAL",
+            "rd_invalid_death_rounds": "INTEGER",
+            "rd_invalid_death_rate": "REAL",
+            "rd_pressure_kpr_ratio": "REAL",
+            "rd_pressure_perf_ratio": "REAL",
+            "rd_pressure_rounds_down3": "INTEGER",
+            "rd_pressure_rounds_normal": "INTEGER",
+            "rd_matchpoint_kpr_ratio": "REAL",
+            "rd_matchpoint_perf_ratio": "REAL",
+            "rd_matchpoint_rounds": "INTEGER",
+            "rd_comeback_kill_share": "REAL",
+            "rd_comeback_rounds": "INTEGER",
+            "rd_trade_response_10s_rate": "REAL",
+            "rd_weapon_top_json": "TEXT",
+            "rd_roundtype_split_json": "TEXT",
+            "map_stability_coef": "REAL",
+            "basic_avg_knife_kill": "REAL",
+            "basic_avg_zeus_kill": "REAL",
+            "basic_zeus_pick_rate": "REAL",
+        },
+    )
+
     conn.commit()
     conn.close()
     logger.info("L3 DB Initialized/Updated with Schema.")