diff options
| -rw-r--r-- | mediagoblin/media_types/stl/models.py | 4 | ||||
| -rw-r--r-- | mediagoblin/media_types/stl/processing.py | 1 |
2 files changed, 4 insertions, 1 deletions
diff --git a/mediagoblin/media_types/stl/models.py b/mediagoblin/media_types/stl/models.py index d0873176..ccb1b9dc 100644 --- a/mediagoblin/media_types/stl/models.py +++ b/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] diff --git a/mediagoblin/media_types/stl/processing.py b/mediagoblin/media_types/stl/processing.py index aabfa9a9..35ac7ea9 100644 --- a/mediagoblin/media_types/stl/processing.py +++ b/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) |