mirror of
https://gitlab.com/MoonTestUse1/AdministrationItDepartmens.git
synced 2025-08-14 00:25:46 +02:00
отправка уведомлений в телеграм
This commit is contained in:
@@ -1,3 +1,3 @@
|
|||||||
|
DATABASE_URL=postgresql://postgres:postgres123@postgres/support_db
|
||||||
TELEGRAM_BOT_TOKEN=7677506032:AAHduD5EePz3bE23DKlo35KoOp2_9lZuS34
|
TELEGRAM_BOT_TOKEN=7677506032:AAHduD5EePz3bE23DKlo35KoOp2_9lZuS34
|
||||||
TELEGRAM_CHAT_ID=5057752127
|
TELEGRAM_CHAT_ID=5057752127
|
||||||
DATABASE_URL=postgresql://postgres:postgres123@postgres/support_db
|
|
@@ -7,6 +7,8 @@ from ..models.request import Request, RequestStatus, RequestPriority
|
|||||||
from ..schemas.request import RequestCreate, RequestResponse, RequestUpdate, RequestStatistics
|
from ..schemas.request import RequestCreate, RequestResponse, RequestUpdate, RequestStatistics
|
||||||
from ..utils.auth import get_current_admin, get_current_employee
|
from ..utils.auth import get_current_admin, get_current_employee
|
||||||
from sqlalchemy import func
|
from sqlalchemy import func
|
||||||
|
from ..models.employee import Employee
|
||||||
|
from ..utils.telegram import notify_new_request
|
||||||
|
|
||||||
router = APIRouter()
|
router = APIRouter()
|
||||||
|
|
||||||
@@ -39,7 +41,7 @@ def get_requests(
|
|||||||
def create_request(
|
def create_request(
|
||||||
request: RequestCreate,
|
request: RequestCreate,
|
||||||
db: Session = Depends(get_db),
|
db: Session = Depends(get_db),
|
||||||
current_employee: dict = Depends(get_current_employee)
|
current_employee: Employee = Depends(get_current_employee)
|
||||||
):
|
):
|
||||||
"""Create new request"""
|
"""Create new request"""
|
||||||
try:
|
try:
|
||||||
@@ -47,14 +49,17 @@ def create_request(
|
|||||||
title=request.title,
|
title=request.title,
|
||||||
description=request.description,
|
description=request.description,
|
||||||
priority=request.priority,
|
priority=request.priority,
|
||||||
status=RequestStatus.NEW.value,
|
status=RequestStatus.NEW,
|
||||||
employee_id=current_employee["id"]
|
employee_id=current_employee.id
|
||||||
)
|
)
|
||||||
|
|
||||||
db.add(db_request)
|
db.add(db_request)
|
||||||
db.commit()
|
db.commit()
|
||||||
db.refresh(db_request)
|
db.refresh(db_request)
|
||||||
|
|
||||||
|
# Отправляем уведомление в Telegram
|
||||||
|
notify_new_request(db_request.id)
|
||||||
|
|
||||||
return request_to_dict(db_request)
|
return request_to_dict(db_request)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
db.rollback()
|
db.rollback()
|
||||||
@@ -63,12 +68,12 @@ def create_request(
|
|||||||
@router.get("/my", response_model=List[RequestResponse])
|
@router.get("/my", response_model=List[RequestResponse])
|
||||||
def get_employee_requests(
|
def get_employee_requests(
|
||||||
db: Session = Depends(get_db),
|
db: Session = Depends(get_db),
|
||||||
current_employee: dict = Depends(get_current_employee)
|
current_employee: Employee = Depends(get_current_employee)
|
||||||
):
|
):
|
||||||
"""Get employee's requests"""
|
"""Get employee's requests"""
|
||||||
try:
|
try:
|
||||||
requests = db.query(Request).filter(
|
requests = db.query(Request).filter(
|
||||||
Request.employee_id == current_employee["id"]
|
Request.employee_id == current_employee.id
|
||||||
).all()
|
).all()
|
||||||
|
|
||||||
# Преобразуем объекты в словари до закрытия сессии
|
# Преобразуем объекты в словари до закрытия сессии
|
||||||
|
@@ -3,11 +3,12 @@ import os
|
|||||||
from fastapi import APIRouter, Request
|
from fastapi import APIRouter, Request
|
||||||
from telebot import TeleBot
|
from telebot import TeleBot
|
||||||
from telebot.types import Update
|
from telebot.types import Update
|
||||||
from ..models.request import RequestStatus
|
from ..models.request import RequestStatus, RequestPriority
|
||||||
from ..database import SessionLocal
|
from ..database import SessionLocal
|
||||||
from ..models.request import Request as DBRequest
|
from ..models.request import Request as DBRequest
|
||||||
|
|
||||||
TELEGRAM_BOT_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
|
TELEGRAM_BOT_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
|
||||||
|
TELEGRAM_CHAT_ID = os.getenv("TELEGRAM_CHAT_ID")
|
||||||
WEBHOOK_URL = "https://itformhelp.ru/telegram/webhook/"
|
WEBHOOK_URL = "https://itformhelp.ru/telegram/webhook/"
|
||||||
WEBHOOK_PATH = "/telegram/webhook/"
|
WEBHOOK_PATH = "/telegram/webhook/"
|
||||||
|
|
||||||
@@ -37,6 +38,34 @@ def handle_message(message):
|
|||||||
"""Handle all messages"""
|
"""Handle all messages"""
|
||||||
bot.reply_to(message, "Я получил ваше сообщение и обязательно обработаю его!")
|
bot.reply_to(message, "Я получил ваше сообщение и обязательно обработаю его!")
|
||||||
|
|
||||||
|
def format_priority(priority: str) -> str:
|
||||||
|
"""Format priority with emoji"""
|
||||||
|
priority_emoji = {
|
||||||
|
RequestPriority.LOW: "🟢",
|
||||||
|
RequestPriority.MEDIUM: "🟡",
|
||||||
|
RequestPriority.HIGH: "🔴"
|
||||||
|
}
|
||||||
|
return f"{priority_emoji.get(priority, '⚪')} {priority.capitalize()}"
|
||||||
|
|
||||||
|
def notify_new_request(request_id: int):
|
||||||
|
"""Send notification about new request"""
|
||||||
|
try:
|
||||||
|
db = SessionLocal()
|
||||||
|
request = db.query(DBRequest).filter(DBRequest.id == request_id).first()
|
||||||
|
if request:
|
||||||
|
message = (
|
||||||
|
f"📋 <b>Новая заявка #{request.id}</b>\n\n"
|
||||||
|
f"📝 <b>Заголовок:</b> {request.title}\n"
|
||||||
|
f"👤 <b>Сотрудник:</b> {request.employee.last_name} {request.employee.first_name}\n"
|
||||||
|
f"❗ <b>Приоритет:</b> {format_priority(request.priority)}\n\n"
|
||||||
|
f"📄 <b>Описание:</b>\n{request.description}"
|
||||||
|
)
|
||||||
|
bot.send_message(TELEGRAM_CHAT_ID, message, parse_mode="HTML")
|
||||||
|
except Exception as e:
|
||||||
|
print(f"Error sending telegram notification: {e}")
|
||||||
|
finally:
|
||||||
|
db.close()
|
||||||
|
|
||||||
def notify_status_change(request_id: int, new_status: RequestStatus):
|
def notify_status_change(request_id: int, new_status: RequestStatus):
|
||||||
"""Notify user about request status change"""
|
"""Notify user about request status change"""
|
||||||
try:
|
try:
|
||||||
|
Reference in New Issue
Block a user