Update modules.py

modules.py

@@ -14,7 +14,7 @@ def get_text_from_html(html_content):
     return all_text
 # get text content type from email
 def get_text(file):
-    message = email.message_from_bytes(file.read())
+    message = email.message_from_bytes(file)
     text_content = ""
     for part in message.walk():
         if part.get_content_type() == 'text/plain':
@@ -25,7 +25,7 @@ def get_text(file):
         return get_text_from_html(get_html_general(file));
 
 def get_email_html(file):
-    content = email.message_from_bytes(file.read())
+    content = email.message_from_bytes(file)
     html_content = ""
     for part in content.walk():
         if part.get_content_type() == 'text/html':
@@ -93,7 +93,7 @@ def check_popWindow(file):
         return False
 
 def check_spf(file):
-    message = email.message_from_bytes(file.read())
+    message = email.message_from_bytes(file)
     received_spf_header = message.get('Received-SPF')
     if received_spf_header == None:
       return 0
@@ -110,7 +110,7 @@ def check_spf(file):
     else:
         return 0
 def check_dkim(file):
-    message = email.message_from_bytes(file.read())
+    message = email.message_from_bytes(file)
     auth = message.get('Authentication-Results')
     if auth == None:
       return 0
@@ -122,7 +122,7 @@ def check_dkim(file):
     else:
       return 0
 def check_dmarc(file):
-    message = email.message_from_bytes(file.read())
+    message = email.message_from_bytes(file)
     auth = message.get('Authentication-Results')
     if auth == None:
       return 0
@@ -134,7 +134,7 @@ def check_dmarc(file):
     else:
       return 0
 def check_deliver_receiver(file):
-    message = email.message_from_bytes(file.read())
+    message = email.message_from_bytes(file)
     deliver = message.get('Delivered-To')
     # print(deliver)
     receiver = message.get('To')
@@ -144,7 +144,7 @@ def check_deliver_receiver(file):
     else:
       return 0
 def check_encript(file):
-    message = email.message_from_bytes(file.read())
+    message = email.message_from_bytes(file)
     received_headers = message.get_all('Received')
     # print(received_headers)
     version_string = 'version'
@@ -178,7 +178,7 @@ def get_urls_from_html(html_content):
                 urls += [href]
     return urls
 def get_text(file):
-        message = email.message_from_bytes(file.read())
+        message = email.message_from_bytes(file)
         text_content = ""
         for part in message.walk():
             if part.get_content_type() == 'text/plain':
@@ -231,7 +231,7 @@ def get_num_FunctionWords(file):
 
 
 def get_email_html(file):
-    content = email.message_from_bytes(file.read())
+    content = email.message_from_bytes(file)
     html_content = ""
     for part in content.walk():
         if part.get_content_type() == 'text/html':
@@ -249,7 +249,7 @@ def get_num_sbj(file):
     count = len(get_subject(file).split())
     return count
 def get_subject(file):
-    message = email.message_from_bytes(file.read())
+    message = email.message_from_bytes(file)
     headers = message.items()
     # Print the headers
     subject = ""
@@ -265,7 +265,7 @@ def get_subject(file):
 
 
 def get_sender(file):
-    message = email.message_from_bytes(file.read())
+    message = email.message_from_bytes(file)
     headers = message.items()
     # Print the headers
     sender = ""