Function to return memory consumption for the script itself.

memory_monitor.py

from resource import getrusage, RUSAGE_SELF

def monitor(unit="M", alert=None):
  unit = unit.upper()
  unit_transformation = {
      "B" : 1./1024.,
      "K" : 1.,
      "M" : 1024.,
      "G" : 1024.*1024.,
    }
  if not unit in unit_transformation.keys():
    unit = "K"  
  mem = getrusage(RUSAGE_SELF).ru_maxrss
  mem /= unit_transformation[unit]
  if not alert is None:
    status = mem > alert
  else:
    status = None
  return (status, mem,)