Compare commits
51 Commits
| Author | SHA1 | Date |
|---|---|---|
|
|
78c23227e4 | |
|
|
7b6d298d68 | |
|
|
1b9649fbc0 | |
|
|
193390ecb8 | |
|
|
55ebed483f | |
|
|
00846331f3 | |
|
|
0adc48209d | |
|
|
d7403bb266 | |
|
|
2b43532a8f | |
|
|
54afe573c1 | |
|
|
b240bc1bad | |
|
|
be996f942b | |
|
|
dda0e8ebb7 | |
|
|
dd78a4a55b | |
|
|
f61e8e1588 | |
|
|
0168036fab | |
|
|
24fdc1014c | |
|
|
5f59c4df92 | |
|
|
63ab836f1a | |
|
|
d57a88321e | |
|
|
5b0eafe695 | |
|
|
35dca573c9 | |
|
|
dc021f6802 | |
|
|
8f8b5d18d1 | |
|
|
7ad9b91a0a | |
|
|
469ab42324 | |
|
|
3547f307b1 | |
|
|
8edd79be72 | |
|
|
e4f596dd82 | |
|
|
b2363215b8 | |
|
|
0c060c79cb | |
|
|
0871c0e251 | |
|
|
acc3c9916c | |
|
|
18629ddffe | |
|
|
a9f3766d9f | |
|
|
ae11fb7588 | |
|
|
6ac9f59716 | |
|
|
c49502dff3 | |
|
|
14ed4ad11f | |
|
|
d263966ded | |
|
|
d159c0f0e4 | |
|
|
dc5c35906f | |
|
|
73c2f1f440 | |
|
|
93e5303ee6 | |
|
|
65e95b8eb2 | |
|
|
3b5ac43c30 | |
|
|
e319ea2e73 | |
|
|
8582bdca9e | |
|
|
a869a07eba | |
|
|
bc99718123 | |
|
|
83aac7cd51 |
|
|
@ -0,0 +1,117 @@
|
|||
pipeline {
|
||||
environment {
|
||||
REGISTRY_URL = "https://proxy.docker.dataekb.ru/local_cache"
|
||||
REGISTRY = "proxy.docker.dataekb.ru/local_cache"
|
||||
BOT_IMAGE_NAME = "bot_open_sesam"
|
||||
TUNNEL_IMAGE_NAME = "tunnel_open_sesam"
|
||||
BOT_IMAGE_TAG = "latest"
|
||||
TUNNEL_IMAGE_TAG = "latest"
|
||||
IMAGE_TAG = "${env.BUILD_NUMBER}"
|
||||
}
|
||||
|
||||
agent { label 'agent_smith'}
|
||||
|
||||
options {
|
||||
buildDiscarder(logRotator(numToKeepStr: '10'))
|
||||
disableConcurrentBuilds() // Prevent cleanup conflicts
|
||||
timeout(time: 30, unit: 'MINUTES')
|
||||
}
|
||||
|
||||
stages {
|
||||
|
||||
stage ('build bot image and push') {
|
||||
steps {
|
||||
script {
|
||||
docker.withRegistry("${REGISTRY_URL}", 'jenkins_harbor') {
|
||||
def BotImage = docker.build(
|
||||
"${REGISTRY}/${BOT_IMAGE_NAME}:${env.BUILD_NUMBER}",
|
||||
"--label build_number=${IMAGE_TAG} " +
|
||||
"--label git_commit=${env.GIT_COMMIT} " +
|
||||
"."
|
||||
)
|
||||
BotImage.push()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
stage ('build tunnel image and push') {
|
||||
steps {
|
||||
dir('tunnel'){
|
||||
script {
|
||||
docker.withRegistry("${REGISTRY_URL}", 'jenkins_harbor') {
|
||||
def TunnelImage = docker.build(
|
||||
"${REGISTRY}/${TUNNEL_IMAGE_NAME}:${env.BUILD_NUMBER}",
|
||||
"--label build_number=${IMAGE_TAG} " +
|
||||
"--label git_commit=${env.GIT_COMMIT} " +
|
||||
"."
|
||||
)
|
||||
TunnelImage.push()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
stage ('clear after build and push') {
|
||||
steps {
|
||||
script {
|
||||
sh "docker image prune --filter label=stage=builder"
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
stage("Deploy") {
|
||||
steps {
|
||||
script {
|
||||
sh '''
|
||||
export BOT_IMAGE_TAG=${IMAGE_TAG}
|
||||
export TUNNEL_IMAGE_TAG=${IMAGE_TAG}
|
||||
|
||||
# Pull new images explicitly
|
||||
docker-compose pull bot_open_sesam tunnel_open_sesam
|
||||
|
||||
# Deploy with specific project name
|
||||
docker-compose up -d --remove-orphans
|
||||
|
||||
'''
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
post {
|
||||
success {
|
||||
echo "Оба образа собраны и задеплоены успешно"
|
||||
}
|
||||
failure {
|
||||
echo "Ошибка сборки!"
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
// Old_version
|
||||
// pipeline {
|
||||
// agent { label 'agent_smith' }
|
||||
//
|
||||
// stages {
|
||||
// stage('Stop and Remove Existing Container') {
|
||||
// steps {
|
||||
// sh '''
|
||||
// docker stop open_sesam || true
|
||||
// docker rm open_sesam || true
|
||||
// docker-compose down || true
|
||||
// '''
|
||||
// }
|
||||
// }
|
||||
//
|
||||
// stage('Build and Run Container') {
|
||||
// steps {
|
||||
// sh '''
|
||||
// docker-compose up --build -d
|
||||
// '''
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
30
README.md
30
README.md
|
|
@ -1,22 +1,14 @@
|
|||
# Usage
|
||||
# USAGE
|
||||
|
||||
Создать файл .env в корневой дирректории проекта, объявить и присвоить значения переменным:
|
||||
Для развертывания бота необходимо внести свой токен в файл
|
||||
.env, запись должна иметь вид: BOT_TOKEN=your_token.
|
||||
|
||||
TOKEN=
|
||||
LOCK_IP=
|
||||
CARD_ID=
|
||||
AUTH_API=
|
||||
Для того чтобы пользователь мог пользоваться ботом, ему
|
||||
необходимо поделиться номером телефона нажав на соответствующую
|
||||
кнопку (номер не будет виден другим пользователям, используется
|
||||
отправляется единожды для сверки с данными в БД).
|
||||
|
||||
, где
|
||||
|
||||
TOKEN - токен телеграм для взаимодействия с ботом
|
||||
LOCK_IP - ip адресс замка
|
||||
CARD_ID - уникальный номер ключ-карты
|
||||
AUTH_API - уникальный набор символов, для взаимодействия с API замка
|
||||
|
||||
Дополнительные подробности можно найти в instruction_http_api_v5.pdf и исходном коде программы (см. main.py)
|
||||
|
||||
## TODO
|
||||
# TODO
|
||||
|
||||
- [x] Написать базовый функционал
|
||||
~~- [ ] Создать соотношение типа "id_пользователя: досутп_к_замку:"~~
|
||||
|
|
@ -24,4 +16,8 @@ AUTH_API - уникальный набор символов, для взаимо
|
|||
- [x] Удалить модуль getenv
|
||||
- [x] Создать и парсить json с информацией о пользователях и их номерах
|
||||
~~- [ ] Проверять, является ли пользователь администратором, если является, выводить дополнительную кнопку, предлагающую добавить номер в БД~~
|
||||
- [ ] Сделать логирование о том, что кто-то открыл дверь в конкретное время
|
||||
- [x] Изменить json хранящийся локально на запросы к API
|
||||
- [x] Изменить названия комнта в локальном json
|
||||
- [x] Ограничить логи внутри докера
|
||||
- [x] Избавиться от json файла, переместить токен в .env файл, информацию от замков получать через запрос по API
|
||||
- [x] Избавиться от модуля requests
|
||||
|
|
|
|||
60
auth.py
60
auth.py
|
|
@ -1,38 +1,48 @@
|
|||
from config import config
|
||||
import re
|
||||
|
||||
|
||||
ALLOWED_PHONE_NUMBERS = list(config.get("users", {}).keys())
|
||||
import aiohttp
|
||||
|
||||
AUTHORIZED_USERS = {}
|
||||
|
||||
|
||||
def check_user_auth(phone: str) -> bool:
|
||||
return phone in ALLOWED_PHONE_NUMBERS
|
||||
|
||||
|
||||
def normalize_phone(phone: str) -> str:
|
||||
phone = phone.strip()
|
||||
phone = re.sub(r"[^\d+]", "", phone)
|
||||
if not phone.startswith("+"):
|
||||
phone = "+" + phone
|
||||
return phone
|
||||
digits = re.sub(r"\D", "", phone)
|
||||
if len(digits) > 10:
|
||||
digits = digits[-10:]
|
||||
return digits
|
||||
|
||||
|
||||
def authorize_user(user_id: int, phone: str) -> bool:
|
||||
normalized_phone = normalize_phone(phone)
|
||||
if normalized_phone in ALLOWED_PHONE_NUMBERS:
|
||||
AUTHORIZED_USERS[user_id] = normalized_phone
|
||||
# if check_user_auth(phone):
|
||||
# AUTHORIZED_USERS[user_id] = phone
|
||||
print(f"{user_id} авторизован с номером: {normalized_phone}")
|
||||
return True
|
||||
else:
|
||||
print(
|
||||
f"Пользователь {user_id} пытался авторизоваться с номером {normalized_phone}"
|
||||
)
|
||||
async def authorize_user(user_id: int, phone: str) -> bool:
|
||||
normalized = normalize_phone(phone)
|
||||
api_url = "https://papi.dataekb.ru/check_access"
|
||||
print("DEBUG: phone (var normalized) is ", normalized)
|
||||
async with aiohttp.ClientSession() as session:
|
||||
try:
|
||||
async with session.get(api_url, params={"tel": normalized}) as response:
|
||||
if response.status != 200:
|
||||
print(f"Ошибка запроса к API: статус {response.status}")
|
||||
return False
|
||||
data = await response.json()
|
||||
except Exception as e:
|
||||
print(f"Исключение при запросе к API: {str(e)}")
|
||||
return False
|
||||
|
||||
if data.get("response") != "1":
|
||||
print(f"Доступ запрещён для номера: {normalized}")
|
||||
return False
|
||||
|
||||
zone_str = data["data"].get("zone", "")
|
||||
zones = [zone.strip() for zone in zone_str.split(";") if zone.strip()]
|
||||
card_code = data["data"].get("card-code", "")
|
||||
if not zones or not card_code:
|
||||
print("Некорректный ответ API: отсутствуют зоны или код карты.")
|
||||
return False
|
||||
|
||||
AUTHORIZED_USERS[user_id] = {"tel": normalized, "card": card_code, "zones": zones}
|
||||
print(
|
||||
f"{user_id} авторизован с номером: {normalized}, карта: {card_code}, зоны: {zones}"
|
||||
)
|
||||
return True
|
||||
|
||||
|
||||
def is_user_auth(user_id: int) -> bool:
|
||||
return user_id in AUTHORIZED_USERS
|
||||
|
|
|
|||
|
|
@ -1,23 +0,0 @@
|
|||
{
|
||||
"bot_token": "",
|
||||
"locks": {
|
||||
"Room418": {
|
||||
"ip": "10.9.1.26",
|
||||
"auth_api": "73B15D12"
|
||||
},
|
||||
"Floor4": {
|
||||
"ip": "10.9.1.27",
|
||||
"auth_api": "F901C40A"
|
||||
}
|
||||
},
|
||||
"users": {
|
||||
"+79000959392": {
|
||||
"access_card": "000D001195DD",
|
||||
"lock_id": ["Room418", "Floor4"]
|
||||
},
|
||||
"+79221716513": {
|
||||
"access_card": "",
|
||||
"lock_id": ["Room418", "Floor4"]
|
||||
}
|
||||
}
|
||||
}
|
||||
27
config.py
27
config.py
|
|
@ -1,11 +1,20 @@
|
|||
import json
|
||||
|
||||
CONFIG_PATH = "bot_config.json"
|
||||
def load_env(path=".env") -> dict:
|
||||
env = {}
|
||||
try:
|
||||
with open(path, "r", encoding="utf-8") as f:
|
||||
for line in f:
|
||||
line = line.strip()
|
||||
# Пропускаем пустые строки и строки-комментарии
|
||||
if not line or line.startswith("#"):
|
||||
continue
|
||||
if "=" in line:
|
||||
key, value = line.split("=", 1)
|
||||
env[key.strip()] = value.strip()
|
||||
except FileNotFoundError:
|
||||
raise Exception(
|
||||
f"Файл {path} не найден. Проверьте наличие файла .env в корне проекта."
|
||||
)
|
||||
return env
|
||||
|
||||
|
||||
def load_config(path: str = CONFIG_PATH) -> dict:
|
||||
with open(path, "r", encoding="utf-8") as f:
|
||||
return json.load(f)
|
||||
|
||||
|
||||
config = load_config()
|
||||
config = load_env()
|
||||
|
|
|
|||
|
|
@ -1,10 +1,52 @@
|
|||
version: '3'
|
||||
services:
|
||||
open_sesam:
|
||||
container_name: open_sesam
|
||||
build: .
|
||||
bot_open_sesam:
|
||||
container_name: bot_open_sesam
|
||||
image: proxy.docker.dataekb.ru/local_cache/bot_open_sesam:${BOT_IMAGE_TAG:-latest}
|
||||
stdin_open: true
|
||||
tty: true
|
||||
volumes:
|
||||
- ./.env:/bot/open_sesam/.env
|
||||
env_file:
|
||||
- ./.env
|
||||
restart: always
|
||||
networks:
|
||||
- bot_open_sesam_network
|
||||
depends_on:
|
||||
tunnel_open_sesam:
|
||||
condition: service_healthy
|
||||
logging:
|
||||
driver: "json-file"
|
||||
options:
|
||||
max-size: "10m"
|
||||
max-file: "3"
|
||||
tunnel_open_sesam:
|
||||
image: proxy.docker.dataekb.ru/local_cache/tunnel_open_sesam:${BOT_IMAGE_TAG:-latest}
|
||||
container_name: tunnel_open_sesam
|
||||
env_file:
|
||||
- ./.env
|
||||
# environment:
|
||||
# - SSH_HOST=91.194.84.91
|
||||
# - SSH_PORT=22
|
||||
# - SSH_USER=root
|
||||
volumes:
|
||||
# SSH ключ
|
||||
- .ssh/id_rsa:/root/.ssh/id_rsa:ro
|
||||
networks:
|
||||
- bot_open_sesam_network
|
||||
restart: always
|
||||
logging:
|
||||
driver: "json-file"
|
||||
options:
|
||||
max-size: "10m"
|
||||
max-file: "3"
|
||||
healthcheck:
|
||||
test: ["CMD", "nc", "-z", "localhost", "1080"]
|
||||
interval: 10s
|
||||
timeout: 5s
|
||||
retries: 5
|
||||
networks:
|
||||
bot_open_sesam_network:
|
||||
driver: bridge
|
||||
ipam:
|
||||
config:
|
||||
- subnet: 172.30.10.0/29
|
||||
gateway: 172.30.10.1
|
||||
|
|
@ -1,4 +1,4 @@
|
|||
FROM python:3.13.3-slim
|
||||
FROM python:3.11-slim-bookworm
|
||||
|
||||
WORKDIR /bot/open_sesam
|
||||
|
||||
|
|
|
|||
|
|
@ -2,35 +2,30 @@ from aiogram import Dispatcher, F
|
|||
from aiogram.types import Message
|
||||
|
||||
from keyboard import get_locks_keyboard, get_contact_keyboard
|
||||
from auth import authorize_user, normalize_phone
|
||||
from config import config
|
||||
from auth import authorize_user, AUTHORIZED_USERS
|
||||
|
||||
|
||||
def register_contact_handler(dp: Dispatcher):
|
||||
@dp.message(F.contact)
|
||||
async def contact_handler(msg: Message):
|
||||
user_id = msg.from_user.id
|
||||
|
||||
if msg.contact is None:
|
||||
await msg.answer("Ошибка: номер телефона не получен")
|
||||
return
|
||||
|
||||
phone = normalize_phone(msg.contact.phone_number)
|
||||
|
||||
if not authorize_user(user_id, phone):
|
||||
await msg.answer("Доступ запрещен, номер не идентифицирован")
|
||||
phone = msg.contact.phone_number
|
||||
if not await authorize_user(user_id, phone):
|
||||
print("DEBUG: phone in contact_handler is ", phone)
|
||||
await msg.answer("Доступ запрещён, номер не идентифицирован")
|
||||
return
|
||||
|
||||
user_conf = config.get("users", {}).get(phone)
|
||||
print(f"***user_conf для {phone}: {user_conf}***")
|
||||
if not user_conf:
|
||||
await msg.answer("Пользователь не опознан")
|
||||
user_data = AUTHORIZED_USERS.get(user_id)
|
||||
if not user_data:
|
||||
await msg.answer("Ошибка авторизации.")
|
||||
return
|
||||
|
||||
allowed_locks = user_conf.get("lock_id", [])
|
||||
print(f"***allowed_locks = {allowed_locks}***")
|
||||
reply_markup = get_locks_keyboard(allowed_locks)
|
||||
|
||||
allowed_zones = user_data["zones"]
|
||||
reply_markup = await get_locks_keyboard(allowed_zones)
|
||||
await msg.answer(
|
||||
"Номер подтвержден. Выберите дверь для открытия", reply_markup=reply_markup
|
||||
"Номер подтверждён. Выберите дверь для открытия", reply_markup=reply_markup
|
||||
)
|
||||
|
|
|
|||
|
|
@ -1,72 +1,44 @@
|
|||
import asyncio
|
||||
import requests
|
||||
|
||||
import aiohttp
|
||||
from aiogram import Dispatcher
|
||||
from aiogram.types import Message
|
||||
|
||||
from auth import AUTHORIZED_USERS, is_user_auth
|
||||
from config import config
|
||||
from locks_api import get_lock_by_label
|
||||
|
||||
|
||||
def register_open_door_handler(dp: Dispatcher):
|
||||
@dp.message()
|
||||
async def open_door_handler(msg: Message):
|
||||
print(
|
||||
f"DEBUG: Получено сообщение от пользователя {msg.from_user.id}: '{msg.text}'"
|
||||
)
|
||||
user_id = msg.from_user.id
|
||||
if not is_user_auth(user_id):
|
||||
await msg.answer(
|
||||
"Доступ запрщен. Необходимо предоставить свой номер телефона."
|
||||
)
|
||||
else:
|
||||
print("OK")
|
||||
|
||||
phone = AUTHORIZED_USERS.get(user_id)
|
||||
print(AUTHORIZED_USERS)
|
||||
print(f"***user_id={user_id},phone={phone}***")
|
||||
user_conf = config.get("users", {}).get(phone)
|
||||
print(f"***phone={phone}, user_conf={user_conf}***")
|
||||
# allowed_locks = user_conf.get("locks", [])
|
||||
# print(f"***allowed_locks={allowed_locks}***")
|
||||
|
||||
# if msg.text not in allowed_locks:
|
||||
# print("**********************")
|
||||
# print(f"***{allowed_locks}***")
|
||||
# print(f"***{msg.text}***")
|
||||
# print("**********************")
|
||||
# return
|
||||
|
||||
lock_conf = config.get("locks", {}).get(msg.text)
|
||||
if not lock_conf:
|
||||
await msg.answer("Информации по замку не найдено")
|
||||
await msg.answer("Доступ запрещён. Предоставьте номер телефона.")
|
||||
return
|
||||
|
||||
url = f"http://{lock_conf['ip']}/cgi-bin/ext"
|
||||
auth_info = ("ext", lock_conf["auth_api"])
|
||||
payload = f"CARD={user_conf['access_card']}&DIR=0"
|
||||
user_data = AUTHORIZED_USERS.get(user_id)
|
||||
if not user_data:
|
||||
await msg.answer("Ошибка авторизации.")
|
||||
return
|
||||
|
||||
lock_info = await get_lock_by_label(msg.text)
|
||||
if not lock_info:
|
||||
await msg.answer("Информация по замку не найдена.")
|
||||
return
|
||||
|
||||
url = f"http://{lock_info['ip']}/cgi-bin/ext"
|
||||
auth_info = aiohttp.BasicAuth(login="ext", password=lock_info["code"])
|
||||
payload = f"CARD={user_data['card']}&DIR=0"
|
||||
headers = {"Content-Type": "application/x-www-form-urlencoded"}
|
||||
|
||||
try:
|
||||
print(
|
||||
f"***DEBUG: Отправляю запрос к {url} c payload: {payload} и auth: {auth_info}"
|
||||
)
|
||||
response = await asyncio.to_thread(
|
||||
requests.post,
|
||||
url,
|
||||
auth=auth_info,
|
||||
data=payload,
|
||||
headers=headers,
|
||||
timeout=5,
|
||||
)
|
||||
print(
|
||||
f"DEBUG: URL: {url}, status: {response.status_code}, response: {response.text}"
|
||||
)
|
||||
if response.status_code == 200:
|
||||
await msg.answer("Открыто")
|
||||
else:
|
||||
await msg.answer(
|
||||
f"Ошибка при открытии замка. Код ошибки: {response.status_code}"
|
||||
)
|
||||
async with aiohttp.ClientSession() as session:
|
||||
async with session.post(
|
||||
url, auth=auth_info, data=payload, headers=headers, timeout=5
|
||||
) as response:
|
||||
if response.status == 200:
|
||||
await msg.answer("Открыто")
|
||||
else:
|
||||
await msg.answer(
|
||||
f"Ошибка при открытии замка. Код ошибки: {response.status}"
|
||||
)
|
||||
except Exception as e:
|
||||
await msg.answer(f"Исключение: {str(e)}")
|
||||
|
|
|
|||
|
|
@ -4,7 +4,6 @@ from aiogram.filters import CommandStart
|
|||
|
||||
from keyboard import get_contact_keyboard, get_locks_keyboard
|
||||
from auth import is_user_auth, AUTHORIZED_USERS
|
||||
from config import config
|
||||
|
||||
|
||||
def register_start_handler(dp: Dispatcher):
|
||||
|
|
@ -12,25 +11,16 @@ def register_start_handler(dp: Dispatcher):
|
|||
async def command_start_handler(msg: Message):
|
||||
user_id = msg.from_user.id
|
||||
if is_user_auth(user_id):
|
||||
phone = AUTHORIZED_USERS.get(user_id)
|
||||
if not phone:
|
||||
await msg.answer("Номер не найден")
|
||||
user_data = AUTHORIZED_USERS.get(user_id)
|
||||
if not user_data:
|
||||
await msg.answer("Ошибка авторизации.")
|
||||
return
|
||||
|
||||
user_conf = config.get("user", {}).get(phone)
|
||||
if not user_conf:
|
||||
await msg.answer("Пользователь не найден в конфигурации")
|
||||
return
|
||||
|
||||
allowed_locks = user_conf.get("locks_id", [])
|
||||
reply_markup = get_locks_keyboard(allowed_locks)
|
||||
await msg.answer(
|
||||
"Авторизация прошла успешно",
|
||||
reply_markup=reply_markup,
|
||||
)
|
||||
allowed_zones = user_data["zones"]
|
||||
reply_markup = await get_locks_keyboard(allowed_zones)
|
||||
await msg.answer("Авторизация прошла успешно", reply_markup=reply_markup)
|
||||
else:
|
||||
reply_markup = get_contact_keyboard()
|
||||
await msg.answer(
|
||||
"Для пользования ботом, предоставьте номер телефона",
|
||||
"Для пользования ботом предоставьте номер телефона",
|
||||
reply_markup=reply_markup,
|
||||
)
|
||||
|
|
|
|||
|
|
@ -1,30 +0,0 @@
|
|||
import os
|
||||
from dotenv import load_dotenv
|
||||
|
||||
|
||||
ENV_FILE = ".env"
|
||||
|
||||
|
||||
def check_env_file():
|
||||
return os.path.exists(ENV_FILE)
|
||||
|
||||
|
||||
def create_env_file():
|
||||
print("Файл .env отсутствует и будет создан автоматически.")
|
||||
|
||||
token = input("Введите TOKEN: ")
|
||||
lock_ip = input("Введите LOCK_IP: ")
|
||||
card_id = input("Введите CARD_ID: ")
|
||||
auth_api = input("Введите AUTH_API: ")
|
||||
|
||||
with open(ENV_FILE, "w") as f:
|
||||
f.write(f"TOKEN={token}\n")
|
||||
f.write(f"LOCK_IP={lock_ip}\n")
|
||||
f.write(f"CARD_ID={card_id}\n")
|
||||
f.write(f"AUTH_API={auth_api}\n")
|
||||
|
||||
print("Файл .env создан. Происходит запуск приложения")
|
||||
|
||||
|
||||
def load_env():
|
||||
load_dotenv()
|
||||
11
keyboard.py
11
keyboard.py
|
|
@ -1,12 +1,15 @@
|
|||
from aiogram.utils.keyboard import ReplyKeyboardBuilder
|
||||
from aiogram.types import KeyboardButton
|
||||
from locks_api import fetch_locks
|
||||
|
||||
|
||||
def get_locks_keyboard(allowed_locks: list):
|
||||
print(f"DEBUG: allowed_locks = {allowed_locks}")
|
||||
async def get_locks_keyboard(allowed_locks: list):
|
||||
kb = ReplyKeyboardBuilder()
|
||||
for lock in allowed_locks:
|
||||
kb.button(text=lock)
|
||||
locks = await fetch_locks()
|
||||
for zone in allowed_locks:
|
||||
lock_info = locks.get(zone)
|
||||
btn_text = lock_info["name"] if lock_info and "name" in lock_info else zone
|
||||
kb.button(text=btn_text)
|
||||
return kb.as_markup(resize_keyboard=True)
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -0,0 +1,35 @@
|
|||
import aiohttp
|
||||
import json
|
||||
|
||||
LOCKS_API_URL = "https://papi.dataekb.ru/get_pacs"
|
||||
|
||||
|
||||
async def fetch_locks() -> dict:
|
||||
async with aiohttp.ClientSession() as session:
|
||||
try:
|
||||
async with session.get(LOCKS_API_URL) as response:
|
||||
if response.status != 200:
|
||||
print(f"Ошибка при получении данных замков: {response.status}")
|
||||
return {}
|
||||
text = await response.text()
|
||||
except Exception as e:
|
||||
print(f"Исключение при запросе данных замков: {str(e)}")
|
||||
return {}
|
||||
try:
|
||||
data = json.loads(text)
|
||||
if not data or not isinstance(data, list):
|
||||
print("Неверный формат данных о замках, ожидался список.")
|
||||
return {}
|
||||
locks_dict = data[0]
|
||||
return locks_dict
|
||||
except Exception as e:
|
||||
print(f"Ошибка при разборе данных замков: {str(e)}")
|
||||
return {}
|
||||
|
||||
|
||||
async def get_lock_by_label(label: str) -> dict:
|
||||
locks = await fetch_locks()
|
||||
for lock_id, details in locks.items():
|
||||
if label == lock_id or details.get("name") == label:
|
||||
return details
|
||||
return {}
|
||||
15
main.py
15
main.py
|
|
@ -1,23 +1,32 @@
|
|||
import os
|
||||
import asyncio
|
||||
import logging
|
||||
import sys
|
||||
|
||||
|
||||
from aiogram import Bot, Dispatcher
|
||||
from aiogram.enums import ParseMode
|
||||
from aiogram.client.default import DefaultBotProperties
|
||||
from aiogram.client.session.aiohttp import AiohttpSession
|
||||
|
||||
from handlers import register_all_handlers
|
||||
from config import config
|
||||
|
||||
|
||||
BOT_TOKEN = config["bot_token"]
|
||||
# BOT_TOKEN = config["BOT_TOKEN"]
|
||||
BOT_TOKEN=os.environ.get('BOT_TOKEN')
|
||||
PROXY_URL = os.environ.get('PROXY_URL')
|
||||
|
||||
dp = Dispatcher()
|
||||
register_all_handlers(dp)
|
||||
|
||||
session = AiohttpSession(proxy=PROXY_URL)
|
||||
|
||||
async def main() -> None:
|
||||
bot = Bot(token=BOT_TOKEN, default=DefaultBotProperties(parse_mode=ParseMode.HTML))
|
||||
bot = Bot(
|
||||
token=BOT_TOKEN
|
||||
, session=session
|
||||
, default=DefaultBotProperties(parse_mode=ParseMode.HTML)
|
||||
)
|
||||
await dp.start_polling(bot)
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -14,8 +14,8 @@ multidict==6.4.3
|
|||
propcache==0.3.1
|
||||
pydantic==2.11.3
|
||||
pydantic_core==2.33.1
|
||||
requests==2.32.3
|
||||
typing-inspection==0.4.0
|
||||
typing_extensions==4.13.2
|
||||
urllib3==2.4.0
|
||||
yarl==1.20.0
|
||||
aiohttp-socks
|
||||
|
|
|
|||
|
|
@ -0,0 +1,15 @@
|
|||
# tunnel/Dockerfile
|
||||
FROM alpine:3.19
|
||||
|
||||
RUN apk add --no-cache \
|
||||
autossh \
|
||||
openssh-client
|
||||
|
||||
# Директория для SSH ключей
|
||||
RUN mkdir -p /root/.ssh && \
|
||||
chmod 700 /root/.ssh
|
||||
|
||||
COPY entrypoint.sh /entrypoint.sh
|
||||
RUN chmod +x /entrypoint.sh
|
||||
|
||||
ENTRYPOINT ["/entrypoint.sh"]
|
||||
|
|
@ -0,0 +1,16 @@
|
|||
#!/bin/sh
|
||||
|
||||
ssh-keyscan -p ${SSH_PORT:-22} ${SSH_HOST} >> /root/.ssh/known_hosts 2>/dev/null
|
||||
|
||||
exec autossh \
|
||||
-M 0 \
|
||||
-N \
|
||||
-D 0.0.0.0:1080 \
|
||||
-o "ServerAliveInterval=30" \
|
||||
-o "ServerAliveCountMax=3" \
|
||||
-o "ExitOnForwardFailure=yes" \
|
||||
-o "StrictHostKeyChecking=no" \
|
||||
-o "ConnectTimeout=10" \
|
||||
-p ${SSH_PORT:-22} \
|
||||
-i /root/.ssh/id_rsa \
|
||||
${SSH_USER}@${SSH_HOST}
|
||||
Loading…
Reference in New Issue