diff --git a/gsitegen/generate.py b/gsitegen/generate.py index e076737..f496835 100644 --- a/gsitegen/generate.py +++ b/gsitegen/generate.py @@ -129,12 +129,11 @@ def main(): pagescount = 0 for file in Path(__file__).parent.iterdir(): - #TODO get file name only if file.is_file(): if file.suffix == ".page": pagescount += 1 with file.open('r') as page: - generateLines(str(file.stem), page.readlines()) + generateLines(file.stem, page.readlines()) print("Found " + str(pagescount) + " pages") #write all pages to files