from wbb import app, OWNER_ID, bot_start_time
import time
from pyrogram import filters
import psutil
from wbb.utils import cust_filter, formater

__MODULE__ = "Stats"
__HELP__ = "/stats - For Bot Owner To Check System Status [Owner Only]"


@app.on_message(filters.user(OWNER_ID) & cust_filter.command(commands=(["stats"])))
async def stats(client, message):
    cpu = psutil.cpu_percent(interval=0.5)
    mem = psutil.virtual_memory().percent
    disk = psutil.disk_usage("/").percent
    stats = (f'''
```Uptime: {formater.get_readable_time((time.time() - bot_start_time))} 
CPU: {cpu}% 
RAM: {mem}% 
Disk: {disk}%```''')
    await message.reply_text(stats)