first commit
This commit is contained in:
0
src/wishlist_bot/__init__.py
Normal file
0
src/wishlist_bot/__init__.py
Normal file
257
src/wishlist_bot/main.py
Normal file
257
src/wishlist_bot/main.py
Normal file
@ -0,0 +1,257 @@
|
||||
import asyncio
|
||||
import sqlite3
|
||||
|
||||
from os import getenv
|
||||
from enum import Enum
|
||||
from dataclasses import dataclass
|
||||
|
||||
from aiogram import Bot, Dispatcher, F
|
||||
from aiogram.filters import Command
|
||||
from aiogram.types import Message
|
||||
from aiogram.fsm.context import FSMContext
|
||||
from aiogram.fsm.state import State, StatesGroup
|
||||
|
||||
|
||||
TOKEN = getenv("BOT_TOKEN")
|
||||
ADMIN_ID = int(getenv("ADMIN_ID"))
|
||||
|
||||
dp = Dispatcher()
|
||||
con = sqlite3.connect("wishlist.db")
|
||||
cur = con.cursor()
|
||||
|
||||
|
||||
class Form(StatesGroup):
|
||||
waiting_for_answer_admin = State()
|
||||
waiting_for_answer_verify = State()
|
||||
|
||||
waiting_for_gift_add = State()
|
||||
waiting_for_gift_add_approve = State()
|
||||
|
||||
waiting_for_gift= State()
|
||||
|
||||
|
||||
class Role(Enum):
|
||||
ADMIN = "ADMIN"
|
||||
MEMBER = "MEMBER"
|
||||
DEFAULT = "DEFAULT"
|
||||
|
||||
|
||||
@dataclass
|
||||
class Gift: # ебаный пиздец я ебал питон
|
||||
name: str | None = None
|
||||
description: str | None = None
|
||||
link: str | None = None
|
||||
picture: str | None = None
|
||||
presenter: int | None = None
|
||||
|
||||
|
||||
def is_admin(id: int) -> bool:
|
||||
res = cur.execute(f"SELECT role FROM users WHERE id={id}").fetchone()[0]
|
||||
|
||||
if (res == Role.ADMIN.value):
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
|
||||
|
||||
def is_member(id: int) -> bool:
|
||||
res = cur.execute(f"SELECT role FROM users WHERE id={id}").fetchone()[0]
|
||||
|
||||
if (res == Role.MEMBER.value):
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
|
||||
|
||||
@dp.message(Command("start"))
|
||||
async def command_start_handler(message: Message) -> None:
|
||||
try:
|
||||
cur.execute(
|
||||
f"INSERT OR IGNORE INTO users(id, name, role) VALUES(?, ?, ?)",
|
||||
(message.from_user.id, message.from_user.full_name, Role.DEFAULT.value)
|
||||
)
|
||||
con.commit()
|
||||
except Exception as e:
|
||||
print(e)
|
||||
|
||||
await message.answer("Привет, теперь попроси подтвердить твое участие именинника 🥳")
|
||||
|
||||
|
||||
@dp.message(Command("add_admin"))
|
||||
async def command_start_handler(message: Message, state: FSMContext) -> None:
|
||||
await state.clear()
|
||||
|
||||
if is_admin(message.from_user.id):
|
||||
res = cur.execute("SELECT name FROM users WHERE role=?", (Role.MEMBER.value,)).fetchall()
|
||||
users = list(map(lambda x: x[0], res))
|
||||
text = "Напиши имя пользователя, которого надо сделать админом: \n"
|
||||
|
||||
for user in users:
|
||||
text += f"{user} \n"
|
||||
|
||||
await message.answer(text)
|
||||
await state.set_state(Form.waiting_for_answer_admin)
|
||||
|
||||
@dp.message(Form.waiting_for_answer_admin)
|
||||
async def process_answer(message: Message, state: FSMContext):
|
||||
cur.execute("UPDATE users SET role=? WHERE name=?", (Role.ADMIN.value, message.text))
|
||||
con.commit()
|
||||
|
||||
answer = message.text
|
||||
await message.answer(f"Теперь {answer} админ!")
|
||||
await state.clear()
|
||||
|
||||
|
||||
@dp.message(Command("verify"))
|
||||
async def command_start_handler(message: Message, state: FSMContext) -> None:
|
||||
await state.clear()
|
||||
|
||||
if is_admin(message.from_user.id):
|
||||
res = cur.execute("SELECT name FROM users WHERE role=?", (Role.DEFAULT.value,)).fetchall()
|
||||
users = list(map(lambda x: x[0], res))
|
||||
text = "Напиши имя пользователя, которого верифицировать: \n"
|
||||
|
||||
for user in users:
|
||||
text += f"{user} \n"
|
||||
|
||||
await message.answer(text)
|
||||
await state.set_state(Form.waiting_for_answer_verify)
|
||||
|
||||
@dp.message(Form.waiting_for_answer_verify)
|
||||
async def process_answer(message: Message, state: FSMContext):
|
||||
cur.execute("UPDATE users SET role=? WHERE name=?", (Role.MEMBER.value, message.text))
|
||||
con.commit()
|
||||
|
||||
answer = message.text
|
||||
await message.answer(f"Теперь {answer} участник вечеринки!")
|
||||
await state.clear()
|
||||
|
||||
|
||||
@dp.message(Command("add_gift"))
|
||||
async def command_start_handler(message: Message, state: FSMContext) -> None:
|
||||
await state.clear()
|
||||
|
||||
if is_admin(message.from_user.id):
|
||||
text = "Хорошо, отправь мне подарок вот так: \n\nНАЗВАНИЕ \nОПИСАНИЕ \nССЫЛКА \n\nА также можешь прикрепить фотографию!"
|
||||
|
||||
await message.answer(text)
|
||||
await state.set_state(Form.waiting_for_gift_add)
|
||||
|
||||
@dp.message(Form.waiting_for_gift_add)
|
||||
async def process_answer(message: Message, state: FSMContext):
|
||||
gift = Gift()
|
||||
|
||||
text = message.text or message.caption
|
||||
|
||||
if len(text.splitlines()) >= 1:
|
||||
gift.name = text.splitlines()[0]
|
||||
|
||||
if len(text.splitlines()) >= 2:
|
||||
gift.description = text.splitlines()[1]
|
||||
|
||||
if len(text.splitlines()) >= 3:
|
||||
gift.link = text.splitlines()[2]
|
||||
|
||||
if message.photo:
|
||||
gift.picture = message.photo[-1].file_id
|
||||
|
||||
|
||||
if gift.picture is None:
|
||||
await message.answer(f"{gift.name}\n{gift.description}\n{gift.link}")
|
||||
try:
|
||||
cur.execute(
|
||||
f"INSERT OR IGNORE INTO gifts(name, description, link) VALUES(?, ?, ?)",
|
||||
(gift.name, gift.description, gift.link)
|
||||
)
|
||||
con.commit()
|
||||
except Exception as e:
|
||||
print(e)
|
||||
else:
|
||||
await message.answer_photo(photo = gift.picture, caption=f"{gift.name}\n{gift.description}\n{gift.link}")
|
||||
try:
|
||||
cur.execute(
|
||||
f"INSERT OR IGNORE INTO gifts(name, description, link, picture) VALUES(?, ?, ?, ?)",
|
||||
(gift.name, gift.description, gift.link, gift.picture)
|
||||
)
|
||||
con.commit()
|
||||
except Exception as e:
|
||||
print(e)
|
||||
|
||||
await message.answer(f'Отлично, подарок добавлен!')
|
||||
await state.clear()
|
||||
|
||||
|
||||
@dp.message(Command("gifts"))
|
||||
async def command_start_handler(message: Message, state: FSMContext) -> None:
|
||||
await state.clear()
|
||||
|
||||
res = cur.execute("SELECT * FROM gifts").fetchall()
|
||||
gifts = list(map(lambda x: Gift(x[0], x[1], x[2], x[3], x[4]),res))
|
||||
|
||||
|
||||
for gift in gifts:
|
||||
text = str()
|
||||
|
||||
if is_admin(message.from_user.id):
|
||||
text = f"Название: {gift.name}\nОписание: {gift.description}\nСсылка: {gift.link}"
|
||||
else:
|
||||
text = f"Название: {gift.name}\nОписание: {gift.description}\nСсылка: {gift.link}\nДарит: {gift.presenter}"
|
||||
|
||||
if gift.picture is None:
|
||||
await message.answer(text)
|
||||
else:
|
||||
await message.answer_photo(photo = gift.picture, caption=text)
|
||||
|
||||
|
||||
@dp.message(Command("gift"))
|
||||
async def command_start_handler(message: Message, state: FSMContext) -> None:
|
||||
await state.clear()
|
||||
|
||||
if is_member(message.from_user.id):
|
||||
res = cur.execute("SELECT * FROM gifts").fetchall()
|
||||
gifts = list(map(lambda x: Gift(x[0], x[1], x[2], x[3], x[4]),res))
|
||||
|
||||
|
||||
for gift in gifts:
|
||||
text = f"Название: {gift.name}\nОписание: {gift.description}\nСсылка: {gift.link}\nДарит: {gift.presenter}"
|
||||
|
||||
if gift.picture is None:
|
||||
await message.answer(text)
|
||||
else:
|
||||
await message.answer_photo(photo = gift.picture, caption=text)
|
||||
|
||||
await message.answer("Напишите название подарка, который хотите подарить:")
|
||||
await state.set_state(Form.waiting_for_gift)
|
||||
else:
|
||||
await message.answer("Вы админ или не участвуете в вечеринке😔")
|
||||
|
||||
|
||||
@dp.message(Form.waiting_for_gift)
|
||||
async def process_answer(message: Message, state: FSMContext):
|
||||
cur.execute("UPDATE gifts SET presenter=? WHERE name=?", (message.from_user.full_name, message.text))
|
||||
con.commit()
|
||||
|
||||
answer = message.text
|
||||
await message.answer(f"Теперь {message.text} дарит {message.from_user.full_name}!")
|
||||
await state.clear()
|
||||
|
||||
|
||||
async def main() -> None:
|
||||
cur.execute(
|
||||
"CREATE TABLE IF NOT EXISTS gifts(name PRIMARY KEY, description TEXT, link TEXT, picture TEXT, presenter TEXT)"
|
||||
)
|
||||
cur.execute("CREATE TABLE IF NOT EXISTS users(id INTEGER PRIMARY KEY, name TEXT NOT NULL, role TEXT NOT NULL)")
|
||||
|
||||
cur.execute(
|
||||
f"INSERT OR IGNORE INTO users(id, name, role) VALUES(?, ?, ?)",
|
||||
(ADMIN_ID, "ADMIN", Role.ADMIN.value)
|
||||
)
|
||||
con.commit()
|
||||
|
||||
bot = Bot(token=TOKEN)
|
||||
await dp.start_polling(bot)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(main())
|
||||
|
||||
Reference in New Issue
Block a user