diff options
Diffstat (limited to 'main.py')
| -rw-r--r-- | main.py | 9 |
1 files changed, 5 insertions, 4 deletions
| @@ -3,6 +3,7 @@ from shutil import copyfile | |||
| 3 | import argparse | 3 | import argparse |
| 4 | 4 | ||
| 5 | from src.parsers import pdf | 5 | from src.parsers import pdf |
| 6 | from src import parser_factory | ||
| 6 | 7 | ||
| 7 | 8 | ||
| 8 | def create_arg_parser(): | 9 | def create_arg_parser(): |
| @@ -19,7 +20,7 @@ def create_arg_parser(): | |||
| 19 | return parser | 20 | return parser |
| 20 | 21 | ||
| 21 | def show_meta(file_name:str): | 22 | def 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") |
