Как измерить общее потребление памяти всех процессов пользователя в Python

Question

Я хотел бы использовать эти команды, чтобы увидеть мое использование памяти на моих серверах:

ps -u $USERNAME -o pid,rss,command | awk '{print $0}{sum+=$2} END {print "Total", sum}' 

Что бы быть лучшим способом, чтобы собрать эту общую сумму в скрипте Python? Я смотрел как psutil, но этот модуль собирает только информацию о глобальной памяти, и нет возможности отфильтровать его пользователем.

Answer 1

Это довольно простое использование psutil. Вы можете просто перебрать все процессы, выбрать те, которые принадлежат вам, и суммировать память, возвращаемую memory_info().

import psutil 
import getpass 

user = getpass.getuser() 

total = sum(p.memory_info()[0] for p in psutil.process_iter() 
           if p.username == user) 

print('Total memory usage in bytes: ', total)