Added a table to store the file extension of a model.

mediagoblin/media_types/stl/models.py

@@ -18,7 +18,7 @@
 from mediagoblin.db.sql.base import Base
 
 from sqlalchemy import (
-    Column, Integer, Float, ForeignKey)
+    Column, Integer, Float, String, ForeignKey)
 from sqlalchemy.orm import relationship, backref
 
 
@@ -39,6 +39,8 @@ class StlData(Base):
     height = Column(Float)
     depth = Column(Float)
 
+    file_type = Column(String)
+
 
 DATA_MODEL = StlData
 MODELS = [StlData]

mediagoblin/media_types/stl/processing.py

@@ -164,6 +164,7 @@ def process_stl(entry):
         "width" : model.width,
         "height" : model.height,
         "depth" : model.depth,
+        "file_type" : ext,
         }
     entry.media_data_init(**dimensions)