diff --git a/app/FSM/states.py b/app/FSM/states.py index 54bc7aa..4ec72a2 100644 --- a/app/FSM/states.py +++ b/app/FSM/states.py @@ -4,6 +4,7 @@ from aiogram.fsm.state import State, StatesGroup class SearchForm(StatesGroup): search_option = State() data_to_search = State() + sent_messages = State() search_result = State() diff --git a/app/database/database_engine.py b/app/database/database_engine.py index 59073e2..15f2335 100644 --- a/app/database/database_engine.py +++ b/app/database/database_engine.py @@ -1,8 +1,8 @@ import os -from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine +from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine, session + DATABASE_URL = (f"postgresql+asyncpg://{os.getenv('DB_USER')}:{os.getenv('DB_PASSWORD')}@" f"{os.getenv('DB_HOST')}/{os.getenv('DATABASE')}") - engine = create_async_engine(DATABASE_URL, echo=True) async_session_ = async_sessionmaker(bind=engine, expire_on_commit=False) diff --git a/app/handlers/orders.py b/app/handlers/orders.py index cbb44a6..bb774a1 100644 --- a/app/handlers/orders.py +++ b/app/handlers/orders.py @@ -132,14 +132,6 @@ async def search_by_item(message: Message, state: FSMContext): async def show_order(callback: CallbackQuery, state: FSMContext, bot: Bot): order_id = int(re.search(r"(\d+)", callback.data).group()) - try: - async with async_session_() as local_session: - result = await local_session.execute(select(Order).filter(Order.id == order_id)) - order = result.scalars().first() - - except Exception as err: - logger.warning(err) - order = next(filter(lambda item: order_id == item.id, await state.get_value("search_result")), None) if order: @@ -154,7 +146,7 @@ async def show_order(callback: CallbackQuery, state: FSMContext, bot: Bot): (f"{clbk}_{order.id}", text) for clbk, text in order_operation_base.items())) ) - for message in SearchForm.sent_messages: + for message in await state.get_value("sent_messages"): await bot.delete_message(chat_id=callback.message.chat.id, message_id=message.message_id) await state.clear() diff --git a/app/main.py b/app/main.py index 0cbd730..b151107 100644 --- a/app/main.py +++ b/app/main.py @@ -1,12 +1,14 @@ import os import asyncio from dotenv import load_dotenv + +load_dotenv(".env") + from aiogram import Dispatcher, Bot from handlers import * from keyboards import set_main_menu from middlewares import SessionMiddleware -load_dotenv(".env") bot = Bot(token=os.getenv("TOKEN")) diff --git a/drone.yaml b/drone.yaml index a098435..74f3b97 100644 --- a/drone.yaml +++ b/drone.yaml @@ -42,8 +42,7 @@ steps: commands: - docker stop myapp || true - docker rm myapp || true - - docker run --name=myapp --network=prod_net -v /srv/prod/telegram_bot/photos:/app/photos/ \ - -v /srv/prod/telegram_bot/.env:/app/.env:ro myapp:${DRONE_COMMIT_BRANCH} + - docker run --name=myapp --network=prod_net -v /srv/prod/telegram_bot/photos:/app/photos/ -v /srv/prod/telegram_bot/.env:/app/.env:ro myapp:${DRONE_COMMIT_BRANCH} volumes: