pdf_to_txt.py

from convert import Converter
from reader import Reader
from writer import Writer
from PyPDF2 import PdfFileReader
import os
import sys

class pdf_to_txt_obj():
	def __init__(self,file_name,target_name,path):
		self.file_name = file_name
		self.target_name = target_name
		self.path = path
		try:
			reader = PdfFileReader(open(self.path+"/"+file_name+".pdf",'rb'))
			self.pg_count = reader.getNumPages()
		except:
			raise ValueError("Could not find the file named {} in {}.".format(self.file_name,self.path))
			sys.exit()

	def convert_to_txt(self):
		converter = Converter(self.file_name+".pdf",self.target_name,self.path)
		converter.convert()
		reader = Reader(self.target_name,self.pg_count,self.path)
		extracted_text = reader.get_text()
		extracted_text = os.linesep.join([s for s in extracted_text.splitlines() if s])
		writer = Writer((self.target_name+".txt"),extracted_text,self.path)
		writer.write()