forust
7ba6bc44f2
- ruff check --fix: 210 auto-fixed errors (import sorting, trailing whitespace, unused imports, f-string fixups, deprecated annotations) - ruff format: 104 files reformatted to consistent style - 268 non-auto-fixable issues remain (S113 requests timeout, etc.)
79 lines
2.6 KiB
Python
79 lines
2.6 KiB
Python
# Dragon-Userbot - telegram userbot
|
|
# Copyright (C) 2020-present Dragon Userbot Organization
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
import datetime
|
|
|
|
from pyrogram import Client, filters, types
|
|
from utils.db import db
|
|
from utils.misc import modules_help, prefix
|
|
|
|
# avoid using global variables
|
|
afk_info = db.get(
|
|
'core.afk',
|
|
'afk_info',
|
|
{
|
|
'start': 0,
|
|
'is_afk': False,
|
|
'reason': '',
|
|
},
|
|
)
|
|
|
|
is_afk = filters.create(lambda _, __, ___: afk_info['is_afk'])
|
|
is_support = filters.create(lambda _, __, message: message.chat.is_support)
|
|
|
|
|
|
@Client.on_message(
|
|
is_afk & (filters.private | filters.mentioned) & ~filters.channel & ~filters.me & ~filters.bot & ~is_support
|
|
)
|
|
async def afk_handler(_, message: types.Message):
|
|
start = datetime.datetime.fromtimestamp(afk_info['start'])
|
|
end = datetime.datetime.now().replace(microsecond=0)
|
|
afk_time = end - start
|
|
await message.reply(f"<b>I'm AFK {afk_time}\nReason:</b> <i>{afk_info['reason']}</i>")
|
|
|
|
|
|
@Client.on_message(filters.command('afk', prefix) & filters.me)
|
|
async def afk(_, message):
|
|
if len(message.text.split()) >= 2:
|
|
reason = message.text.split(' ', maxsplit=1)[1]
|
|
else:
|
|
reason = 'None'
|
|
|
|
afk_info['start'] = int(datetime.datetime.now().timestamp())
|
|
afk_info['is_afk'] = True
|
|
afk_info['reason'] = reason
|
|
|
|
await message.edit(f"<b>I'm going AFK.\nReason:</b> <i>{reason}</i>")
|
|
|
|
db.set('core.afk', 'afk_info', afk_info)
|
|
|
|
|
|
@Client.on_message(filters.command('unafk', prefix) & filters.me)
|
|
async def unafk(_, message):
|
|
if afk_info['is_afk']:
|
|
start = datetime.datetime.fromtimestamp(afk_info['start'])
|
|
end = datetime.datetime.now().replace(microsecond=0)
|
|
afk_time = end - start
|
|
await message.edit(f"<b>I'm not AFK anymore.\nI was afk {afk_time}</b>")
|
|
afk_info['is_afk'] = False
|
|
else:
|
|
await message.edit("<b>You weren't afk</b>")
|
|
|
|
db.set('core.afk', 'afk_info', afk_info)
|
|
|
|
|
|
modules_help['afk'] = {'afk [reason]': 'Go to afk', 'unafk': 'Get out of AFK'}
|