21 lines
566 B
Python
21 lines
566 B
Python
from asyncpg.connection import Connection
|
|
|
|
|
|
class Repo:
|
|
"""Db abstraction layer"""
|
|
|
|
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
|
|
)
|
|
|
|
async def get_user(self, user_id: int):
|
|
return await self.conn.fetchval(
|
|
"SELECT user_id FROM users WHERE user_id = $1", user_id
|
|
)
|