Build out graph logic into different classes on the back-end

Graph.get_rule_options was getting too complex, so I've built out a
class-based system for organizing the logic of the various graph types.

Author: nikolas

econplayground/main/graphs.py

@@ -0,0 +1,111 @@
+"""
+Graph-specific logic
+"""
+from abc import ABC, abstractmethod
+
+
+GRAPH_TYPES = (
+    (0, 'Linear Demand and Supply'),
+    (1, 'Input Markets'),
+    (3, 'Cobb-Douglas Production Graph'),
+    (5, 'Consumption-Leisure: Constraint'),
+    (7, 'Consumption-Saving: Constraint'),
+    (8, 'Linear Demand and Supply: 3 Functions'),
+
+    # Area Under Curve graphs
+    (9, 'Linear Demand and Supply: Areas'),
+    (10, 'Input Markets: Areas'),
+
+    (11, 'Consumption-Saving: Optimal Choice'),
+    (15, 'Consumption-Leisure: Optimal Choice'),
+
+    # Joint Graphs
+    (12, 'Input-Output Illustrations'),
+    (13, 'Linear Demand and Supply: 2 Diagrams'),
+    (14, 'Input Markets: 2 Diagrams'),
+
+    (16, 'Template Graph'),
+
+    (17, 'Optimal Choice: Consumption with 2 Goods'),
+    (18, 'Cost Functions'),
+    (20, 'Price Elasticity of Demand and Revenue'),
+    (21, 'Optimal Choice: Cost-Minimizing Production Inputs'),
+
+    (22, 'Tax Rate and Revenue'),
+    (23, 'Taxation in Linear Demand and Supply'),
+    (24, 'Tax Supply and Demand vs. Tax Revenue'),
+
+    (25, 'Linear Demand and Supply - Surplus Policies'),
+
+    # Externalities
+    (26, 'Negative Production Externality - Producer'),
+    (27, 'Negative Production Externality - Industry'),
+    (28, 'Positive Externality - Industry'),
+)
+
+
+class Graph(ABC):
+    line_actions = [
+        'up',
+        'down',
+        'slope up',
+        'slope down',
+    ]
+    variable_actions = [
+        'up',
+        'down',
+        '<exact value>',
+    ]
+
+    @classmethod
+    def get_rule_options(cls) -> dict:
+        return {
+            'line1': {
+                'name': 'Orange line',
+                'possible_values': cls.line_actions,
+            },
+
+            'line2': {
+                'name': 'Blue line',
+                'possible_values': cls.line_actions,
+            },
+
+            'line1 label': 'Orange line label',
+            'line2 label': 'Blue line label',
+            'intersectionLabel': 'Intersection label',
+            'intersectionHorizLineLabel':
+            'Orange-Blue intersection horizontal label',
+            'intersectionVertLineLabel':
+            'Orange-Blue intersection vertical label',
+            'x-axis label': 'X-axis label',
+            'y-axis label': 'Y-axis label',
+        }
+
+
+class Graph0(Graph):
+    name = GRAPH_TYPES[0][1]
+    graph_type = 0
+
+    @classmethod
+    def get_rule_options(cls) -> dict:
+        return super().get_rule_options()
+
+
+class Graph1(Graph):
+    name = GRAPH_TYPES[1][1]
+    graph_type = 1
+
+
+class Graph3(Graph):
+    name = GRAPH_TYPES[2][1]
+    graph_type = 3
+
+
+class Graph5(Graph):
+    name = GRAPH_TYPES[3][1]
+    graph_type = 5
+
+
+class Graph7(Graph):
+    name = GRAPH_TYPES[4][1]
+    graph_type = 7

econplayground/main/models.py

@@ -9,45 +9,8 @@
 from django.dispatch import receiver
 from django.db.models.signals import post_save, pre_delete
 
+from econplayground.main.graphs import GRAPH_TYPES, Graph0
 
-GRAPH_TYPES = (
-    (0, 'Linear Demand and Supply'),
-    (1, 'Input Markets'),
-    (3, 'Cobb-Douglas Production Graph'),
-    (5, 'Consumption-Leisure: Constraint'),
-    (7, 'Consumption-Saving: Constraint'),
-    (8, 'Linear Demand and Supply: 3 Functions'),
-
-    # Area Under Curve graphs
-    (9, 'Linear Demand and Supply: Areas'),
-    (10, 'Input Markets: Areas'),
-
-    (11, 'Consumption-Saving: Optimal Choice'),
-    (15, 'Consumption-Leisure: Optimal Choice'),
-
-    # Joint Graphs
-    (12, 'Input-Output Illustrations'),
-    (13, 'Linear Demand and Supply: 2 Diagrams'),
-    (14, 'Input Markets: 2 Diagrams'),
-
-    (16, 'Template Graph'),
-
-    (17, 'Optimal Choice: Consumption with 2 Goods'),
-    (18, 'Cost Functions'),
-    (20, 'Price Elasticity of Demand and Revenue'),
-    (21, 'Optimal Choice: Cost-Minimizing Production Inputs'),
-
-    (22, 'Tax Rate and Revenue'),
-    (23, 'Taxation in Linear Demand and Supply'),
-    (24, 'Tax Supply and Demand vs. Tax Revenue'),
-
-    (25, 'Linear Demand and Supply - Surplus Policies'),
-
-    # Externalities
-    (26, 'Negative Production Externality - Producer'),
-    (27, 'Negative Production Externality - Industry'),
-    (28, 'Positive Externality - Industry'),
-)
 
 ASSIGNMENT_TYPES = (
     (0, 'Template'),
@@ -475,6 +438,33 @@ def get_rule_options(graph_type: int = 0) -> dict:
                     'possible_values': line_actions,
                 }
             })
+        elif graph_type == 25:
+            rules = {
+                'a1': {
+                    'name': 'Choke Price',
+                    'possible_values': variable_actions,
+                },
+                'a2': {
+                    'name': 'Demand Slope',
+                    'possible_values': variable_actions,
+                },
+                'a3': {
+                    'name': 'Reservation Price',
+                    'possible_values': variable_actions,
+                },
+                'a4': {
+                    'name': 'Supply Slope',
+                    'possible_values': variable_actions,
+                },
+                'a5': {
+                    'name': 'Global/Minimum/Maximum Price or Production Quota',
+                    'possible_values': variable_actions,
+                },
+                'a6': {
+                    'name': 'Tariff',
+                    'possible_values': variable_actions,
+                },
+            }
         elif graph_type == 26:
             rules = {
                 'a1': {