From eadc644e0da40a6975966794dd22f9b2b9b9425f Mon Sep 17 00:00:00 2001
From: Satyam Koshta <58166232+satyamkoshta340@users.noreply.github.com>
Date: Mon, 19 Oct 2020 14:32:43 +0530
Subject: [PATCH] Update mathgen.py

---
 mathgen.py | 23 +++++++++++++++++++++--
 1 file changed, 21 insertions(+), 2 deletions(-)

diff --git a/mathgen.py b/mathgen.py
index b0a426a..85a8f3d 100644
--- a/mathgen.py
+++ b/mathgen.py
@@ -1,5 +1,24 @@
-#!/usr/bin/python
-# -*- coding: utf-8 -*-
+import random
+
+
+genList = []
+
+
+# || Generator class
+class Generator:
+    def __init__(self, title, id, generalProb, generalSol, func):
+        self.title = title
+        self.id = id
+        self.generalProb = generalProb
+        self.generalSol = generalSol
+        self.func = func
+        genList.append([id, title, self])
+
+    def __str__(self):
+        return str(self.id) + " " + self.title + " " + self.generalProb + " " + self.generalSol
+
+    def __call__(self, **kwargs):
+        return self.func(**kwargs)
 
 
 def isprime(max_a=100):