Compare commits
2 Commits
8e408554d5
...
e2fcedf2c9
Author | SHA1 | Date | |
---|---|---|---|
e2fcedf2c9 | |||
8a06d7b7e8 |
16
main.py
16
main.py
@ -10,6 +10,7 @@ import datetime
|
|||||||
import logging
|
import logging
|
||||||
import socks
|
import socks
|
||||||
import socket
|
import socket
|
||||||
|
import re
|
||||||
|
|
||||||
from email.header import decode_header
|
from email.header import decode_header
|
||||||
|
|
||||||
@ -20,14 +21,14 @@ def decode_mime_words(s):
|
|||||||
for fragment, encoding in decoded_fragments
|
for fragment, encoding in decoded_fragments
|
||||||
)
|
)
|
||||||
|
|
||||||
def add_mask(original_msg, content):
|
def add_mask(original_msg, content, is_html):
|
||||||
original_subject = decode_mime_words(original_msg['Subject'])
|
original_subject = decode_mime_words(original_msg['Subject'])
|
||||||
from_name, from_address = parseaddr(original_msg['From'])
|
from_name, from_address = parseaddr(original_msg['From'])
|
||||||
from_name = decode_mime_words(from_name)
|
from_name = decode_mime_words(from_name)
|
||||||
to_name, to_address = parseaddr(original_msg['To'])
|
to_name, to_address = parseaddr(original_msg['To'])
|
||||||
to_name = decode_mime_words(to_name)
|
to_name = decode_mime_words(to_name)
|
||||||
header = f"""<table align=center style="background:#3d3d3d;padding:8px 16px;margin-top:30px;margin-bottom:30px;width:96%;border-radius:6px;max-width:1200px"width=100% bgcolor=#3D3D3D><tr><td style=font-size:xx-large;font-weight:bolder;color:#fff width=50% align=left>Forwarded Email<td style=color:#fff;text-align:right width=50% align=right><p>From: {from_name} <{from_address}><p>To: {to_name} <{to_address}><p>Subject: {original_subject}</table><table align=center style=padding:0;max-width:850px width=100%><tr><td style=padding-left:15px;padding-right:15px width=100%>"""
|
header = f"""<html><head></head><body><table style="background:#3d3d3d;padding:8px 16px;margin-top:30px;margin-bottom:30px;width:96%;border-radius:6px;max-width:1200px" width="100%" bgcolor="#3D3D3D" align="center"><tbody><tr><td width="50%" align="left" style="font-weight:bolder;color:#fff"><p style="font-size:x-large;margin-block:.2em">Forwarded Email</p>{'' if is_html else '<p style="font-size: medium; margin-block: 0.2em;">This email is plain text, it may have display issues</p>'}</td><td width="50%" align="right" style="color:#fff;text-align:right"><p style="margin-block:.1em">From: {from_name} <{from_address}></p><p style="margin-block:.1em">To: {to_name} <{to_address}></p><p style="margin-block:.1em">Subject: {original_subject}</p></td></tr></tbody></table><table style="padding:0;max-width:850px" width="100%" align="center"><tbody><tr><td style="padding-left:15px;padding-right:15px" width="100%">"""
|
||||||
footer = f"""<table align=center bgcolor=#3D3D3D style="background:#3d3d3d;padding:8px 16px;margin-top:30px;margin-bottom:30px;width:96%;border-radius:6px;max-width:1200px"width=100%><tr><td align=left style=color:#d22;font-size:xx-large;font-weight:bolder width=50%>FORWARDED<td align=right style=color:#fff;text-align:left width=50%><p style=font-size:x-large;font-weight:700;margin-block:0>Notice:<p style=margin-block:.1em> This is a automatically forwarded email, which means it may contains something bad.<p style=margin-block:.1em> You shouldn't reply directly to this email, it will never reach your destination!</table>"""
|
footer = f"""</td></tr></tbody></table><table style="background:#3d3d3d;padding:8px 16px;margin-top:30px;margin-bottom:30px;width:96%;border-radius:6px;max-width:1200px" width="100%" bgcolor="#3D3D3D" align="center"><tbody><tr><td width="50%" align="left" style="color:#d22;font-size:xx-large;font-weight:bolder">FORWARDED</td><td width="50%" align="right" style="color:#fff;text-align:left"><p style="font-size:x-large;font-weight:700;margin-block:0">Notice:</p><p style="margin-block:.1em"> This is a automatically forwarded email, which means it may contains something bad.</p><p style="margin-block:.1em"> You shouldn't reply directly to this email, it will never reach your destination!</p></td></tr></tbody></table></body></html>"""
|
||||||
return header + content + footer
|
return header + content + footer
|
||||||
|
|
||||||
def load_config(config_file='config.json'):
|
def load_config(config_file='config.json'):
|
||||||
@ -48,7 +49,7 @@ def setup_logging(filename):
|
|||||||
chlr = logging.StreamHandler()
|
chlr = logging.StreamHandler()
|
||||||
chlr.setFormatter(formatter)
|
chlr.setFormatter(formatter)
|
||||||
chlr.setLevel('INFO')
|
chlr.setLevel('INFO')
|
||||||
fhlr = logging.FileHandler(filename)
|
fhlr = logging.FileHandler(filename, encoding='utf-8')
|
||||||
fhlr.setFormatter(formatter)
|
fhlr.setFormatter(formatter)
|
||||||
logger.addHandler(chlr)
|
logger.addHandler(chlr)
|
||||||
logger.addHandler(fhlr)
|
logger.addHandler(fhlr)
|
||||||
@ -133,7 +134,12 @@ def forward_emails(account_config, emails, logger):
|
|||||||
logger.error(f"Failed to extract body from email {email_id}")
|
logger.error(f"Failed to extract body from email {email_id}")
|
||||||
continue
|
continue
|
||||||
|
|
||||||
html_content = add_mask(original_msg, body)
|
is_html = bool(re.compile(r'<[^>]+>').search(body))
|
||||||
|
|
||||||
|
if not is_html:
|
||||||
|
body = body.replace('\n', '<br>')
|
||||||
|
|
||||||
|
html_content = add_mask(original_msg, body, is_html)
|
||||||
msg.attach(MIMEText(html_content, 'html'))
|
msg.attach(MIMEText(html_content, 'html'))
|
||||||
|
|
||||||
for attachment in attachments:
|
for attachment in attachments:
|
||||||
|
149
template.html
149
template.html
@ -1,71 +1,78 @@
|
|||||||
<table
|
<html>
|
||||||
style="
|
<head></head>
|
||||||
background: #3d3d3d;
|
<body>
|
||||||
padding: 8px 16px;
|
<table
|
||||||
margin-top: 30px;
|
style="
|
||||||
margin-bottom: 30px;
|
background: #3d3d3d;
|
||||||
width: 96%;
|
padding: 8px 16px;
|
||||||
border-radius: 6px;
|
margin-top: 30px;
|
||||||
max-width: 1200px;
|
margin-bottom: 30px;
|
||||||
"
|
width: 96%;
|
||||||
width="100%"
|
border-radius: 6px;
|
||||||
bgcolor="#3D3D3D"
|
max-width: 1200px;
|
||||||
align="center">
|
"
|
||||||
<tbody>
|
width="100%"
|
||||||
<tr>
|
bgcolor="#3D3D3D"
|
||||||
<td
|
align="center">
|
||||||
width="50%"
|
<tbody>
|
||||||
align="left"
|
<tr>
|
||||||
style="
|
<td width="50%" align="left" style="font-weight: bolder; color: #ffffff">
|
||||||
font-size: xx-large;
|
<p style="font-size: x-large; margin-block: 0.2em">Forwarded Email</p>
|
||||||
font-weight: bolder;
|
<!-- <p style="font-size: medium; margin-block: 0.2em;">This email is plain text, it may have display issues</p> -->
|
||||||
color: #ffffff;
|
</td>
|
||||||
">
|
<td width="50%" align="right" style="color: #ffffff; text-align: right">
|
||||||
Forwarded Email
|
<p style="margin-block: 0.1em">From: {from_name} <{from_address}></p>
|
||||||
</td>
|
<p style="margin-block: 0.1em">To: {to_name} <{to_address}></p>
|
||||||
<td width="50%" align="right" style="color: #ffffff; text-align: right;">
|
<p style="margin-block: 0.1em">Subject: {original_subject}</p>
|
||||||
<p>From: {from_name} <{from_address}></p>
|
</td>
|
||||||
<p>To: {to_name} <{to_address}></p>
|
</tr>
|
||||||
<p>Subject: {original_subject}</p>
|
</tbody>
|
||||||
</td>
|
</table>
|
||||||
</tr>
|
<table style="padding: 0; max-width: 850px" width="100%" align="center">
|
||||||
</tbody>
|
<tbody>
|
||||||
</table>
|
<tr>
|
||||||
<table style="padding: 0; max-width: 850px" width="100%" align="center">
|
<td style="padding-left: 15px; padding-right: 15px" width="100%">
|
||||||
<tbody>
|
<!-- content -->
|
||||||
<tr>
|
</td>
|
||||||
<td style="padding-left: 15px; padding-right: 15px" width="100%">
|
</tr>
|
||||||
<!-- content -->
|
</tbody>
|
||||||
</td>
|
</table>
|
||||||
</tr>
|
<table
|
||||||
</tbody>
|
style="
|
||||||
</table>
|
background: #3d3d3d;
|
||||||
<table
|
padding: 8px 16px;
|
||||||
style="
|
margin-top: 30px;
|
||||||
background: #3d3d3d;
|
margin-bottom: 30px;
|
||||||
padding: 8px 16px;
|
width: 96%;
|
||||||
margin-top: 30px;
|
border-radius: 6px;
|
||||||
margin-bottom: 30px;
|
max-width: 1200px;
|
||||||
width: 96%;
|
"
|
||||||
border-radius: 6px;
|
width="100%"
|
||||||
max-width: 1200px;
|
bgcolor="#3D3D3D"
|
||||||
"
|
align="center">
|
||||||
width="100%"
|
<tbody>
|
||||||
bgcolor="#3D3D3D"
|
<tr>
|
||||||
align="center">
|
<td
|
||||||
<tbody>
|
width="50%"
|
||||||
<tr>
|
align="left"
|
||||||
<td
|
style="color: #dd2222; font-size: xx-large; font-weight: bolder">
|
||||||
width="50%"
|
FORWARDED
|
||||||
align="left"
|
</td>
|
||||||
style="color: #dd2222; font-size: xx-large; font-weight: bolder">
|
<td width="50%" align="right" style="color: #ffffff; text-align: left">
|
||||||
FORWARDED
|
<p style="font-size: x-large; font-weight: bold; margin-block: 0">
|
||||||
</td>
|
Notice:
|
||||||
<td width="50%" align="right" style="color: #ffffff; text-align: left;">
|
</p>
|
||||||
<p style="font-size: x-large; font-weight: bold; margin-block: 0;">Notice:</p>
|
<p style="margin-block: 0.1em">
|
||||||
<p style="margin-block: 0.1em;"> This is a automatically forwarded email, which means it may contains something bad.</p>
|
 This is a automatically forwarded email, which means it may
|
||||||
<p style="margin-block: 0.1em;"> You shouldn't reply directly to this email, it will never reach your destination!</p>
|
contains something bad.
|
||||||
</td>
|
</p>
|
||||||
</tr>
|
<p style="margin-block: 0.1em">
|
||||||
</tbody>
|
 You shouldn't reply directly to this email, it will never reach
|
||||||
</table>
|
your destination!
|
||||||
|
</p>
|
||||||
|
</td>
|
||||||
|
</tr>
|
||||||
|
</tbody>
|
||||||
|
</table>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
|
Loading…
x
Reference in New Issue
Block a user