From 25b3f26d6bb0e71d4d52d2c8e5fceb26cf44555e Mon Sep 17 00:00:00 2001 From: j Date: Tue, 29 Jan 2019 18:46:12 +0530 Subject: [PATCH] only try first image --- oml/media/cbr.py | 11 ++++------- oml/media/cbz.py | 14 ++++++-------- 2 files changed, 10 insertions(+), 15 deletions(-) diff --git a/oml/media/cbr.py b/oml/media/cbr.py index fdade82..0c29fe2 100644 --- a/oml/media/cbr.py +++ b/oml/media/cbr.py @@ -10,13 +10,10 @@ def cover(path): try: from unrar import rarfile rar = rarfile.RarFile(path) - for cover in ox.sorted_strings(rar.namelist()): - try: - data = rar.read(cover) - except: - data = None - finally: - return data + files = rar.namelist() + if files: + cover = ox.sorted_strings(files)[0] + data = rar.read(cover) except: data = None return data diff --git a/oml/media/cbz.py b/oml/media/cbz.py index 83fba49..b08c972 100644 --- a/oml/media/cbz.py +++ b/oml/media/cbz.py @@ -15,14 +15,12 @@ def cover(path): return data files = [f.filename for f in z.filelist] if files: - for cover in ox.sortedstrings(files): - try: - data = z.read(cover) - except: - data = None - finally: - return data - + cover = ox.sortedstrings(files)[0] + try: + data = z.read(cover) + except: + data = None + return data def info(path): data = {}