1
0
mirror of synced 2024-06-12 17:19:51 +02:00
mat2/main.py
2018-03-06 23:20:18 +01:00

47 lines
1.3 KiB
Python

import sys
from shutil import copyfile
import argparse
from src.parsers import pdf
def create_arg_parser():
parser = argparse.ArgumentParser(description='Metadata anonymisation toolkit 2')
parser.add_argument('files', nargs='*')
info = parser.add_argument_group('Information')
info.add_argument('-c', '--check', action='store_true',
help='check if a file is free of harmful metadatas')
info.add_argument('-l', '--list', action='store_true',
help='list all supported fileformats')
info.add_argument('-s', '--show', action='store_true',
help='list all the harmful metadata of a file without removing them')
return parser
def show_meta(file_name:str):
p = pdf.PDFParser(file_name)
for k,v in p.get_meta().items():
print("%s: %s" % (k, v))
def main():
argparser = create_arg_parser()
args = argparser.parse_args()
if args.show:
for f in args.files:
show_meta(f)
return 0
elif not args.files:
return parser.show_help()
copyfile(sys.argv[1] + '.bak', sys.argv[1])
p = pdf.PDFParser(sys.argv[1])
p.remove_all()
p = pdf.PDFParser('OUT_clean.pdf')
print("ok")
if __name__ == '__main__':
main()