diff --git a/nhentai/utils.py b/nhentai/utils.py index 9cafe37..260edb1 100644 --- a/nhentai/utils.py +++ b/nhentai/utils.py @@ -81,10 +81,14 @@ def generate_html(output_dir='.', doujinshi_obj=None): except Exception as e: logger.warning('Writen HTML Viewer failed ({})'.format(str(e))) -def generate_main_html(output_dir='.'): - """Generete a main html to show all the contain doujinshi. - With a link to thier `index.html`. - Default output folder will be the CLI path.""" + +def generate_main_html(output_dir='./'): + """ + Generate a main html to show all the contain doujinshi. + With a link to their `index.html`. + Default output folder will be the CLI path. + """ + count = 0 image_html = '' main = readfile('viewer/main.html') @@ -99,31 +103,26 @@ def generate_main_html(output_dir='.'): \n\ \n' - if output_dir == '': - os.chdir('.') - else: - os.chdir(output_dir) - # switch to given dir + os.chdir(output_dir) doujinshi_dirs = next(os.walk('.'))[1] - # 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) + files.sort() + if 'index.html' in files: count += 1 + logger.info('Add doujinshi \'{}\''.format(folder)) else: - logger.warning('{} folder does not have index.html (try use --html arg first).' - .format(folder)) continue + image = files[0] # 001.jpg or 001.png if folder is not None: title = folder.replace('_', ' ') - # if sys.version_info > (3, 0): - # title = title.encode('utf-8') else: title = 'nHentai HTML Viewer' + image_html += element.format(FOLDER=folder, IMAGE=image, TITLE=title) if image_html == '': @@ -142,6 +141,7 @@ def generate_main_html(output_dir='.'): except Exception as e: logger.warning('Writen Main Viewer failed ({})'.format(str(e))) + def generate_cbz(output_dir='.', doujinshi_obj=None, rm_origin_dir=False): if doujinshi_obj is not None: doujinshi_dir = os.path.join(output_dir, doujinshi_obj.filename)