trial
This commit is contained in:
@@ -4,7 +4,6 @@ from random import choice
|
||||
|
||||
from telebot.types import Message, ReplyKeyboardRemove
|
||||
|
||||
from daemons.bot import bot
|
||||
from daemons.fetch import fetch_schedule_for_user
|
||||
from helpers import get_next_daily_notify_time
|
||||
from helpers.keyboards import main_keyboard, notify_keyboard, yes_no_keyboard, again_keyboard, no_daily_notify, \
|
||||
@@ -12,6 +11,7 @@ from helpers.keyboards import main_keyboard, notify_keyboard, yes_no_keyboard, a
|
||||
from helpers.mongo import mongo
|
||||
from helpers.sprint_platform import platform
|
||||
from helpers.ruz import ruz
|
||||
from utils import queues
|
||||
|
||||
|
||||
class User:
|
||||
@@ -32,6 +32,8 @@ class Answer:
|
||||
def __init__(self, message: Message):
|
||||
self.message = message
|
||||
self.message_text = message.text or message.caption or ""
|
||||
if self.message_text.startswith('/start'):
|
||||
mongo.users_collection.delete_many({"chat_id": message.chat.id})
|
||||
user = mongo.users_collection.find_one({"chat_id": message.chat.id})
|
||||
if user is None:
|
||||
user = {
|
||||
@@ -72,7 +74,10 @@ class Answer:
|
||||
def send_message(self, text, reply_markup=None, remove_keyboard=True, **kwargs):
|
||||
if reply_markup is None and remove_keyboard:
|
||||
reply_markup = ReplyKeyboardRemove()
|
||||
bot.send_message(self.user['chat_id'], text, reply_markup=reply_markup, **kwargs)
|
||||
body = {'text': text, 'chat_id': self.user['chat_id']}
|
||||
if reply_markup:
|
||||
body['reply_markup'] = reply_markup.to_json()
|
||||
queues.set_task('ruz_bot_mailbox', body, 1)
|
||||
|
||||
def set_state(self, state: str):
|
||||
self.user['state'] = state
|
||||
|
||||
Reference in New Issue
Block a user