From 27899241119a25c2b57669de4fe67d74c8b0ad9f Mon Sep 17 00:00:00 2001 From: infosecB Date: Tue, 4 Feb 2020 09:16:02 -0500 Subject: [PATCH] updated for use with Python 3 --- normalize_atpsafelink.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/normalize_atpsafelink.py b/normalize_atpsafelink.py index f70d822..d89a345 100644 --- a/normalize_atpsafelink.py +++ b/normalize_atpsafelink.py @@ -1,21 +1,21 @@ from argparse import ArgumentParser -import urllib +from urllib import parse def normalize_atpsafelink(safelink_url): safelink_url = safelink_url.split("=")[1] - safelink_url = urllib.unquote(safelink_url).decode('utf8') - safelink_url = safelink_url.replace('&data', '') + safelink_url = parse.unquote(safelink_url) + safelink_url = safelink_url.split('&')[0] return safelink_url def main(url): - print normalize_atpsafelink(url) + print(normalize_atpsafelink(url)) if __name__ == "__main__": args = ArgumentParser() - args.add_argument('-u', '--url', help="Enter the URL.", dest='url') + args.add_argument('-u', '--url', help="Enter the URL.", dest='url', required=True) args = args.parse_args() - + main(args.url)