first commit
This commit is contained in:
parent
44d5d25cef
commit
df4f642c2b
2 changed files with 117 additions and 1 deletions
3
.gitignore
vendored
Normal file
3
.gitignore
vendored
Normal file
|
@ -0,0 +1,3 @@
|
|||
.env
|
||||
admins.db
|
||||
.idea/
|
115
main.py
115
main.py
|
@ -1 +1,114 @@
|
|||
print("start")
|
||||
import asyncio
|
||||
import logging
|
||||
import sys
|
||||
from os import getenv
|
||||
|
||||
from aiogram import Bot, Dispatcher, html
|
||||
from aiogram.client.default import DefaultBotProperties
|
||||
from aiogram.enums import ParseMode
|
||||
from aiogram.filters import CommandStart, Command
|
||||
from aiogram.types import Message
|
||||
|
||||
import sqlite3
|
||||
|
||||
from dotenv import load_dotenv
|
||||
|
||||
load_dotenv()
|
||||
|
||||
# Bot token can be obtained via https://t.me/BotFather
|
||||
TOKEN = getenv("BOT_TOKEN")
|
||||
|
||||
# All handlers should be attached to the Router (or Dispatcher)
|
||||
|
||||
dp = Dispatcher()
|
||||
|
||||
|
||||
@dp.message(CommandStart())
|
||||
async def command_start_handler(message: Message) -> None:
|
||||
"""
|
||||
This handler receives messages with `/start` command
|
||||
"""
|
||||
# Most event objects have aliases for API methods that can be called in events' context
|
||||
# For example if you want to answer to incoming message you can use `message.answer(...)` alias
|
||||
# and the target chat will be passed to :ref:`aiogram.methods.send_message.SendMessage`
|
||||
# method automatically or call API method directly via
|
||||
# Bot instance: `bot.send_message(chat_id=message.chat.id, ...)`
|
||||
await message.answer(f"Hello, {html.bold(message.from_user.full_name)}!")
|
||||
|
||||
@dp.message(Command("add_admin"))
|
||||
async def command_add_admin_handler(message: Message) -> None:
|
||||
"""
|
||||
This handles adding new admins
|
||||
|
||||
# TODO: add a state
|
||||
"""
|
||||
con = sqlite3.connect("admins.db")
|
||||
|
||||
|
||||
id = message.from_user.id
|
||||
username = message.from_user.first_name
|
||||
|
||||
|
||||
try:
|
||||
con.execute("""
|
||||
INSERT INTO admins VALUES
|
||||
(?, ?);
|
||||
""", (id, username))
|
||||
|
||||
con.commit()
|
||||
except Exception as err:
|
||||
print(err)
|
||||
|
||||
print("addmint added")
|
||||
|
||||
await message.answer(f"yoooo, added an admin lmao")
|
||||
|
||||
@dp.message(Command("am_i_admin"))
|
||||
async def command_check_admin_handler(message: Message) -> None:
|
||||
con = sqlite3.connect("admins.db")
|
||||
|
||||
user_id = message.from_user.id
|
||||
|
||||
check_id = con.execute("SELECT id FROM admins where id=?", (user_id,))
|
||||
|
||||
if check_id.fetchone() != None:
|
||||
await message.answer(f"yes you are, nigga")
|
||||
else:
|
||||
await message.answer(f"no, lolz")
|
||||
|
||||
|
||||
@dp.message()
|
||||
async def echo_handler(message: Message) -> None:
|
||||
"""
|
||||
Handler will forward receive a message back to the sender
|
||||
|
||||
By default, message handler will handle all message types (like a text, photo, sticker etc.)
|
||||
"""
|
||||
try:
|
||||
# Send a copy of the received message
|
||||
await message.send_copy(chat_id=message.chat.id)
|
||||
except TypeError:
|
||||
# But not all the types is supported to be copied so need to handle it
|
||||
await message.answer("Nice try!")
|
||||
|
||||
|
||||
async def main() -> None:
|
||||
# Initialize Bot instance with default bot properties which will be passed to all API calls
|
||||
bot = Bot(token=TOKEN, default=DefaultBotProperties(parse_mode=ParseMode.HTML))
|
||||
|
||||
con = sqlite3.connect("admins.db")
|
||||
|
||||
con.execute("""CREATE TABLE IF NOT EXISTS admins (
|
||||
id INTEGER PRIMARY KEY,
|
||||
name TEXT NOT NULL
|
||||
);""")
|
||||
|
||||
con.commit()
|
||||
|
||||
# And the run events dispatching
|
||||
await dp.start_polling(bot)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
logging.basicConfig(level=logging.INFO, stream=sys.stdout)
|
||||
asyncio.run(main())
|
Loading…
Reference in a new issue