From ed03295563e4a0c6d0f80ec85f98fdf7487f0ec0 Mon Sep 17 00:00:00 2001
From: lukew3 <lukew25073@gmail.com>
Date: Thu, 15 Oct 2020 11:11:48 -0400
Subject: [PATCH] Add getGenList function

---
 mathgenerator/mathgen.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/mathgenerator/mathgen.py b/mathgenerator/mathgen.py
index e2c486f..a7063b2 100644
--- a/mathgenerator/mathgen.py
+++ b/mathgenerator/mathgen.py
@@ -1,8 +1,8 @@
 import random
 
 genList = []
-# || Generator class
 
+# || Generator class
 class Generator:
     def __init__(self, title, id, generalProb, generalSol, func):
         self.title = title
@@ -10,7 +10,7 @@ class Generator:
         self.generalProb = generalProb
         self.generalSol = generalSol
         self.func = func
-        genList.append(self)
+        genList.append([id, title, self])
 
     def __str__(self):
         return str(self.id) + " " + self.title + " " + self.generalProb + " " + self.generalSol
@@ -18,12 +18,15 @@ class Generator:
     def __call__(self):
         return self.func()
 
-# || CallbyId
+# || Non-generator Functions
 def genById(id):
-    generator = genList[id]
+    generator = genList[id][2]
     return(generator())
 
-# || Functions
+def getGenList():
+    return(genList)
+
+# || Generator Functions
 
 def additionFunc(maxSum = 99, maxAddend = 50):
     a = random.randint(0, maxAddend)