Add utility function to eject drives.

[cura.git] / Cura / util / removableStorage.py

diff --git a/Cura/util/removableStorage.py b/Cura/util/removableStorage.py
index 27d5c7055fdcd1e060e280f0e67395c161e810b4..5039503b8c2bb1653d3f10e8633a513eb7d39a13 100644 (file)
--- a/Cura/util/removableStorage.py
+++ b/Cura/util/removableStorage.py
@@ -95,5 +95,17 @@ def getPossibleSDcardDrives():
        _removeableCacheTime = time.time()
        return drives
 
+def ejectDrive(driveName):
+       if platform.system() == "Windows":
+               cmd = '"%s" %s>NUL' % (os.path.abspath(os.path.join(os.path.dirname(__file__), '..', 'EjectMedia.exe')), driveName)
+       elif platform.system() == "Darwin":
+               cmd = "diskutil eject '%s' > /dev/null 2>&1" % (driveName)
+       else:
+               cmd = "umount '%s' > /dev/null 2>&1" % (driveName)
+       if os.system(cmd):
+               return False
+       else:
+               return True
+
 if __name__ == '__main__':
        print getPossibleSDcardDrives()