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 = {}