Adding stuff for disk endpoint

2020-01-28 14:19:48 -05:00 · 2020-01-28 14:19:48 -05:00 · 0cec28e3f7
parent 9683f2d6f3
commit 0cec28e3f7
1 changed files with 5 additions and 1 deletions
--- a/endpoints/disk.py
+++ b/endpoints/disk.py
@ -1,5 +1,9 @@
 from flask_restful import Resource, abort
+import psutil

 class Disk(Resource):
    def get(self):
-        abort(501, message="Not currently implemented.")
+        disk_usage = psutil.disk_usage('/')
+        payload = {"total":disk_usage.total,"used":disk_usage.used,"percentused":disk_usage.percent}
+        return payload
+        ##abort(501, message="Not currently implemented.")