leo/Stan-Ansible
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add database

Browse Source
This commit is contained in:
Rykov7
2023-02-25 19:33:39 +03:00
parent d0cf0762c8
commit fd069a9615
7 changed files with 88 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/stan.py
+33 -37
View File
@@ -4,14 +4,14 @@ import threading
from time import sleep
from .config import bot, types
from .filters import is_white, is_nongrata
from .database import session
from .models import Quote
def speak(chance_of):
def speak(chance_of, group_id):
number = random.randint(0, chance_of)
if number == 0:
return random.choice(
[i.rstrip() for i in open("Stan.txt", "r", encoding="utf8") if i]
)
return random.choice([i[0] for i in session.query(Quote.text).filter(Quote.chat_id == group_id).all()])
def send_quote(after_sec, message, quote):
@@ -25,8 +25,8 @@ def send_quote(after_sec, message, quote):
bot.send_message(message.chat.id, quote)
def act(message):
quote = speak(50)
def act(message: types.Message):
quote = speak(50, message.chat.id)
if quote:
threading.Thread(
target=send_quote, args=(len(quote) * 0.13, message, quote)
@@ -34,49 +34,45 @@ def act(message):
@bot.message_handler(func=is_white, commands=["add"])
def add_stan_quote(message):
def add_stan_quote(message: types.Message):
if message.reply_to_message and message.reply_to_message.text:
with open("Stan.txt", "a", encoding="utf8") as stan_quotes:
if message.reply_to_message.text not in (
i.rstrip() for i in open("Stan.txt", "r", encoding="utf8")
):
stan_quotes.write(
message.reply_to_message.text.replace("\n", " ") + "\n"
)
bot.send_message(
if message.reply_to_message.text not in [i[0] for i in session.query(Quote.text).all()]:
session.add(Quote(chat_id=message.chat.id, text=message.reply_to_message.text.replace("\n", " ")))
session.commit()
bot.send_message(
message.chat.id,
"✅ <b>Добавил</b>\n └ <i>"
+ message.reply_to_message.text.replace("\n", " ")
+ "</i>",
)
else:
bot.send_message(
message.chat.id,
f"⛔️ <b>Не добавил</b>, есть токое\n └ <i>{message.reply_to_message.text}</i>",
)
@bot.message_handler(func=is_white, commands=["remove"])
def remove_stan_quote(message):
if message.reply_to_message and message.reply_to_message.text:
if message.reply_to_message.text in (
i.rstrip() for i in open("Stan.txt", "r", encoding="utf8")
):
quotes = list(open("Stan.txt", "r", encoding="utf8"))
with open("Stan.txt", "w", encoding="utf8") as stan_quotes:
quotes.remove(message.reply_to_message.text + "\n")
stan_quotes.writelines(quotes)
bot.send_message(
message.chat.id,
f"✅ <b>Удалил</b>\n └ <i>{message.reply_to_message.text}</i>",
)
else:
bot.send_message(
message.chat.id,
f"⛔️ <b>Нет такого</b>\n └ <i>{message.reply_to_message.text}</i>",
f"⛔️ <b>Не добавил</b>, есть токое\n └ <i>{message.reply_to_message.text}</i>",
)
# @bot.message_handler(func=is_white, commands=["remove"])
# def remove_stan_quote(message):
# if message.reply_to_message and message.reply_to_message.text:
# if message.reply_to_message.text in (
# i.rstrip() for i in open("Stan.txt", "r", encoding="utf8")
# ):
# quotes = list(open("Stan.txt", "r", encoding="utf8"))
# with open("Stan.txt", "w", encoding="utf8") as stan_quotes:
# quotes.remove(message.reply_to_message.text + "\n")
# stan_quotes.writelines(quotes)
# bot.send_message(
# message.chat.id,
# f"✅ <b>Удалил</b>\n └ <i>{message.reply_to_message.text}</i>",
# )
# else:
# bot.send_message(
# message.chat.id,
# f"⛔️ <b>Нет такого</b>\n └ <i>{message.reply_to_message.text}</i>",
# )
@bot.message_handler(func=is_nongrata)
def tease_nongrata(message: types.Message):
"""Reply to non grata mentions."""