PyPDF2 support for #30

xpmethod · May 26, 2014 · 7e11bda · 7e11bda
1 parent c9c36ce
commit 7e11bda
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/extractor_research/extractors/pdf2.py b/extractor_research/extractors/pdf2.py
@@ -0,0 +1,16 @@
+#!/usr/bin/python
+from PyPDF2 import PdfFileReader
+
+class PDF2:
+    def __init__(self, pdf_file, txt_file):
+        self.doc = PdfFileReader(open(pdf_file, 'rb'))
+        self.output = open(txt_file, 'w')
+
+    def extract(self):
+        for page in self.doc.pages:
+            self.output.write(page.extractText())
+        self.output.close()
+
+if __name__ == '__main__': 
+    pdf = PDF2('../input/pride_and_prej/1.pdf', '../output/pride_and_prej/pdf2/1.txt')
+    pdf.extract()
diff --git a/extractor_research/output/pride_and_prej/pdf2/1.txt b/extractor_research/output/pride_and_prej/pdf2/1.txt