refactoring
This commit is contained in:
@@ -0,0 +1,25 @@
|
||||
from urllib.request import urlopen
|
||||
|
||||
|
||||
def my_ip():
|
||||
ip_service = 'https://icanhazip.com'
|
||||
ip = urlopen(ip_service, )
|
||||
return ip.read().decode('utf-8')
|
||||
|
||||
|
||||
def me(message):
|
||||
msg = f"<b>Telegram ID:</b> <code>{message.from_user.id}</code>\n\
|
||||
├ <b>Is bot:</b> {message.from_user.is_bot}\n\
|
||||
├ <b>First name:</b> {message.from_user.first_name}\n"
|
||||
if message.from_user.last_name:
|
||||
msg += f" ├ <b>Last name:</b> {message.from_user.last_name}\n"
|
||||
if message.from_user.username:
|
||||
msg += f" ├ <b>Username:</b> {message.from_user.username}\n"
|
||||
msg += f" ├ <b>Language code:</b> {message.from_user.language_code}\n"
|
||||
if message.from_user.is_premium is True:
|
||||
msg += f" ├ <b>Is Premium</b>: ⭐️\n"
|
||||
else:
|
||||
msg += f" ├ <b>Is Premium</b>: No\n"
|
||||
msg += f" ├ <b>Chat ID:</b> {message.chat.id}"
|
||||
|
||||
return msg
|
||||
@@ -0,0 +1,70 @@
|
||||
""" 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')
|
||||
Reference in New Issue
Block a user