78 lines
2.3 KiB
Python
78 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")
|