wiki/flask-wikipages
Archived
1
0
Fork 0
Code Issues 1 Pull Requests Projects Releases Wiki Activity

Fix non-working code

Browse Source
This commit is contained in:
thematdev 2022-04-04 23:14:14 +03:00
parent 227c042642
commit dfa7138091
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
app/__init__.py
View File

@ -29,7 +29,8 @@ def render_page(page):
path = f'{PANDOC_PATH}/{page}' path = f'{PANDOC_PATH}/{page}'
whitelist = WHITELIST_PATH whitelist = WHITELIST_PATH
with open(whitelist, 'r') as f: with open(whitelist, 'r') as f:
if page not in f.readlines(): lines = f.read().splitlines()
if page not in lines:
raise Exception("Page doesn't exist!") raise Exception("Page doesn't exist!")
if not os.path.exists(f'{path}'): if not os.path.exists(f'{path}'):
raise Exception("Page doesn't exist!") raise Exception("Page doesn't exist!")
@ -45,7 +46,9 @@ def get_pandoc_page(page):
path = f'{PANDOC_PATH}/{page}' path = f'{PANDOC_PATH}/{page}'
whitelist = WHITELIST_PATH whitelist = WHITELIST_PATH
with open(whitelist, 'r') as f: with open(whitelist, 'r') as f:
if page not in f.readlines(): lines = f.read().splitlines()
if page not in lines:
print('Access to page not in list {page}')
return 'This page does not exist' return 'This page does not exist'
if not os.path.exists(f'{path}'): if not os.path.exists(f'{path}'):
# TODO: Add 404 handler # TODO: Add 404 handler