Initial. Слил данные, но там всего один коммит был, тк я ленивая жопа
This commit is contained in:
commit
747f0bf64b
5 changed files with 215 additions and 0 deletions
8
.gitignore
vendored
Normal file
8
.gitignore
vendored
Normal file
|
|
@ -0,0 +1,8 @@
|
||||||
|
NoVagonBot.session
|
||||||
|
NoVagonBot.session.save
|
||||||
|
db.sqlite
|
||||||
|
.env
|
||||||
|
venv/
|
||||||
|
.idea/
|
||||||
|
__pycache__/
|
||||||
|
df_key.json
|
||||||
57
db.py
Normal file
57
db.py
Normal file
|
|
@ -0,0 +1,57 @@
|
||||||
|
import aiosqlite
|
||||||
|
from enum import Enum, auto
|
||||||
|
|
||||||
|
|
||||||
|
class DBAction(Enum):
|
||||||
|
fetchone = auto()
|
||||||
|
fetchall = auto()
|
||||||
|
commit = auto()
|
||||||
|
|
||||||
|
|
||||||
|
async def db_action(sql: str, args: tuple, action: DBAction):
|
||||||
|
"""Runs a query on the database"""
|
||||||
|
conn = await aiosqlite.connect('db.sqlite')
|
||||||
|
cursor = await conn.cursor()
|
||||||
|
|
||||||
|
await cursor.execute(sql, args)
|
||||||
|
if action == DBAction.fetchone:
|
||||||
|
result = await cursor.fetchone()
|
||||||
|
elif action == DBAction.fetchall:
|
||||||
|
result = await cursor.fetchall()
|
||||||
|
elif action == DBAction.commit:
|
||||||
|
conn.commit()
|
||||||
|
result = cursor.lastrowid
|
||||||
|
|
||||||
|
await cursor.close()
|
||||||
|
await conn.close()
|
||||||
|
|
||||||
|
return result
|
||||||
|
|
||||||
|
|
||||||
|
async def create_table():
|
||||||
|
"""Creates a table in the database"""
|
||||||
|
conn = await aiosqlite.connect('db.sqlite')
|
||||||
|
cursor = await conn.cursor()
|
||||||
|
|
||||||
|
await cursor.execute('''CREATE TABLE IF NOT EXISTS chats (
|
||||||
|
id INTEGER PRIMARY KEY,
|
||||||
|
chat_id INTEGER NOT NULL
|
||||||
|
)''')
|
||||||
|
|
||||||
|
await cursor.execute("""CREATE TABLE IF NOT EXISTS user_exceptions (
|
||||||
|
id INTEGER PRIMARY KEY,
|
||||||
|
user_id INTEGER NOT NULL,
|
||||||
|
chat_id INTEGER NOT NULL,
|
||||||
|
FOREIGN KEY(chat_id) REFERENCES chats(chat_id)
|
||||||
|
)""")
|
||||||
|
|
||||||
|
await cursor.execute("""CREATE TABLE IF NOT EXISTS banned_stickerpacks (
|
||||||
|
id INTEGER PRIMARY KEY,
|
||||||
|
pack_id INTEGER NOT NULL,
|
||||||
|
chat_id INTEGER NOT NULL,
|
||||||
|
FOREIGN KEY(chat_id) REFERENCES chats(chat_id)
|
||||||
|
)""")
|
||||||
|
|
||||||
|
await cursor.close()
|
||||||
|
await conn.close()
|
||||||
|
|
||||||
20
df_utils.py
Normal file
20
df_utils.py
Normal file
|
|
@ -0,0 +1,20 @@
|
||||||
|
from google.cloud import dialogflow_v2beta1 as dialogflow
|
||||||
|
|
||||||
|
from dotenv import load_dotenv
|
||||||
|
|
||||||
|
load_dotenv()
|
||||||
|
|
||||||
|
|
||||||
|
async def detect_intent_text(project_id, session_id, text, language_code):
|
||||||
|
session_client = dialogflow.SessionsClient()
|
||||||
|
session = session_client.session_path(project_id, session_id)
|
||||||
|
|
||||||
|
text_input = dialogflow.types.TextInput(text=text, language_code=language_code)
|
||||||
|
query_input = dialogflow.types.QueryInput(text=text_input)
|
||||||
|
|
||||||
|
response = session_client.detect_intent(session=session, query_input=query_input)
|
||||||
|
|
||||||
|
if response.query_result.intent.is_fallback:
|
||||||
|
return None
|
||||||
|
else:
|
||||||
|
return response.query_result.fulfillment_text
|
||||||
125
main.py
Normal file
125
main.py
Normal file
|
|
@ -0,0 +1,125 @@
|
||||||
|
from telethon import TelegramClient, events
|
||||||
|
from telethon.tl.types import ChannelParticipantsAdmins, DocumentAttributeSticker
|
||||||
|
from datetime import timedelta
|
||||||
|
|
||||||
|
from db import DBAction, db_action, create_table
|
||||||
|
from df_utils import detect_intent_text
|
||||||
|
|
||||||
|
import random
|
||||||
|
import asyncio
|
||||||
|
|
||||||
|
import os
|
||||||
|
from dotenv import load_dotenv
|
||||||
|
|
||||||
|
|
||||||
|
bot = TelegramClient(os.environ['SESSION_NAME'], int(os.environ['API_ID']),
|
||||||
|
os.environ['API_HASH']).start(bot_token=os.environ['BOT_TOKEN'])
|
||||||
|
|
||||||
|
load_dotenv()
|
||||||
|
asyncio.get_event_loop().run_until_complete(create_table())
|
||||||
|
|
||||||
|
|
||||||
|
async def check_admin(msg):
|
||||||
|
if not any([user.id == msg.sender_id async for user in bot.iter_participants(msg.chat_id,
|
||||||
|
filter=ChannelParticipantsAdmins)]):
|
||||||
|
await msg.reply('Эту команду может только администратор использовать')
|
||||||
|
raise events.StopPropagation
|
||||||
|
|
||||||
|
|
||||||
|
@bot.on(events.NewMessage())
|
||||||
|
async def on_message(msg):
|
||||||
|
chats = await db_action('SELECT chat_id FROM chats', (), DBAction.fetchall)
|
||||||
|
if msg.chat_id not in chats:
|
||||||
|
await db_action('INSERT INTO chats (chat_id) VALUES (?)', (msg.chat_id,), DBAction.commit)
|
||||||
|
|
||||||
|
|
||||||
|
async def check_user_in_exceptions(user_id, chat_id):
|
||||||
|
"""Checks if a user is in the exceptions list"""
|
||||||
|
return db_action('SELECT id FROM user_exceptions WHERE user_id = ? AND chat_id = ?', (user_id, chat_id),
|
||||||
|
DBAction.fetchone) is not None
|
||||||
|
|
||||||
|
|
||||||
|
@bot.on(events.NewMessage(pattern='/exception'))
|
||||||
|
async def on_exception(msg):
|
||||||
|
await check_admin(msg)
|
||||||
|
|
||||||
|
args = msg.message.text.split(' ')
|
||||||
|
if len(args) < 4:
|
||||||
|
await msg.reply('Неверный формат')
|
||||||
|
raise events.StopPropagation
|
||||||
|
|
||||||
|
if args[2]:
|
||||||
|
if args[2] == 'me':
|
||||||
|
user_id = msg.sender_id
|
||||||
|
else:
|
||||||
|
user_id = args[2]
|
||||||
|
elif msg.reply_to_message:
|
||||||
|
user_id = msg.reply_to_message.sender_id
|
||||||
|
else:
|
||||||
|
await msg.reply('Допиши к команде айди пользователя, "me", чтобы добавить себя, или ответь командой на сообщение нужного пользователя')
|
||||||
|
raise events.StopPropagation
|
||||||
|
|
||||||
|
match args[1]:
|
||||||
|
case 'add':
|
||||||
|
if await check_user_in_exceptions(user_id, msg.chat_id):
|
||||||
|
await msg.reply('Пользователь уже есть в списке исключений')
|
||||||
|
raise events.StopPropagation
|
||||||
|
|
||||||
|
await db_action('INSERT INTO user_exceptions (user_id) VALUES (?)', (user_id,), DBAction.commit)
|
||||||
|
await msg.reply('Пользователь добавлен в исключения, теперь его сообщения не будут проверяться ботом')
|
||||||
|
case 'remove':
|
||||||
|
if not await check_user_in_exceptions(user_id, msg.chat_id):
|
||||||
|
await msg.reply('Пользователя пока нет в исключениях')
|
||||||
|
raise events.StopPropagation
|
||||||
|
|
||||||
|
await db_action('DELETE FROM user_exceptions WHERE user_id = ?', (user_id,), DBAction.commit)
|
||||||
|
await msg.reply('Пользователь удален из списка исключений')
|
||||||
|
|
||||||
|
|
||||||
|
@bot.on(events.NewMessage(pattern='/addstickerpack'))
|
||||||
|
async def add_stickerpack(msg):
|
||||||
|
await check_admin(msg)
|
||||||
|
|
||||||
|
args = msg.text.split()
|
||||||
|
if args[1]:
|
||||||
|
stickerpack_id = args[1]
|
||||||
|
elif msg.reply_to_message.sticker:
|
||||||
|
for attr in msg.sticker.attributes:
|
||||||
|
if isinstance(attr, DocumentAttributeSticker):
|
||||||
|
stickerpack_id = attr.stickerset.id
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
msg.reply('Допиши к команде айди набора стикеров или ответь командой на стикер из набора')
|
||||||
|
raise events.StopPropagation
|
||||||
|
|
||||||
|
if await db_action("SELECT * FROM banned_stickerpacks WHERE pack_id = ? AND chat_id = ?",
|
||||||
|
(stickerpack_id, msg.chat_id,), DBAction.fetchone) is None:
|
||||||
|
await db_action("INSERT INTO banned_stickerpacks (pack_id, chat_id) VALUES (?, ?)",
|
||||||
|
(stickerpack_id, msg.chat_id,), DBAction.commit)
|
||||||
|
await msg.reply(f"Стикерпак с айди {msg.text.split(' ')[1]} добавлен в банлист")
|
||||||
|
else:
|
||||||
|
await msg.reply(f"Стикерпак с айди {msg.text.split(' ')[1]} уже в банлисте")
|
||||||
|
|
||||||
|
raise events.StopPropagation
|
||||||
|
|
||||||
|
|
||||||
|
async def on_banword(msg):
|
||||||
|
await bot.edit_permissions(msg.chat_id, msg.sender_id, timedelta(minutes=5), send_messages=False)
|
||||||
|
await msg.reply(random.choice(['Харе про вагоны писать', 'Надоел уже со своими номерными', 'Никаких вагонов тут нет']))
|
||||||
|
|
||||||
|
|
||||||
|
@bot.on(events.NewMessage())
|
||||||
|
async def handler(msg):
|
||||||
|
if msg.is_private:
|
||||||
|
await msg.respond('Hello! Add me to a group to start using me!')
|
||||||
|
return
|
||||||
|
|
||||||
|
await msg.reply(await detect_intent_text(os.environ['GOOGLE_PROJECT_ID'], msg.chat_id, msg.raw_text, 'ru'))
|
||||||
|
|
||||||
|
banstickerpacks = [stickerpack[0] for stickerpack in db_action('SELECT pack_id FROM banned_stickerpacks', (), DBAction.fetchall)]
|
||||||
|
|
||||||
|
if msg.sticker and any([x.stickerset.id in banstickerpacks for x in msg.sticker.attributes if isinstance(x, DocumentAttributeSticker)]):
|
||||||
|
await on_banword(msg)
|
||||||
|
|
||||||
|
|
||||||
|
bot.run_until_disconnected()
|
||||||
5
requirements.txt
Normal file
5
requirements.txt
Normal file
|
|
@ -0,0 +1,5 @@
|
||||||
|
telethon
|
||||||
|
pymorphy2
|
||||||
|
pyenchant
|
||||||
|
google-cloud-dialogflow
|
||||||
|
python-dotenv
|
||||||
Loading…
Add table
Reference in a new issue