From 4730fb646b3bf1ddcd95b2aae5697a0d160d1f33 Mon Sep 17 00:00:00 2001 From: LoRiot Date: Sun, 16 Oct 2022 16:30:57 +0300 Subject: [PATCH] Add add_user method --- handlers/commands.py | 12 +++++++++--- services/repository.py | 7 +++++++ 2 files changed, 16 insertions(+), 3 deletions(-) diff --git a/handlers/commands.py b/handlers/commands.py index 69202bc..fa4032a 100644 --- a/handlers/commands.py +++ b/handlers/commands.py @@ -1,12 +1,18 @@ +from loguru import logger + from aiogram import types, Dispatcher +from services.repository import Repo -async def create_project(message: types.Message): +async def request_invite(message: types.Message, repo: Repo): + await repo.add_user(message.from_user.id, message.chat.id) + logger.debug(f"Added new user {message.from_user.full_name}") + bot_info = await message.bot.get_me() await message.reply( - f"Перейдите в лс к боту @{bot_info.username}" + f"Чтобы создать Ваш проект, перейдите в лс к боту @{bot_info.username}" ) def register_command_handlers(dp: Dispatcher): - dp.register_message_handler(create_project, commands="create_project") + dp.register_message_handler(request_invite, commands="inviteme") diff --git a/services/repository.py b/services/repository.py index 6bf0d1a..cde7404 100644 --- a/services/repository.py +++ b/services/repository.py @@ -6,3 +6,10 @@ class Repo: def __init__(self, conn: Connection): self.conn = conn + + async def add_user(self, user_id: int, chat_id: int): + await self.conn.execute( + "INSERT INTO users (user_id, chat_id) " + "VALUES ($1, $2) ON CONFLICT (user_id) DO NOTHING", + user_id, chat_id + )