Cleanup working files when done with a particular file, rather than end of process

This commit is contained in:
James R. Barlow
2019-05-23 01:25:08 -07:00
parent 22298b31be
commit d0efdf643c
+2 -4
View File
@@ -15,7 +15,6 @@
# You should have received a copy of the GNU General Public License
# along with OCRmyPDF. If not, see <http://www.gnu.org/licenses/>.
import atexit
import logging
import logging.handlers
import multiprocessing
@@ -264,9 +263,6 @@ def run_pipeline(options, api=False):
os.environ.setdefault('OMP_THREAD_LIMIT', '1')
work_folder = mkdtemp(prefix="com.github.ocrmypdf.")
atexit.register(cleanup_working_files, work_folder, options)
try:
check_requested_output_file(options)
start_input_file = create_input_file(options, work_folder)
@@ -303,6 +299,8 @@ def run_pipeline(options, api=False):
raise
log.exception("An exception occurred while executing the pipeline")
return ExitCode.other_error
finally:
cleanup_working_files(work_folder, options)
if options.output_file == '-':
log.info("Output sent to stdout")