From 31d86cd0a27922c71c362ddf518ce211f5ed580e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gina=20H=C3=A4u=C3=9Fge?= <osd@foosel.net>
Date: Sat, 20 Jul 2013 16:01:17 +0200
Subject: [PATCH] Added left space on filesystem holding the uploads folder to
 the gcode files list

Closes #186
---
 octoprint/server.py                 |  2 +-
 octoprint/static/css/octoprint.less |  4 ++++
 octoprint/static/js/ui.js           |  4 ++++
 octoprint/templates/index.jinja2    |  3 +++
 octoprint/util/__init__.py          | 19 +++++++++++++++++--
 5 files changed, 29 insertions(+), 3 deletions(-)

diff --git a/octoprint/server.py b/octoprint/server.py
index 88b553b..2601e89 100644
--- a/octoprint/server.py
+++ b/octoprint/server.py
@@ -301,7 +301,7 @@ def readGcodeFiles():
 				"date": "n/a",
 				"origin": "sd"
 			})
-	return jsonify(files=files)
+	return jsonify(files=files, free=util.getFormattedSize(util.getFreeBytes(settings().getBaseFolder("uploads"))))
 
 @app.route(BASEURL + "gcodefiles/<path:filename>", methods=["GET"])
 def readGcodeFile(filename):
diff --git a/octoprint/static/css/octoprint.less b/octoprint/static/css/octoprint.less
index 187a9c6..05ad83e 100644
--- a/octoprint/static/css/octoprint.less
+++ b/octoprint/static/css/octoprint.less
@@ -385,6 +385,10 @@ ul.dropdown-menu li a {
       margin-bottom: 0;
     }
   }
+
+  table {
+    margin-bottom: 0;
+  }
 }
 
 /** Control tab */
diff --git a/octoprint/static/js/ui.js b/octoprint/static/js/ui.js
index 4040f32..c11ab6c 100644
--- a/octoprint/static/js/ui.js
+++ b/octoprint/static/js/ui.js
@@ -784,6 +784,8 @@ function GcodeFilesViewModel(printerStateViewModel, loginStateViewModel) {
     self.isLoading = ko.observable(undefined);
     self.isSdReady = ko.observable(undefined);
 
+    self.freeSpace = ko.observable(undefined);
+
     // initialize list helper
     self.listHelper = new ItemListHelper(
         "gcodeFiles",
@@ -884,6 +886,8 @@ function GcodeFilesViewModel(printerStateViewModel, loginStateViewModel) {
             self.listHelper.switchToItem(function(item) {return item.name == response.filename});
         }
 
+        self.freeSpace(response.free);
+
         self.highlightFilename(self.printerState.filename());
     }
 
diff --git a/octoprint/templates/index.jinja2 b/octoprint/templates/index.jinja2
index 3cfa071..936d836 100644
--- a/octoprint/templates/index.jinja2
+++ b/octoprint/templates/index.jinja2
@@ -186,6 +186,9 @@
                                         </tr>
                                     </tbody>
                                 </table>
+                                <div class="muted text-right">
+                                    <small>Free: <span data-bind="text: freeSpace"></span></small>
+                                </div>
                                 <div class="pagination pagination-mini pagination-centered">
                                     <ul>
                                         <li data-bind="css: {disabled: listHelper.currentPage() === 0}"><a href="#" data-bind="click: listHelper.prevPage">«</a></li>
diff --git a/octoprint/util/__init__.py b/octoprint/util/__init__.py
index 845cce5..77b046f 100644
--- a/octoprint/util/__init__.py
+++ b/octoprint/util/__init__.py
@@ -11,7 +11,7 @@ from octoprint.settings import settings
 
 def getFormattedSize(num):
 	"""
-	 Taken from http://stackoverflow.com/questions/1094841/reusable-library-to-get-human-readable-version-of-file-size
+	Taken from http://stackoverflow.com/a/1094933/2028598
 	"""
 	for x in ["bytes","KB","MB","GB"]:
 		if num < 1024.0:
@@ -42,7 +42,7 @@ def getFormattedDateTime(d):
 
 def getClass(name):
 	"""
-	 Taken from http://stackoverflow.com/a/452981/2028598
+	Taken from http://stackoverflow.com/a/452981/2028598
 	"""
 	parts = name.split(".")
 	module = ".".join(parts[:-1])
@@ -89,3 +89,18 @@ def getNewTimeout(type):
 		return now # timeout immediately for unknown timeout type
 
 	return now + settings().getFloat(["serial", "timeout", type])
+
+
+def getFreeBytes(path):
+	"""
+	Taken from http://stackoverflow.com/a/2372171/2028598
+	"""
+	if sys.platform == "win32":
+		import ctypes
+		freeBytes = ctypes.c_ulonglong(0)
+		ctypes.windll.kernel32.GetDiskFreeSpaceExW(ctypes.c_wchar_p(path), None, None, ctypes.pointer(freeBytes))
+		return freeBytes.value
+	else:
+		st = os.statvfs(path)
+		return st.f_bavail * st.f_frsize
+