updated for use with Python 3

infosecB · Feb 4, 2020 · 2789924 · 2789924
1 parent d5019c7
commit 2789924
Showing 1 changed file with 6 additions and 6 deletions.
diff --git 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('&amp')[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)