Initial commit: добавление проекта predictV1

Включает модели ML для предсказаний, API маршруты, скрипты обучения и данные. Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
2026-02-21 17:22:58 +03:00
commit 8a134239d7
42 changed files with 12831 additions and 0 deletions
--- a/start/parse_pro_details_match.py
+++ b/start/parse_pro_details_match.py
@@ -0,0 +1,106 @@
+import requests
+import psycopg2
+import time
+from psycopg2.extras import RealDictCursor, execute_values
+
+# Подключение к базе данных
+conn = psycopg2.connect(
+    host="localhost",
+    port=5432,
+    database="korobka_db",
+    user="postgres",
+    password="postgres"
+)
+cursor = conn.cursor(cursor_factory=RealDictCursor)
+
+# Функция для определения позиции по lane и lane_role
+def get_position(lane, lane_role):
+    """
+    lane 2, lane_role 2 -> pos 2 (Mid Core)
+    lane 1, lane_role 1 -> pos 1 (Safe Lane Carry)
+    lane 1, lane_role 3 -> pos 5 (Safe Lane Support)
+    lane 3, lane_role 1 -> pos 3 (Offlane Core)
+    lane 3, lane_role 3 -> pos 4 (Offlane Support)
+    """
+    if lane == 2 and lane_role == 2:
+        return 2
+    elif lane == 1 and lane_role == 1:
+        return 1
+    elif lane == 1 and lane_role == 3:
+        return 5
+    elif lane == 3 and lane_role == 1:
+        return 3
+    elif lane == 3 and lane_role == 3:
+        return 4
+    else:
+        return -1  # Неизвестная комбинация
+
+# Получение ID матчей из БД
+cursor.execute("""
+select m.id
+from matches m
+left join details_match dm on dm.match_id = m.id
+where m."source" = 'pro' and dm.match_id is null
+""")
+matches = cursor.fetchall()
+match_ids = [match['id'] for match in matches]
+
+print(f"Получено {len(match_ids)} ID матчей из БД")
+
+# Запрос деталей для каждого матча
+for idx, match_id in enumerate(match_ids, 1):
+    url = f"https://api.opendota.com/api/matches/{match_id}"
+
+    try:
+        response = requests.get(url, timeout=30)
+
+        if response.status_code == 200:
+            match_data = response.json()
+            picks_bans = match_data.get('picks_bans', [])
+            players = match_data.get('players', [])
+
+            print(f"[{idx}/{len(match_ids)}] Match {match_id}: {len(picks_bans)} picks/bans, {len(players)} players")
+
+            # Создаем словари из players
+            hero_to_account = {player.get('hero_id'): player.get('account_id') for player in players if player.get('hero_id')}
+            hero_to_pos = {player.get('hero_id'): get_position(player.get('lane'), player.get('lane_role')) for player in players if player.get('hero_id')}
+
+            # Сохранение picks_bans в БД (только пики) с позициями
+            data = [
+                (
+                    match_id,
+                    pick_ban.get('hero_id'),
+                    pick_ban.get('team'),
+                    pick_ban.get('order'),
+                    hero_to_account.get(pick_ban.get('hero_id')),  # players_id
+                    hero_to_pos.get(pick_ban.get('hero_id'), -1),  # pos
+                    'pro'  # source
+                )
+                for pick_ban in picks_bans
+                if pick_ban.get('is_pick')
+            ]
+
+            print(f"Вставка {len(data)} записей для матча {match_id}")
+
+            execute_values(
+                cursor,
+                """INSERT INTO details_match (match_id, hero_id, team, "order", players_id, pos, source)
+                    VALUES %s""",
+                data
+            )
+            conn.commit()
+            time.sleep(1)  # пауза между запросами
+
+        else:
+            print(f"[{idx}/{len(match_ids)}] Ошибка для матча {match_id}: {response.status_code}")
+            if response.status_code == 429:
+                print("Превышен лимит запросов, ждем 10 секунд...")
+                time.sleep(10)
+
+    except Exception as e:
+        print(f"[{idx}/{len(match_ids)}] Исключение для матча {match_id}: {e}")
+
+cursor.close()
+conn.close()
+
+print("Завершено сохранение деталей матчей")