NussNougate/nhentai
4
0
Fork 0
mirror of https://github.com/RicterZ/nhentai.git synced 2025-04-19 18:31:24 +02:00
Code Issues Packages Projects Releases Wiki Activity

ignore other folders

Browse Source
This commit is contained in:
symant233 2019-05-26 15:57:50 +08:00
parent 710cc86eaf
commit 86b3a092c7
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
nhentai/utils.py
View File

@ -109,6 +109,8 @@ def generate_main_html(output_dir='.'):
# https://stackoverflow.com/questions/141291/how-to-list-only-top-level-directories-in-python
for folder in doujinshi_dirs:
if folder[0] is not '[':
continue
files = os.listdir(folder)
if 'index.html' in files:
count += 1
@ -125,6 +127,9 @@ def generate_main_html(output_dir='.'):
title = 'nHentai HTML Viewer'
image_html += element.format(FOLDER=folder, IMAGE=image, TITLE=title)
if image_html == '':
logger.warning('None index.html found, --gen-main paused.')
return
try:
data = main.format(STYLES=css, COUNT=count, PICTURE=image_html)
if sys.version_info < (3, 0):
@ -133,7 +138,7 @@ def generate_main_html(output_dir='.'):
else:
with open('./main.html', 'wb') as f:
f.write(data.encode('utf-8'))
logger.log(15, 'Main Viewer has been write to \'{0}\''.format('./main.html'))
logger.log(15, 'Main Viewer has been write to \'{0}/main.html\''.format(output_dir))
except Exception as e:
logger.warning('Writen Main Viewer failed ({})'.format(str(e)))
logger.info('==>Process finished.')