Commit 24bad335 authored by Aram Verstegen's avatar Aram Verstegen
Browse files

Use fast storage if available

parent ccc7c575
...@@ -4,7 +4,6 @@ import sys ...@@ -4,7 +4,6 @@ import sys
import argparse import argparse
from collections import OrderedDict from collections import OrderedDict
#from hocr.parse import hocr_page_iterator
import hocr.parse import hocr.parse
from ebooklib import epub from ebooklib import epub
from abbyy_to_epub3.verify_epub import EpubVerify from abbyy_to_epub3.verify_epub import EpubVerify
...@@ -19,7 +18,10 @@ import os ...@@ -19,7 +18,10 @@ import os
import shutil import shutil
import subprocess import subprocess
WORKING_DIR = '/tmp/' if os.path.exists('/var/tmp/fast'):
WORKING_DIR = '/var/tmp/fast/'
else:
WORKING_DIR = '/tmp/'
class ImageStack(object): class ImageStack(object):
filenames = [] filenames = []
...@@ -55,7 +57,7 @@ class ImageStack(object): ...@@ -55,7 +57,7 @@ class ImageStack(object):
print("%s - Cropping page %u to box %s" % (datetime.now(), page, box)) print("%s - Cropping page %u to box %s" % (datetime.now(), page, box))
# Extract the image from the zipfile # Extract the image from the zipfile
tempfile_tiff = os.path.join(WORKING_DIR, 'temp_%u.tiff' % page) tempfile_tiff = os.path.join(WORKING_DIR, 'page_%u.tiff' % page)
if tempfile_tiff not in self.temp_files: if tempfile_tiff not in self.temp_files:
self.zf.extract(self.filenames[page], self.tempdir_zip) self.zf.extract(self.filenames[page], self.tempdir_zip)
extracted_file_path = os.path.join(self.tempdir_zip, self.filenames[page]) extracted_file_path = os.path.join(self.tempdir_zip, self.filenames[page])
...@@ -196,7 +198,7 @@ class EpubGenerator(object): ...@@ -196,7 +198,7 @@ class EpubGenerator(object):
if not self.scandata_xml_file_path: if not self.scandata_xml_file_path:
self.scandata_xml_file_path = self.hocr_xml_file_path.replace('_hocr.html', '_scandata.xml') self.scandata_xml_file_path = self.hocr_xml_file_path.replace('_hocr.html', '_scandata.xml')
self.img_stack = ImageStack(self.image_stack_zip_file_path, os.path.join(WORKING_DIR, "test_img")) self.img_stack = ImageStack(self.image_stack_zip_file_path, os.path.join(WORKING_DIR, "epub_img"))
self.epub_zip_file_path = self.hocr_xml_file_path.replace('_hocr.html', '_ebook.epub') self.epub_zip_file_path = self.hocr_xml_file_path.replace('_hocr.html', '_ebook.epub')
try: try:
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment