Fixed issue with some epubs not getting correct metadata when uploading

8 years ago · 2e5021e4ef
parent 41a20fe3e1
commit 2e5021e4ef
1 changed files with 4 additions and 1 deletions
--- a/cps/epub.py
+++ b/cps/epub.py
@ -11,7 +11,10 @@ def extractCover(zip, coverFile, tmp_file_name):
    if coverFile is None:
        return None
    else:
-        cf = zip.read("OPS/" + coverFile)
+        try:
+            cf = zip.read("OPS/" + coverFile)
+        except KeyError, e:
+            cf = zip.read(coverFile)
        prefix = os.path.splitext(tmp_file_name)[0]
        tmp_cover_name = prefix + "." + coverFile
        image = open(tmp_cover_name, 'wb')