AbbyyPdfTextExtractorNetworkTest.py

#!/usr/local/bin/python

import unittest
import sys
import os.path
import glob
import ConfigParser
from pdftools.PdfSeparate import *
from abbyy.AbbyyPdfTextExtractor import *
from urllib2 import HTTPError

class AbbyyPdfTextExtractorNetworkTest(unittest.TestCase):
    def setUp(self):
        self.outdir = "tests/out/abbyy/text"
        self.indir = "tests/out/abbyy/pdf"
        self.createOrCleanDir(self.outdir)
        self.createOrCleanDir(self.indir)
        self.configParser = ConfigParser.RawConfigParser()
        self.configParser.read('settings.config')


    def createOrCleanDir(self, directory):
        if not os.path.exists(directory):
            os.makedirs(directory)
        else:
            files = glob.glob(directory)
            for f in files:
                if os.path.isfile(f):
                    os.remove(f)

    def testScanned44PdfPageForNetwork(self):
        pdfSeparate = PdfSeparate('tests/sample-scanned-44pages.pdf', self.indir)
        pdfSeparate.extractPages()
        self.assertTrue(os.path.isfile(os.path.join(self.indir,"1.pdf")))

        try:
            abbyyPdf = AbbyyPdfTextExtractor(self.indir, self.outdir, 44, "english")
            abbyyPdf.setApplicationCredentials(self.configParser.get('abbyy','appid'), self.configParser.get('abbyy','password'))
            abbyyPdf.extractPages();
            self.assertTrue(os.path.isfile(os.path.join(self.outdir,"1.txt")))
            self.assertTrue(os.path.isfile(os.path.join(self.outdir,"44.txt")))
        except Exception:
            pass