Compare commits
2 Commits
681eea5e6e
...
fadcc6cd8b
| Author | SHA1 | Date | |
|---|---|---|---|
| fadcc6cd8b | |||
| 797176ccb3 |
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
@@ -0,0 +1 @@
|
||||
.env
|
||||
BIN
__pycache__/get_data.cpython-312.pyc
Normal file
BIN
__pycache__/get_data.cpython-312.pyc
Normal file
Binary file not shown.
489
deploy.sh
Normal file
489
deploy.sh
Normal file
@@ -0,0 +1,489 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Настройка цветов для вывода
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
YELLOW='\033[1;33m'
|
||||
NC='\033[0m' # No Color
|
||||
|
||||
show_help() {
|
||||
echo "Использование: $0 -t <домашняя команда> -r <релиз> [-l <лига>]"
|
||||
echo " -t Домашняя команда"
|
||||
echo " -r Релиз (тег или ветка в git)"
|
||||
echo " -l Лига (опционально)"
|
||||
echo " -h Показать эту справку"
|
||||
echo ""
|
||||
echo "Пример: $0 -t cska -r main"
|
||||
echo "Пример: $0 -t zenit -r Barabanov_TEST -l vtb"
|
||||
echo "Пример: $0 -t avtodor -r main"
|
||||
echo ""
|
||||
exit 0
|
||||
}
|
||||
|
||||
# Функция для вывода цветных сообщений
|
||||
log_info() {
|
||||
echo -e "${GREEN}[INFO]${NC} $1" >&2
|
||||
}
|
||||
|
||||
log_warn() {
|
||||
echo -e "${YELLOW}[WARN]${NC} $1" >&2
|
||||
}
|
||||
|
||||
log_error() {
|
||||
echo -e "${RED}[ERROR]${NC} $1" >&2
|
||||
}
|
||||
|
||||
# Функция проверки и установки пакетов
|
||||
install_packages() {
|
||||
log_info "Обновление списка пакетов..."
|
||||
apt-get update
|
||||
|
||||
log_info "Установка необходимых пакетов..."
|
||||
apt-get install -y python3 python3-pip python3-venv git net-tools
|
||||
|
||||
# Проверка установки
|
||||
if ! command -v python3 &> /dev/null; then
|
||||
log_error "Python3 не установлен!"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if ! command -v pip3 &> /dev/null; then
|
||||
log_error "pip3 не установлен!"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
log_info "Версия Python: $(python3 --version)"
|
||||
log_info "Версия pip: $(pip3 --version)"
|
||||
}
|
||||
|
||||
# Функция загрузки кода
|
||||
download_code() {
|
||||
local repo_url="https://git.tvstart.ru/ychernenko/RFB.git"
|
||||
local release="$1"
|
||||
local target_dir="/root/RFB"
|
||||
|
||||
log_info "Создание рабочей директории $target_dir..."
|
||||
mkdir -p $target_dir
|
||||
|
||||
if [ -d "$target_dir/.git" ]; then
|
||||
log_info "Обновление существующего репозитория..."
|
||||
cd $target_dir
|
||||
git fetch --all
|
||||
git checkout $release
|
||||
git pull origin $release
|
||||
else
|
||||
log_info "Клонирование репозитория $repo_url..."
|
||||
git clone $repo_url $target_dir
|
||||
cd $target_dir
|
||||
git checkout $release
|
||||
fi
|
||||
|
||||
# Проверка успешности checkout
|
||||
if [ $? -ne 0 ]; then
|
||||
log_error "Не удалось переключиться на релиз $release"
|
||||
log_error "Доступные ветки:"
|
||||
git branch -r
|
||||
exit 1
|
||||
fi
|
||||
|
||||
log_info "Код успешно загружен (релиз: $release)"
|
||||
log_info "Текущая ветка: $(git branch --show-current)"
|
||||
log_info "Последний коммит: $(git log --oneline -1)"
|
||||
}
|
||||
|
||||
# Функция настройки виртуального окружения
|
||||
setup_venv() {
|
||||
local target_dir="/root/RFB"
|
||||
cd $target_dir
|
||||
|
||||
log_info "Создание виртуального окружения..."
|
||||
python3 -m venv .venv
|
||||
|
||||
if [ ! -f ".venv/bin/activate" ]; then
|
||||
log_error "Не удалось создать виртуальное окружение"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
log_info "Активация виртуального окружения и установка зависимостей..."
|
||||
source .venv/bin/activate
|
||||
|
||||
# Обновление pip
|
||||
pip install --upgrade pip
|
||||
|
||||
# Проверка наличия requirements.txt
|
||||
if [ -f "requirements.txt" ]; then
|
||||
log_info "Установка зависимостей из requirements.txt..."
|
||||
if pip install -r requirements.txt; then
|
||||
log_info "Все зависимости успешно установлены"
|
||||
else
|
||||
log_error "Ошибка при установке зависимостей из requirements.txt"
|
||||
log_warn "Попытка установки базовых зависимостей..."
|
||||
pip install requests pandas numpy
|
||||
fi
|
||||
else
|
||||
log_warn "Файл requirements.txt не найден, устанавливаем базовые зависимости..."
|
||||
pip install requests pandas numpy
|
||||
fi
|
||||
|
||||
# Проверка основных пакетов
|
||||
log_info "Проверка установки основных пакетов..."
|
||||
python -c "
|
||||
try:
|
||||
import requests, pandas, numpy
|
||||
print('✓ Все основные пакеты успешно импортируются')
|
||||
except ImportError as e:
|
||||
print(f'✗ Ошибка импорта: {e}')
|
||||
exit(1)
|
||||
"
|
||||
|
||||
log_info "Установленные пакеты:"
|
||||
pip list --format=columns
|
||||
}
|
||||
|
||||
# Функция создания systemd сервиса
|
||||
create_systemd_service() {
|
||||
local team="$1"
|
||||
local league="$2"
|
||||
|
||||
log_info "Создание systemd сервиса для команды: $team"
|
||||
if [ -n "$league" ]; then
|
||||
log_info "Лига: $league"
|
||||
fi
|
||||
|
||||
# Останавливаем и отключаем старый сервис если он есть
|
||||
if systemctl is-active --quiet rfb-data.service; then
|
||||
log_info "Остановка rfb-data.service..."
|
||||
systemctl stop rfb-data.service
|
||||
fi
|
||||
|
||||
if systemctl is-enabled --quiet rfb-data.service; then
|
||||
log_info "Отключение rfb-data.service..."
|
||||
systemctl disable rfb-data.service
|
||||
fi
|
||||
|
||||
# Формируем команду для data сервиса
|
||||
local data_command="/root/RFB/.venv/bin/python3 /root/RFB/get_data.py --team \"$team\""
|
||||
if [ -n "$league" ]; then
|
||||
data_command="$data_command --league \"$league\""
|
||||
fi
|
||||
|
||||
# Создаем сервисный файл в правильной директории
|
||||
local data_service_file="/etc/systemd/system/rfb-data.service"
|
||||
|
||||
log_info "Создание файла сервиса: $data_service_file"
|
||||
cat > "$data_service_file" << EOF
|
||||
[Unit]
|
||||
Description=RFB Data Service
|
||||
After=network.target
|
||||
|
||||
[Service]
|
||||
Type=simple
|
||||
User=root
|
||||
WorkingDirectory=/root/RFB
|
||||
Environment=PATH=/root/RFB/.venv/bin
|
||||
ExecStart=$data_command
|
||||
Restart=always
|
||||
RestartSec=30
|
||||
StandardOutput=journal
|
||||
StandardError=journal
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
||||
EOF
|
||||
|
||||
# Настройка прав
|
||||
chmod 644 "$data_service_file"
|
||||
|
||||
log_info "Перезагрузка systemd демона..."
|
||||
systemctl daemon-reload
|
||||
|
||||
log_info "Включение сервиса rfb-data.service..."
|
||||
systemctl enable rfb-data.service
|
||||
}
|
||||
|
||||
# Функция проверки файлов
|
||||
check_required_files() {
|
||||
local target_dir="/root/RFB"
|
||||
cd $target_dir
|
||||
|
||||
local missing_files=()
|
||||
local required_files=("get_data.py")
|
||||
|
||||
for file in "${required_files[@]}"; do
|
||||
if [ ! -f "$file" ]; then
|
||||
missing_files+=("$file")
|
||||
fi
|
||||
done
|
||||
|
||||
if [ ${#missing_files[@]} -ne 0 ]; then
|
||||
log_error "Отсутствуют необходимые файлы: ${missing_files[*]}"
|
||||
log_error "Содержимое директории $target_dir:"
|
||||
ls -la
|
||||
exit 1
|
||||
fi
|
||||
|
||||
log_info "Все необходимые файлы присутствуют"
|
||||
|
||||
# Дополнительная проверка на исполняемость Python файла
|
||||
if ! python3 -c "import ast; ast.parse(open('get_data.py').read())" 2>/dev/null; then
|
||||
log_warn "Файл get_data.py содержит синтаксические ошибки Python"
|
||||
fi
|
||||
}
|
||||
|
||||
# Функция определения IP и команды (исправленная)
|
||||
detect_team() {
|
||||
local team_arg="$1"
|
||||
|
||||
# Определение IP адреса
|
||||
if [[ "$(uname -s)" == "Linux" ]]; then
|
||||
ip_address=$(ip route get 1 2>/dev/null | awk '{print $7; exit}' ||
|
||||
ip a 2>/dev/null | grep -Eo 'inet (addr:)?([0-9]*\.){3}[0-9]*' |
|
||||
grep -Eo '([0-9]*\.){3}[0-9]*' | grep -v '127.0.0.1' | head -n1)
|
||||
else
|
||||
log_error "Скрипт работает только на Linux"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
log_info "Определен IP адрес: $ip_address"
|
||||
|
||||
# База данных хостов в виде пар IP-команда
|
||||
declare -A hosts=(
|
||||
["10.10.35.21"]="cska"
|
||||
["10.10.35.22"]="Lokomotiv Kuban"
|
||||
["10.10.35.23"]="uralmash"
|
||||
["10.10.35.24"]="betcity parma"
|
||||
["10.10.35.25"]="avtodor"
|
||||
["10.10.35.26"]="zenit"
|
||||
["10.10.35.27"]="samara"
|
||||
["10.10.35.28"]="mba-mai"
|
||||
["10.10.35.29"]="Pari Nizhny Novgorod"
|
||||
["10.10.35.30"]="unics"
|
||||
)
|
||||
|
||||
# Определение команды по IP
|
||||
detected_team="${hosts[$ip_address]}"
|
||||
|
||||
# Определение финальной команды (приоритет у аргумента, затем у автоопределения)
|
||||
if [[ -n "$team_arg" ]]; then
|
||||
final_team="$team_arg"
|
||||
log_info "Используется команда из аргумента: $final_team"
|
||||
elif [[ -n "$detected_team" ]]; then
|
||||
final_team="$detected_team"
|
||||
log_info "Используется автоопределенная команда: $final_team"
|
||||
else
|
||||
log_error "Не удалось определить команду. Укажите явно через -t"
|
||||
echo "Доступные команды:" >&2
|
||||
for ip in "${!hosts[@]}"; do
|
||||
echo " $ip: ${hosts[$ip]}" >&2
|
||||
done
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Возвращаем только чистую строку с названием команды (без цветовых кодов)
|
||||
echo "$final_team"
|
||||
}
|
||||
|
||||
# Функция настройки firewall
|
||||
setup_firewall() {
|
||||
log_info "Настройка firewall (открытие порта 8000)..."
|
||||
|
||||
# Проверяем наличие ufw
|
||||
if command -v ufw &> /dev/null && systemctl is-active --quiet ufw; then
|
||||
ufw allow 8000/tcp
|
||||
log_info "Порт 8000 открыт в ufw"
|
||||
fi
|
||||
|
||||
# Для firewalld (CentOS/RHEL)
|
||||
if command -v firewall-cmd &> /dev/null && systemctl is-active --quiet firewalld; then
|
||||
firewall-cmd --permanent --add-port=8000/tcp
|
||||
firewall-cmd --reload
|
||||
log_info "Порт 8000 открыт в firewalld"
|
||||
fi
|
||||
}
|
||||
|
||||
# Функция проверки порта
|
||||
check_port() {
|
||||
local port=8000
|
||||
local occupied=false
|
||||
|
||||
if command -v netstat &> /dev/null && netstat -tuln 2>/dev/null | grep ":$port " > /dev/null; then
|
||||
occupied=true
|
||||
elif command -v ss &> /dev/null && ss -tuln 2>/dev/null | grep ":$port " > /dev/null; then
|
||||
occupied=true
|
||||
fi
|
||||
|
||||
if [ "$occupied" = true ]; then
|
||||
log_warn "Порт $port уже занят. Возможно, приложение уже запущено."
|
||||
return 1
|
||||
fi
|
||||
return 0
|
||||
}
|
||||
|
||||
# Функция управления сервисами
|
||||
manage_services() {
|
||||
local action="$1"
|
||||
|
||||
case $action in
|
||||
"start")
|
||||
log_info "Запуск сервиса rfb-data.service..."
|
||||
systemctl start rfb-data.service
|
||||
;;
|
||||
"restart")
|
||||
log_info "Перезапуск сервиса rfb-data.service..."
|
||||
systemctl restart rfb-data.service
|
||||
;;
|
||||
"stop")
|
||||
log_info "Остановка сервиса rfb-data.service..."
|
||||
systemctl stop rfb-data.service
|
||||
;;
|
||||
esac
|
||||
|
||||
# Даем время сервису на запуск/остановку
|
||||
sleep 3
|
||||
}
|
||||
|
||||
# Функция проверки статуса сервиса
|
||||
check_service_status() {
|
||||
log_info "Статус сервиса rfb-data.service:"
|
||||
if systemctl is-active rfb-data.service; then
|
||||
systemctl status rfb-data.service --no-pager -l
|
||||
else
|
||||
log_warn "Сервис rfb-data.service не запущен"
|
||||
fi
|
||||
}
|
||||
|
||||
# Функция проверки системы
|
||||
check_system() {
|
||||
log_info "Проверка системы..."
|
||||
|
||||
# Проверка дистрибутива
|
||||
if [ -f /etc/os-release ]; then
|
||||
source /etc/os-release
|
||||
log_info "Дистрибутив: $PRETTY_NAME"
|
||||
fi
|
||||
|
||||
# Проверка свободного места
|
||||
local available_space=$(df /root --output=avail | tail -1)
|
||||
if [ "$available_space" -lt 1048576 ]; then # Меньше 1GB
|
||||
log_warn "Мало свободного места: $((available_space / 1024)) MB"
|
||||
else
|
||||
log_info "Свободное место: $((available_space / 1024 / 1024)) GB"
|
||||
fi
|
||||
|
||||
# Проверка памяти
|
||||
local total_mem=$(grep MemTotal /proc/meminfo | awk '{print $2}')
|
||||
log_info "Оперативная память: $((total_mem / 1024)) MB"
|
||||
}
|
||||
|
||||
# Основная функция
|
||||
main() {
|
||||
local team=""
|
||||
local release="main" # значение по умолчанию
|
||||
local league="" # переменная для лиги
|
||||
|
||||
# Обработка аргументов командной строки
|
||||
while getopts "t:r:l:h" opt; do
|
||||
case $opt in
|
||||
t) team="$OPTARG" ;;
|
||||
r) release="$OPTARG" ;;
|
||||
l) league="$OPTARG" ;;
|
||||
h) show_help ;;
|
||||
*) log_error "Неверный аргумент"; exit 1 ;;
|
||||
esac
|
||||
done
|
||||
|
||||
# Проверка обязательных аргументов
|
||||
if [ -z "$team" ] && [ -z "$release" ]; then
|
||||
log_error "Необходимо указать команду (-t) и релиз (-r)"
|
||||
show_help
|
||||
fi
|
||||
|
||||
log_info "Начало установки RFB Data Service..."
|
||||
log_info "Команда: $team, Релиз: $release"
|
||||
if [ -n "$league" ]; then
|
||||
log_info "Лига: $league"
|
||||
else
|
||||
log_info "Лига: не указана (будет использовано значение по умолчанию)"
|
||||
fi
|
||||
|
||||
# Проверка прав root
|
||||
if [[ $EUID -ne 0 ]]; then
|
||||
log_error "Этот скрипт должен запускаться с правами root"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Проверка системы
|
||||
check_system
|
||||
|
||||
# Установка пакетов
|
||||
install_packages
|
||||
|
||||
# Загрузка кода
|
||||
download_code "$release"
|
||||
|
||||
# Проверка файлов
|
||||
check_required_files
|
||||
|
||||
# Настройка виртуального окружения
|
||||
setup_venv
|
||||
|
||||
# Определение команды - ТЕПЕРЬ БЕЗ временного файла
|
||||
final_team=$(detect_team "$team")
|
||||
log_info "Финальная команда: '$final_team'"
|
||||
|
||||
# Проверка что команда не пустая и не содержит спецсимволов
|
||||
if [[ -z "$final_team" || "$final_team" =~ [^a-zA-Z0-9[:space:]-] ]]; then
|
||||
log_error "Некорректное название команды: '$final_team'"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Настройка firewall
|
||||
setup_firewall
|
||||
|
||||
# Проверка порта
|
||||
check_port
|
||||
|
||||
# Создание systemd сервиса
|
||||
create_systemd_service "$final_team" "$league"
|
||||
|
||||
log_info "Настройка завершена!"
|
||||
|
||||
# Запуск сервиса
|
||||
manage_services "start"
|
||||
|
||||
# Проверка статуса
|
||||
check_service_status
|
||||
|
||||
# Получаем IP для вывода
|
||||
ip_address=$(ip route get 1 2>/dev/null | awk '{print $7; exit}' ||
|
||||
ip a 2>/dev/null | grep -Eo 'inet (addr:)?([0-9]*\.){3}[0-9]*' |
|
||||
grep -Eo '([0-9]*\.){3}[0-9]*' | grep -v '127.0.0.1' | head -n1)
|
||||
|
||||
log_info "=================================================="
|
||||
log_info "Установка завершена успешно!"
|
||||
log_info "Команда: $final_team"
|
||||
if [ -n "$league" ]; then
|
||||
log_info "Лига: $league"
|
||||
fi
|
||||
log_info "Режим: $release"
|
||||
log_info ""
|
||||
log_info "Для просмотра логов:"
|
||||
log_info " journalctl -u rfb-data.service -f"
|
||||
log_info ""
|
||||
log_info "Управление сервисом:"
|
||||
log_info " Перезапуск: systemctl restart rfb-data.service"
|
||||
log_info " Остановка: systemctl stop rfb-data.service"
|
||||
log_info " Статус: systemctl status rfb-data.service"
|
||||
log_info " Логи: journalctl -u rfb-data.service"
|
||||
log_info ""
|
||||
log_info "Проверка работы:"
|
||||
log_info " Проверить процессы: ps aux | grep get_data.py"
|
||||
log_info " Проверить логи: tail -f /root/RFB/logs/*.log 2>/dev/null || echo 'Директория логов не найдена'"
|
||||
log_info "=================================================="
|
||||
}
|
||||
|
||||
# Обработка прерывания
|
||||
trap 'log_error "Прервано пользователем"; exit 1' INT TERM
|
||||
|
||||
# Запуск основной функции
|
||||
main "$@"
|
||||
330
get_data.py
Normal file
330
get_data.py
Normal file
@@ -0,0 +1,330 @@
|
||||
from fastapi import FastAPI
|
||||
from fastapi.responses import Response, JSONResponse, HTMLResponse
|
||||
import pandas as pd
|
||||
import requests, io, os
|
||||
from requests.auth import HTTPBasicAuth
|
||||
from dotenv import load_dotenv
|
||||
from datetime import datetime
|
||||
import uvicorn
|
||||
from threading import Thread, Event, Lock
|
||||
import time
|
||||
from contextlib import asynccontextmanager
|
||||
|
||||
|
||||
# --- Глобальные переменные ---
|
||||
selected_game_id: int | None = None
|
||||
current_tournament_id: int | None = None # будем обновлять при загрузке расписания
|
||||
latest_game_data: dict | None = None # сюда кладём последние данные по матчу
|
||||
latest_game_error: str | None = None
|
||||
_latest_lock = Lock()
|
||||
_stop_event = Event()
|
||||
_worker_thread: Thread | None = None
|
||||
|
||||
# Загружаем переменные из .env
|
||||
load_dotenv()
|
||||
api_user = os.getenv("API_USER")
|
||||
api_pass = os.getenv("API_PASS")
|
||||
league = os.getenv("LEAGUE")
|
||||
POLL_SEC = int(os.getenv("GAME_POLL_SECONDS"))
|
||||
|
||||
|
||||
|
||||
def load_today_schedule():
|
||||
"""Возвращает DataFrame матчей на сегодня с нужными колонками (или пустой DF)."""
|
||||
url_tournaments = "http://stat2tv.khl.ru/tournaments.xml"
|
||||
r = requests.get(
|
||||
url_tournaments, auth=HTTPBasicAuth(api_user, api_pass), verify=False
|
||||
)
|
||||
df = pd.read_xml(io.StringIO(r.text))
|
||||
|
||||
df["startDate"] = pd.to_datetime(df["startDate"], errors="coerce")
|
||||
df["endDate"] = pd.to_datetime(df["endDate"], errors="coerce")
|
||||
now = datetime.now()
|
||||
|
||||
filtered = df[
|
||||
(df["level"] == league)
|
||||
& (df["startDate"] <= now)
|
||||
& (df["endDate"] >= now)
|
||||
& (df["seasonPart"] == "regular")
|
||||
]
|
||||
if filtered.empty:
|
||||
return pd.DataFrame()
|
||||
|
||||
tournament_id = int(filtered.iloc[0]["id"])
|
||||
global current_tournament_id
|
||||
current_tournament_id = tournament_id
|
||||
url_schedule = f"http://stat2tv.khl.ru/{tournament_id}/schedule-{tournament_id}.xml"
|
||||
r = requests.get(url_schedule, auth=HTTPBasicAuth(api_user, api_pass), verify=False)
|
||||
schedule_df = pd.read_xml(io.StringIO(r.text))
|
||||
|
||||
# Нужные колонки (скорректируй под реальные имена из XML)
|
||||
needed_columns = ["id", "date", "time", "homeName_en", "visitorName_en", "arena"]
|
||||
exist = [c for c in needed_columns if c in schedule_df.columns]
|
||||
schedule_df = schedule_df[exist].copy()
|
||||
|
||||
# Преобразуем дату и время
|
||||
schedule_df["date"] = pd.to_datetime(schedule_df["date"], errors="coerce")
|
||||
today = now.date()
|
||||
schedule_today = schedule_df[schedule_df["date"].dt.date == today].copy()
|
||||
|
||||
# --- Нормализуем время и строим единый datetime для сортировки ---
|
||||
# 1) берём из time только HH:MM (на случай '19:30', '19:30:00', '19:30 MSK', и т.п.)
|
||||
time_clean = (
|
||||
schedule_today.get("time")
|
||||
.astype(str)
|
||||
.str.extract(r"(?P<hhmm>\d{1,2}:\d{2})", expand=True)["hhmm"]
|
||||
)
|
||||
|
||||
# 2) собираем строку "YYYY-MM-DD HH:MM" и парсим в Timestamp
|
||||
date_str = schedule_today["date"].dt.strftime("%Y-%m-%d")
|
||||
kickoff_str = (date_str + " " + time_clean.fillna("")).str.strip()
|
||||
schedule_today["kickoff_dt"] = pd.to_datetime(kickoff_str, errors="coerce")
|
||||
|
||||
# 3) сортировка: сначала по времени (NaT в конец), потом по id
|
||||
schedule_today = schedule_today.sort_values(
|
||||
by=["kickoff_dt", "id"], ascending=[True, True], na_position="last"
|
||||
)
|
||||
|
||||
# 4) человекочитаемая строка даты/времени для таблицы
|
||||
schedule_today["datetime_str"] = schedule_today["kickoff_dt"].dt.strftime(
|
||||
"%d.%m.%Y %H:%M"
|
||||
)
|
||||
# если time отсутствует и kickoff_dt = NaT — показываем просто дату
|
||||
mask_nat = schedule_today["kickoff_dt"].isna()
|
||||
schedule_today.loc[mask_nat, "datetime_str"] = schedule_today.loc[
|
||||
mask_nat, "date"
|
||||
].dt.strftime("%d.%m.%Y")
|
||||
|
||||
return schedule_today
|
||||
|
||||
|
||||
def _build_game_url(tournament_id: int, game_id: int) -> str:
|
||||
# URL по аналогии с расписанием: .../{tournament_id}/json_en/{game_id}.json
|
||||
# Если у тебя другой шаблон — просто поменяй строку ниже.
|
||||
return f"http://stat2tv.khl.ru/{tournament_id}/json_en/{game_id}.json"
|
||||
|
||||
def _fetch_game_once(tournament_id: int, game_id: int) -> dict:
|
||||
"""Один запрос к API матча -> чистый JSON из API."""
|
||||
url = _build_game_url(tournament_id, game_id)
|
||||
r = requests.get(
|
||||
url,
|
||||
auth=HTTPBasicAuth(api_user, api_pass),
|
||||
verify=False,
|
||||
timeout=10
|
||||
)
|
||||
r.raise_for_status()
|
||||
|
||||
# Пробуем распарсить JSON
|
||||
try:
|
||||
data = r.json()
|
||||
except ValueError:
|
||||
# если это не JSON — вернём текст
|
||||
data = {"raw": r.text}
|
||||
|
||||
return {"url": url, "json": data}
|
||||
|
||||
|
||||
def _game_poll_worker():
|
||||
"""Фоновый цикл: опрашивает API для выбранного game_id."""
|
||||
global latest_game_data, latest_game_error
|
||||
while not _stop_event.is_set():
|
||||
gid = selected_game_id
|
||||
tid = current_tournament_id
|
||||
if gid and tid:
|
||||
try:
|
||||
data = _fetch_game_once(tid, gid)
|
||||
with _latest_lock:
|
||||
latest_game_data = {
|
||||
"tournament_id": tid,
|
||||
"game_id": gid,
|
||||
"fetched_at": datetime.now().isoformat(),
|
||||
"data": data,
|
||||
}
|
||||
latest_game_error = None
|
||||
except Exception as e:
|
||||
with _latest_lock:
|
||||
latest_game_error = f"{type(e).__name__}: {e}"
|
||||
# Ждём интервал (с возможностью ранней остановки)
|
||||
_stop_event.wait(POLL_SEC)
|
||||
|
||||
|
||||
@asynccontextmanager
|
||||
async def lifespan(app: FastAPI):
|
||||
"""Запускаем и останавливаем поток чтения данных при старте/остановке приложения."""
|
||||
global _worker_thread
|
||||
_stop_event.clear()
|
||||
_worker_thread = Thread(target=_game_poll_worker, daemon=True)
|
||||
_worker_thread.start()
|
||||
print("✅ Background thread started")
|
||||
|
||||
# Отдаём управление FastAPI
|
||||
yield
|
||||
|
||||
# Останавливаем при завершении
|
||||
_stop_event.set()
|
||||
if _worker_thread.is_alive():
|
||||
_worker_thread.join(timeout=2)
|
||||
print("🛑 Background thread stopped")
|
||||
|
||||
app = FastAPI(lifespan=lifespan)
|
||||
|
||||
|
||||
@app.get("/games")
|
||||
async def games():
|
||||
df = load_today_schedule()
|
||||
if df.empty:
|
||||
return JSONResponse({"message": "Сегодня матчей нет"})
|
||||
|
||||
json_schedule = df.to_json(orient="records", force_ascii=False, date_format="iso")
|
||||
return Response(content=json_schedule, media_type="application/json")
|
||||
|
||||
|
||||
@app.get("/games/html")
|
||||
async def games_html():
|
||||
df = load_today_schedule()
|
||||
if df.empty:
|
||||
return HTMLResponse(
|
||||
"<h2 style='font-family:Inter,system-ui'>Сегодня матчей нет</h2>"
|
||||
)
|
||||
|
||||
# Строим строки таблицы
|
||||
rows_html = []
|
||||
for _, row in df.iterrows():
|
||||
gid = int(row["id"])
|
||||
home = row.get("homeName_en", "")
|
||||
away = row.get("visitorName_en", "")
|
||||
when = row.get("datetime_str", "")
|
||||
arena = row.get("arena", "")
|
||||
rows_html.append(
|
||||
f"""
|
||||
<tr data-id="{gid}">
|
||||
<td>{gid}</td>
|
||||
<td>{when}</td>
|
||||
<td>{home}</td>
|
||||
<td>{away}</td>
|
||||
<td>{arena}</td>
|
||||
<td><button class="pick">Выбрать</button></td>
|
||||
</tr>
|
||||
"""
|
||||
)
|
||||
|
||||
# ✅ Весь HTML, включая JS, внутри тройных кавычек
|
||||
html = f"""
|
||||
<!doctype html>
|
||||
<html lang="ru">
|
||||
<meta charset="utf-8">
|
||||
<title>Выбор матча</title>
|
||||
<style>
|
||||
body {{ font-family: Inter, system-ui, -apple-system, Segoe UI, Roboto, Arial; background:#0b0f14; color:#e6edf3; margin:0; padding:24px; }}
|
||||
h1 {{ margin:0 0 16px; font-size:20px; }}
|
||||
table {{ border-collapse: collapse; width:100%; background:#0f1620; border:1px solid #1f2a36; border-radius:12px; overflow:hidden; }}
|
||||
thead th {{ text-align:left; padding:12px 14px; font-weight:600; background:#121b26; border-bottom:1px solid #1f2a36; }}
|
||||
tbody td {{ padding:12px 14px; border-top:1px solid #16212c; }}
|
||||
tr:hover {{ background:#13202d; }}
|
||||
tr.selected {{ outline:2px solid #6aa6ff; outline-offset:-2px; background:#132c49; }}
|
||||
button.pick {{
|
||||
border:1px solid #2a3a4d; background:#1a2635; padding:8px 12px; border-radius:10px;
|
||||
color:#e6edf3; cursor:pointer; font-weight:600;
|
||||
}}
|
||||
button.pick:hover {{ background:#223243; }}
|
||||
.status {{ margin-top:14px; opacity:0.9; }}
|
||||
</style>
|
||||
<h1>Матчи сегодня</h1>
|
||||
<table id="tbl">
|
||||
<thead>
|
||||
<tr>
|
||||
<th>ID</th><th>Дата/время</th><th>Хозяева</th><th>Гости</th><th>Арена</th><th></th>
|
||||
</tr>
|
||||
</thead>
|
||||
<tbody>
|
||||
{''.join(rows_html)}
|
||||
</tbody>
|
||||
</table>
|
||||
<div class="status" id="status">Ничего не выбрано</div>
|
||||
<script>
|
||||
const tbl = document.getElementById('tbl');
|
||||
const status = document.getElementById('status');
|
||||
let selectedRow = null;
|
||||
|
||||
tbl.addEventListener('click', async (e) => {{
|
||||
const btn = e.target.closest('.pick');
|
||||
if (!btn) return;
|
||||
const row = btn.closest('tr');
|
||||
const gameId = row.getAttribute('data-id');
|
||||
|
||||
// визуально выделяем выбранную строку
|
||||
if (selectedRow) selectedRow.classList.remove('selected');
|
||||
row.classList.add('selected');
|
||||
selectedRow = row;
|
||||
|
||||
status.textContent = 'Сохраняю выбор...';
|
||||
|
||||
try {{
|
||||
const res = await fetch(`/select-game/${{gameId}}`, {{
|
||||
method: 'POST'
|
||||
}});
|
||||
if (!res.ok) throw new Error('HTTP ' + res.status);
|
||||
const data = await res.json();
|
||||
status.textContent = 'Выбран матч ID: ' + data.selected_id;
|
||||
}} catch (err) {{
|
||||
status.textContent = 'Ошибка: ' + err.message;
|
||||
}}
|
||||
}});
|
||||
</script>
|
||||
</html>
|
||||
"""
|
||||
return HTMLResponse(html)
|
||||
|
||||
|
||||
@app.post("/select-game/{game_id}")
|
||||
async def select_game(game_id: int):
|
||||
global selected_game_id, latest_game_data, latest_game_error
|
||||
selected_game_id = game_id
|
||||
|
||||
# моментально подтянуть первое состояние, если известен турнир
|
||||
if current_tournament_id:
|
||||
try:
|
||||
data = _fetch_game_once(current_tournament_id, selected_game_id)
|
||||
with _latest_lock:
|
||||
latest_game_data = {
|
||||
"tournament_id": current_tournament_id,
|
||||
"game_id": selected_game_id,
|
||||
"fetched_at": datetime.now().isoformat(),
|
||||
"data": data,
|
||||
}
|
||||
latest_game_error = None
|
||||
except Exception as e:
|
||||
with _latest_lock:
|
||||
latest_game_error = f"{type(e).__name__}: {e}"
|
||||
|
||||
return JSONResponse({"selected_id": selected_game_id})
|
||||
|
||||
|
||||
@app.get("/selected-game")
|
||||
async def get_selected_game():
|
||||
return JSONResponse({"selected_id": selected_game_id})
|
||||
|
||||
|
||||
@app.get("/game/url")
|
||||
async def game_url():
|
||||
if not (selected_game_id and current_tournament_id):
|
||||
return JSONResponse({"message": "game_id или tournament_id не задан"})
|
||||
return JSONResponse({
|
||||
"url": _build_game_url(current_tournament_id, selected_game_id),
|
||||
"game_id": selected_game_id,
|
||||
"tournament_id": current_tournament_id
|
||||
})
|
||||
|
||||
@app.get("/game/data")
|
||||
async def game_data():
|
||||
with _latest_lock:
|
||||
if latest_game_data:
|
||||
return JSONResponse(latest_game_data)
|
||||
if latest_game_error:
|
||||
return JSONResponse({"error": latest_game_error}, status_code=502)
|
||||
return JSONResponse({"message": "Ещё нет данных. Выберите матч и подождите первое обновление."})
|
||||
|
||||
if __name__ == "__main__":
|
||||
uvicorn.run(
|
||||
"get_data:app", host="0.0.0.0", port=8000, reload=True, log_level="debug"
|
||||
)
|
||||
8
requirements.txt
Normal file
8
requirements.txt
Normal file
@@ -0,0 +1,8 @@
|
||||
requests>=2.31.0
|
||||
pandas>=2.0.0
|
||||
numpy>=1.24.0
|
||||
fastapi>=0.115.0
|
||||
uvicorn>=0.30.0
|
||||
requests>=2.31.0
|
||||
python-telegram-handler
|
||||
python-dotenv>=1.1.0
|
||||
Reference in New Issue
Block a user