Compare commits
13 Commits
5ba5040513
...
main
| Author | SHA1 | Date | |
|---|---|---|---|
f49138614d
|
|||
|
ec2b4aaa29
|
|||
|
76b36d6141
|
|||
|
0ce2c5fb24
|
|||
|
e54d35a9d7
|
|||
|
993ca50352
|
|||
|
936cffd0ed
|
|||
|
d7d2b86a9f
|
|||
|
e2fcedf2c9
|
|||
|
8a06d7b7e8
|
|||
|
8e408554d5
|
|||
|
4b4121ebd8
|
|||
|
f33e2caabd
|
4
.gitignore
vendored
4
.gitignore
vendored
@ -1 +1,3 @@
|
||||
config.json
|
||||
config.json
|
||||
*.log
|
||||
.vscode/
|
||||
@ -38,6 +38,7 @@ pip install pysocks
|
||||
|
||||
```json
|
||||
{
|
||||
"log": "output.log",
|
||||
"check_interval": 60,
|
||||
"accounts": [
|
||||
{
|
||||
@ -70,6 +71,7 @@ pip install pysocks
|
||||
|
||||
### 配置项说明
|
||||
|
||||
- `log`: 日志文件路径。
|
||||
- `check_interval`: 检查邮件的时间间隔，单位为秒。
|
||||
- `accounts`: 配置多个邮箱账户。
|
||||
- `email`: 邮箱地址。
|
||||
|
||||
@ -1,4 +1,5 @@
|
||||
{
|
||||
"log": "output.log",
|
||||
"accounts": [
|
||||
{
|
||||
"enabled": false,
|
||||
|
||||
135
main.py
135
main.py
@ -1,3 +1,4 @@
|
||||
# pylint: disable=C0301, C0114
|
||||
import json
|
||||
import imaplib
|
||||
import smtplib
|
||||
@ -5,106 +6,137 @@ import email
|
||||
from email.mime.multipart import MIMEMultipart
|
||||
from email.mime.text import MIMEText
|
||||
from email.utils import parseaddr
|
||||
import time
|
||||
import logging
|
||||
import socks
|
||||
import socket
|
||||
|
||||
from email.header import decode_header
|
||||
import time
|
||||
import datetime
|
||||
import logging
|
||||
import socket
|
||||
import re
|
||||
import socks
|
||||
|
||||
def provider_in(address, provider_list):
|
||||
"""Check if the provider is in the provider list"""
|
||||
for provider in provider_list:
|
||||
if address.endswith(provider):
|
||||
return True
|
||||
return False
|
||||
|
||||
def decode_mime_words(s):
|
||||
"""Decodes a string containing multiple MIME words"""
|
||||
decoded_fragments = decode_header(s)
|
||||
return ''.join(
|
||||
str(fragment, encoding or 'utf-8') if isinstance(fragment, bytes) else fragment
|
||||
for fragment, encoding in decoded_fragments
|
||||
)
|
||||
|
||||
def add_mask(original_msg, content):
|
||||
def add_mask(original_msg, content, is_html):
|
||||
"""Adds a mask to the content"""
|
||||
original_subject = decode_mime_words(original_msg['Subject'])
|
||||
from_name, from_address = parseaddr(original_msg['From'])
|
||||
from_name = decode_mime_words(from_name)
|
||||
to_name, to_address = parseaddr(original_msg['To'])
|
||||
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 id=relay-email-header><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%>"""
|
||||
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>"""
|
||||
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: 0.1em">From: {from_name} <<a href="mailto:{from_address}">{from_address}</a>></p><p style="margin-block: 0.1em">To: {to_name} <<a href="mailto:{to_address}">{to_address}</a>></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 = """</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
|
||||
|
||||
def load_config(config_file='config.json'):
|
||||
with open(config_file, 'r') as file:
|
||||
"""Loads the configuration file"""
|
||||
with open(config_file, 'r', encoding='utf-8') as file:
|
||||
config = json.load(file)
|
||||
return config
|
||||
|
||||
def set_proxy(proxy_config):
|
||||
"""Sets the proxy"""
|
||||
socks.setdefaultproxy(socks.PROXY_TYPE_SOCKS5, proxy_config['server'], proxy_config['port'])
|
||||
socket.socket = socks.socksocket
|
||||
|
||||
def setup_logging():
|
||||
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
|
||||
def setup_logging(filename):
|
||||
"""Sets up the logging"""
|
||||
logger = logging.getLogger()
|
||||
logger.setLevel('DEBUG')
|
||||
basic_format = "%(asctime)s >> %(levelname)s - %(message)s"
|
||||
date_format = '%Y-%m-%d %H:%M:%S'
|
||||
formatter = logging.Formatter(basic_format, date_format)
|
||||
chlr = logging.StreamHandler()
|
||||
chlr.setFormatter(formatter)
|
||||
chlr.setLevel('INFO')
|
||||
fhlr = logging.FileHandler(filename, encoding='utf-8')
|
||||
fhlr.setFormatter(formatter)
|
||||
logger.addHandler(chlr)
|
||||
logger.addHandler(fhlr)
|
||||
return logger
|
||||
|
||||
def get_unforwarded_emails(account_config, logger):
|
||||
"""Gets the unforwarded emails"""
|
||||
if account_config['proxy']['enabled']:
|
||||
set_proxy(account_config['proxy'])
|
||||
|
||||
|
||||
if account_config['imap'].get('use_ssl', True):
|
||||
imap = imaplib.IMAP4_SSL(account_config['imap']['server'], account_config['imap']['port'])
|
||||
imap = imaplib.IMAP4_SSL(account_config['imap']['server'], account_config['imap']['port'], timeout=10)
|
||||
else:
|
||||
imap = imaplib.IMAP4(account_config['imap']['server'], account_config['imap']['port'])
|
||||
imap = imaplib.IMAP4(account_config['imap']['server'], account_config['imap']['port'], timeout=10)
|
||||
|
||||
imap.login(account_config['email'], account_config['password'])
|
||||
|
||||
if "163.com" in account_config['email'] or "126.com" in account_config['email']:
|
||||
imaplib.Commands['ID'] = ('AUTH')
|
||||
|
||||
if provider_in(account_config['email'], ["163.com", "126.com"]):
|
||||
imaplib.Commands['ID'] = 'AUTH'
|
||||
args = ("name","XXXX","contact","XXXX@163.com","version","1.0.0","vendor","myclient")
|
||||
typ, dat = imap._simple_command('ID', '("' + '" "'.join(args) + '")')
|
||||
imap._simple_command('ID', '("' + '" "'.join(args) + '")') # pylint: disable=W0212
|
||||
|
||||
imap.select()
|
||||
|
||||
status, messages = imap.search(None, 'UNSEEN')
|
||||
|
||||
_, messages = imap.search(None, 'UNSEEN')
|
||||
email_ids = messages[0].split()
|
||||
|
||||
|
||||
emails = []
|
||||
for email_id in email_ids:
|
||||
status, data = imap.fetch(email_id, '(RFC822)')
|
||||
_, data = imap.fetch(email_id, '(RFC822)')
|
||||
for response_part in data:
|
||||
if isinstance(response_part, tuple):
|
||||
msg = email.message_from_bytes(response_part[1])
|
||||
if 'Forwarded' not in msg['Subject']:
|
||||
emails.append((email_id, msg))
|
||||
msg = email.message_from_bytes(response_part[1]) # pylint: disable=E1136
|
||||
emails.append((email_id, msg))
|
||||
imap.store(email_id, '+FLAGS', '\\Seen')
|
||||
|
||||
imap.logout()
|
||||
logger.info(f"Retrieved {len(emails)} new emails from {account_config['email']}")
|
||||
if len(emails) > 0:
|
||||
logger.info(f"Retrieved {len(emails)} new emails from {account_config['email']}")
|
||||
return emails
|
||||
|
||||
def forward_emails(account_config, emails, logger):
|
||||
"""Forwards the emails"""
|
||||
if account_config['proxy']['enabled']:
|
||||
set_proxy(account_config['proxy'])
|
||||
|
||||
|
||||
smtp = None
|
||||
if account_config['smtp'].get('use_ssl', False):
|
||||
smtp = smtplib.SMTP_SSL(account_config['smtp']['server'], account_config['smtp']['port'])
|
||||
smtp = smtplib.SMTP_SSL(account_config['smtp']['server'], account_config['smtp']['port'], timeout=10)
|
||||
else:
|
||||
smtp = smtplib.SMTP(account_config['smtp']['server'], account_config['smtp']['port'])
|
||||
smtp = smtplib.SMTP(account_config['smtp']['server'], account_config['smtp']['port'], timeout=10)
|
||||
if account_config['smtp'].get('use_starttls', False):
|
||||
smtp.starttls()
|
||||
|
||||
|
||||
smtp.login(account_config['email'], account_config['password'])
|
||||
|
||||
|
||||
for email_id, original_msg in emails:
|
||||
for recipient in account_config['forward']['to']:
|
||||
from_name, from_address = parseaddr(original_msg['From'])
|
||||
from_name = decode_mime_words(from_name)
|
||||
to_name, to_address = parseaddr(original_msg['To'])
|
||||
to_name = decode_mime_words(to_name)
|
||||
msg = MIMEMultipart('mixed')
|
||||
msg['From'] = account_config['email']
|
||||
msg['To'] = recipient
|
||||
msg['From'] = f'"{from_name} ({from_address}) via Forwarder" <{account_config["email"]}>'
|
||||
msg['To'] = f'"{to_name} ({to_address}) via Forwarder" <{recipient}>'
|
||||
original_subject = decode_mime_words(original_msg['Subject'])
|
||||
msg['Subject'] = original_subject
|
||||
|
||||
|
||||
body = ""
|
||||
attachments = []
|
||||
if original_msg.is_multipart():
|
||||
for part in original_msg.walk():
|
||||
content_type = part.get_content_type()
|
||||
content_disposition = str(part.get("Content-Disposition"))
|
||||
|
||||
|
||||
if "attachment" in content_disposition or part.get_filename():
|
||||
attachments.append(part)
|
||||
elif content_type == 'text/html':
|
||||
@ -113,12 +145,17 @@ def forward_emails(account_config, emails, logger):
|
||||
body = part.get_payload(decode=True).decode()
|
||||
else:
|
||||
body = original_msg.get_payload(decode=True).decode()
|
||||
|
||||
|
||||
if not body:
|
||||
logger.error(f"Failed to extract body from email {email_id}")
|
||||
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'))
|
||||
|
||||
for attachment in attachments:
|
||||
@ -127,16 +164,17 @@ def forward_emails(account_config, emails, logger):
|
||||
filename = decode_mime_words(filename)
|
||||
attachment.add_header('Content-Disposition', 'attachment', filename=filename)
|
||||
msg.attach(attachment)
|
||||
|
||||
|
||||
smtp.sendmail(account_config['email'], recipient, msg.as_string())
|
||||
logger.info(f"Forwarded email {original_subject} to {recipient}")
|
||||
|
||||
logger.info(f"Forwarded email {original_subject} from {account_config['email']} to {recipient}")
|
||||
|
||||
smtp.quit()
|
||||
|
||||
def main():
|
||||
logger = setup_logging()
|
||||
"""Main function"""
|
||||
config = load_config()
|
||||
|
||||
logger = setup_logging(config['log'])
|
||||
|
||||
while True:
|
||||
for account in config['accounts']:
|
||||
if account['enabled']:
|
||||
@ -144,13 +182,10 @@ def main():
|
||||
emails = get_unforwarded_emails(account, logger)
|
||||
if emails:
|
||||
forward_emails(account, emails, logger)
|
||||
else:
|
||||
logger.info(f"No new emails to forward for {account['email']}.")
|
||||
except Exception as e:
|
||||
logger.error(f"Error processing account {account['email']}: {str(e)}")
|
||||
else:
|
||||
logger.info(f"Account {account['email']} is disabled.")
|
||||
|
||||
except Exception as e: # pylint: disable=W0718
|
||||
logger.error(f"Error processing account {account['email']}: {str(e)}") # pylint: disable=W1203
|
||||
|
||||
logger.info(datetime.datetime.now().strftime("Check finished at %Y-%m-%d %H:%M:%S"))
|
||||
time.sleep(config.get('check_interval', 60))
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
150
template.html
150
template.html
@ -1,72 +1,78 @@
|
||||
<table
|
||||
id="relay-email-header"
|
||||
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-size: xx-large;
|
||||
font-weight: bolder;
|
||||
color: #ffffff;
|
||||
">
|
||||
Forwarded Email
|
||||
</td>
|
||||
<td width="50%" align="right" style="color: #ffffff; text-align: right;">
|
||||
<p>From: {from_name} <{from_address}></p>
|
||||
<p>To: {to_name} <{to_address}></p>
|
||||
<p>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%">
|
||||
<!-- content -->
|
||||
</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: #dd2222; font-size: xx-large; font-weight: bolder">
|
||||
FORWARDED
|
||||
</td>
|
||||
<td width="50%" align="right" style="color: #ffffff; text-align: left;">
|
||||
<p style="font-size: x-large; font-weight: bold; margin-block: 0;">Notice:</p>
|
||||
<p style="margin-block: 0.1em;"> This is a automatically forwarded email, which means it may contains something bad.</p>
|
||||
<p style="margin-block: 0.1em;"> You shouldn't reply directly to this email, it will never reach your destination!</p>
|
||||
</td>
|
||||
</tr>
|
||||
</tbody>
|
||||
</table>
|
||||
<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: #ffffff">
|
||||
<p style="font-size: x-large; margin-block: 0.2em">Forwarded Email</p>
|
||||
<!-- <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: #ffffff; text-align: right">
|
||||
<p style="margin-block: 0.1em">From: {from_name} <<a href="mailto:{from_address}">{from_address}</a>></p>
|
||||
<p style="margin-block: 0.1em">To: {to_name} <<a href="mailto:{to_address}">{to_address}</a>></p>
|
||||
<p style="margin-block: 0.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%">
|
||||
<!-- content -->
|
||||
</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: #dd2222; font-size: xx-large; font-weight: bolder">
|
||||
FORWARDED
|
||||
</td>
|
||||
<td width="50%" align="right" style="color: #ffffff; text-align: left">
|
||||
<p style="font-size: x-large; font-weight: bold; margin-block: 0">
|
||||
Notice:
|
||||
</p>
|
||||
<p style="margin-block: 0.1em">
|
||||
 This is a automatically forwarded email, which means it may
|
||||
contains something bad.
|
||||
</p>
|
||||
<p style="margin-block: 0.1em">
|
||||
 You shouldn't reply directly to this email, it will never reach
|
||||
your destination!
|
||||
</p>
|
||||
</td>
|
||||
</tr>
|
||||
</tbody>
|
||||
</table>
|
||||
</body>
|
||||
</html>
|
||||
|
||||
Reference in New Issue
Block a user