# -*- coding: utf-8 -*- import os from utils import find_isbns import tempfile import subprocess def cover(path): image = tempfile.mkstemp('.jpg')[1] cmd = ['python3', '../reader/txt.js/txt.py', '-i', path, '-o', image] p = subprocess.Popen(cmd, close_fds=True) p.wait() with open(image, 'rb') as fd: data = fd.read() os.unlink(image) return data def info(path): data = {} data['title'] = os.path.splitext(os.path.basename(path))[0] text = extract_text(path) isbn = extract_isbn(text) if isbn: data['isbn'] = isbn data['textsize'] = len(text) return data def extract_text(path): with open(path, 'rb') as fd: data = fd.read().decode('utf-8', errors='replace') return data def extract_isbn(text): isbns = find_isbns(text) if isbns: return isbns[0]