From ac352c0f0eea77ba7d6f3c603618b1b8fec4f6c8 Mon Sep 17 00:00:00 2001 From: lukew3 Date: Fri, 20 Nov 2020 20:57:37 -0500 Subject: [PATCH] Add groupname item to genList --- mathgenerator/__init__.py | 6 ++++-- test.py | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/mathgenerator/__init__.py b/mathgenerator/__init__.py index 381125f..b4bfb54 100644 --- a/mathgenerator/__init__.py +++ b/mathgenerator/__init__.py @@ -21,8 +21,10 @@ class Generator: text) = traceback.extract_stack()[-2] funcname = filename[filename.rfind('/'):].strip() funcname = funcname[1:-3] - # print(funcname) - genList.append([id, title, self, funcname]) + groupname = filename[:filename.rfind('/')].strip() + groupname = groupname[groupname.rfind('/'):].strip() + groupname = groupname[1:] + genList.append([id, title, self, funcname, groupname]) def __str__(self): return str( diff --git a/test.py b/test.py index e4388cb..2c51d56 100644 --- a/test.py +++ b/test.py @@ -10,7 +10,7 @@ for item in list: print(item[2]) # print(mathgen.getGenList()) -print(mathgen.genById(1)) +print(mathgen.genById(10)) #Make a pdf with 10 problems of generator id 1 # mathgen.makePdf(0, 10)