buff1.0/buff/sendMail-api/main.py

import sys

from PyQt5.QtWidgets import QMessageBox
from PyQt5.QtWidgets import QApplication, QWidget, QLabel, QLineEdit, QPushButton, QVBoxLayout, QHBoxLayout, QWidgetItem
from PyQt5.QtCore import Qt
import sqlite3
import threading
import time
from extract_and_save import run
import logging

logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(name)s - %(message)s')
logger = logging.getLogger(__name__)

class MailSenderApp(QWidget):
    def __init__(self):
        super().__init__()
        self.initUI()


    def initUI(self):
        self.setWindowTitle('Mail Sender')
        self.setGeometry(100, 100, 200, 200)

        main_layout = QVBoxLayout()
        main_layout.setAlignment(Qt.AlignmentFlag.AlignLeft)  # 设置垂直布局居中

        # 创建标签和输入框，并将它们放在一个水平布局中
        def create_row(label_text):
            layout = QHBoxLayout()
            label = QLabel(label_text, self)
            edit = QLineEdit(self)
            edit.setFixedWidth(300)  # 设置输入框的固定宽度为500
            layout.addWidget(label)
            layout.addWidget(edit)
            return layout, edit

        self.phone_layout, self.phone_edit = create_row('手机号码:')
        self.session_layout, self.session_edit = create_row('Session:')
        self.sender_email_layout, self.sender_email_edit = create_row('发送人邮箱号:')
        self.receiver_email_layout, self.receiver_email_edit = create_row('接受人邮箱号:')
        self.smtp_proxy_layout, self.smtp_proxy_edit = create_row('邮箱SMTP代理:')
        self.smtp_account_layout, self.smtp_account_edit = create_row('SMTP账号:')
        self.smtp_password_layout, self.smtp_password_edit = create_row('SMTP密码:')
        self.email_alias_layout, self.email_alias_edit = create_row('邮箱特殊代号:')
        self.scan_interval_layout, self.scan_interval_edit = create_row('扫描间隔时间:')
        self.error_count_layout, self.error_count_edit = create_row('异常次数\n(超出则退出扫描):')
        self.success_scan_interval_layout, self.success_scan_interval_edit = create_row('没有短信验证\n中间间隔时间(秒):')

        # 将水平布局添加到垂直布局中
        for layout in [
            self.phone_layout,
            self.session_layout,
            self.sender_email_layout,
            self.receiver_email_layout,
            self.smtp_proxy_layout,
            self.smtp_account_layout,
            self.smtp_password_layout,
            self.email_alias_layout,
            self.scan_interval_layout,
            self.error_count_layout,
            self.success_scan_interval_layout
        ]:
            main_layout.addLayout(layout)

        # 创建按钮
        self.save_button = QPushButton('保存', self)
        self.save_button.clicked.connect(self.save_settings)

        self.start_scan_button = QPushButton('开启扫描', self)
        self.start_scan_button.clicked.connect(self.start_scan)

        self.stop_scan_button = QPushButton('关闭扫描', self)
        self.stop_scan_button.clicked.connect(self.stop_scan)

        # 创建一个水平布局来放置按钮
        button_layout = QHBoxLayout()
        button_layout.addWidget(self.save_button)
        button_layout.addWidget(self.start_scan_button)
        button_layout.addWidget(self.stop_scan_button)

        # 将按钮布局添加到主布局中
        main_layout.addLayout(button_layout)

        # 设置主窗口的布局
        container = QWidget()
        container.setLayout(main_layout)
        container.setMaximumWidth(300)  # 设置容器的最大宽度为1000
        main_window_layout = QVBoxLayout()
        main_window_layout.addWidget(container, alignment=Qt.AlignmentFlag.AlignLeft)
        self.setLayout(main_window_layout)

        # 从数据库中读取数据并填充到输入框中
        self.load_settings()

    def save_settings(self):
        phone = self.phone_edit.text()
        session = self.session_edit.text()
        sender_email = self.sender_email_edit.text()
        receiver_email = self.receiver_email_edit.text()
        smtp_proxy = self.smtp_proxy_edit.text()
        smtp_account = self.smtp_account_edit.text()
        smtp_password = self.smtp_password_edit.text()
        email_alias = self.email_alias_edit.text()
        scan_interval = self.scan_interval_edit.text()
        error_count = self.error_count_edit.text()
        success_scan_interval = self.success_scan_interval_edit.text()

        # 保存到SQLite数据库
        conn = sqlite3.connect('settings.db')
        cursor = conn.cursor()
        cursor.execute('''CREATE TABLE IF NOT EXISTS settings (
                            phone TEXT,
                            session TEXT,
                            sender_email TEXT,
                            receiver_email TEXT,
                            smtp_proxy TEXT,
                            smtp_account TEXT,
                            smtp_password TEXT,
                            email_alias TEXT,
                            scan_interval TEXT,
                            error_count TEXT,
                            success_scan_interval TEXT
                        )''')
        cursor.execute('''DELETE FROM settings''')  # 删除现有数据
        cursor.execute('''INSERT INTO settings (
                            phone,
                            session,
                            sender_email,
                            receiver_email,
                            smtp_proxy,
                            smtp_account,
                            smtp_password,
                            email_alias,
                            scan_interval,
                            error_count,
                            success_scan_interval
                        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?,?,?)''',
                       (phone, session, sender_email, receiver_email, smtp_proxy, smtp_account, smtp_password, email_alias, scan_interval,error_count,success_scan_interval))
        conn.commit()
        conn.close()
        # 提示保存成功
        QMessageBox.information(self, '保存成功', '设置已保存到数据库。')

    def load_settings(self):
        # 从SQLite数据库中读取数据
        conn = sqlite3.connect('settings.db')
        cursor = conn.cursor()
        cursor.execute('''CREATE TABLE IF NOT EXISTS settings (
                            phone TEXT,
                            session TEXT,
                            sender_email TEXT,
                            receiver_email TEXT,
                            smtp_proxy TEXT,
                            smtp_account TEXT,
                            smtp_password TEXT,
                            email_alias TEXT,
                            scan_interval TEXT,
                            error_count TEXT,
                            success_scan_interval TEXT
                        )''')
        cursor.execute('''SELECT * FROM settings''')
        row = cursor.fetchone()
        conn.close()

        if row:
            self.phone_edit.setText(row[0])
            self.session_edit.setText(row[1])
            self.sender_email_edit.setText(row[2])
            self.receiver_email_edit.setText(row[3])
            self.smtp_proxy_edit.setText(row[4])
            self.smtp_account_edit.setText(row[5])
            self.smtp_password_edit.setText(row[6])
            self.email_alias_edit.setText(row[7])
            self.scan_interval_edit.setText(row[8])
            self.error_count_edit.setText(row[9])
            self.success_scan_interval_edit.setText(row[10])


    def start_scan(self):
        # 实现开启扫描的逻辑
        # 获取session的值
        session = self.session_edit.text()
        # 获取发送人邮箱号
        sender_email = self.sender_email_edit.text()
        # 获取接受人邮箱号
        receiver_email = self.receiver_email_edit.text()
        smtp_proxy = self.smtp_proxy_edit.text()
        smtp_account = self.smtp_account_edit.text()
        smtp_password = self.smtp_password_edit.text()
        email_alias = self.email_alias_edit.text()
        # 获取间隔时间
        scan_interval = self.scan_interval_edit.text()
        # 异常次数
        error_count = self.error_count_edit.text()
        # 获取成功间隔时间
        scan_interval = self.scan_interval_edit.text()
        # 获取成功间隔时间
        success_scan_interval = self.success_scan_interval_edit.text()

        logger.info(f'session:{session}')
        logger.info(f'发送人邮箱号：sender_email:{sender_email}')
        logger.info(f'接收人邮箱号：receiver_email:{receiver_email}')
        logger.info(f'smtp代理：smtp_proxy:{smtp_proxy}')
        logger.info(f'smtp账号：smtp_account:{smtp_account}')
        logger.info(f'smtp密码：smtp_password:{smtp_password}')
        logger.info(f'邮箱特殊代号：email_alias:{email_alias}')
        logger.info(f'间隔时长（秒）：scan_interval:{scan_interval}')
        logger.info(f'异常次数：error_count:{int(error_count)}')
        logger.info(f'没有短信验证中间间隔时长（秒）：success_scan_interval:{success_scan_interval}')

        global stop_flag
        stop_flag = False
        # 清空表
        logger.info("清空表sms_codes")
        conn = sqlite3.connect('settings.db')
        cursor = conn.cursor()
        cursor.execute('''CREATE TABLE IF NOT EXISTS sms_codes (edit_content TEXT, send_content_short TEXT)''')
        cursor.execute('delete from sms_codes')
        conn.commit()
        conn.close()
        # 创建一个线程，并启动线程，线程每间隔10时间执行一下extract_and_save.py中的run方法
        thread = threading.Thread(target=thread_function, args=(session, sender_email, receiver_email, smtp_proxy, smtp_account, smtp_password, email_alias,scan_interval,error_count,success_scan_interval))
        thread.start()
        logger.info("扫描已开启")
        QMessageBox.information(self, '扫描已开启', '扫描已开启。')

    def stop_scan(self):
        # 获取全局的变量 stop_flag
        global stop_flag
        stop_flag = True
        logger.info("扫描已关闭")
        QMessageBox.information(self, '扫描已关闭', '扫描已关闭。')

def thread_function(session, sender_email, receiver_email, smtp_proxy, smtp_account, smtp_password, email_alias,scan_interval,error_count,success_scan_interval):
    #获取 全局的变量 stop_flag
    global stop_flag
    while True:
        # logger.info(f'扫描停止标志：stop_flag = {stop_flag}')
        # stop_flag 为true 则跳出
        if stop_flag:
            return
        stop1_flag = run(
            session=session,
            sender_email=sender_email,
            receiver_email=receiver_email,
            smtp_proxy=smtp_proxy,
            smtp_account=smtp_account,
            smtp_password=smtp_password,
            email_alias=email_alias,
            error_count=error_count
        )
        logger.info(f'扫描停止标志：stop1_flag = {stop1_flag}')
        if stop1_flag == 'stop':
            return
        elif stop1_flag == 'success':
            logger.info(f"没有短信验证时，间隔{float(success_scan_interval)}秒")
            time.sleep(float(success_scan_interval))
        time.sleep(float(scan_interval))

if __name__ == "__main__":
    app = QApplication(sys.argv)
    ex = MailSenderApp()
    ex.show()
    sys.exit(app.exec())