Compare commits
25 Commits
6de3607a02
...
main
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
5994e18efa | ||
|
|
5205dd5c5c | ||
|
|
ead3683e4c | ||
|
|
8f300e0154 | ||
| 6e92022e67 | |||
| 74801ad710 | |||
| 28b0f09ec2 | |||
|
|
c60caaa8aa | ||
|
|
9cbf32f831 | ||
|
|
d4d9584bb3 | ||
|
|
216138ceed | ||
|
|
0d06097181 | ||
| 71f1e62630 | |||
| d120630548 | |||
| 13c86660ac | |||
| df0b6f8678 | |||
| 3b7b447f40 | |||
| 711e217d6e | |||
| 9778efef99 | |||
|
|
fb12719e60 | ||
| e81eb5b0fc | |||
| 5894d2d8a5 | |||
| 78e28eaa92 | |||
| 17b1c9de8b | |||
| 6d1e83d6fe |
7
.gitignore
vendored
7
.gitignore
vendored
@@ -1 +1,8 @@
|
|||||||
.env
|
.env
|
||||||
|
/.venv
|
||||||
|
logs/
|
||||||
|
/logs/*
|
||||||
|
__pycache__/
|
||||||
|
*.pyc
|
||||||
|
*.pyd
|
||||||
|
*.pyo
|
||||||
Binary file not shown.
Binary file not shown.
395
deploy.sh
Normal file → Executable file
395
deploy.sh
Normal file → Executable file
@@ -4,15 +4,21 @@
|
|||||||
RED='\033[0;31m'
|
RED='\033[0;31m'
|
||||||
GREEN='\033[0;32m'
|
GREEN='\033[0;32m'
|
||||||
YELLOW='\033[1;33m'
|
YELLOW='\033[1;33m'
|
||||||
|
BLUE='\033[0;34m'
|
||||||
NC='\033[0m' # No Color
|
NC='\033[0m' # No Color
|
||||||
|
|
||||||
|
# Конфигурация
|
||||||
|
REPO_URL="https://git.tvstart.ru/ychernenko/KHL.git"
|
||||||
|
TARGET_DIR="/root/KHL"
|
||||||
|
SERVICE_NAME="khl-data.service"
|
||||||
|
TARGET_ENV="/mnt/khl/.env"
|
||||||
|
|
||||||
show_help() {
|
show_help() {
|
||||||
echo "Использование: $0 -t <домашняя команда> -r <релиз> [-l <лига>]"
|
echo "Использование: $0 -r <релиз> [-h]"
|
||||||
echo " -r Релиз (тег или ветка в git)"
|
echo " -r Релиз (тег или ветка в git)"
|
||||||
echo " -h Показать эту справку"
|
echo " -h Показать эту справку"
|
||||||
echo ""
|
echo ""
|
||||||
echo "Пример: $0 -r Barabanov_TEST"
|
echo "Пример: $0 -r Barabanov_TEST"
|
||||||
|
|
||||||
echo ""
|
echo ""
|
||||||
exit 0
|
exit 0
|
||||||
}
|
}
|
||||||
@@ -30,13 +36,41 @@ log_error() {
|
|||||||
echo -e "${RED}[ERROR]${NC} $1" >&2
|
echo -e "${RED}[ERROR]${NC} $1" >&2
|
||||||
}
|
}
|
||||||
|
|
||||||
|
log_debug() {
|
||||||
|
echo -e "${BLUE}[DEBUG]${NC} $1" >&2
|
||||||
|
}
|
||||||
|
|
||||||
|
# Функция проверки зависимостей системы
|
||||||
|
check_dependencies() {
|
||||||
|
local deps=("git" "python3" "pip" "netstat" "systemctl")
|
||||||
|
local missing=()
|
||||||
|
|
||||||
|
for dep in "${deps[@]}"; do
|
||||||
|
if ! command -v "$dep" &> /dev/null; then
|
||||||
|
missing+=("$dep")
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
if [ ${#missing[@]} -ne 0 ]; then
|
||||||
|
log_warn "Отсутствуют некоторые команды: ${missing[*]}"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
# Функция проверки и установки пакетов
|
# Функция проверки и установки пакетов
|
||||||
install_packages() {
|
install_packages() {
|
||||||
log_info "Обновление списка пакетов..."
|
log_info "Обновление списка пакетов..."
|
||||||
apt-get update
|
if ! apt-get update; then
|
||||||
|
log_error "Ошибка при обновлении списка пакетов"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
log_info "Установка необходимых пакетов..."
|
log_info "Установка необходимых пакетов..."
|
||||||
apt-get install -y python3 python3-pip python3-venv git net-tools
|
if ! apt-get install -y python3 python3-pip python3-venv git net-tools; then
|
||||||
|
log_error "Ошибка при установке пакетов"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
# Проверка установки
|
# Проверка установки
|
||||||
if ! command -v python3 &> /dev/null; then
|
if ! command -v python3 &> /dev/null; then
|
||||||
@@ -44,43 +78,47 @@ install_packages() {
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
if ! command -v pip3 &> /dev/null; then
|
if ! command -v pip &> /dev/null; then
|
||||||
log_error "pip3 не установлен!"
|
log_error "pip не установлен!"
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
log_info "Версия Python: $(python3 --version)"
|
log_info "Версия Python: $(python3 --version)"
|
||||||
log_info "Версия pip: $(pip3 --version)"
|
log_info "Версия pip: $(pip --version)"
|
||||||
}
|
}
|
||||||
|
|
||||||
# Функция загрузки кода
|
# Функция загрузки кода
|
||||||
download_code() {
|
download_code() {
|
||||||
local repo_url="https://git.tvstart.ru/ychernenko/KHL.git"
|
|
||||||
local release="$1"
|
local release="$1"
|
||||||
local target_dir="/root/KHL"
|
local target_dir="$TARGET_DIR"
|
||||||
|
|
||||||
log_info "Создание рабочей директории $target_dir..."
|
log_info "Создание рабочей директории $target_dir..."
|
||||||
mkdir -p $target_dir
|
mkdir -p "$target_dir"
|
||||||
|
|
||||||
if [ -d "$target_dir/.git" ]; then
|
if [ -d "$target_dir/.git" ]; then
|
||||||
log_info "Обновление существующего репозитория..."
|
log_info "Обновление существующего репозитория..."
|
||||||
cd $target_dir
|
cd "$target_dir" || exit 1
|
||||||
git fetch --all
|
git fetch --all
|
||||||
git checkout $release
|
if ! git checkout "$release"; then
|
||||||
git pull origin $release
|
log_error "Не удалось переключиться на релиз $release"
|
||||||
|
log_info "Доступные ветки:"
|
||||||
|
git branch -r
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
git pull origin "$release"
|
||||||
else
|
else
|
||||||
log_info "Клонирование репозитория $repo_url..."
|
log_info "Клонирование репозитория $REPO_URL..."
|
||||||
git clone $repo_url $target_dir
|
if ! git clone "$REPO_URL" "$target_dir"; then
|
||||||
cd $target_dir
|
log_error "Ошибка при клонировании репозитория"
|
||||||
git checkout $release
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
cd "$target_dir" || exit 1
|
||||||
# Проверка успешности checkout
|
if ! git checkout "$release"; then
|
||||||
if [ $? -ne 0 ]; then
|
log_error "Не удалось переключиться на релиз $release"
|
||||||
log_error "Не удалось переключиться на релиз $release"
|
log_info "Доступные ветки:"
|
||||||
log_error "Доступные ветки:"
|
git branch -r
|
||||||
git branch -r
|
exit 1
|
||||||
exit 1
|
fi
|
||||||
fi
|
fi
|
||||||
|
|
||||||
log_info "Код успешно загружен (релиз: $release)"
|
log_info "Код успешно загружен (релиз: $release)"
|
||||||
@@ -90,11 +128,20 @@ download_code() {
|
|||||||
|
|
||||||
# Функция настройки виртуального окружения
|
# Функция настройки виртуального окружения
|
||||||
setup_venv() {
|
setup_venv() {
|
||||||
local target_dir="/root/KHL"
|
local target_dir="$TARGET_DIR"
|
||||||
cd $target_dir
|
cd "$target_dir" || exit 1
|
||||||
|
|
||||||
|
# Удаляем существующее виртуальное окружение для чистоты
|
||||||
|
if [ -d ".venv" ]; then
|
||||||
|
log_info "Удаление существующего виртуального окружения..."
|
||||||
|
rm -rf .venv
|
||||||
|
fi
|
||||||
|
|
||||||
log_info "Создание виртуального окружения..."
|
log_info "Создание виртуального окружения..."
|
||||||
python3 -m venv .venv
|
if ! python3 -m venv .venv; then
|
||||||
|
log_error "Не удалось создать виртуальное окружение"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
if [ ! -f ".venv/bin/activate" ]; then
|
if [ ! -f ".venv/bin/activate" ]; then
|
||||||
log_error "Не удалось создать виртуальное окружение"
|
log_error "Не удалось создать виртуальное окружение"
|
||||||
@@ -105,7 +152,9 @@ setup_venv() {
|
|||||||
source .venv/bin/activate
|
source .venv/bin/activate
|
||||||
|
|
||||||
# Обновление pip
|
# Обновление pip
|
||||||
pip install --upgrade pip
|
if ! pip install --upgrade pip; then
|
||||||
|
log_warn "Не удалось обновить pip"
|
||||||
|
fi
|
||||||
|
|
||||||
# Проверка наличия requirements.txt
|
# Проверка наличия requirements.txt
|
||||||
if [ -f "requirements.txt" ]; then
|
if [ -f "requirements.txt" ]; then
|
||||||
@@ -114,24 +163,29 @@ setup_venv() {
|
|||||||
log_info "Все зависимости успешно установлены"
|
log_info "Все зависимости успешно установлены"
|
||||||
else
|
else
|
||||||
log_error "Ошибка при установке зависимостей из requirements.txt"
|
log_error "Ошибка при установке зависимостей из requirements.txt"
|
||||||
log_warn "Попытка установки базовых зависимостей..."
|
exit 1
|
||||||
pip install requests pandas numpy
|
|
||||||
fi
|
fi
|
||||||
else
|
else
|
||||||
log_warn "Файл requirements.txt не найден, устанавливаем базовые зависимости..."
|
log_warn "Файл requirements.txt не найден, устанавливаем базовые зависимости..."
|
||||||
pip install requests pandas numpy
|
if ! pip install requests pandas numpy fastapi uvicorn python-telegram-handler python-dotenv; then
|
||||||
|
log_error "Ошибка при установке базовых зависимостей"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# Проверка основных пакетов
|
# Проверка основных пакетов
|
||||||
log_info "Проверка установки основных пакетов..."
|
log_info "Проверка установки основных пакетов..."
|
||||||
python -c "
|
if ! python -c "
|
||||||
try:
|
try:
|
||||||
import requests, pandas, numpy
|
import requests, pandas, numpy
|
||||||
print('✓ Все основные пакеты успешно импортируются')
|
print('✓ Все основные пакеты успешно импортируются')
|
||||||
except ImportError as e:
|
except ImportError as e:
|
||||||
print(f'✗ Ошибка импорта: {e}')
|
print(f'✗ Ошибка импорта: {e}')
|
||||||
exit(1)
|
exit(1)
|
||||||
"
|
"; then
|
||||||
|
log_error "Ошибка при проверке пакетов"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
log_info "Установленные пакеты:"
|
log_info "Установленные пакеты:"
|
||||||
pip list --format=columns
|
pip list --format=columns
|
||||||
@@ -139,50 +193,52 @@ except ImportError as e:
|
|||||||
|
|
||||||
# Функция создания systemd сервиса
|
# Функция создания systemd сервиса
|
||||||
create_systemd_service() {
|
create_systemd_service() {
|
||||||
local team="$1"
|
log_info "Создание systemd сервиса"
|
||||||
local league="$2"
|
|
||||||
|
|
||||||
log_info "Создание systemd сервиса для команды: $team"
|
|
||||||
if [ -n "$league" ]; then
|
|
||||||
log_info "Лига: $league"
|
|
||||||
fi
|
|
||||||
|
|
||||||
# Останавливаем и отключаем старый сервис если он есть
|
# Останавливаем и отключаем старый сервис если он есть
|
||||||
if systemctl is-active --quiet khl-data.service; then
|
if systemctl is-active --quiet "$SERVICE_NAME"; then
|
||||||
log_info "Остановка khl-data.service..."
|
log_info "Остановка $SERVICE_NAME..."
|
||||||
systemctl stop khl-data.service
|
systemctl stop "$SERVICE_NAME"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
if systemctl is-enabled --quiet khl-data.service; then
|
if systemctl is-enabled --quiet "$SERVICE_NAME"; then
|
||||||
log_info "Отключение khl-data.service..."
|
log_info "Отключение $SERVICE_NAME..."
|
||||||
systemctl disable khl-data.service
|
systemctl disable "$SERVICE_NAME"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# Формируем команду для data сервиса
|
# Формируем команду для data сервиса
|
||||||
local data_command="/root/KHL/.venv/bin/python3 /root/KHL/get_data.py""
|
local data_service_file="/etc/systemd/system/$SERVICE_NAME"
|
||||||
if [ -n "$league" ]; then
|
|
||||||
data_command="$data_command --league \"$league\""
|
|
||||||
fi
|
|
||||||
|
|
||||||
# Создаем сервисный файл в правильной директории
|
|
||||||
local data_service_file="/etc/systemd/system/khl-data.service"
|
|
||||||
|
|
||||||
log_info "Создание файла сервиса: $data_service_file"
|
log_info "Создание файла сервиса: $data_service_file"
|
||||||
cat > "$data_service_file" << EOF
|
cat > "$data_service_file" << EOF
|
||||||
[Unit]
|
[Unit]
|
||||||
Description=khl Data Service
|
Description=KHL Data Service
|
||||||
|
Documentation=https://git.tvstart.ru/ychernenko/KHL
|
||||||
After=network.target
|
After=network.target
|
||||||
|
Wants=network.target
|
||||||
|
|
||||||
[Service]
|
[Service]
|
||||||
Type=simple
|
Type=simple
|
||||||
User=root
|
User=root
|
||||||
WorkingDirectory=/root/KHL
|
WorkingDirectory=$TARGET_DIR
|
||||||
Environment=PATH=/root/KHL/.venv/bin
|
Environment=PATH=$TARGET_DIR/.venv/bin
|
||||||
ExecStart=$data_command
|
EnvironmentFile=$TARGET_ENV
|
||||||
Restart=always
|
ExecStart=python3 $TARGET_DIR/get_data.py
|
||||||
RestartSec=30
|
|
||||||
|
# Лимиты ресурсов
|
||||||
|
MemoryMax=1G
|
||||||
|
CPUQuota=80%
|
||||||
|
|
||||||
|
# Логирование
|
||||||
StandardOutput=journal
|
StandardOutput=journal
|
||||||
StandardError=journal
|
StandardError=journal
|
||||||
|
SyslogIdentifier=KHL
|
||||||
|
|
||||||
|
# Поведение при перезапуске
|
||||||
|
Restart=always
|
||||||
|
RestartSec=10
|
||||||
|
StartLimitInterval=300
|
||||||
|
StartLimitBurst=5
|
||||||
|
|
||||||
[Install]
|
[Install]
|
||||||
WantedBy=multi-user.target
|
WantedBy=multi-user.target
|
||||||
@@ -192,16 +248,22 @@ EOF
|
|||||||
chmod 644 "$data_service_file"
|
chmod 644 "$data_service_file"
|
||||||
|
|
||||||
log_info "Перезагрузка systemd демона..."
|
log_info "Перезагрузка systemd демона..."
|
||||||
systemctl daemon-reload
|
if ! systemctl daemon-reload; then
|
||||||
|
log_error "Ошибка при перезагрузке systemd демона"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
log_info "Включение сервиса khl-data.service..."
|
log_info "Включение сервиса $SERVICE_NAME..."
|
||||||
systemctl enable khl-data.service
|
if ! systemctl enable "$SERVICE_NAME"; then
|
||||||
|
log_error "Ошибка при включении сервиса"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
# Функция проверки файлов
|
# Функция проверки файлов
|
||||||
check_required_files() {
|
check_required_files() {
|
||||||
local target_dir="/root/KHL"
|
local target_dir="$TARGET_DIR"
|
||||||
cd $target_dir
|
cd "$target_dir" || exit 1
|
||||||
|
|
||||||
local missing_files=()
|
local missing_files=()
|
||||||
local required_files=("get_data.py")
|
local required_files=("get_data.py")
|
||||||
@@ -227,74 +289,26 @@ check_required_files() {
|
|||||||
fi
|
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
|
# Функция настройки firewall
|
||||||
setup_firewall() {
|
setup_firewall() {
|
||||||
log_info "Настройка firewall (открытие порта 8000)..."
|
log_info "Настройка firewall (открытие порта 8000)..."
|
||||||
|
|
||||||
# Проверяем наличие ufw
|
# Проверяем наличие ufw
|
||||||
if command -v ufw &> /dev/null && systemctl is-active --quiet ufw; then
|
if command -v ufw &> /dev/null && systemctl is-active --quiet ufw; then
|
||||||
ufw allow 8000/tcp
|
if ufw allow 8000/tcp; then
|
||||||
log_info "Порт 8000 открыт в ufw"
|
log_info "Порт 8000 открыт в ufw"
|
||||||
|
else
|
||||||
|
log_warn "Не удалось открыть порт 8000 в ufw"
|
||||||
|
fi
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# Для firewalld (CentOS/RHEL)
|
# Для firewalld (CentOS/RHEL)
|
||||||
if command -v firewall-cmd &> /dev/null && systemctl is-active --quiet firewalld; then
|
if command -v firewall-cmd &> /dev/null && systemctl is-active --quiet firewalld; then
|
||||||
firewall-cmd --permanent --add-port=8000/tcp
|
if firewall-cmd --permanent --add-port=8000/tcp && firewall-cmd --reload; then
|
||||||
firewall-cmd --reload
|
log_info "Порт 8000 открыт в firewalld"
|
||||||
log_info "Порт 8000 открыт в firewalld"
|
else
|
||||||
|
log_warn "Не удалось открыть порт 8000 в firewalld"
|
||||||
|
fi
|
||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -303,9 +317,9 @@ check_port() {
|
|||||||
local port=8000
|
local port=8000
|
||||||
local occupied=false
|
local occupied=false
|
||||||
|
|
||||||
if command -v netstat &> /dev/null && netstat -tuln 2>/dev/null | grep ":$port " > /dev/null; then
|
if command -v netstat &> /dev/null && netstat -tuln 2>/dev/null | grep -q ":$port "; then
|
||||||
occupied=true
|
occupied=true
|
||||||
elif command -v ss &> /dev/null && ss -tuln 2>/dev/null | grep ":$port " > /dev/null; then
|
elif command -v ss &> /dev/null && ss -tuln 2>/dev/null | grep -q ":$port "; then
|
||||||
occupied=true
|
occupied=true
|
||||||
fi
|
fi
|
||||||
|
|
||||||
@@ -322,30 +336,42 @@ manage_services() {
|
|||||||
|
|
||||||
case $action in
|
case $action in
|
||||||
"start")
|
"start")
|
||||||
log_info "Запуск сервиса khl-data.service..."
|
log_info "Запуск сервиса $SERVICE_NAME..."
|
||||||
systemctl start khl-data.service
|
if ! systemctl start "$SERVICE_NAME"; then
|
||||||
|
log_error "Ошибка при запуске сервиса"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
;;
|
;;
|
||||||
"restart")
|
"restart")
|
||||||
log_info "Перезапуск сервиса khl-data.service..."
|
log_info "Перезапуск сервиса $SERVICE_NAME..."
|
||||||
systemctl restart khl-data.service
|
if ! systemctl restart "$SERVICE_NAME"; then
|
||||||
|
log_error "Ошибка при перезапуске сервиса"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
;;
|
;;
|
||||||
"stop")
|
"stop")
|
||||||
log_info "Остановка сервиса khl-data.service..."
|
log_info "Остановка сервиса $SERVICE_NAME..."
|
||||||
systemctl stop khl-data.service
|
if ! systemctl stop "$SERVICE_NAME"; then
|
||||||
|
log_error "Ошибка при остановке сервиса"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
;;
|
;;
|
||||||
esac
|
esac
|
||||||
|
|
||||||
# Даем время сервису на запуск/остановку
|
# Даем время сервису на запуск/остановку
|
||||||
sleep 3
|
sleep 3
|
||||||
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
# Функция проверки статуса сервиса
|
# Функция проверки статуса сервиса
|
||||||
check_service_status() {
|
check_service_status() {
|
||||||
log_info "Статус сервиса khl-data.service:"
|
log_info "Статус сервиса $SERVICE_NAME:"
|
||||||
if systemctl is-active khl-data.service; then
|
if systemctl is-active "$SERVICE_NAME" &>/dev/null; then
|
||||||
systemctl status khl-data.service --no-pager -l
|
systemctl status "$SERVICE_NAME" --no-pager -l
|
||||||
|
return 0
|
||||||
else
|
else
|
||||||
log_warn "Сервис khl-data.service не запущен"
|
log_warn "Сервис $SERVICE_NAME не запущен"
|
||||||
|
return 1
|
||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -360,7 +386,12 @@ check_system() {
|
|||||||
fi
|
fi
|
||||||
|
|
||||||
# Проверка свободного места
|
# Проверка свободного места
|
||||||
local available_space=$(df /root --output=avail | tail -1)
|
local available_space
|
||||||
|
available_space=$(df "$TARGET_DIR" --output=avail 2>/dev/null | tail -1)
|
||||||
|
if [ -z "$available_space" ]; then
|
||||||
|
available_space=$(df / --output=avail | tail -1)
|
||||||
|
fi
|
||||||
|
|
||||||
if [ "$available_space" -lt 1048576 ]; then # Меньше 1GB
|
if [ "$available_space" -lt 1048576 ]; then # Меньше 1GB
|
||||||
log_warn "Мало свободного места: $((available_space / 1024)) MB"
|
log_warn "Мало свободного места: $((available_space / 1024)) MB"
|
||||||
else
|
else
|
||||||
@@ -368,39 +399,46 @@ check_system() {
|
|||||||
fi
|
fi
|
||||||
|
|
||||||
# Проверка памяти
|
# Проверка памяти
|
||||||
local total_mem=$(grep MemTotal /proc/meminfo | awk '{print $2}')
|
if [ -f /proc/meminfo ]; then
|
||||||
log_info "Оперативная память: $((total_mem / 1024)) MB"
|
local total_mem
|
||||||
|
total_mem=$(grep MemTotal /proc/meminfo | awk '{print $2}')
|
||||||
|
log_info "Оперативная память: $((total_mem / 1024)) MB"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Проверка архитектуры
|
||||||
|
log_info "Архитектура: $(uname -m)"
|
||||||
|
}
|
||||||
|
|
||||||
|
# Функция очистки при прерывании
|
||||||
|
cleanup() {
|
||||||
|
log_info "Очистка..."
|
||||||
|
if [ -f "/etc/systemd/system/$SERVICE_NAME" ]; then
|
||||||
|
manage_services "stop"
|
||||||
|
systemctl disable "$SERVICE_NAME" 2>/dev/null || true
|
||||||
|
rm -f "/etc/systemd/system/$SERVICE_NAME"
|
||||||
|
systemctl daemon-reload
|
||||||
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
# Основная функция
|
# Основная функция
|
||||||
main() {
|
main() {
|
||||||
local team=""
|
|
||||||
local release="main" # значение по умолчанию
|
local release="main" # значение по умолчанию
|
||||||
local league="" # переменная для лиги
|
|
||||||
|
|
||||||
# Обработка аргументов командной строки
|
# Обработка аргументов командной строки
|
||||||
while getopts "t:r:l:h" opt; do
|
while getopts "r:h" opt; do
|
||||||
case $opt in
|
case $opt in
|
||||||
t) team="$OPTARG" ;;
|
|
||||||
r) release="$OPTARG" ;;
|
r) release="$OPTARG" ;;
|
||||||
l) league="$OPTARG" ;;
|
|
||||||
h) show_help ;;
|
h) show_help ;;
|
||||||
*) log_error "Неверный аргумент"; exit 1 ;;
|
*) log_error "Неверный аргумент"; exit 1 ;;
|
||||||
esac
|
esac
|
||||||
done
|
done
|
||||||
|
|
||||||
# Проверка обязательных аргументов
|
log_info "Начало установки KHL Data Service..."
|
||||||
if [ -z "$team" ] && [ -z "$release" ]; then
|
if [ -n "$release" ]; then
|
||||||
log_error "Необходимо указать команду (-t) и релиз (-r)"
|
log_info "Релиз: $release"
|
||||||
show_help
|
|
||||||
fi
|
|
||||||
|
|
||||||
log_info "Начало установки khl Data Service..."
|
|
||||||
log_info "Команда: $team, Релиз: $release"
|
|
||||||
if [ -n "$league" ]; then
|
|
||||||
log_info "Лига: $league"
|
|
||||||
else
|
else
|
||||||
log_info "Лига: не указана (будет использовано значение по умолчанию)"
|
log_info "Релиз: не указан (будет использовано значение по умолчанию: main)"
|
||||||
|
release="main"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# Проверка прав root
|
# Проверка прав root
|
||||||
@@ -409,9 +447,15 @@ main() {
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
# Установка обработчика прерывания
|
||||||
|
trap 'log_error "Прервано пользователем"; cleanup; exit 1' INT TERM
|
||||||
|
|
||||||
# Проверка системы
|
# Проверка системы
|
||||||
check_system
|
check_system
|
||||||
|
|
||||||
|
# Проверка зависимостей
|
||||||
|
check_dependencies
|
||||||
|
|
||||||
# Установка пакетов
|
# Установка пакетов
|
||||||
install_packages
|
install_packages
|
||||||
|
|
||||||
@@ -424,16 +468,6 @@ main() {
|
|||||||
# Настройка виртуального окружения
|
# Настройка виртуального окружения
|
||||||
setup_venv
|
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
|
# Настройка firewall
|
||||||
setup_firewall
|
setup_firewall
|
||||||
|
|
||||||
@@ -441,46 +475,47 @@ main() {
|
|||||||
check_port
|
check_port
|
||||||
|
|
||||||
# Создание systemd сервиса
|
# Создание systemd сервиса
|
||||||
create_systemd_service "$final_team" "$league"
|
create_systemd_service
|
||||||
|
|
||||||
log_info "Настройка завершена!"
|
log_info "Настройка завершена!"
|
||||||
|
|
||||||
# Запуск сервиса
|
# Запуск сервиса
|
||||||
manage_services "start"
|
if manage_services "start"; then
|
||||||
|
log_info "Сервис успешно запущен"
|
||||||
|
else
|
||||||
|
log_error "Не удалось запустить сервис"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
# Проверка статуса
|
# Проверка статуса
|
||||||
check_service_status
|
check_service_status
|
||||||
|
|
||||||
# Получаем IP для вывода
|
# Получаем IP для вывода
|
||||||
|
local ip_address
|
||||||
ip_address=$(ip route get 1 2>/dev/null | awk '{print $7; exit}' ||
|
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]*' |
|
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)
|
grep -Eo '([0-9]*\.){3}[0-9]*' | grep -v '127.0.0.1' | head -n1)
|
||||||
|
|
||||||
log_info "=================================================="
|
log_info "=================================================="
|
||||||
log_info "Установка завершена успешно!"
|
log_info "Установка завершена успешно!"
|
||||||
log_info "Команда: $final_team"
|
log_info "Релиз: $release"
|
||||||
if [ -n "$league" ]; then
|
log_info "Рабочая директория: $TARGET_DIR"
|
||||||
log_info "Лига: $league"
|
|
||||||
fi
|
|
||||||
log_info "Режим: $release"
|
|
||||||
log_info ""
|
log_info ""
|
||||||
log_info "Для просмотра логов:"
|
log_info "Для просмотра логов:"
|
||||||
log_info " journalctl -u khl-data.service -f"
|
log_info " journalctl -u $SERVICE_NAME -f"
|
||||||
|
log_info " journalctl -t KHL -f"
|
||||||
log_info ""
|
log_info ""
|
||||||
log_info "Управление сервисом:"
|
log_info "Управление сервисом:"
|
||||||
log_info " Перезапуск: systemctl restart khl-data.service"
|
log_info " Перезапуск: systemctl restart $SERVICE_NAME"
|
||||||
log_info " Остановка: systemctl stop khl-data.service"
|
log_info " Остановка: systemctl stop $SERVICE_NAME"
|
||||||
log_info " Статус: systemctl status khl-data.service"
|
log_info " Статус: systemctl status $SERVICE_NAME"
|
||||||
log_info " Логи: journalctl -u khl-data.service"
|
log_info " Логи: journalctl -u $SERVICE_NAME"
|
||||||
log_info ""
|
log_info ""
|
||||||
log_info "Проверка работы:"
|
log_info "Проверка работы:"
|
||||||
log_info " Проверить процессы: ps aux | grep get_data.py"
|
log_info " Проверить процессы: ps aux | grep get_data.py"
|
||||||
log_info " Проверить логи: tail -f /root/KHL/logs/*.log 2>/dev/null || echo 'Директория логов не найдена'"
|
log_info " Проверить логи: tail -f $TARGET_DIR/logs/*.log 2>/dev/null || echo 'Директория логов не найдена'"
|
||||||
log_info "=================================================="
|
log_info "=================================================="
|
||||||
}
|
}
|
||||||
|
|
||||||
# Обработка прерывания
|
|
||||||
trap 'log_error "Прервано пользователем"; exit 1' INT TERM
|
|
||||||
|
|
||||||
# Запуск основной функции
|
# Запуск основной функции
|
||||||
main "$@"
|
main "$@"
|
||||||
348
get_data.py
348
get_data.py
@@ -1,5 +1,5 @@
|
|||||||
from fastapi import FastAPI, HTTPException, Query
|
from fastapi import FastAPI, HTTPException, Query, Request
|
||||||
from fastapi.responses import Response, JSONResponse, HTMLResponse
|
from fastapi.responses import Response, JSONResponse, HTMLResponse, StreamingResponse
|
||||||
from fastapi.encoders import jsonable_encoder
|
from fastapi.encoders import jsonable_encoder
|
||||||
import pandas as pd
|
import pandas as pd
|
||||||
import requests, io, os
|
import requests, io, os
|
||||||
@@ -10,39 +10,39 @@ import uvicorn
|
|||||||
from threading import Thread, Event, Lock
|
from threading import Thread, Event, Lock
|
||||||
import time
|
import time
|
||||||
from contextlib import asynccontextmanager
|
from contextlib import asynccontextmanager
|
||||||
from smb_excel import read_excel_from_smb, fetch_smb_file
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
import nasio
|
||||||
|
import logging
|
||||||
|
import logging.config
|
||||||
|
import platform
|
||||||
|
import json
|
||||||
|
from pprint import pprint
|
||||||
|
|
||||||
|
|
||||||
# --- Глобальные переменные ---
|
# --- Глобальные переменные ---
|
||||||
selected_game_id: int | None = None
|
selected_game_id: int | None = None
|
||||||
current_tournament_id: int | None = None # будем обновлять при загрузке расписания
|
current_tournament_id: int | None = None # будем обновлять при загрузке расписания
|
||||||
current_season: str | None = None # будем обновлять при загрузке расписания
|
current_season: str | None = None # будем обновлять при загрузке расписания
|
||||||
latest_game_data: dict | None = None # сюда кладём последние данные по матчу
|
latest_game_data: dict | None = None # сюда кладём последние данные по матчу
|
||||||
latest_game_error: str | None = None
|
latest_game_error: str | None = None
|
||||||
_latest_lock = Lock()
|
_latest_lock = Lock()
|
||||||
_stop_event = Event()
|
_stop_event = Event()
|
||||||
_worker_thread: Thread | None = None
|
_worker_thread: Thread | None = None
|
||||||
|
|
||||||
# Загружаем переменные из .env
|
pprint(f"Локальный файл окружения ={load_dotenv(verbose=True)}")
|
||||||
load_dotenv()
|
|
||||||
api_user = os.getenv("API_USER")
|
api_user = os.getenv("API_USER")
|
||||||
api_pass = os.getenv("API_PASS")
|
api_pass = os.getenv("API_PASS")
|
||||||
league = os.getenv("LEAGUE")
|
league = os.getenv("LEAGUE")
|
||||||
|
api_base_url = os.getenv("API_BASE_URL")
|
||||||
POLL_SEC = int(os.getenv("GAME_POLL_SECONDS"))
|
POLL_SEC = int(os.getenv("GAME_POLL_SECONDS"))
|
||||||
SERVER_NAME = os.getenv("SERVER_NAME")
|
SERVER_NAME = os.getenv("SYNO_URL")
|
||||||
SERVER_IP = os.getenv("SERVER_IP")
|
USER = os.getenv("SYNO_USERNAME")
|
||||||
SHARE = os.getenv("SHARE")
|
PASSWORD = os.getenv("SYNO_PASSWORD")
|
||||||
PATH_IN_SHARE = os.getenv("PATH_IN_SHARE")
|
PATH = f'{os.getenv("SYNO_PATH")}MATCH.xlsm'
|
||||||
USER = os.getenv("USER")
|
|
||||||
PASSWORD = os.getenv("PASSWORD")
|
|
||||||
DOMAIN = os.getenv("DOMAIN")
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def load_today_schedule():
|
def load_today_schedule():
|
||||||
"""Возвращает DataFrame матчей на сегодня с нужными колонками (или пустой DF)."""
|
"""Возвращает DataFrame матчей на сегодня с нужными колонками (или пустой DF)."""
|
||||||
url_tournaments = "http://stat2tv.khl.ru/tournaments.xml"
|
url_tournaments = f"{api_base_url}tournaments.xml"
|
||||||
r = requests.get(
|
r = requests.get(
|
||||||
url_tournaments, auth=HTTPBasicAuth(api_user, api_pass), verify=False
|
url_tournaments, auth=HTTPBasicAuth(api_user, api_pass), verify=False
|
||||||
)
|
)
|
||||||
@@ -66,12 +66,22 @@ def load_today_schedule():
|
|||||||
global current_tournament_id, current_season
|
global current_tournament_id, current_season
|
||||||
current_tournament_id = tournament_id
|
current_tournament_id = tournament_id
|
||||||
current_season = season
|
current_season = season
|
||||||
url_schedule = f"http://stat2tv.khl.ru/{tournament_id}/schedule-{tournament_id}.xml"
|
url_schedule = f"{api_base_url}{tournament_id}/schedule-{tournament_id}.xml"
|
||||||
r = requests.get(url_schedule, auth=HTTPBasicAuth(api_user, api_pass), verify=False)
|
r = requests.get(url_schedule, auth=HTTPBasicAuth(api_user, api_pass), verify=False)
|
||||||
schedule_df = pd.read_xml(io.StringIO(r.text))
|
schedule_df = pd.read_xml(io.StringIO(r.text))
|
||||||
|
|
||||||
# Нужные колонки (скорректируй под реальные имена из XML)
|
# Нужные колонки (скорректируй под реальные имена из XML)
|
||||||
needed_columns = ["id", "date", "time", "homeName_en", "visitorName_en", "arena_en", "arena_city_en", "homeCity_en", "visitorCity_en"]
|
needed_columns = [
|
||||||
|
"id",
|
||||||
|
"date",
|
||||||
|
"time",
|
||||||
|
"homeName_en",
|
||||||
|
"visitorName_en",
|
||||||
|
"arena_en",
|
||||||
|
"arena_city_en",
|
||||||
|
"homeCity_en",
|
||||||
|
"visitorCity_en",
|
||||||
|
]
|
||||||
exist = [c for c in needed_columns if c in schedule_df.columns]
|
exist = [c for c in needed_columns if c in schedule_df.columns]
|
||||||
schedule_df = schedule_df[exist].copy()
|
schedule_df = schedule_df[exist].copy()
|
||||||
|
|
||||||
@@ -114,17 +124,14 @@ def load_today_schedule():
|
|||||||
def _build_game_url(tournament_id: int, game_id: int) -> str:
|
def _build_game_url(tournament_id: int, game_id: int) -> str:
|
||||||
# URL по аналогии с расписанием: .../{tournament_id}/json_en/{game_id}.json
|
# URL по аналогии с расписанием: .../{tournament_id}/json_en/{game_id}.json
|
||||||
# Если у тебя другой шаблон — просто поменяй строку ниже.
|
# Если у тебя другой шаблон — просто поменяй строку ниже.
|
||||||
return f"http://stat2tv.khl.ru/{tournament_id}/json_en/{game_id}.json"
|
return f"{api_base_url}{tournament_id}/json_en/{game_id}.json"
|
||||||
|
|
||||||
|
|
||||||
def _fetch_game_once(tournament_id: int, game_id: int) -> dict:
|
def _fetch_game_once(tournament_id: int, game_id: int) -> dict:
|
||||||
"""Один запрос к API матча -> чистый JSON из API."""
|
"""Один запрос к API матча -> чистый JSON из API."""
|
||||||
url = _build_game_url(tournament_id, game_id)
|
url = _build_game_url(tournament_id, game_id)
|
||||||
r = requests.get(
|
r = requests.get(
|
||||||
url,
|
url, auth=HTTPBasicAuth(api_user, api_pass), verify=False, timeout=10
|
||||||
auth=HTTPBasicAuth(api_user, api_pass),
|
|
||||||
verify=False,
|
|
||||||
timeout=10
|
|
||||||
)
|
)
|
||||||
r.raise_for_status()
|
r.raise_for_status()
|
||||||
|
|
||||||
@@ -180,11 +187,12 @@ async def lifespan(app: FastAPI):
|
|||||||
_worker_thread.join(timeout=2)
|
_worker_thread.join(timeout=2)
|
||||||
print("🛑 Background thread stopped")
|
print("🛑 Background thread stopped")
|
||||||
|
|
||||||
|
|
||||||
app = FastAPI(
|
app = FastAPI(
|
||||||
lifespan=lifespan,
|
lifespan=lifespan,
|
||||||
docs_url=None, # ❌ отключает /docs
|
docs_url=None, # ❌ отключает /docs
|
||||||
redoc_url=None, # ❌ отключает /redoc
|
redoc_url=None, # ❌ отключает /redoc
|
||||||
openapi_url=None # ❌ отключает /openapi.json
|
openapi_url=None, # ❌ отключает /openapi.json
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@@ -199,8 +207,6 @@ async def games():
|
|||||||
return Response(content=json_schedule, media_type="application/json")
|
return Response(content=json_schedule, media_type="application/json")
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@app.get("/select")
|
@app.get("/select")
|
||||||
async def select():
|
async def select():
|
||||||
df = load_today_schedule()
|
df = load_today_schedule()
|
||||||
@@ -333,11 +339,13 @@ async def get_selected_game():
|
|||||||
async def game_url():
|
async def game_url():
|
||||||
if not (selected_game_id and current_tournament_id):
|
if not (selected_game_id and current_tournament_id):
|
||||||
return JSONResponse({"message": "game_id или tournament_id не задан"})
|
return JSONResponse({"message": "game_id или tournament_id не задан"})
|
||||||
return JSONResponse({
|
return JSONResponse(
|
||||||
"url": _build_game_url(current_tournament_id, selected_game_id),
|
{
|
||||||
"game_id": selected_game_id,
|
"url": _build_game_url(current_tournament_id, selected_game_id),
|
||||||
"tournament_id": current_tournament_id
|
"game_id": selected_game_id,
|
||||||
})
|
"tournament_id": current_tournament_id,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
# @app.get("/info")
|
# @app.get("/info")
|
||||||
@@ -353,7 +361,9 @@ async def game_data():
|
|||||||
return JSONResponse(latest_game_data)
|
return JSONResponse(latest_game_data)
|
||||||
if latest_game_error:
|
if latest_game_error:
|
||||||
return JSONResponse({"error": latest_game_error}, status_code=502)
|
return JSONResponse({"error": latest_game_error}, status_code=502)
|
||||||
return JSONResponse({"message": "Ещё нет данных. Выберите матч и подождите первое обновление."})
|
return JSONResponse(
|
||||||
|
{"message": "Ещё нет данных. Выберите матч и подождите первое обновление."}
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@app.get("/referee")
|
@app.get("/referee")
|
||||||
@@ -380,14 +390,13 @@ async def referee():
|
|||||||
return data_referees
|
return data_referees
|
||||||
|
|
||||||
|
|
||||||
|
async def team(who: str):
|
||||||
async def team(who:str):
|
""" "who: A - домашняя команда, B - гостевая"""
|
||||||
""""who: A - домашняя команда, B - гостевая"""
|
|
||||||
with _latest_lock:
|
with _latest_lock:
|
||||||
lgd = latest_game_data
|
lgd = latest_game_data
|
||||||
# print(lgd)
|
# print(lgd)
|
||||||
if not lgd or "data" not in lgd:
|
if not lgd or "data" not in lgd:
|
||||||
return [{"details":"Нет данных по матчу!"}]
|
return [{"details": "Нет данных по матчу!"}]
|
||||||
|
|
||||||
players1_temp = lgd["data"]["players"][who]
|
players1_temp = lgd["data"]["players"][who]
|
||||||
players1 = []
|
players1 = []
|
||||||
@@ -486,10 +495,12 @@ async def team(who:str):
|
|||||||
async def team1():
|
async def team1():
|
||||||
return await team("A")
|
return await team("A")
|
||||||
|
|
||||||
|
|
||||||
@app.get("/team2")
|
@app.get("/team2")
|
||||||
async def team2():
|
async def team2():
|
||||||
return await team("B")
|
return await team("B")
|
||||||
|
|
||||||
|
|
||||||
# 👉 метка для первой строки (period row)
|
# 👉 метка для первой строки (period row)
|
||||||
def _period_label(period: str | int) -> str:
|
def _period_label(period: str | int) -> str:
|
||||||
s = str(period).strip().upper()
|
s = str(period).strip().upper()
|
||||||
@@ -513,6 +524,7 @@ def _period_label(period: str | int) -> str:
|
|||||||
return "FINAL"
|
return "FINAL"
|
||||||
return ""
|
return ""
|
||||||
|
|
||||||
|
|
||||||
# 👉 сортировка периодов: 1,2,3, затем OT/OT2/...
|
# 👉 сортировка периодов: 1,2,3, затем OT/OT2/...
|
||||||
def _period_sort_key(k: str) -> tuple[int, int]:
|
def _period_sort_key(k: str) -> tuple[int, int]:
|
||||||
s = str(k).strip().upper()
|
s = str(k).strip().upper()
|
||||||
@@ -579,15 +591,21 @@ def format_team_stat(team1: dict, team2: dict, period: str | None = None) -> lis
|
|||||||
formatted = []
|
formatted = []
|
||||||
|
|
||||||
if period is not None:
|
if period is not None:
|
||||||
formatted.append({
|
formatted.append(
|
||||||
"name0": "period",
|
{
|
||||||
"name1": str(period),
|
"name0": "period",
|
||||||
"name2": "",
|
"name1": str(period),
|
||||||
"StatParameterGFX": _period_label(period) or "Period"
|
"name2": "",
|
||||||
})
|
"StatParameterGFX": _period_label(period) or "Period",
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
for key in keys:
|
for key in keys:
|
||||||
row = {"name0": key, "name1": teams[0].get(key, ""), "name2": teams[1].get(key, "")}
|
row = {
|
||||||
|
"name0": key,
|
||||||
|
"name1": teams[0].get(key, ""),
|
||||||
|
"name2": teams[1].get(key, ""),
|
||||||
|
}
|
||||||
# подписи
|
# подписи
|
||||||
for code, eng, _ru in stat_list:
|
for code, eng, _ru in stat_list:
|
||||||
if key == code:
|
if key == code:
|
||||||
@@ -605,6 +623,7 @@ def format_team_stat(team1: dict, team2: dict, period: str | None = None) -> lis
|
|||||||
def flip(s: str) -> str:
|
def flip(s: str) -> str:
|
||||||
parts = s.split()
|
parts = s.split()
|
||||||
return f"{parts[1]} {parts[0]}" if len(parts) >= 2 else s
|
return f"{parts[1]} {parts[0]}" if len(parts) >= 2 else s
|
||||||
|
|
||||||
r["name1"] = flip(r["name1"])
|
r["name1"] = flip(r["name1"])
|
||||||
r["name2"] = flip(r["name2"])
|
r["name2"] = flip(r["name2"])
|
||||||
|
|
||||||
@@ -619,11 +638,13 @@ def _iter_period_pairs(teams_periods: dict):
|
|||||||
"""
|
"""
|
||||||
a = teams_periods.get("A", {})
|
a = teams_periods.get("A", {})
|
||||||
b = teams_periods.get("B", {})
|
b = teams_periods.get("B", {})
|
||||||
|
|
||||||
def _key(k):
|
def _key(k):
|
||||||
try:
|
try:
|
||||||
return int(k)
|
return int(k)
|
||||||
except (TypeError, ValueError):
|
except (TypeError, ValueError):
|
||||||
return k
|
return k
|
||||||
|
|
||||||
for k in sorted(a.keys(), key=_key):
|
for k in sorted(a.keys(), key=_key):
|
||||||
if k in b:
|
if k in b:
|
||||||
yield k, a[k], b[k]
|
yield k, a[k], b[k]
|
||||||
@@ -635,41 +656,20 @@ def _build_all_stats(payload: dict) -> dict:
|
|||||||
"""
|
"""
|
||||||
total_a = payload["teams"]["A"]
|
total_a = payload["teams"]["A"]
|
||||||
total_b = payload["teams"]["B"]
|
total_b = payload["teams"]["B"]
|
||||||
result = {
|
result = {"total": format_team_stat(total_a, total_b), "periods": []}
|
||||||
"total": format_team_stat(total_a, total_b),
|
|
||||||
"periods": []
|
|
||||||
}
|
|
||||||
for period_key, a_stat, b_stat in _iter_period_pairs(payload["teams_periods"]):
|
for period_key, a_stat, b_stat in _iter_period_pairs(payload["teams_periods"]):
|
||||||
result["periods"].append({
|
result["periods"].append(
|
||||||
"period": period_key,
|
{"period": period_key, "stats": format_team_stat(a_stat, b_stat)}
|
||||||
"stats": format_team_stat(a_stat, b_stat)
|
)
|
||||||
})
|
|
||||||
return result
|
return result
|
||||||
|
|
||||||
def xl():
|
|
||||||
df = read_excel_from_smb(
|
|
||||||
server_name=SERVER_NAME,
|
|
||||||
server_ip=SERVER_IP,
|
|
||||||
share_name=SHARE,
|
|
||||||
file_path_in_share=PATH_IN_SHARE,
|
|
||||||
username=USER,
|
|
||||||
password=PASSWORD,
|
|
||||||
domain=DOMAIN,
|
|
||||||
client_machine_name="KHL_SOFT",
|
|
||||||
sheet_name="TEAMS", # или None, или список листов
|
|
||||||
)
|
|
||||||
df = df.replace([float("inf"), float("-inf")], pd.NA)
|
|
||||||
columns_to_keep = [
|
|
||||||
"Team", "Logo", "Short", "HexPodl", # ← укажи свои
|
|
||||||
]
|
|
||||||
df = df.loc[:, [c for c in columns_to_keep if c in df.columns]]
|
|
||||||
json_text = df.to_json(orient="records", force_ascii=False) # NaN/NA -> null
|
|
||||||
return Response(content=json_text, media_type="application/json; charset=utf-8")
|
|
||||||
|
|
||||||
@app.get("/teams/stats")
|
@app.get("/teams/stats")
|
||||||
async def teams_stats(
|
async def teams_stats(
|
||||||
scope: str = Query("all", pattern="^(all|total|period)$"),
|
scope: str = Query("all", pattern="^(all|total|period)$"),
|
||||||
n: str | None = Query(None, description="Номер периода (строка или число) при scope=period"),
|
n: str | None = Query(
|
||||||
|
None, description="Номер периода (строка или число) при scope=period"
|
||||||
|
),
|
||||||
):
|
):
|
||||||
"""
|
"""
|
||||||
Все-в-одном: GET /teams/stats?scope=all
|
Все-в-одном: GET /teams/stats?scope=all
|
||||||
@@ -687,7 +687,9 @@ async def teams_stats(
|
|||||||
payload = lgd["data"]
|
payload = lgd["data"]
|
||||||
|
|
||||||
if scope == "total":
|
if scope == "total":
|
||||||
data = format_team_stat(payload["teams"]["A"], payload["teams"]["B"], period="Total")
|
data = format_team_stat(
|
||||||
|
payload["teams"]["A"], payload["teams"]["B"], period="Total"
|
||||||
|
)
|
||||||
return JSONResponse({"scope": "total", "data": data})
|
return JSONResponse({"scope": "total", "data": data})
|
||||||
|
|
||||||
if scope == "period":
|
if scope == "period":
|
||||||
@@ -704,21 +706,23 @@ async def teams_stats(
|
|||||||
a = payload["teams_periods"]["A"].get(period_key)
|
a = payload["teams_periods"]["A"].get(period_key)
|
||||||
b = payload["teams_periods"]["B"].get(period_key)
|
b = payload["teams_periods"]["B"].get(period_key)
|
||||||
if a is None or b is None:
|
if a is None or b is None:
|
||||||
raise HTTPException(status_code=404, detail=f"Период {period_key} не найден.")
|
raise HTTPException(
|
||||||
return JSONResponse({
|
status_code=404, detail=f"Период {period_key} не найден."
|
||||||
"scope": "period",
|
)
|
||||||
"period": period_key,
|
return JSONResponse(
|
||||||
"is_current": period_key == _current_period_key(payload),
|
{
|
||||||
"data": format_team_stat(a, b, period=period_key)
|
"scope": "period",
|
||||||
})
|
"period": period_key,
|
||||||
|
"is_current": period_key == _current_period_key(payload),
|
||||||
|
"data": format_team_stat(a, b, period=period_key),
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
# scope == "all"
|
# scope == "all"
|
||||||
cur = _current_period_key(payload)
|
cur = _current_period_key(payload)
|
||||||
return JSONResponse({
|
return JSONResponse(
|
||||||
"scope": "all",
|
{"scope": "all", "current_period": cur, "data": _build_all_stats(payload)}
|
||||||
"current_period": cur,
|
)
|
||||||
"data": _build_all_stats(payload)
|
|
||||||
})
|
|
||||||
|
|
||||||
|
|
||||||
def _norm_name(s: str | None) -> str:
|
def _norm_name(s: str | None) -> str:
|
||||||
@@ -727,8 +731,16 @@ def _norm_name(s: str | None) -> str:
|
|||||||
return ""
|
return ""
|
||||||
return str(s).strip().casefold()
|
return str(s).strip().casefold()
|
||||||
|
|
||||||
|
def _load_buf():
|
||||||
|
buf = nasio.load_bio(user=USER, password=PASSWORD,
|
||||||
|
nas_ip=SERVER_NAME, nas_port="443", path=PATH)
|
||||||
|
if isinstance(buf, (bytes, bytearray, memoryview)):
|
||||||
|
buf = io.BytesIO(buf)
|
||||||
|
buf.seek(0)
|
||||||
|
return buf
|
||||||
|
|
||||||
@app.get("/info")
|
@app.get("/info")
|
||||||
async def info():
|
async def info(format: str = "xlsx", sheet: str = "TEAMS"):
|
||||||
# 1) Проверяем, выбран ли матч
|
# 1) Проверяем, выбран ли матч
|
||||||
global current_season
|
global current_season
|
||||||
if not selected_game_id:
|
if not selected_game_id:
|
||||||
@@ -737,76 +749,128 @@ async def info():
|
|||||||
# 2) Берём расписание и ищем строку по выбранному ID
|
# 2) Берём расписание и ищем строку по выбранному ID
|
||||||
df = load_today_schedule()
|
df = load_today_schedule()
|
||||||
if df.empty:
|
if df.empty:
|
||||||
return JSONResponse({"message": "Сегодня матчей нет", "selected_id": selected_game_id})
|
return JSONResponse(
|
||||||
|
{"message": "Сегодня матчей нет", "selected_id": selected_game_id}
|
||||||
|
)
|
||||||
|
|
||||||
# безопасно приводим id к int и ищем
|
# безопасно приводим id к int и ищем
|
||||||
try:
|
try:
|
||||||
row = df.loc[df["id"].astype(int) == int(selected_game_id)].iloc[0]
|
row = df.loc[df["id"].astype(int) == int(selected_game_id)].iloc[0]
|
||||||
except Exception:
|
except Exception:
|
||||||
return JSONResponse({"message": "Выбранный матч не найден в расписании на сегодня",
|
return JSONResponse(
|
||||||
"selected_id": selected_game_id}, status_code=404)
|
{
|
||||||
|
"message": "Выбранный матч не найден в расписании на сегодня",
|
||||||
|
"selected_id": selected_game_id,
|
||||||
|
},
|
||||||
|
status_code=404,
|
||||||
|
)
|
||||||
|
|
||||||
home_name = str(row.get("homeName_en", "")).strip()
|
home_name = str(row.get("homeName_en", "")).strip()
|
||||||
away_name = str(row.get("visitorName_en", "")).strip()
|
away_name = str(row.get("visitorName_en", "")).strip()
|
||||||
|
|
||||||
# 3) Подтягиваем справочник команд из Excel (лист TEAMS)
|
# 3) Подтягиваем справочник команд из Excel (лист TEAMS)
|
||||||
teams_df = read_excel_from_smb(
|
src = _load_buf()
|
||||||
server_name=SERVER_NAME,
|
|
||||||
server_ip=SERVER_IP,
|
|
||||||
share_name=SHARE,
|
|
||||||
file_path_in_share=PATH_IN_SHARE,
|
|
||||||
username=USER,
|
|
||||||
password=PASSWORD,
|
|
||||||
domain=DOMAIN,
|
|
||||||
client_machine_name="KHL_SOFT",
|
|
||||||
sheet_name="TEAMS",
|
|
||||||
)
|
|
||||||
|
|
||||||
# Оставляем только полезные поля (подгони под свой файл)
|
if format == "xlsx":
|
||||||
keep = [ "Team", "Logo", "Short", "HexPodl", "HexBase", "HexText" ]
|
# читаем нужный лист из исходного XLSM
|
||||||
keep = [c for c in keep if c in teams_df.columns]
|
df = pd.read_excel(src, sheet_name=sheet, engine="openpyxl")
|
||||||
teams_df = teams_df.loc[:, keep].copy()
|
|
||||||
|
|
||||||
# 4) Нормализованные ключи для джоина по имени
|
# пишем НОВЫЙ XLSX (без макросов) — это то, что понимает vMix
|
||||||
teams_df["__key"] = teams_df["Team"].apply(_norm_name)
|
out = io.BytesIO()
|
||||||
|
with pd.ExcelWriter(out, engine="openpyxl") as writer:
|
||||||
|
df.to_excel(writer, sheet_name=sheet, index=False)
|
||||||
|
out.seek(0)
|
||||||
|
|
||||||
def _pick_team_info(name: str) -> dict:
|
return StreamingResponse(
|
||||||
key = _norm_name(name)
|
out,
|
||||||
hit = teams_df.loc[teams_df["__key"] == key]
|
media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
|
||||||
if hit.empty:
|
headers={
|
||||||
# не нашли точное совпадение — вернём только название
|
# стабильное имя файла, чтобы vMix не путался
|
||||||
return {"Team": name}
|
"Content-Disposition": "inline; filename=vmix.xlsx",
|
||||||
rec = hit.iloc[0].to_dict()
|
# отключаем кэш браузера/прокси, vMix сам опрашивает по интервалу
|
||||||
rec.pop("__key", None)
|
"Cache-Control": "no-cache, no-store, must-revalidate",
|
||||||
# заменим NaN/inf на None, чтобы JSON не падал
|
"Pragma": "no-cache",
|
||||||
for k, v in list(rec.items()):
|
},
|
||||||
if pd.isna(v) or v in (np.inf, -np.inf):
|
)
|
||||||
rec[k] = None
|
|
||||||
return rec
|
|
||||||
|
|
||||||
home_info = _pick_team_info(home_name)
|
elif format == "csv":
|
||||||
away_info = _pick_team_info(away_name)
|
df = pd.read_excel(src, sheet_name=sheet, engine="openpyxl")
|
||||||
date_obj = datetime.strptime(row.get("datetime_str", ""), "%d.%m.%Y %H:%M")
|
csv_bytes = df.to_csv(index=False, encoding="utf-8-sig").encode("utf-8")
|
||||||
try:
|
return StreamingResponse(
|
||||||
full_format = date_obj.strftime("%B %-d, %Y")
|
io.BytesIO(csv_bytes),
|
||||||
except ValueError:
|
media_type="text/csv; charset=utf-8",
|
||||||
full_format = date_obj.strftime("%B %#d, %Y")
|
headers={
|
||||||
|
"Content-Disposition": "inline; filename=vmix.csv",
|
||||||
|
"Cache-Control": "no-cache, no-store, must-revalidate",
|
||||||
|
"Pragma": "no-cache",
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
elif format == "json":
|
||||||
|
df = pd.read_excel(src, sheet_name=sheet, engine="openpyxl")
|
||||||
|
payload = json.dumps(df.to_dict(orient="records"), ensure_ascii=False)
|
||||||
|
return Response(
|
||||||
|
content=payload,
|
||||||
|
media_type="application/json; charset=utf-8",
|
||||||
|
headers={
|
||||||
|
"Cache-Control": "no-cache, no-store, must-revalidate",
|
||||||
|
"Pragma": "no-cache",
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
payload = [{
|
return Response("Unsupported format", status_code=400)
|
||||||
"selected_id": int(selected_game_id),
|
|
||||||
"tournament_id": int(current_tournament_id) if current_tournament_id else None,
|
# # Оставляем только полезные поля (подгони под свой файл)
|
||||||
"datetime": str(full_format),
|
# keep = ["Team", "Logo", "Short", "HexPodl", "HexBase", "HexText"]
|
||||||
"arena": str(row.get("arena_en", "")),
|
# keep = [c for c in keep if c in teams_df.columns]
|
||||||
"arena_city": str(row.get("arena_city_en", "")),
|
# teams_df = teams_df.loc[:, keep].copy()
|
||||||
"home": home_info,
|
|
||||||
"home_city": str(row.get("homeCity_en", "")),
|
# # 4) Нормализованные ключи для джоина по имени
|
||||||
"away": away_info,
|
# teams_df["__key"] = teams_df["Team"].apply(_norm_name)
|
||||||
"away_city": str(row.get("visitorCity_en", "")),
|
|
||||||
"season": current_season,
|
# def _pick_team_info(name: str) -> dict:
|
||||||
}]
|
# key = _norm_name(name)
|
||||||
|
# hit = teams_df.loc[teams_df["__key"] == key]
|
||||||
|
# if hit.empty:
|
||||||
|
# # не нашли точное совпадение — вернём только название
|
||||||
|
# return {"Team": name}
|
||||||
|
# rec = hit.iloc[0].to_dict()
|
||||||
|
# rec.pop("__key", None)
|
||||||
|
# # заменим NaN/inf на None, чтобы JSON не падал
|
||||||
|
# for k, v in list(rec.items()):
|
||||||
|
# if pd.isna(v) or v in (np.inf, -np.inf):
|
||||||
|
# rec[k] = None
|
||||||
|
# return rec
|
||||||
|
|
||||||
|
# home_info = _pick_team_info(home_name)
|
||||||
|
# away_info = _pick_team_info(away_name)
|
||||||
|
# date_obj = datetime.strptime(row.get("datetime_str", ""), "%d.%m.%Y %H:%M")
|
||||||
|
# try:
|
||||||
|
# full_format = date_obj.strftime("%B %-d, %Y")
|
||||||
|
# except ValueError:
|
||||||
|
# full_format = date_obj.strftime("%B %#d, %Y")
|
||||||
|
|
||||||
|
# payload = [
|
||||||
|
# {
|
||||||
|
# "selected_id": int(selected_game_id),
|
||||||
|
# "tournament_id": (
|
||||||
|
# int(current_tournament_id) if current_tournament_id else None
|
||||||
|
# ),
|
||||||
|
# "datetime": str(full_format),
|
||||||
|
# "arena": str(row.get("arena_en", "")),
|
||||||
|
# "arena_city": str(row.get("arena_city_en", "")),
|
||||||
|
# "home": home_info,
|
||||||
|
# "home_city": str(row.get("homeCity_en", "")),
|
||||||
|
# "away": away_info,
|
||||||
|
# "away_city": str(row.get("visitorCity_en", "")),
|
||||||
|
# "season": current_season,
|
||||||
|
# }
|
||||||
|
# ]
|
||||||
|
|
||||||
|
# return JSONResponse(content=payload)
|
||||||
|
# except Exception as ex:
|
||||||
|
# pprint(ex)
|
||||||
|
|
||||||
return JSONResponse(content=payload)
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
uvicorn.run(
|
uvicorn.run(
|
||||||
|
|||||||
@@ -6,3 +6,7 @@ uvicorn>=0.30.0
|
|||||||
requests>=2.31.0
|
requests>=2.31.0
|
||||||
python-telegram-handler
|
python-telegram-handler
|
||||||
python-dotenv>=1.1.0
|
python-dotenv>=1.1.0
|
||||||
|
lxml
|
||||||
|
python-dotenv
|
||||||
|
--extra-index-url https://git.tvstart.ru/api/packages/lexx/pypi/simple
|
||||||
|
nasio
|
||||||
154
smb_excel.py
154
smb_excel.py
@@ -1,154 +0,0 @@
|
|||||||
# smb_excel.py (обновлённый фрагмент)
|
|
||||||
from io import BytesIO
|
|
||||||
from typing import Optional, Union, Any, Dict
|
|
||||||
from smb.SMBConnection import SMBConnection
|
|
||||||
import pandas as pd
|
|
||||||
import re
|
|
||||||
|
|
||||||
class SMBDownloadError(Exception):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def _normalize_path(p: str) -> str:
|
|
||||||
"""Убирает случайные r"..."/"...", лишние кавычки и ставит прямые слэши."""
|
|
||||||
if p is None:
|
|
||||||
return p
|
|
||||||
# уберём обрамляющие кавычки/префикс r"..."
|
|
||||||
m = re.fullmatch(r"""r?["'](.*)["']""", p.strip())
|
|
||||||
if m:
|
|
||||||
p = m.group(1)
|
|
||||||
# заменим \ на /
|
|
||||||
p = p.replace("\\", "/").lstrip("/") # путь внутри шары не должен начинаться с /
|
|
||||||
return p
|
|
||||||
|
|
||||||
def _try_connect(
|
|
||||||
remote_name: str,
|
|
||||||
server_ip: str,
|
|
||||||
username: str,
|
|
||||||
password: str,
|
|
||||||
*,
|
|
||||||
client_machine_name: str,
|
|
||||||
domain: str,
|
|
||||||
port: int
|
|
||||||
) -> Optional[SMBConnection]:
|
|
||||||
conn = SMBConnection(
|
|
||||||
username,
|
|
||||||
password,
|
|
||||||
client_machine_name,
|
|
||||||
remote_name, # ВАЖНО: remote_name — имя сервера (или IP как fallback)
|
|
||||||
domain=domain,
|
|
||||||
use_ntlm_v2=True,
|
|
||||||
is_direct_tcp=True
|
|
||||||
)
|
|
||||||
try:
|
|
||||||
if conn.connect(server_ip, port):
|
|
||||||
return conn
|
|
||||||
except Exception:
|
|
||||||
pass
|
|
||||||
return None
|
|
||||||
|
|
||||||
def _connect_smb(
|
|
||||||
server_name: str,
|
|
||||||
server_ip: str,
|
|
||||||
username: str,
|
|
||||||
password: str,
|
|
||||||
*,
|
|
||||||
client_machine_name: str = "client",
|
|
||||||
domain: str = "",
|
|
||||||
port: int = 445,
|
|
||||||
) -> SMBConnection:
|
|
||||||
"""
|
|
||||||
Пробуем два варианта remote_name: (1) реальное имя сервера, (2) IP.
|
|
||||||
"""
|
|
||||||
for remote_name in (server_name, server_ip):
|
|
||||||
if not remote_name:
|
|
||||||
continue
|
|
||||||
conn = _try_connect(
|
|
||||||
remote_name=remote_name,
|
|
||||||
server_ip=server_ip,
|
|
||||||
username=username,
|
|
||||||
password=password,
|
|
||||||
client_machine_name=client_machine_name,
|
|
||||||
domain=domain,
|
|
||||||
port=port,
|
|
||||||
)
|
|
||||||
if conn:
|
|
||||||
return conn
|
|
||||||
raise SMBDownloadError(
|
|
||||||
f"Не удалось подключиться к SMB {server_ip}:{port}. "
|
|
||||||
f"Проверь server_name (реальное имя хоста), домен/логин и доступность порта 445."
|
|
||||||
)
|
|
||||||
|
|
||||||
def fetch_smb_file(
|
|
||||||
*,
|
|
||||||
server_name: str,
|
|
||||||
server_ip: str,
|
|
||||||
share_name: str,
|
|
||||||
file_path_in_share: str,
|
|
||||||
username: str,
|
|
||||||
password: str,
|
|
||||||
client_machine_name: str = "client",
|
|
||||||
domain: str = "",
|
|
||||||
port: int = 445,
|
|
||||||
) -> BytesIO:
|
|
||||||
"""
|
|
||||||
Возвращает содержимое файла из SMB как BytesIO.
|
|
||||||
"""
|
|
||||||
file_path_in_share = _normalize_path(file_path_in_share)
|
|
||||||
conn: Optional[SMBConnection] = None
|
|
||||||
try:
|
|
||||||
conn = _connect_smb(
|
|
||||||
server_name=server_name,
|
|
||||||
server_ip=server_ip,
|
|
||||||
username=username,
|
|
||||||
password=password,
|
|
||||||
client_machine_name=client_machine_name,
|
|
||||||
domain=domain,
|
|
||||||
port=port,
|
|
||||||
)
|
|
||||||
# Быстрая проверка существования каталога/файла
|
|
||||||
parent = "/".join(file_path_in_share.split("/")[:-1]) or ""
|
|
||||||
# listPath кидает исключение, если каталога/шары нет — получим понятную ошибку
|
|
||||||
if parent:
|
|
||||||
conn.listPath(share_name, parent)
|
|
||||||
|
|
||||||
buf = BytesIO()
|
|
||||||
conn.retrieveFile(share_name, file_path_in_share, buf)
|
|
||||||
buf.seek(0)
|
|
||||||
return buf
|
|
||||||
except Exception as e:
|
|
||||||
raise SMBDownloadError(
|
|
||||||
f"Ошибка скачивания {share_name}/{file_path_in_share}: {e}"
|
|
||||||
) from e
|
|
||||||
finally:
|
|
||||||
if conn:
|
|
||||||
try:
|
|
||||||
conn.close()
|
|
||||||
except Exception:
|
|
||||||
pass
|
|
||||||
|
|
||||||
def read_excel_from_smb(
|
|
||||||
*,
|
|
||||||
server_name: str,
|
|
||||||
server_ip: str,
|
|
||||||
share_name: str,
|
|
||||||
file_path_in_share: str,
|
|
||||||
username: str,
|
|
||||||
password: str,
|
|
||||||
client_machine_name: str = "client",
|
|
||||||
domain: str = "",
|
|
||||||
port: int = 445,
|
|
||||||
sheet_name: Optional[Union[str, int, list]] = None,
|
|
||||||
**read_excel_kwargs: Dict[str, Any],
|
|
||||||
) -> pd.DataFrame:
|
|
||||||
file_obj = fetch_smb_file(
|
|
||||||
server_name=server_name,
|
|
||||||
server_ip=server_ip,
|
|
||||||
share_name=share_name,
|
|
||||||
file_path_in_share=file_path_in_share,
|
|
||||||
username=username,
|
|
||||||
password=password,
|
|
||||||
client_machine_name=client_machine_name,
|
|
||||||
domain=domain,
|
|
||||||
port=port,
|
|
||||||
)
|
|
||||||
return pd.read_excel(file_obj, sheet_name=sheet_name, **read_excel_kwargs)
|
|
||||||
Reference in New Issue
Block a user