From 6d703d236a8d0ce0d7a70d57fdb8e27afb504f8a Mon Sep 17 00:00:00 2001 From: Ross Hendrickson Date: Sun, 21 Jul 2013 19:07:57 -0600 Subject: [PATCH] Minor improvement to getFileData --- octoprint/gcodefiles.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/octoprint/gcodefiles.py b/octoprint/gcodefiles.py index 992ed267..32b304ec 100644 --- a/octoprint/gcodefiles.py +++ b/octoprint/gcodefiles.py @@ -217,20 +217,15 @@ class GcodeManager: return files def getFileData(self, filename): - from octoprint.filemanager.types import FileTypes - + from octoprint.util import isSTLFileName + if not filename: return filename = self._getBasicFilename(filename) - fileType = filename.rsplit(".", 1)[1] - - if not fileType: - return None - # TODO: Make this more robust when STLs will be viewable from the client - if fileType != FileTypes.GCODE: + if isSTLFileName(filename): return absolutePath = self.getAbsolutePath(filename)