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())
|
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))
|
@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()
|
chat_id = re.search(r'(\d+)', message.text).group()
|
||||||
print(chat_id)
|
print(chat_id)
|
||||||
await bot.send_message(text='Ronis->' + message.text.strip('@msg_' + chat_id), chat_id=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
|
import re
|
||||||
from aiogram import Router, Bot, F
|
from aiogram import Router, Bot, F
|
||||||
from aiogram.filters import CommandStart, Command
|
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.exceptions import AiogramError
|
||||||
from aiogram.fsm.state import State, StatesGroup
|
from aiogram.fsm.state import State, StatesGroup
|
||||||
from aiogram.fsm.context import FSMContext
|
from aiogram.fsm.context import FSMContext
|
||||||
|
|
||||||
from sqlalchemy import select, insert
|
from sqlalchemy import select, insert
|
||||||
from loguru import logger
|
from loguru import logger
|
||||||
from filters import IsAdmin
|
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"))
|
@orders_router.callback_query(lambda x: x.data.startswith("create_order"))
|
||||||
async def get_order_worker_id(callback: CallbackQuery, state: FSMContext):
|
async def get_order_worker_id(callback: CallbackQuery, state: FSMContext):
|
||||||
await state.set_state(OrderForm.worker_id)
|
await state.set_state(OrderForm.worker_id)
|
||||||
await callback.message.answer("Введите id сборщика который будет собирать заказ:",
|
async with async_session_() as session:
|
||||||
reply_markup=create_inline_kb(**{f"{callback.from_user.id}": "Ввести мой id"}))
|
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()
|
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)
|
@orders_router.callback_query(OrderForm.worker_id)
|
||||||
async def get_order_counterparty(message: Message | CallbackQuery, state: FSMContext):
|
async def get_order_customer(callback: CallbackQuery, state: FSMContext):
|
||||||
if isinstance(message, Message):
|
worker_tg_id = int(re.search(r"(\d+)", callback.data).group())
|
||||||
worker_id = int(message.text)
|
await callback.message.answer("Введите данные заказчика ",
|
||||||
msg = message
|
reply_markup=button_create(["❌ Прервать создание заказа"]))
|
||||||
else:
|
await callback.message.delete()
|
||||||
worker_id = int(message.data)
|
await state.update_data(worker_id=worker_tg_id)
|
||||||
msg = message.message
|
|
||||||
await msg.answer("Введите данные заказчика ")
|
|
||||||
await msg.delete()
|
|
||||||
await state.update_data(worker_id=worker_id)
|
|
||||||
await state.set_state(OrderForm.customer)
|
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)
|
@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.answer("Введите описание заказа в виде ключевых слов (АВР, ПСС, НКУ и т.д.) )")
|
||||||
|
await message.delete()
|
||||||
await state.update_data(customer=message.text)
|
await state.update_data(customer=message.text)
|
||||||
await state.set_state(OrderForm.description)
|
await state.set_state(OrderForm.description)
|
||||||
|
|
||||||
@@ -99,7 +108,7 @@ async def create_order(message: Message, state: FSMContext):
|
|||||||
session.add(Order(**order_))
|
session.add(Order(**order_))
|
||||||
|
|
||||||
await state.clear()
|
await state.clear()
|
||||||
await message.answer("Заказ успешно создан ")
|
await message.answer("Заказ успешно создан ", reply_markup=ReplyKeyboardRemove())
|
||||||
|
|
||||||
|
|
||||||
@orders_router.callback_query(lambda x: x.data == "find_orders")
|
@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:
|
async with async_session_() as local_session:
|
||||||
search_opt = await state.get_value("search_option")
|
search_opt = await state.get_value("search_option")
|
||||||
col = getattr(Order, search_opt)
|
col = getattr(Order, search_opt)
|
||||||
await message.answer(message.text)
|
|
||||||
result = await local_session.execute(
|
result = await local_session.execute(
|
||||||
select(Order).where(col.ilike(f"%{message.text}%") if search_opt != "id" else col == int(message.text)))
|
select(Order).where(col.ilike(f"%{message.text}%") if search_opt != "id" else col == int(message.text)))
|
||||||
selected_orders = result.scalars().all()
|
selected_orders = result.scalars().all()
|
||||||
|
|
||||||
if selected_orders:
|
if selected_orders:
|
||||||
await message.answer(text="Список заказов", reply_markup=create_inline_kb(width=1, **dict(
|
await message.answer(text="Список найденных заказов: ",
|
||||||
(f"show_order_{order.id}", order.description or "Отсутствует") for order in selected_orders)))
|
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.update_data(search_result=selected_orders)
|
||||||
await state.set_state(SearchForm.search_result)
|
await state.set_state(SearchForm.search_result)
|
||||||
else:
|
else:
|
||||||
|
@@ -1,12 +1,11 @@
|
|||||||
import os
|
import os
|
||||||
from asyncio import Event
|
from asyncio import Event, wait_for, TimeoutError
|
||||||
|
|
||||||
from aiogram import Router, Bot
|
from aiogram import Router, Bot
|
||||||
from aiogram.filters import CommandStart
|
from aiogram.filters import CommandStart
|
||||||
from aiogram.types import Message, User
|
from aiogram.types import Message, User
|
||||||
|
|
||||||
from sqlalchemy import insert, select
|
from sqlalchemy import insert, select
|
||||||
from sqlalchemy.orm import selectinload
|
|
||||||
|
|
||||||
from keyboards import create_inline_kb
|
from keyboards import create_inline_kb
|
||||||
from database import async_session_, Worker
|
from database import async_session_, Worker
|
||||||
@@ -20,10 +19,11 @@ user_info_template = ("Новый пользователь ждет регист
|
|||||||
"Юзернейм: @{}\n"
|
"Юзернейм: @{}\n"
|
||||||
"ID: @msg_{}\n")
|
"ID: @msg_{}\n")
|
||||||
|
|
||||||
|
admins_ids = list(map(int, os.getenv("BOT_ADMINS").split(",")))
|
||||||
|
|
||||||
|
|
||||||
@registration_router.message(CommandStart())
|
@registration_router.message(CommandStart())
|
||||||
async def registration_command(message: Message, bot: Bot):
|
async def registration_command(message: Message, bot: Bot):
|
||||||
admins_ids = os.getenv("BOT_ADMINS").split(",")
|
|
||||||
async with async_session_() as session:
|
async with async_session_() as session:
|
||||||
async with session.begin():
|
async with session.begin():
|
||||||
result = await session.execute(select(Worker).where(Worker.telegram_id == message.from_user.id))
|
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:
|
if not user:
|
||||||
|
|
||||||
user = message.from_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_info = user_info_template.format(user.first_name, user.last_name if user.last_name else 'Не указана',
|
||||||
user.username if user.username else 'Не указан', user.id)
|
user.username if user.username else 'Не указан', user.id)
|
||||||
for admin in admins_ids:
|
for admin in admins_ids:
|
||||||
|
try:
|
||||||
await bot.send_message(chat_id=admin, text=user_info)
|
await bot.send_message(chat_id=admin, text=user_info)
|
||||||
await bot.send_message(chat_id=admin, text='Зарегистрировать пользователя',
|
await bot.send_message(chat_id=admin, text='Зарегистрировать пользователя',
|
||||||
reply_markup=create_inline_kb(width=2, **dict_for_inline))
|
reply_markup=create_inline_kb(width=2, **dict_for_inline))
|
||||||
|
except Exception as err:
|
||||||
|
pass
|
||||||
reg_confirm = Event()
|
reg_confirm = Event()
|
||||||
registration_confirm[user.id] = reg_confirm
|
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 async_session_() as local_session:
|
||||||
async with local_session.begin():
|
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]
|
del registration_confirm[user.id]
|
||||||
|
|
||||||
else:
|
else:
|
||||||
|
@@ -1,7 +1,7 @@
|
|||||||
CREATE TABLE workers
|
CREATE TABLE workers
|
||||||
(
|
(
|
||||||
id SERIAL PRIMARY KEY,
|
id SERIAL PRIMARY KEY,
|
||||||
telegram_id INTEGER UNIQUE NOT NULL,
|
telegram_id BIGINT UNIQUE NOT NULL,
|
||||||
name VARCHAR NOT NULL,
|
name VARCHAR NOT NULL,
|
||||||
email VARCHAR(50),
|
email VARCHAR(50),
|
||||||
phone_number VARCHAR(20) NOT NULL,
|
phone_number VARCHAR(20) NOT NULL,
|
||||||
@@ -15,7 +15,7 @@ CREATE TABLE orders
|
|||||||
(
|
(
|
||||||
id SERIAL PRIMARY KEY,
|
id SERIAL PRIMARY KEY,
|
||||||
name VARCHAR,
|
name VARCHAR,
|
||||||
worker_id INTEGER REFERENCES workers (telegram_id),
|
worker_id BIGINT REFERENCES workers (telegram_id),
|
||||||
status_id status DEFAULT 'Создано',
|
status_id status DEFAULT 'Создано',
|
||||||
counterparty VARCHAR(50),
|
counterparty VARCHAR(50),
|
||||||
customer VARCHAR NOT NULL,
|
customer VARCHAR NOT NULL,
|
||||||
|
Reference in New Issue
Block a user