import requests
from bs4 import BeautifulSoup
import time
import smtplib
import playsound
from email.mime.text import MIMEText
import yaml
import logging
from concurrent.futures import ThreadPoolExecutor

class WebsiteMonitor:
    def __init__(self):
        self.load_config()
        self.setup_logging()
        
    def load_config(self):
        with open('config.yml', 'r', encoding='utf-8') as f:
            self.config = yaml.safe_load(f)
        self.banned_words = self.config['banned_words']
        self.websites = self.config['websites']
        self.email_config = self.config['email']
        
    def setup_logging(self):
        logging.basicConfig(
            level=logging.INFO,
            format='%(asctime)s - %(levelname)s - %(message)s',
            filename='monitor.log'
        )
        
    def check_website(self, url):
        try:
            response = requests.get(url, timeout=10)
            soup = BeautifulSoup(response.text, 'html.parser')
            text_content = soup.get_text().lower()
            
            found_words = []
            for word in self.banned_words:
                if word.lower() in text_content:
                    found_words.append(word)
                    
            if found_words:
                self.alert(url, found_words)
                
        except Exception as e:
            logging.error(f"检查网站 {url} 时发生错误: {str(e)}")
            
    def alert(self, url, found_words):
        message = f"警告！在网站 {url} 中发现违禁词: {', '.join(found_words)}"
        logging.warning(message)
        
        # 发送邮件提醒
        self.send_email(message)
        
        # 播放声音提醒
        self.play_alert_sound()
        
    def send_email(self, message):
        try:
            msg = MIMEText(message)
            msg['Subject'] = '违禁词监测警告'
            msg['From'] = self.email_config['sender']
            msg['To'] = self.email_config['receiver']
            
            with smtplib.SMTP_SSL(self.email_config['smtp_server'], self.email_config['smtp_port']) as server:
                server.login(self.email_config['sender'], self.email_config['password'])
                server.send_message(msg)
                
        except Exception as e:
            logging.error(f"发送邮件时发生错误: {str(e)}")
            
    def play_alert_sound(self):
        try:
            playsound.playsound('alert.mp3')
        except Exception as e:
            logging.error(f"播放提醒声音时发生错误: {str(e)}")
            
    def start_monitoring(self):
        logging.info("开始监测网站...")
        while True:
            with ThreadPoolExecutor(max_workers=5) as executor:
                executor.map(self.check_website, self.websites)
            time.sleep(self.config['check_interval'])

if __name__ == '__main__':
    monitor = WebsiteMonitor()
    monitor.start_monitoring()