openmedialibrary/oml/media/txt.py

# -*- coding: utf-8 -*-
# vi:si:et:sw=4:sts=4:ts=4


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]