class EmailConfig:
    """
    郵箱配置
    """
    # 你的郵箱
    user: str = '138xxxx8888@163.com'
    # 郵箱授權(quán)碼
    password: str = 'TXYuclse932VzHJ'
    # 郵箱服務(wù)器
    host: str = 'smtp.163.com'
    # 端口
    port: int = '587'

class SenderInfo:
    """
    發(fā)件人信息
    """
    # 顯示給客戶的發(fā)件人名稱
    name: str = '起風(fēng)了-dev'
    # 職位
    position: str = '開發(fā)人員'
    # 公司
    company: str = 'xxx科技有限責(zé)任公司'
    # 郵箱
    email: str = '138xxxx8888@163.com'
    # 簽名
    signature: str = '簽名測試'

    def get_subject(self, client: dict):
        """
        獲取郵件主題
        :param client: 客戶信息, 與客戶信息文件內(nèi)一致
        :return: 郵件主題
        """
        return f"{self.name}致{client['name']}的郵件"

    def to_json(self):
        return {
            'name': self.name,
            'position': self.position,
            'company': self.company,
            'email': self.email,
            'signature': self.signature
        }

class SendSettings:
    """
    發(fā)送設(shè)置
    """
    # 郵件間隔(秒)，避免被限制
    delay_between_emails: int = 2
    # 測試模式（只發(fā)給自己）
    test_mode: bool = False
    # 每批最多發(fā)送量
    max_emails_per_batch: int = 20

    def load_clients(self) -> list[dict]:
        """加載客戶數(shù)據(jù)"""
        df = pd.read_excel(self.client_data_file)
        client_list = df.to_dict(orient='records')
        if not client_list:
            raise ValueError("客戶數(shù)據(jù)為空")
        # 客戶數(shù)據(jù)校驗并清洗
        for client in client_list:
            if not client['email']:
                raise ValueError(f"客戶 {json.dumps(client, ensure_ascii=False)} 的郵箱為空")
            if not client['name']:
                raise ValueError(f"客戶 {json.dumps(client, ensure_ascii=False)} 的姓名為空")
            client['name'] = client['name'].strip()
            client['email'] = client['email'].strip()
        print(f"已加載 {len(client_list)} 個客戶數(shù)據(jù)")
        return client_list

    def load_template(self):
        """加載郵件模板"""
        template = None
        if self.template_file:
            with open(self.template_file, 'r', encoding='utf-8') as f:
                template = Template(f.read())
        txt_template = None
        if self.txt_template_file:
            with open(self.txt_template_file, 'r', encoding='utf-8') as f:
                txt_template = f.read()
        return template, txt_template

    def _convert_html_to_text(self, html):
        """
        智能轉(zhuǎn)換HTML到純文本
        保留重要格式（列表、段落、強調(diào)等）
        """
        # 自定義轉(zhuǎn)換規(guī)則
        replacements = [
            (r'<br\s*/?>', '\n'),
            (r'<p.*?>', '\n\n'),
            (r'</p>', ''),
            (r'<li.*?>', '? '),
            (r'</li>', '\n'),
            (r'<strong.*?>', '*'),
            (r'</strong>', '*'),
            (r'<em.*?>', '_'),
            (r'</em>', '_'),
            (r'<h[1-6].*?>', '\n\n'),
            (r'</h[1-6]>', '\n\n'),
            (r'<[^>]+>', ''),  # 移除其他所有HTML標(biāo)簽
        ]

        text = html
        for pattern, replacement in replacements:
            text = re.sub(pattern, replacement, text)

        # 清理多余的空行
        text = re.sub(r'\n\s*\n\s*\n', '\n\n', text)
        print(f'_convert_html_to_text:{text}')
        return text.strip()

    def personalize_content(self, client, template, txt_template: str):
        """個性化郵件內(nèi)容"""

        html_content = template.render(client=client, sender=self.sender_info.to_json()) if template else None

        if txt_template:
            txt_content = txt_template
            for key in client.keys():
                txt_content = txt_content.replace('{client_' + key + '}', str(client[key]))
            sender_json = self.sender_info.to_json()
            for key in sender_json:
                txt_content = txt_content.replace('{sender_' + key + '}', str(sender_json[key]))
        else:
            txt_content = self._convert_html_to_text(html_content)

        return html_content, txt_content

    def send_to_client(self, client, content):
        """發(fā)送給單個客戶"""
        try:
            # 構(gòu)建郵件
            subject = self.sender_info.get_subject(client=client)

            # 如果是測試模式，發(fā)給自己
            to_email = client['email']
            if self.send_settings.test_mode:
                to_email = self.email_config.user
                subject = f"[測試] {subject}"

            # 發(fā)送
            self.yag.send(
                to=to_email,
                subject=subject,
                contents=content,
                # attachments=['attachments/sample.pdf'],  # 可選附件
                headers={
                    # 'X-Priority': '1',  #  高優(yōu)先級 ??（顯示感嘆號）
                    'X-Priority': '3',  # 普通優(yōu)先級（默認，不顯示特殊標(biāo)記）
                    # 'X-Priority': '5',  # 低優(yōu)先級
                    'X-Mailer': 'PersonalEmailSender'
                }
            )

            # 記錄日志
            log_entry = {
                'client': client['name'],
                'email': client['email'],
                'time': datetime.now().isoformat(),
                'status': '成功'
            }
            self.sent_log.append(log_entry)

            print(f"? 已發(fā)送給 {client['name']} <{client['email']}>")
            return True

        except Exception as e:
            print(f"? 發(fā)送失敗 {client['name']}: {e}")
            self.sent_log.append({
                'client': client['name'],
                'email': client['email'],
                'time': datetime.now().isoformat(),
                'status': f'失敗: {str(e)}'
            })
            return False

    def save_log(self):
        """保存發(fā)送記錄"""
        if not self.sent_log:
            return

        log_df = pd.DataFrame(self.sent_log)
        timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
        log_file = f'sent_logs/sent_{timestamp}.csv'

        # 確保目錄存在
        os.makedirs('sent_logs', exist_ok=True)

        log_df.to_csv(log_file, index=False, encoding='utf-8-sig')
        print(f"\n?? 發(fā)送記錄已保存: {log_file}")

    def run(self):
        """主運行函數(shù)"""
        print("開始發(fā)送個性化郵件...\n")

        # 加載數(shù)據(jù)
        client_list = self.load_clients()
        template, txt_template = self.load_template()

        sent_count = 0
        total_clients = len(client_list)

        for i, client in enumerate(client_list, 1):
            print(f"\n[{i}/{total_clients}] 處理: {client['name']}")

            # 個性化內(nèi)容
            html_content, text_content = self.personalize_content(client, template, txt_template)

            if not text_content:
                print(f"? 無內(nèi)容可發(fā)送給 {client['name']}")
                # 記錄日志
                log_entry = {
                    'client': client['name'],
                    'email': client['email'],
                    'time': datetime.now().isoformat(),
                    'status': '失敗',
                    '原因': '無內(nèi)容'
                }
                self.sent_log.append(log_entry)
                continue

            content = [text_content, html_content] if html_content else text_content
            # 發(fā)送
            if self.send_to_client(client, content):
                sent_count += 1

            # 延遲，避免被限制
            if i < total_clients:
                delay = self.send_settings.delay_between_emails
                print(f"等待 {delay} 秒...")
                time.sleep(delay)

            # 分批暫停（如果發(fā)送量大）
            if i % self.send_settings.max_emails_per_batch == 0:
                print(f"\n已發(fā)送 {i} 封，暫停 60 秒...")
                time.sleep(60)

        # 保存日志
        self.save_log()

        print(f"\n? 完成！成功發(fā)送 {sent_count}/{total_clients} 封郵件")

        # 關(guān)閉連接
        if self.yag:
            self.yag.close()

if __name__ == "__main__":
    # 客戶數(shù)據(jù)文件
    client_data_file = './clients/clients.xlsx'
    # 郵件模板文件
    txt_template_file = './templates/christmas.txt'
    sender = EmailSender(client_data_file=client_data_file,
                         txt_template_file=txt_template_file)
    sender.run()

# 配置信息
class EmailConfig:
    """
    郵箱配置
    """
    # 你的郵箱
    user: str = '138xxxx8888@163.com'
    # 郵箱授權(quán)碼
    password: str = 'TXYuclse932VzHJ'
    # 郵箱服務(wù)器
    host: str = 'smtp.163.com'
    # 端口
    port: int = '587'
class SenderInfo:
    """
    發(fā)件人信息
    """
    # 顯示給客戶的發(fā)件人名稱
    name: str = '起風(fēng)了-dev'
    # 職位
    position: str = '開發(fā)人員'
    # 公司
    company: str = 'xxx科技有限責(zé)任公司'
    # 郵箱
    email: str = '138xxxx8888@163.com'
    # 簽名
    signature: str = '簽名測試'

    def get_subject(self, client: dict):
        """
        獲取郵件主題
        :param client: 客戶信息, 與客戶信息文件內(nèi)一致
        :return: 郵件主題
        """
        return f"{self.name}致{client['name']}的郵件"

    def to_json(self):
        return {
            'name': self.name,
            'position': self.position,
            'company': self.company,
            'email': self.email,
            'signature': self.signature
        }
class SendSettings:
    """
    發(fā)送設(shè)置
    """
    # 郵件間隔(秒)，避免被限制
    delay_between_emails: int = 2
    # 測試模式（只發(fā)給自己）
    test_mode: bool = False
    # 每批最多發(fā)送量
    max_emails_per_batch: int = 20

# 主功能代碼
import json
import re

import yagmail
import pandas as pd
import time
import os
from datetime import datetime
from jinja2 import Template
from config.config import EmailConfig, SenderInfo, SendSettings


class EmailSender:
    """郵件發(fā)送器"""

    def __init__(self, client_data_file: str, txt_template_file: str | None):
        """初始化"""
        # 客戶數(shù)據(jù)
        self.client_data_file = client_data_file
        # 郵件模板
        self.template_file = None
        self.txt_template_file = txt_template_file
        # 郵件配置
        self.email_config = EmailConfig()
        self.sender_info = SenderInfo()
        self.send_settings = SendSettings()
        self.yag = yagmail.SMTP(
            user=self.email_config.user,
            password=self.email_config.password,
            host=self.email_config.host,
            port=self.email_config.port
        )
        self.sent_log = []

    def load_clients(self) -> list[dict]:
        """加載客戶數(shù)據(jù)"""
        df = pd.read_excel(self.client_data_file)
        client_list = df.to_dict(orient='records')
        if not client_list:
            raise ValueError("客戶數(shù)據(jù)為空")
        # 客戶數(shù)據(jù)校驗并清洗
        for client in client_list:
            if not client['email']:
                raise ValueError(f"客戶 {json.dumps(client, ensure_ascii=False)} 的郵箱為空")
            if not client['name']:
                raise ValueError(f"客戶 {json.dumps(client, ensure_ascii=False)} 的姓名為空")
            client['name'] = client['name'].strip()
            client['email'] = client['email'].strip()
        print(f"已加載 {len(client_list)} 個客戶數(shù)據(jù)")
        return client_list

    def load_template(self):
        """加載郵件模板"""
        template = None
        if self.template_file:
            with open(self.template_file, 'r', encoding='utf-8') as f:
                template = Template(f.read())
        txt_template = None
        if self.txt_template_file:
            with open(self.txt_template_file, 'r', encoding='utf-8') as f:
                txt_template = f.read()
        return template, txt_template

    def _convert_html_to_text(self, html):
        """
        智能轉(zhuǎn)換HTML到純文本
        保留重要格式（列表、段落、強調(diào)等）
        """
        # 自定義轉(zhuǎn)換規(guī)則
        replacements = [
            (r'<br\s*/?>', '\n'),
            (r'<p.*?>', '\n\n'),
            (r'</p>', ''),
            (r'<li.*?>', '? '),
            (r'</li>', '\n'),
            (r'<strong.*?>', '*'),
            (r'</strong>', '*'),
            (r'<em.*?>', '_'),
            (r'</em>', '_'),
            (r'<h[1-6].*?>', '\n\n'),
            (r'</h[1-6]>', '\n\n'),
            (r'<[^>]+>', ''),  # 移除其他所有HTML標(biāo)簽
        ]

        text = html
        for pattern, replacement in replacements:
            text = re.sub(pattern, replacement, text)

        # 清理多余的空行
        text = re.sub(r'\n\s*\n\s*\n', '\n\n', text)
        print(f'_convert_html_to_text:{text}')
        return text.strip()

    def personalize_content(self, client, template, txt_template: str):
        """個性化郵件內(nèi)容"""

        html_content = template.render(client=client, sender=self.sender_info.to_json()) if template else None

        if txt_template:
            txt_content = txt_template
            for key in client.keys():
                txt_content = txt_content.replace('{client_' + key + '}', str(client[key]))
            sender_json = self.sender_info.to_json()
            for key in sender_json:
                txt_content = txt_content.replace('{sender_' + key + '}', str(sender_json[key]))
        else:
            txt_content = self._convert_html_to_text(html_content)

        return html_content, txt_content

    def send_to_client(self, client, content):
        """發(fā)送給單個客戶"""
        try:
            # 構(gòu)建郵件
            subject = self.sender_info.get_subject(client=client)

            # 如果是測試模式，發(fā)給自己
            to_email = client['email']
            if self.send_settings.test_mode:
                to_email = self.email_config.user
                subject = f"[測試] {subject}"

            # 發(fā)送
            self.yag.send(
                to=to_email,
                subject=subject,
                contents=content,
                # attachments=['attachments/sample.pdf'],  # 可選附件
                headers={
                    # 'X-Priority': '1',  #  高優(yōu)先級 ??（顯示感嘆號）
                    'X-Priority': '3',  # 普通優(yōu)先級（默認，不顯示特殊標(biāo)記）
                    # 'X-Priority': '5',  # 低優(yōu)先級
                    'X-Mailer': 'PersonalEmailSender'
                }
            )

            # 記錄日志
            log_entry = {
                'client': client['name'],
                'email': client['email'],
                'time': datetime.now().isoformat(),
                'status': '成功'
            }
            self.sent_log.append(log_entry)

            print(f"? 已發(fā)送給 {client['name']} <{client['email']}>")
            return True

        except Exception as e:
            print(f"? 發(fā)送失敗 {client['name']}: {e}")
            self.sent_log.append({
                'client': client['name'],
                'email': client['email'],
                'time': datetime.now().isoformat(),
                'status': f'失敗: {str(e)}'
            })
            return False

    def save_log(self):
        """保存發(fā)送記錄"""
        if not self.sent_log:
            return

        log_df = pd.DataFrame(self.sent_log)
        timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
        log_file = f'sent_logs/sent_{timestamp}.csv'

        # 確保目錄存在
        os.makedirs('sent_logs', exist_ok=True)

        log_df.to_csv(log_file, index=False, encoding='utf-8-sig')
        print(f"\n?? 發(fā)送記錄已保存: {log_file}")

    def run(self):
        """主運行函數(shù)"""
        print("開始發(fā)送個性化郵件...\n")

        # 加載數(shù)據(jù)
        client_list = self.load_clients()
        template, txt_template = self.load_template()

        sent_count = 0
        total_clients = len(client_list)

        for i, client in enumerate(client_list, 1):
            print(f"\n[{i}/{total_clients}] 處理: {client['name']}")

            # 個性化內(nèi)容
            html_content, text_content = self.personalize_content(client, template, txt_template)

            if not text_content:
                print(f"? 無內(nèi)容可發(fā)送給 {client['name']}")
                # 記錄日志
                log_entry = {
                    'client': client['name'],
                    'email': client['email'],
                    'time': datetime.now().isoformat(),
                    'status': '失敗',
                    '原因': '無內(nèi)容'
                }
                self.sent_log.append(log_entry)
                continue

            content = [text_content, html_content] if html_content else text_content
            # 發(fā)送
            if self.send_to_client(client, content):
                sent_count += 1

            # 延遲，避免被限制
            if i < total_clients:
                delay = self.send_settings.delay_between_emails
                print(f"等待 {delay} 秒...")
                time.sleep(delay)

            # 分批暫停（如果發(fā)送量大）
            if i % self.send_settings.max_emails_per_batch == 0:
                print(f"\n已發(fā)送 {i} 封，暫停 60 秒...")
                time.sleep(60)

        # 保存日志
        self.save_log()

        print(f"\n? 完成！成功發(fā)送 {sent_count}/{total_clients} 封郵件")

        # 關(guān)閉連接
        if self.yag:
            self.yag.close()


if __name__ == "__main__":
    # 客戶數(shù)據(jù)文件
    client_data_file = './clients/clients.xlsx'
    # 郵件模板文件
    txt_template_file = './templates/christmas.txt'
    sender = EmailSender(client_data_file=client_data_file,
                         txt_template_file=txt_template_file)
    sender.run()