Compare commits
4 Commits
5662d8877a
...
c0f4be1179
Author | SHA1 | Date | |
---|---|---|---|
c0f4be1179 | |||
4748127399 | |||
1148d91598 | |||
cbb33f6da8 |
@@ -11,8 +11,7 @@ admin_router = Router()
|
||||
|
||||
admin_router.message.filter(IsAdmin())
|
||||
|
||||
|
||||
regex = re.compile(r'(del|reg) @.+')
|
||||
regex = re.compile(r'(del|reg)_@.+')
|
||||
|
||||
|
||||
@admin_router.callback_query(lambda x: re.fullmatch(regex, x.data))
|
||||
@@ -30,5 +29,3 @@ async def send_message_command(message: Message, bot: Bot):
|
||||
chat_id = re.search(r'(\d+)', message.text).group()
|
||||
print(chat_id)
|
||||
await bot.send_message(text='Ronis->' + message.text.strip('@msg_' + chat_id), chat_id=chat_id)
|
||||
|
||||
|
||||
|
@@ -4,10 +4,11 @@ from pathlib import Path
|
||||
import re
|
||||
from aiogram import Router, Bot, F
|
||||
from aiogram.filters import CommandStart, Command
|
||||
from aiogram.types import Message, CallbackQuery, FSInputFile, InputMediaPhoto, InputMediaVideo
|
||||
from aiogram.types import Message, CallbackQuery, FSInputFile, InputMediaPhoto, InputMediaVideo, ReplyKeyboardRemove
|
||||
from aiogram.exceptions import AiogramError
|
||||
from aiogram.fsm.state import State, StatesGroup
|
||||
from aiogram.fsm.context import FSMContext
|
||||
|
||||
from sqlalchemy import select, insert
|
||||
from loguru import logger
|
||||
from filters import IsAdmin
|
||||
@@ -60,22 +61,29 @@ async def orders_menu(message: Message):
|
||||
@orders_router.callback_query(lambda x: x.data.startswith("create_order"))
|
||||
async def get_order_worker_id(callback: CallbackQuery, state: FSMContext):
|
||||
await state.set_state(OrderForm.worker_id)
|
||||
await callback.message.answer("Введите id сборщика который будет собирать заказ:",
|
||||
reply_markup=create_inline_kb(**{f"{callback.from_user.id}": "Ввести мой id"}))
|
||||
async with async_session_() as session:
|
||||
async with session.begin():
|
||||
result = await session.execute(select(Worker.name, Worker.telegram_id))
|
||||
workers = result.all()
|
||||
await callback.message.answer("Введите сборщика ответственного за заказ:",
|
||||
reply_markup=create_inline_kb(
|
||||
**{f"set_worker_in_order_{tg_id}": name for name, tg_id in workers}))
|
||||
await callback.message.delete()
|
||||
|
||||
|
||||
@orders_router.message(F.text == "❌ Прервать создание заказа")
|
||||
async def order_description(message: Message, state: FSMContext):
|
||||
await state.clear()
|
||||
await message.answer("Создание заказа прервано", reply_markup=ReplyKeyboardRemove())
|
||||
|
||||
|
||||
@orders_router.callback_query(OrderForm.worker_id)
|
||||
async def get_order_counterparty(message: Message | CallbackQuery, state: FSMContext):
|
||||
if isinstance(message, Message):
|
||||
worker_id = int(message.text)
|
||||
msg = message
|
||||
else:
|
||||
worker_id = int(message.data)
|
||||
msg = message.message
|
||||
await msg.answer("Введите данные заказчика ")
|
||||
await msg.delete()
|
||||
await state.update_data(worker_id=worker_id)
|
||||
async def get_order_customer(callback: CallbackQuery, state: FSMContext):
|
||||
worker_tg_id = int(re.search(r"(\d+)", callback.data).group())
|
||||
await callback.message.answer("Введите данные заказчика ",
|
||||
reply_markup=button_create(["❌ Прервать создание заказа"]))
|
||||
await callback.message.delete()
|
||||
await state.update_data(worker_id=worker_tg_id)
|
||||
await state.set_state(OrderForm.customer)
|
||||
|
||||
|
||||
@@ -84,8 +92,9 @@ async def get_order_counterparty(message: Message | CallbackQuery, state: FSMCon
|
||||
#
|
||||
|
||||
@orders_router.message(OrderForm.customer)
|
||||
async def create_order(message: Message, state: FSMContext):
|
||||
async def order_description(message: Message, state: FSMContext):
|
||||
await message.answer("Введите описание заказа в виде ключевых слов (АВР, ПСС, НКУ и т.д.) )")
|
||||
await message.delete()
|
||||
await state.update_data(customer=message.text)
|
||||
await state.set_state(OrderForm.description)
|
||||
|
||||
@@ -99,7 +108,7 @@ async def create_order(message: Message, state: FSMContext):
|
||||
session.add(Order(**order_))
|
||||
|
||||
await state.clear()
|
||||
await message.answer("Заказ успешно создан ")
|
||||
await message.answer("Заказ успешно создан ", reply_markup=ReplyKeyboardRemove())
|
||||
|
||||
|
||||
@orders_router.callback_query(lambda x: x.data == "find_orders")
|
||||
@@ -123,13 +132,18 @@ async def search_by_item(message: Message, state: FSMContext):
|
||||
async with async_session_() as local_session:
|
||||
search_opt = await state.get_value("search_option")
|
||||
col = getattr(Order, search_opt)
|
||||
await message.answer(message.text)
|
||||
result = await local_session.execute(
|
||||
select(Order).where(col.ilike(f"%{message.text}%") if search_opt != "id" else col == int(message.text)))
|
||||
selected_orders = result.scalars().all()
|
||||
|
||||
if selected_orders:
|
||||
await message.answer(text="Список заказов", reply_markup=create_inline_kb(width=1, **dict(
|
||||
(f"show_order_{order.id}", order.description or "Отсутствует") for order in selected_orders)))
|
||||
await message.answer(text="Список найденных заказов: ",
|
||||
reply_markup=create_inline_kb(width=1, **dict(
|
||||
(f"show_order_{order.id}", f"Описание: {order.description or "Отсутствует"}"
|
||||
f"Заказ: №{order.id}\n"
|
||||
f"Заказчик: {order.customer}\n"
|
||||
) for order in
|
||||
selected_orders)))
|
||||
await state.update_data(search_result=selected_orders)
|
||||
await state.set_state(SearchForm.search_result)
|
||||
else:
|
||||
|
@@ -1,12 +1,11 @@
|
||||
import os
|
||||
from asyncio import Event
|
||||
from asyncio import Event, wait_for, TimeoutError
|
||||
|
||||
from aiogram import Router, Bot
|
||||
from aiogram.filters import CommandStart
|
||||
from aiogram.types import Message, User
|
||||
|
||||
from sqlalchemy import insert, select
|
||||
from sqlalchemy.orm import selectinload
|
||||
|
||||
from keyboards import create_inline_kb
|
||||
from database import async_session_, Worker
|
||||
@@ -20,10 +19,11 @@ user_info_template = ("Новый пользователь ждет регист
|
||||
"Юзернейм: @{}\n"
|
||||
"ID: @msg_{}\n")
|
||||
|
||||
admins_ids = list(map(int, os.getenv("BOT_ADMINS").split(",")))
|
||||
|
||||
|
||||
@registration_router.message(CommandStart())
|
||||
async def registration_command(message: Message, bot: Bot):
|
||||
admins_ids = os.getenv("BOT_ADMINS").split(",")
|
||||
async with async_session_() as session:
|
||||
async with session.begin():
|
||||
result = await session.execute(select(Worker).where(Worker.telegram_id == message.from_user.id))
|
||||
@@ -31,19 +31,28 @@ async def registration_command(message: Message, bot: Bot):
|
||||
if not user:
|
||||
|
||||
user = message.from_user
|
||||
dict_for_inline = {f'reg @{user.id}': 'Allow', f'del @{user.id}': 'Reject'}
|
||||
dict_for_inline = {f'reg_@{user.id}': 'Allow', f'del_@{user.id}': 'Reject'}
|
||||
user_info = user_info_template.format(user.first_name, user.last_name if user.last_name else 'Не указана',
|
||||
user.username if user.username else 'Не указан', user.id)
|
||||
for admin in admins_ids:
|
||||
try:
|
||||
await bot.send_message(chat_id=admin, text=user_info)
|
||||
await bot.send_message(chat_id=admin, text='Зарегистрировать пользователя',
|
||||
reply_markup=create_inline_kb(width=2, **dict_for_inline))
|
||||
except Exception as err:
|
||||
pass
|
||||
reg_confirm = Event()
|
||||
registration_confirm[user.id] = reg_confirm
|
||||
if await reg_confirm:
|
||||
try:
|
||||
await wait_for(reg_confirm.wait(), timeout=60)
|
||||
async with async_session_() as local_session:
|
||||
async with local_session.begin():
|
||||
local_session.add(Worker(telegram_id=int(user.id), name=user.first_name))
|
||||
local_session.add(Worker(telegram_id=user.id, name=user.first_name))
|
||||
|
||||
await message.answer("Регистрация подтверждена")
|
||||
except TimeoutError:
|
||||
await message.answer("Время ожидания истекло.")
|
||||
|
||||
del registration_confirm[user.id]
|
||||
|
||||
else:
|
||||
|
@@ -1,7 +1,7 @@
|
||||
CREATE TABLE workers
|
||||
(
|
||||
id SERIAL PRIMARY KEY,
|
||||
telegram_id INTEGER UNIQUE NOT NULL,
|
||||
telegram_id BIGINT UNIQUE NOT NULL,
|
||||
name VARCHAR NOT NULL,
|
||||
email VARCHAR(50),
|
||||
phone_number VARCHAR(20) NOT NULL,
|
||||
@@ -15,7 +15,7 @@ CREATE TABLE orders
|
||||
(
|
||||
id SERIAL PRIMARY KEY,
|
||||
name VARCHAR,
|
||||
worker_id INTEGER REFERENCES workers (telegram_id),
|
||||
worker_id BIGINT REFERENCES workers (telegram_id),
|
||||
status_id status DEFAULT 'Создано',
|
||||
counterparty VARCHAR(50),
|
||||
customer VARCHAR NOT NULL,
|
||||
|
Reference in New Issue
Block a user