Browse Source

drop banner in text part too if found and insert at end

Pi 1 year ago
parent
commit
33d2b791c5
1 changed files with 5 additions and 2 deletions
  1. 5 2
      mime-inject.py

+ 5 - 2
mime-inject.py

@@ -48,8 +48,11 @@ def inject(m):
     e = m.get('Content-Transfer-Encoding','').lower()
     payload = m.get_payload(decode=True).decode(m.get_content_charset("utf-8"))
 
-    if c == 'text/plain' and txt not in payload:
-        payload += f"\n{txt}\n"
+    if c == 'text/plain':
+        if txt.strip() in payload:
+           payload = payload.replace(txt.strip(), '')
+
+        payload = f"{payload.strip()}\n\n{txt}\n"
 
     elif c == 'text/html':
         h = HTML.fromstring(payload)