Moving to more accurate disk usage monitor.

This commit is contained in:
Samuel Clay 2016-03-16 11:40:11 -07:00
parent a1e23eb8d5
commit 21e5da0ecf

View file

@ -3,25 +3,28 @@
import sys import sys
sys.path.append('/srv/newsblur') sys.path.append('/srv/newsblur')
import subprocess
import psutil import psutil
import requests import requests
import settings import settings
import socket import socket
def main(): def main():
usage = psutil.disk_usage('/') df = subprocess.Popen(["df", "/"], stdout=subprocess.PIPE)
output = df.communicate()[0]
device, size, used, available, percent, mountpoint = output.split("\n")[1].split()
hostname = socket.gethostname() hostname = socket.gethostname()
if usage.percent > 95: if int(percent) > 95:
requests.post( requests.post(
"https://api.mailgun.net/v2/%s/messages" % settings.MAILGUN_SERVER_NAME, "https://api.mailgun.net/v2/%s/messages" % settings.MAILGUN_SERVER_NAME,
auth=("api", settings.MAILGUN_ACCESS_KEY), auth=("api", settings.MAILGUN_ACCESS_KEY),
data={"from": "NewsBlur Monitor: %s <admin@%s.newsblur.com>" % (hostname, hostname), data={"from": "NewsBlur Monitor: %s <admin@%s.newsblur.com>" % (hostname, hostname),
"to": [settings.ADMINS[0][1]], "to": [settings.ADMINS[0][1]],
"subject": "%s hit %s%% disk usage!" % (hostname, usage.percent), "subject": "%s hit %s%% disk usage!" % (hostname, int(percent)),
"text": "Usage on %s: %s" % (hostname, usage)}) "text": "Usage on %s: %s" % (hostname, output)})
else: else:
print " ---> Disk usage is fine: %s / %s%% used" % (hostname, usage.percent) print " ---> Disk usage is fine: %s / %s%% used" % (hostname, int(percent))
if __name__ == '__main__': if __name__ == '__main__':
main() main()