From 120c347864038fc8b930a74ec34d47285c98fb9b Mon Sep 17 00:00:00 2001 From: Rykov7 Date: Tue, 3 Jan 2023 01:35:52 +0300 Subject: [PATCH] mode refactoring --- app/bot.py | 4 ++-- app/stan.py | 3 +-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/app/bot.py b/app/bot.py index 1a12d78..f1e95ee 100644 --- a/app/bot.py +++ b/app/bot.py @@ -266,8 +266,8 @@ def list_jobs(message): @bot.message_handler(func=is_white, commands=['add']) def add_stan_quote(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 stan_quotes): + 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('\n' + message.reply_to_message.text) bot.send_message(message.chat.id, f'✅ Добавил\n └ {message.reply_to_message.text}') else: diff --git a/app/stan.py b/app/stan.py index d061767..1cf3f3a 100644 --- a/app/stan.py +++ b/app/stan.py @@ -5,5 +5,4 @@ import random def speak(chance_of): number = random.randint(0, chance_of) if number == 0: - quotes = (i.rstrip() for i in open('Stan.txt', 'r', encoding='utf8')) - return random.choice(quotes) + return random.choice(i.rstrip() for i in open('Stan.txt', 'r', encoding='utf8'))