main.py

from dublib.Methods.Filesystem import MakeRootDirectories, ReadJSON
from dublib.TelebotUtils import TeleCache, TeleMaster, UsersManager
from dublib.Methods.System import CheckPythonMinimalVersion
from dublib.Engine.GetText import GetText
from dublib.Methods.System import Clear

from threading import Thread
import random
import os

from telebot import TeleBot, types

#==========================================================================================#
# >>>>> ИНИЦИАЛИЗАЦИЯ СКРИПТА <<<<< #
#==========================================================================================#

CheckPythonMinimalVersion(3, 10)
MakeRootDirectories(["Data/Horoscopes"])
Clear()

#==========================================================================================#
# >>>>> ЧТЕНИЕ НАСТРОЕК И СОЗДАНИЕ ОБЪЕКТОВ <<<<< #
#==========================================================================================#

Settings = ReadJSON("Settings.json")
GetText.initialize("HoroscopeBot", Settings["language"])
_ = GetText.gettext

from Source.Core.Horoscope import Horoscoper, Zodiacs
from Source.UI import InlineKeyboards, ReplyKeyboards
from Source.TeleBotAdminPanel import Panel, Modules
from Source.Core.Scheduler import Scheduler
from Source import PanelAdditional

Bot = TeleBot(Settings["bot_token"])
MasterBot = TeleMaster(Bot)
Users = UsersManager("Data/Users")

AdminPanel = Panel(Bot, Users, Settings["password"])

TBAP_TREE = {
	"📊 Статистика": Modules.SM_Statistics,
	"❌ Закрыть": Modules.SM_Close
}

AdminPanel.set_tree(TBAP_TREE)
AdminPanel.set_close_callback(PanelAdditional.close_callback, (Bot,))

SM_Statistics: Modules.SM_Statistics = AdminPanel.get_module_object(Modules.SM_Statistics.__name__)
SM_Statistics.columns["Zodiac"] = PanelAdditional.get_zodiac

Cacher = TeleCache()
Cacher.set_bot(Bot)
Cacher.set_chat_id(Settings["cache_chat_id"])

Horoscopes = Horoscoper(Cacher, Settings)

SchedulerObject = Scheduler(Bot, Users, Horoscopes)
if Settings["update_on_restart"]: Thread(target = SchedulerObject.update_horoscopes).start()
SchedulerObject.run()

for CurrentUser in Users.users:
	Data = Bot.get_chat(CurrentUser.id)
	NameParts = list()
	if Data.first_name: NameParts.append(Data.first_name)
	if Data.last_name: NameParts.append(Data.last_name)
	Name = " ".join(NameParts)
	if Name: CurrentUser.set_property("name", Name)

#==========================================================================================#
# >>>>> ОБРАБОТКА КОММАНД <<<<< #
#==========================================================================================#

@Bot.message_handler(commands = ["admin"])
def Command(Message: types.Message):
	User = Users.auth(Message.from_user)
	AdminPanel.open(User, "Панель управления открыта.")

@Bot.message_handler(commands = ["mailing", "mailset"])
def Command(Message: types.Message):
	User = Users.auth(Message.from_user)
	Bot.send_message(User.id, _("Желаете настроить/отключить утреннюю рассылку <b>Гороскопа дня</b>?"), parse_mode = "HTML", reply_markup = InlineKeyboards.notifications())

@Bot.message_handler(commands = ["share"])
def Command(Message: types.Message):
	User = Users.auth(Message.from_user)
	
	QrPath = "Data/Images/qr.jpg"
	BotName = Bot.get_me().username
	BotNames = f"@{BotName}\n@{BotName}\n@{BotName}\n\n"
	Caption = BotNames + _("<b>🌟 Гороскоп дня</b>\nНайди свой знак зодиака и узнай, что для тебя на сегодня приготовили звезды!\n\n<b><i>Пользуйся и делись с друзьями!</i></b>")

	if os.path.exists(QrPath):
		FileID = Cacher.get_real_cached_file(QrPath, types.InputMediaPhoto).file_id
		Bot.send_photo(User.id, FileID, Caption, parse_mode = "HTML", reply_markup = InlineKeyboards().share(BotName))

	else:
		Bot.send_message(User.id, Caption, parse_mode = "HTML")

@Bot.message_handler(commands = ["start"])
def Command(Message: types.Message):
	User = Users.auth(Message.from_user)
	User.set_property("zodiac", None, force = False)

	Bot.send_message(
		chat_id = Message.chat.id,
		text = _("<b>Добро пожаловать в Гороскоп дня!</b>\n\nСамый большой и популярный бот-астролог в Telegram 💫\n\nВыбирай свой знак зодиака и смело начинай этот день!"),
		parse_mode = "HTML",
		reply_markup = ReplyKeyboards.zodiac_menu()
	)

#==========================================================================================#
# >>>>> ОБРАБОТКА ВВОДА ТЕКСТА <<<<< #
#==========================================================================================#

@Bot.message_handler(content_types = ["text"])
def Text(Message: types.Message):
	User = Users.auth(Message.from_user)
	if AdminPanel.procedures.text(Message): return
	Bot.send_chat_action(User.id, "typing")

	ErrorMessages = [
		_("Пожалуйста, используйте кнопки ниже, для выбора своего знака зодиака"),
		_("Немножко некорректный запрос. Для работы со мной используйте меню внизу)"),
		_("Не могу обработать эту команду. Буду рад, если вы нажмете на свой знак зодиака"),
		_("Очень интересно, но, к сожалению, не знаю, что на это ответить. Попробуйте использовать меню внизу)"),
		_("Это неизвестная для меня команда. Для получения прогноза вам лучше воспользоваться кнопками ниже")
	]

	ErrorMessage = random.choice(ErrorMessages)
	Words = Message.text.split(" ")

	if len(Words) != 2:
		Bot.send_message(User.id, ErrorMessage)
		return

	Zodiac = Words[-1].lower()

	if Zodiac not in [Element.value for Element in Zodiacs]:
		Bot.send_message(User.id, ErrorMessage)
		return
	
	Zodiac = Zodiacs(Zodiac)
	SchedulerObject.send_horoscope(User, Zodiac)

	if Settings["enable_notifications_after_first_use"]:
		if User.has_property("is_first") and User.get_property("is_first"): User.set_property("zodiac", Zodiac.name)
		User.set_property("is_first", False)

#==========================================================================================#
# >>>>> ОБРАБОТКА INLINE-КНОПОК <<<<< #
#==========================================================================================#

AdminPanel.decorators.inline_keyboards()

@Bot.callback_query_handler(func = lambda Callback: Callback.data.startswith("delete"))
def InlineButton(Call: types.CallbackQuery):
	Bot.delete_message(Call.message.chat.id, Call.message.id)

@Bot.callback_query_handler(func = lambda Callback: Callback.data.startswith("notifications"))
def InlineButton(Call: types.CallbackQuery):
	User = Users.auth(Call.from_user)
	Command = Call.data.split("_")[-1]

	match Command:

		case "enable":
			Bot.edit_message_text(_("Выберите свой знак зодиака из представленного списка ниже:"), User.id, Call.message.id, reply_markup = InlineKeyboards.zodiac_selector())

		case "disable":
			User.set_property("zodiac", None)
			Bot.edit_message_text(
				text = _("Хорошо! Вы в любой момент сможете посмотреть предсказания, выбрав свой знак зодиака из меню ниже 💫"),
				chat_id = User.id,
				message_id = Call.message.id,
				reply_markup = InlineKeyboards.delete(_("Благодарю!"))
			)

@Bot.callback_query_handler(func = lambda Callback: Callback.data.startswith("select"))
def InlineButton(Call: types.CallbackQuery):
	User = Users.auth(Call.from_user)
	Value = Call.data.split("_")[-1]
	User.set_property("zodiac", Value)
	Bot.edit_message_text(
		text = _("Спасибо! Теперь вы будете просыпаться вместе со звездами! ✨️"),
		chat_id = User.id,
		message_id = Call.message.id,
		reply_markup = InlineKeyboards.delete(_("Хотелось бы!"))
	)
	
Bot.infinity_polling()