diff --git a/mathgenerator/__init__.py b/mathgenerator/__init__.py index d42c0fe..1e88f8b 100644 --- a/mathgenerator/__init__.py +++ b/mathgenerator/__init__.py @@ -1,3 +1,4 @@ +import os import sys import traceback @@ -8,7 +9,7 @@ import scipy genList = [] - +SEP = os.sep class Generator: def __init__(self, title, id, func, kwargs): self.title = title @@ -18,10 +19,10 @@ class Generator: (filename, line_number, function_name, text) = traceback.extract_stack()[-2] - funcname = filename[filename.rfind('/'):].strip() + funcname = filename[filename.rfind(SEP):].strip() funcname = funcname[1:-3] - subjectname = filename[:filename.rfind('/')].strip() - subjectname = subjectname[subjectname.rfind('/'):].strip() + subjectname = filename[:filename.rfind(SEP)].strip() + subjectname = subjectname[subjectname.rfind(SEP):].strip() subjectname = subjectname[1:] genList.append([id, title, self, funcname, subjectname, kwargs]) diff --git a/scripts/makeReadme.py b/scripts/makeReadme.py index 887e8c2..154a28a 100644 --- a/scripts/makeReadme.py +++ b/scripts/makeReadme.py @@ -144,7 +144,7 @@ def main(): for write_line in write_list: lines.append(write_line) - with open('../README.md', "w") as g: + with open('../README.md', "w", encoding = 'utf-8') as g: g.writelines(lines) print("New README.md table generated")