Merge pull request #1 from keraattin/development

Header extracting added

email-analyzer.py

@@ -7,6 +7,17 @@
 
 SUPPORTED_FILE_TYPES = ["eml"]
 
+def get_headers(mail_data : str):
+    '''Get & Print Headers from mail data'''
+    print(pyfiglet.figlet_format("Headers")) # Print Banner
+    # Get Headers from mail data
+    headers = HeaderParser().parsestr(mail_data, headersonly=True)
+    # Print Headers
+    for key,val in headers.items():
+        print("_"*70)
+        print(key+":")
+        print(val)
+        print("_"*70)
 
 # Main
 if __name__ == '__main__':
@@ -18,8 +29,15 @@
         help="Name of file", 
         required=True
     )
+    parser.add_argument(
+        "-H",
+        "--headers",
+        help="Headers of the eml file", 
+        required=False,
+        action="store_true"
+    )
     args = parser.parse_args()
-    
+
     # Filename
     if args.filename:
         # Get Filename
@@ -31,4 +49,9 @@
             sys.exit(-1) #Exit with error code
 
     with open(filename,"r",encoding="utf-8") as file:
-        data = file.read().rstrip(
+        data = file.read().rstrip()
+
+    # Headers
+    if args.headers:
+        # Get & Print Headers
+        get_headers(data)