update: locale detection & mail subject check

邮件批量发送.py

@@ -12,6 +12,7 @@ from selenium.webdriver.support import expected_conditions as EC
 from selenium.webdriver.support.wait import WebDriverWait
 from selenium.webdriver.remote.webelement import WebElement
 
+from pathlib import Path
 from datetime import datetime, timedelta
 from zoneinfo import ZoneInfo
 from nameparser import HumanName
@@ -38,11 +39,6 @@ errors = 0
 warnings = 0
 
 greetings = [
-    {
-        "locale": "**",
-        "default": "不使用问候语",
-        "registry": []
-    },
     {
         "locale": "en",
         "default": "Hello",
@@ -189,7 +185,8 @@ def main():
     while True:
         try:
             print('[信息] 请选择数据源')
-            dialog = wx.FileDialog(None, 'Open', wildcard='*.xlsx', style=wx.FD_OPEN | wx.FD_FILE_MUST_EXIST)
+            title = 'Open (%s)' % args.address
+            dialog = wx.FileDialog(None, title, wildcard='*.xlsx', style=wx.FD_OPEN | wx.FD_FILE_MUST_EXIST)
 
             if dialog.ShowModal() == wx.ID_OK:
                 filepath = dialog.GetPath()
@@ -201,6 +198,9 @@ def main():
             workbook = pandas.read_excel(filepath)
             data = workbook.where(pandas.notnull(workbook), None).to_dict(orient='list')
             recipients = data.get(args.column_address, [])
+
+            parts = Path(filepath).stem.split('.', 2)
+            suffix = parts[1].lower() if len(parts) > 1 else None
         except Exception as e:
             print(f'[警告] 读取数据表失败：{e}')
             continue
@@ -219,42 +219,38 @@ def main():
             # 打开草稿箱
             click("li[data-id='default0/Brouillons']", condition=EC.presence_of_element_located)
             click("button[data-id='default0/Brouillons']", condition=EC.presence_of_element_located)
-
             # 打开邮件
             click("ul[aria-label='List view'] li[data-index='0']", condition=EC.presence_of_element_located)
-
-            subject = locate("h1.subject")
-            print(f'[信息] 已读取邮件：{subject.text}')
-
-            sender = locate("header div.from")
-            print(f'[信息] 指定发件人：{sender.text[6:].replace('\n', ' ')}')
         except Exception as e:
-            print(f'[!!!!] 读取邮件时发生了错误：{e}')
-            return 5
+            print(f'[警告] 打开草稿邮件时发生了错误：{e}')
+            print(f'[信息] 提示：请手动选择邮件')
+            pass
 
         rate = 60 / (args.interval + 3)
         length = list.count(sents, None)
-        command = None
         timezone = ZoneInfo(args.timezone) if args.timezone else None
 
         print(f'[信息] 当前发送速率 {round(rate, 2)} 封/分钟')
         if rate > 8.33: print('[警告] 当前发送速率已超出限制 8.33 封/分钟')
         print(f'[信息] 预计使用时间 {timedelta(minutes=length / rate)}')
         print(f'[信息] 当前时区：{args.timezone or '无'}')
-        print(f'[信息] 已读取可用问候语 {len(greetings)} 条：', end='\n\n')
 
-        for index, item in enumerate(greetings):
-            print(f'\t[{index}] {item.get('locale')}', end=' ')
-            print('- %s' % (', '.join(filter(None, item.get('registry'))) or item.get('default')))
+        entries = list(filter(lambda it: it.locale == suffix, greetings))
+        locale = dict(entries[0]) if len(entries) > 0 else None
+        print(f'[信息] 当前语言：{str(suffix).upper() if locale else '无'}', end='\n\n')
 
-        while command is None:
-            match input('\n[????] 请选择 (留空取消操作): ').strip():
-                case keys if not keys:
+        if input('[????] 确认操作：(y) 确定 / (N) 取消') not in ['Y', 'y']:
             print('[信息] 操作取消')
-                    return 0
-                case keys if keys.isdigit():
-                    number = int(keys)
-                    if number < len(greetings): command = number
+            continue
+
+        subject = locate("h1.subject").text.strip()
+        sender = locate("header div.from").text[6:].replace('\n', ' ')
+        print(f'[信息] 已读取邮件：{subject}')
+        print(f'[信息] 指定发件人：{sender}')
+
+        if sender.lower().find(f'<{str(args.address).lower()}>'):
+            print(f'[警告] 检测到发件人与设定不一致')
+            print(f'[信息] 提示：请检查邮件是否正确')
 
         global date
         date = datetime.now()
@@ -296,20 +292,27 @@ def main():
                     print('[信息] 正在发送：%s (%.2f %%)' % (recipient, current / limit * 100))
                     click("button[aria-label='Edit copy']")
 
+                    if (target := locate("h1.subject").text.strip()) != subject:
+                        print(f'[警告] 邮件主题与设定不一致：{target}')
+                        if input('[????] 是否继续：(y) / (N): ') in ['Y', 'y']:
+                            subject = target
+                            print(f'[信息] 已更新邮件主题设定')
+                        else:
+                            raise Exception('操作取消')
+
                     ready(driver, lambda x: x.find_element(By.CSS_SELECTOR, ".io-ox-busy"))
                     locate("div.io-ox-mail-compose-window iframe", condition=EC.frame_to_be_available_and_switch_to_it)
 
-                    if command > 0 and (entry := greetings[command]):
+                    if locale and (registry := locale.get('registry')):
                         match datetime.now(timezone).hour if timezone else -1:
-                            case hour if 6  <= hour < 12: registry = entry.get('registry')[0]
-                            case hour if 12 <= hour < 18: registry = entry.get('registry')[1]
-                            case hour if 18 <= hour < 21: registry = entry.get('registry')[2]
-                            case _: registry = None
+                            case hour if 6  <= hour < 12: hello = registry[0]
+                            case hour if 12 <= hour < 18: hello = registry[1]
+                            case hour if 18 <= hour < 21: hello = registry[2]
+                            case _: hello = locale.get('default')
 
                         iframe = driver.switch_to.active_element
                         action = ActionChains(driver)
                         clean = False
-                        hello = registry or entry.get('default')
 
                         if name is not None and (name := str(name).strip()) and not contains_non_latin_alphabet(name):
                             parts = HumanName(name)