Send mail at the end of program

src/mail.py

@@ -0,0 +1,37 @@
+import smtplib
+
+import reserve
+
+
+class Mail:
+    def __init__(self, host, user, passwd, message):
+        self.host = host
+        self.user = user
+        self.passwd = passwd
+        self.message = message
+
+    def send(self, count):
+        print(f'Sending email#${count}...')
+        reserve.logging.info(f'Sending email#${count}...')
+        try:
+            smtp_obj = smtplib.SMTP_SSL(self.host, 465)
+            smtp_obj.login(self.user, self.passwd)
+            smtp_obj.sendmail(self.user, self.user, self.message)
+            smtp_obj.quit()
+            print(f'Email #${count} succeed')
+            reserve.logging.info(f'Email #${count} succeed')
+        except smtplib.SMTPException as e:
+            print(f'Email #${count} error', e)
+            reserve.logging.error(f'Email #${count} error. ' + str(e))
+
+
+mails: list[Mail] = []
+
+
+def send_all():
+    if len(mails) == 0:
+        return
+    count = 1
+    for mail in mails:
+        mail.send(count)
+        count += 1

src/main.py

@@ -3,12 +3,12 @@
 # import
 import os
 import platform
-import smtplib
 import sys
 import time
 from email.mime.text import MIMEText
 
 import reserve
+from mail import Mail, mails, send_all
 from utils import has_more
 
 # pre-define
@@ -36,6 +36,12 @@
                    '202': '181'}
             }
 
+
+def send_and_exit():
+    send_all()
+    reserve.logging.info('Exiting...')
+
+
 # Read config
 reserve.logging.info('Importing data from config.conf')
 with open("config.conf", "r") as config:
@@ -75,8 +81,8 @@ def send_email(seat=None, success=False, error=None):
         mail_user, mail_pass = mail_data[0], mail_data[1]
         mail_host = f"smtp.{mail_user.split('@')[1]}"
 
-        print('Sending email...')
-        reserve.logging.info('Sending email...')
+        print('Generating email...')
+        reserve.logging.info('Generating email...')
 
         if success:
             reserve.logging.info('Success = \'True\'')
@@ -90,16 +96,7 @@ def send_email(seat=None, success=False, error=None):
         message['From'] = mail_user
         message['To'] = mail_user
 
-        try:
-            smtp_obj = smtplib.SMTP_SSL(mail_host, 465)
-            smtp_obj.login(mail_user, mail_pass)
-            smtp_obj.sendmail(mail_user, mail_user, message.as_string())
-            smtp_obj.quit()
-            print('Email succeed')
-            reserve.logging.info('Email succeed')
-        except smtplib.SMTPException as e:
-            print('Email error', e)
-            reserve.logging.error('Email error. ' + str(e))
+        mails.append(Mail(mail_host, mail_user, mail_pass, message.as_string()))
 
 
     # main
@@ -119,6 +116,6 @@ def send_email(seat=None, success=False, error=None):
                 reserve.logging.info('Detected. Returning...')
                 reserve.logging.info('')
             else:
-                reserve.logging.info('Exiting...')
+                send_and_exit()
     else:
-        reserve.logging.info('Exiting...')
+        send_and_exit()