From ec2b4aaa29e13ae61abee1164add1ca2bcb5bb82 Mon Sep 17 00:00:00 2001 From: Zichao Lin Date: Fri, 26 Jul 2024 13:32:03 +0800 Subject: [PATCH] fix: add timeout --- main.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/main.py b/main.py index 0fe82c2..0fdd548 100644 --- a/main.py +++ b/main.py @@ -66,9 +66,9 @@ def get_unforwarded_emails(account_config, logger): 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']) @@ -102,9 +102,9 @@ def forward_emails(account_config, emails, logger): 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()