37 lines
856 B
Python
37 lines
856 B
Python
# -*- 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]
|