71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
""" Service commands for Admin purposes. """
|
|
|
|
import logging
|
|
from datetime import datetime as dt
|
|
from ..config import bot, ADMIN_ID, types
|
|
from .. import reminder
|
|
from .. import reloader
|
|
from .. import report
|
|
from ..filters import is_admin
|
|
from .get import me, my_ip
|
|
|
|
|
|
@bot.message_handler(commands=['me'])
|
|
def command_me(message):
|
|
""" Send info about user and chat id [Service]. """
|
|
bot.send_message(message.chat.id, me(message))
|
|
|
|
|
|
@bot.message_handler(func=is_admin, commands=['ip'])
|
|
def get_ip(message):
|
|
bot.send_message(message.chat.id, my_ip())
|
|
|
|
|
|
@bot.message_handler(commands=['remind'])
|
|
def remind_manually(message):
|
|
""" Remind holidays manually. """
|
|
args = message.text.split()
|
|
if len(args) > 1:
|
|
try:
|
|
today = dt.strptime(args[1], "%m-%d-%Y")
|
|
except ValueError as ve:
|
|
bot.send_message(message.chat.id, f"Не удалось разобрать дату!\n{ve}")
|
|
else:
|
|
reminder.remind(message.chat.id, today)
|
|
else:
|
|
bot.send_message(message.chat.id, f"<b>Формат даты: MM-DD-YYYY</b>\n\n"
|
|
f"Примеры:\n"
|
|
f"/remind 09-12-2024\n"
|
|
f"/remind 09-13-2022")
|
|
|
|
|
|
@bot.message_handler(func=is_admin, commands=['jobs'])
|
|
def list_jobs(message):
|
|
""" List all the jobs in schedule. """
|
|
bot.send_message(ADMIN_ID, reminder.print_get_jobs())
|
|
|
|
|
|
@bot.message_handler(func=is_admin, commands=['stats'])
|
|
def send_stats(message: types.Message):
|
|
if len(message.text.split()) == 1:
|
|
rep = report.create_report_text(message.chat.id)
|
|
if rep:
|
|
bot.send_message(message.chat.id, rep)
|
|
else:
|
|
bot.send_message(message.chat.id, report.create_report_text(message.text.split()[-1]))
|
|
|
|
|
|
@bot.message_handler(func=is_admin, commands=['reset_stats'], chat_types=['supergroup', 'group'])
|
|
def send_stats(message: types.Message):
|
|
logging.warning('reset_stats')
|
|
report.reset_report_stats(message.chat.id)
|
|
bot.send_message(message.chat.id, report.reset_report_stats(message.chat.id))
|
|
|
|
|
|
@bot.message_handler(func=is_admin, commands=['reload'])
|
|
def send_stats(message):
|
|
logging.warning('Reloading...')
|
|
reloader.reload_modules()
|
|
logging.warning('Reloaded!!!')
|
|
bot.send_message(message.chat.id, 'Reloaded successfully')
|