stefano/NotiMail
1
0
Fork 0
mirror of https://github.com/draga79/NotiMail.git synced 2024-09-19 11:04:20 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Some server send a BYE or drop the connection after less than 10 minutes (for example, Gmail). The socket timeout has been reduced to 8 minutes and there's now some logic to deal with the BYE IMAP message

Browse source
This commit is contained in:
Stefano Marinelli 2023-10-22 19:54:25 +02:00
parent 028e77c9ac
commit b85545bb9c
1 changed files with 17 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
NotiMail.py
View file

@ -288,6 +288,8 @@ class IMAPHandler:
line = self.mail.readline()
if line:
print(line.decode('utf-8'))
if b'BYE' in line:
raise ConnectionAbortedError("Received BYE from server. Trying to reconnect...")
if b'EXISTS' in line:
break
self.mail.send(b'DONE\r\n')
@ -330,23 +332,24 @@ class MultiIMAPHandler:
for thread in threads:
thread.join()
@staticmethod
def monitor_account(handler):
print(f"Monitoring {handler.email_user}")
logging.info(f"Monitoring {handler.email_user}")
try:
handler.connect()
while True:
handler.idle()
handler.process_emails()
except ConnectionAbortedError as e:
print(str(e))
time.sleep(30)
except Exception as e:
print(f"An unexpected error occurred: {str(e)}")
logging.error(f"An unexpected error occurred: {str(e)}")
notifier.send_notification("Script Error", f"An unexpected error occurred: {str(e)}")
while True: # Add a loop to keep retrying on connection loss
try:
handler.connect()
while True:
handler.idle()
handler.process_emails()
except ConnectionAbortedError as e:
print(str(e))
time.sleep(30) # Sleep for 30 seconds before retrying
except Exception as e:
print(f"An unexpected error occurred: {str(e)}")
logging.error(f"An unexpected error occurred: {str(e)}")
notifier.send_notification("Script Error", f"An unexpected error occurred: {str(e)}")
break
def shutdown_handler(signum, frame):
print("Shutdown signal received. Cleaning up...")
@ -410,7 +413,7 @@ def multi_account_main():
global notifier
notifier = Notifier(providers)
socket.setdefaulttimeout(600)
socket.setdefaulttimeout(480)
signal.signal(signal.SIGTERM, shutdown_handler)
signal.signal(signal.SIGINT, shutdown_handler)