Add Graph.get_assessment_names()

This should facilitate dynamic rule display for the various graph types.

Author: nikolas

econplayground/main/models.py

@@ -404,6 +404,28 @@ def clone(self):
         g.save()
         return g
 
+    def get_assessment_names(self):
+        assessment_names = {
+            'line1': 'Orange line',
+            'line2': 'Blue line',
+            'line1 label': 'Orange line label',
+            'line2 label': 'Blue line label',
+            'intersectionLabel': 'Intersection label',
+            'intersectionHorizLineLabel':
+            'Orange-Blue intersection horizontal',
+            'intersectionVertLineLabel': 'Orange-Blue intersection vertical',
+            'x-axis label': 'X-axis label',
+            'y-axis label': 'Y-axis label',
+        }
+
+        # TODO: extend possible rules for the various other graph types
+        if self.graph_type == 8:
+            assessment_names.update({
+                'line3': 'Green line'
+            })
+
+        return assessment_names
+
 
 class JXGLine(models.Model):
     class Meta: