Merge pull request #238 from theonion/infographics

Infographics

CHANGELOG.md

@@ -2,6 +2,10 @@
 
 ## Development
 
+## Version 3.1.3
+
+- Added `AbstractInfographic.get_infographic_type_name` method.
+
 ## Version 3.1.2
 
 - Added styling for mobile ads in bulbs-styles

bulbs/__init__.py

@@ -1 +1 @@
-__version__ = "3.1.2"
+__version__ = "3.1.3"

bulbs/infographics/enum.py

@@ -7,3 +7,11 @@ class InfographicType(enum.Enum):
     STRONGSIDE_WEAKSIDE = 2
     PRO_CON = 3
     COMPARISON = 4
+
+    labels = {
+        LIST: "List",
+        TIMELINE: "Timeline",
+        STRONGSIDE_WEAKSIDE: "Strongside_Weakside",
+        PRO_CON: "Pro_Con",
+        COMPARISON: "Comparison"
+    }

bulbs/infographics/models.py

@@ -17,6 +17,9 @@ class AbstractInfographic(models.Model):
     class Meta:
         abstract = True
 
+    def get_infographic_type_name(self):
+        return InfographicType.label(self.infographic_type)
+
     def get_data_serializer(self):
         return get_data_serializer(self.infographic_type)
 

tests/infographics/test_models.py

@@ -66,3 +66,17 @@ def test_timeline_validator(self):
                 }]
             }
         )
+
+    def test_get_infographic_type_name(self):
+        instance = BaseInfographic.objects.create(
+            title="A GOD BOY ADVENTURE",
+            infographic_type=InfographicType.TIMELINE,
+            data={
+                "entries": [{
+                    "title": "Sickass",
+                    "copy": "heya",
+                    "image": {"id": 50}
+                }]
+            }
+        )
+        self.assertEqual(instance.get_infographic_type_name(), "Timeline")