summaryrefslogtreecommitdiff
path: root/main.py
diff options
context:
space:
mode:
Diffstat (limited to 'main.py')
-rw-r--r--main.py9
1 files changed, 5 insertions, 4 deletions
diff --git a/main.py b/main.py
index e4157e6..4b965b4 100644
--- a/main.py
+++ b/main.py
@@ -3,6 +3,7 @@ from shutil import copyfile
3import argparse 3import argparse
4 4
5from src.parsers import pdf 5from src.parsers import pdf
6from src import parser_factory
6 7
7 8
8def create_arg_parser(): 9def create_arg_parser():
@@ -19,7 +20,7 @@ def create_arg_parser():
19 return parser 20 return parser
20 21
21def show_meta(file_name:str): 22def show_meta(file_name:str):
22 p = pdf.PDFParser(file_name) 23 p = parser_factory(file_name)
23 for k,v in p.get_meta().items(): 24 for k,v in p.get_meta().items():
24 print("%s: %s" % (k, v)) 25 print("%s: %s" % (k, v))
25 26
@@ -32,10 +33,10 @@ def main():
32 show_meta(f) 33 show_meta(f)
33 return 0 34 return 0
34 elif not args.files: 35 elif not args.files:
35 return parser.show_help() 36 return argparser.show_help()
36 37
37 copyfile(sys.argv[1] + '.bak', sys.argv[1]) 38 #p = pdf.PDFParser(sys.argv[1])
38 p = pdf.PDFParser(sys.argv[1]) 39 p = parser_factory.get_parser(sys.argv[1])
39 p.remove_all() 40 p.remove_all()
40 p = pdf.PDFParser('OUT_clean.pdf') 41 p = pdf.PDFParser('OUT_clean.pdf')
41 print("ok") 42 print("ok")