diff --git a/RUN.sh b/RUN.sh deleted file mode 100755 index 0113c68..0000000 --- a/RUN.sh +++ /dev/null @@ -1,102 +0,0 @@ -#!/bin/bash - -show_help() { - echo "Использование: $0 -t <домашняя команда>" - echo " -t Домашняя команда" - echo " -h Показать эту справку" - echo "" - echo "Пример: $0 -t \"cska\"" - exit 0 -} - -# Определение IP адреса -if [[ "$(uname -s)" != "Darwin" ]]; then - ip_address=$(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 - ip_address=$(ifconfig | grep -Eo 'inet (addr:)?([0-9]*\.){3}[0-9]*' | grep -Eo '([0-9]*\.){3}[0-9]*' | grep -v '127.0.0.1' | head -n1) -fi - -echo "IP: $ip_address" - -# База данных хостов в виде пар IP-команда -hosts=( - "10.10.35.21:cska" - "10.10.35.22: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="" -for host in "${hosts[@]}"; do - host_ip="${host%%:*}" - host_team="${host#*:}" - if [[ "$ip_address" == "$host_ip" ]]; then - detected_team="$host_team" - break - fi -done - -# Обработка аргументов командной строки -team="" -while getopts "t:h" opt; do - case $opt in - t) team="$OPTARG" ;; - h) show_help ;; - *) echo "Неверный аргумент"; exit 1 ;; - esac -done - -# Определение финальной команды (приоритет у аргумента, затем у автоопределения) -if [[ -n "$team" ]]; then - final_team="$team" - echo "Используется команда из аргумента: $final_team" -elif [[ -n "$detected_team" ]]; then - final_team="$detected_team" - echo "Используется автоопределенная команда: $final_team" -else - echo "Ошибка: Не удалось определить команду. Укажите явно через -t или проверьте IP адрес" - echo "Доступные команды:" - for ip in "${!hosts[@]}"; do - echo " $ip: ${hosts[$ip]}" - done - exit 1 -fi - -# Проверка существования файлов -if [[ ! -f "$(pwd)/.venv/bin/streamlit" ]]; then - echo "Ошибка: streamlit не найден в $(pwd)/.venv/bin/streamlit" - exit 1 -fi - -if [[ ! -f "$(pwd)/visual.py" ]]; then - echo "Ошибка: visual.py не найден в $(pwd)/visual.py" - exit 1 -fi - -if [[ ! -f "$(pwd)/get_data.py" ]]; then - echo "Ошибка: get_data.py не найден в $(pwd)/get_data.py" - exit 1 -fi - -#Запуск сбора данных -echo "Запуск get_data.py..." -"$(pwd)/.venv/bin/python $(pwd)/get_data.py" --team "$final_team" & - -# Небольшая задержка перед запуском второго процесса -sleep 2 - -# Запуск веб-морды -echo "Запуск приложения для команды: $final_team" -echo "Запуск streamlit..." -"$(pwd)/.venv/bin/streamlit" run "$(pwd)/visual.py" - -# Ожидание завершения процессов -echo "Оба процесса запущены. Для остановки нажмите Ctrl+C" -wait \ No newline at end of file diff --git a/setup.sh b/setup.sh new file mode 100755 index 0000000..1e951a4 --- /dev/null +++ b/setup.sh @@ -0,0 +1,340 @@ +#!/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 <релиз>" + echo " -t Домашняя команда" + echo " -r Релиз (тег или ветка в git)" + echo " -h Показать эту справку" + echo "" + echo "Пример: $0 -t cska -r main" + echo "Пример: $0 -t zenit -r Barabanov_TEST" + echo "" + exit 0 +} + +# Функция для вывода цветных сообщений +log_info() { + echo -e "${GREEN}[INFO]${NC} $1" +} + +log_warn() { + echo -e "${YELLOW}[WARN]${NC} $1" +} + +log_error() { + echo -e "${RED}[ERROR]${NC} $1" +} + +# Функция проверки и установки пакетов +install_packages() { + log_info "Обновление списка пакетов..." + apt-get update + + log_info "Установка необходимых пакетов..." + apt-get install -y python3 python3-pip python3-venv git systemd + + # Проверка установки + 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="/opt/RFB" + + log_info "Создание рабочей директории $target_dir..." + mkdir -p $target_dir + cd $target_dir + + if [ -d ".git" ]; then + log_info "Обновление существующего репозитория..." + git fetch --all + git checkout $release + git pull origin $release + else + log_info "Клонирование репозитория $repo_url..." + git clone $repo_url . + 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)" +} + +# Функция настройки виртуального окружения +setup_venv() { + local target_dir="/opt/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 + + if [ -f "requirements.txt" ]; then + log_info "Установка зависимостей из requirements.txt..." + pip install -r requirements.txt + else + log_warn "Файл requirements.txt не найден, устанавливаем базовые зависимости..." + pip install streamlit requests pandas numpy plotly + fi + + # Установка дополнительных зависимостей которые могут понадобиться + pip install watchdog pillow + + # Проверка установки streamlit + if ! [ -f ".venv/bin/streamlit" ]; then + log_error "Streamlit не установлен в виртуальном окружении" + exit 1 + fi + + log_info "Установленные пакеты:" + pip list +} + +# Функция создания systemd сервиса +create_systemd_service() { + local team="$1" + local service_file="/etc/systemd/system/rfb-stat.service" + + log_info "Создание systemd сервиса для команды: $team..." + + cat > $service_file << EOF +[Unit] +Description=RFB Stat v.1.0 +After=network.target + +[Service] +Type=simple +User=root +WorkingDirectory=/opt/RFB +Environment=PATH=/opt/RFB/.venv/bin +ExecStart=/bin/bash -c 'cd /opt/RFB && source /opt/RFB/.venv/bin/activate && python3 get_data.py --team "$team" & streamlit run visual.py --server.port 8501 --server.address 0.0.0.0' +Restart=always +RestartSec=3 +StandardOutput=journal +StandardError=journal + +[Install] +WantedBy=multi-user.target +EOF + + # Настройка прав + chmod 644 $service_file + + log_info "Перезагрузка systemd..." + systemctl daemon-reload + + log_info "Включение сервиса..." + systemctl enable rfb-stat.service +} + +# Функция проверки файлов +check_required_files() { + local target_dir="/opt/RFB" + cd $target_dir + + local missing_files=() + + if [ ! -f "visual.py" ]; then + missing_files+=("visual.py") + fi + + if [ ! -f "get_data.py" ]; then + missing_files+=("get_data.py") + fi + + if [ ${#missing_files[@]} -ne 0 ]; then + log_error "Отсутствуют необходимые файлы: ${missing_files[*]}" + log_error "Содержимое директории $target_dir:" + ls -la + exit 1 + fi + + log_info "Все необходимые файлы присутствуют" + + # Проверка прав на выполнение + chmod +x visual.py get_data.py +} + +# Функция определения IP и команды +detect_team() { + local team_arg="$1" + + # Определение IP адреса + if [[ "$(uname -s)" == "Linux" ]]; then + ip_address=$(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"]="Kuban" + ["10.10.35.23"]="uralmash" + ["10.10.35.24"]="betcity" + ["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 "Доступные команды:" + for ip in "${!hosts[@]}"; do + echo " $ip: ${hosts[$ip]}" + done + exit 1 + fi + + echo "$final_team" +} + +# Функция настройки firewall +setup_firewall() { + log_info "Настройка firewall (открытие порта 8501)..." + + # Проверяем наличие ufw + if command -v ufw &> /dev/null; then + ufw allow 8501/tcp + log_info "Порт 8501 открыт в ufw" + fi + + # Для firewalld (CentOS/RHEL) + if command -v firewall-cmd &> /dev/null; then + firewall-cmd --permanent --add-port=8501/tcp + firewall-cmd --reload + log_info "Порт 8501 открыт в firewalld" + fi +} + +# Функция проверки порта +check_port() { + local port=8501 + if netstat -tuln | grep ":$port " > /dev/null; then + log_warn "Порт $port уже занят. Возможно, приложение уже запущено." + return 1 + fi + return 0 +} + +# Основная функция +main() { + local team="" + local release="main" # значение по умолчанию + + # Обработка аргументов командной строки + while getopts "t:r:h" opt; do + case $opt in + t) team="$OPTARG" ;; + r) release="$OPTARG" ;; + h) show_help ;; + *) log_error "Неверный аргумент"; exit 1 ;; + esac + done + + log_info "Начало установки RFB Stat..." + log_info "Команда: $team, Релиз: $release" + + # Проверка прав root + if [[ $EUID -ne 0 ]]; then + log_error "Этот скрипт должен запускаться с правами root" + exit 1 + fi + + # Установка пакетов + install_packages + + # Загрузка кода + download_code "$release" + + # Настройка виртуального окружения + setup_venv + + # Определение команды + final_team=$(detect_team "$team") + + # Проверка файлов + check_required_files + + # Настройка firewall + setup_firewall + + # Проверка порта + check_port + + # Создание systemd сервиса + create_systemd_service "$final_team" + + log_info "Настройка завершена!" + log_info "Запуск сервиса..." + systemctl start rfb-stat.service + + # Проверка статуса + sleep 3 + log_info "Статус сервиса:" + systemctl status rfb-stat.service --no-pager + + log_info "==================================================" + log_info "Установка завершена успешно!" + log_info "Приложение должно быть доступно по адресу: http://$ip_address:8501" + log_info "Команда: $final_team" + log_info "Режим: $release" + log_info "" + log_info "Для просмотра логов: journalctl -u rfb-stat.service -f" + log_info "Для перезапуска: systemctl restart rfb-stat.service" + log_info "Для остановки: systemctl stop rfb-stat.service" + log_info "==================================================" +} + +# Запуск основной функции +main "$@" \ No newline at end of file