From 53c34db027d5a52af63d9d775dad32a180c94a98 Mon Sep 17 00:00:00 2001
From: jaskaranSM <beerabassian32@gmail.com>
Date: Sun, 21 Jun 2020 14:02:24 +0530
Subject: [PATCH] Added cpu and ram usage in stats

---
 bot/__main__.py  | 9 +++++++--
 requirements.txt | 1 +
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/bot/__main__.py b/bot/__main__.py
index 34df995..bc432bb 100644
--- a/bot/__main__.py
+++ b/bot/__main__.py
@@ -1,9 +1,10 @@
-import shutil
+import shutil, psutil
 import signal
 import pickle
 
 from os import execl, path, remove
 from sys import executable
+import time
 
 from telegram.ext import CommandHandler, run_async
 from bot import dispatcher, updater, botStartTime
@@ -22,10 +23,14 @@ def stats(update, context):
     total = get_readable_file_size(total)
     used = get_readable_file_size(used)
     free = get_readable_file_size(free)
+    cpuUsage = psutil.cpu_percent(interval=0.5)
+    memory = psutil.virtual_memory().percent
     stats = f'Bot Uptime: {currentTime}\n' \
             f'Total disk space: {total}\n' \
             f'Used: {used}\n' \
-            f'Free: {free}'
+            f'Free: {free}\n' \
+            f'CPU: {cpuUsage}%\n' \
+            f'RAM: {memory}%'
     sendMessage(stats, context.bot, update)
 
 
diff --git a/requirements.txt b/requirements.txt
index 078a15e..9a915ff 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,5 @@
 requests
+psutil
 python-telegram-bot==12.6.1 
 google-api-python-client>=1.7.11,<1.7.20
 google-auth-httplib2>=0.0.3,<0.1.0