I add command /analytics and I improve code
It's version 0.2.3
This commit is contained in:
+48
-9
@@ -1,27 +1,66 @@
|
||||
import asyncio
|
||||
import os
|
||||
|
||||
async def get_macbook_battery_level():
|
||||
"""
|
||||
Асинхронная функция для получения уровня заряда батареи на macOS.
|
||||
Использует утилиту pmset.
|
||||
"""
|
||||
# Запускаем команду pmset -g batt
|
||||
process = await asyncio.create_subprocess_exec(
|
||||
"pmset", "-g", "batt",
|
||||
stdout=asyncio.subprocess.PIPE,
|
||||
stderr=asyncio.subprocess.PIPE
|
||||
)
|
||||
|
||||
stdout, stderr = await process.communicate()
|
||||
|
||||
if process.returncode != 0:
|
||||
raise RuntimeError(f"Ошибка при выполнении pmset: {stderr.decode().strip()}")
|
||||
|
||||
output = stdout.decode().strip()
|
||||
# Пример строки: " - InternalBattery-0 (id=1234567) 85%; charging; ..."
|
||||
for part in output.splitlines():
|
||||
if "%" in part:
|
||||
percent_str = part.split("\t")[-1].split(";")[0].strip()
|
||||
return int(percent_str.replace("%", ""))
|
||||
raise ValueError("Не удалось определить уровень заряда")
|
||||
|
||||
raise ValueError("Не удалось определить уровень заряда")
|
||||
|
||||
async def get_process_usage(pid=None):
|
||||
"""
|
||||
Возвращает CPU, MEM% и RSS в мегабайтах для указанного процесса.
|
||||
По умолчанию берёт текущий процесс (os.getpid()).
|
||||
"""
|
||||
if pid is None:
|
||||
pid = os.getpid()
|
||||
|
||||
process = await asyncio.create_subprocess_exec(
|
||||
"ps", "-p", str(pid), "-o", "%cpu,%mem,rss,comm",
|
||||
stdout=asyncio.subprocess.PIPE,
|
||||
stderr=asyncio.subprocess.PIPE
|
||||
)
|
||||
stdout, stderr = await process.communicate()
|
||||
if process.returncode != 0:
|
||||
raise RuntimeError(f"Ошибка ps: {stderr.decode().strip()}")
|
||||
|
||||
lines = stdout.decode().strip().splitlines()
|
||||
if len(lines) < 2:
|
||||
return None
|
||||
|
||||
cpu, mem_percent, rss_kb, comm = lines[1].split(None, 3)
|
||||
|
||||
return {
|
||||
"pid": pid,
|
||||
"command": comm,
|
||||
"cpu_percent": float(cpu),
|
||||
"mem_percent": float(mem_percent),
|
||||
"rss_mb": int(rss_kb) / 1024 # переводим КБ → МБ
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
async def main():
|
||||
battery = await get_macbook_battery_level()
|
||||
usage = await get_process_usage()
|
||||
|
||||
print(f"🔋 Батарея: {battery}%")
|
||||
print(f"🖥 CPU: {usage['cpu_percent']}% | MEM: {usage['mem_percent']}% | RSS: {usage['rss_mb']:.2f} MB")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(main())
|
||||
|
||||
Reference in New Issue
Block a user