From d25ed5dd4c364ab3c3f6cd28f5c64b55592630a7 Mon Sep 17 00:00:00 2001
From: Aeva Ntsc <aeva.ntsc@gmail.com>
Date: Tue, 16 Oct 2012 01:01:34 -0500
Subject: [PATCH] Added a table to store the file extension of a model.

---
 mediagoblin/media_types/stl/models.py     | 4 +++-
 mediagoblin/media_types/stl/processing.py | 1 +
 2 files changed, 4 insertions(+), 1 deletion(-)

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)