Python 自动截取屏幕信息发送至企业微信群，自动化监控数据状态

import requests
import json
import time
import os
from datetime import datetime
from PIL import ImageGrab


class WeChatMonitor:
    def __init__(self, webhook_url, screenshot_interval=60):
        """
        初始化监控器
        :param webhook_url: 企业微信群机器人的Webhook地址
        :param screenshot_interval: 截图间隔时间（秒），默认60秒
        """
        self.webhook_url = webhook_url
        self.interval = screenshot_interval
        self.screenshot_dir = "monitor_screenshots"
        # 创建截图保存目录
        if not os.path.exists(self.screenshot_dir):
            os.makedirs(self.screenshot_dir)


    def take_screenshot(self):
        """捕获屏幕截图并保存为时间戳命名的文件"""
        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
        filename = f"screenshot_{timestamp}.jpg"
        filepath = os.path.join(self.screenshot_dir, filename)
        
        # 使用PIL截图
        screenshot = ImageGrab.grab()
        screenshot.save(filepath, "JPEG")
        print(f"截图已保存: {filepath}")
        return filepath


    def upload_to_wechat(self, image_path):
        """将图片上传到企业微信并获取media_id"""
        upload_url = " https://qyapi.weixin.qq.com/cgi-bin/webhook/upload_media "
        params = {
            "key": self.webhook_url.split("key=")[-1],  # 从Webhook URL提取key
            "type": "file"
        }
        with open(image_path, "rb") as f:
            files = {"media": f}
            response = requests.post(upload_url, params=params, files=files)
        if response.status_code == 200:
            result = response.json()
            return result.get("media_id")
        else:
            print(f"上传失败: {response.text}")
            return None


    def send_wechat_message(self, media_id, text="自动监控截图"):
        """通过Webhook发送图片消息到企业微信群"""
        message_data = {
            "msgtype": "image",
            "image": {
                "media_id": media_id
            },
            "text": {
                "content": text
            }
        }
        headers = {"Content-Type": "application/json"}
        response = requests.post(self.webhook_url, data=json.dumps(message_data), headers=headers)
        if response.status_code == 200:
            print("消息发送成功")
        else:
            print(f"发送失败: {response.text}")


    def start_monitoring(self, duration_minutes=60):
        """启动监控，持续运行指定分钟数"""
        total_seconds = duration_minutes * 60
        start_time = time.time()
        
        while time.time() - start_time < total_seconds:
            try:
                # 截图并发送
                screenshot_path = self.take_screenshot()
                media_id = self.upload_to_wechat(screenshot_path)
                if media_id:
                    self.send_wechat_message(media_id, f"监控时间: {datetime.now().strftime('%H:%M:%S')}")
                else:
                    print("图片上传失败，跳过发送")
            except Exception as e:
                print(f"执行错误: {e}")
            
            # 等待间隔时间
            time.sleep(self.interval)


if __name__ == "__main__":
    # 替换为你的企业微信群机器人Webhook URL
    WEBHOOK_URL = " https://qyapi.weixin.qq.com/cgi-bin/webhook/send?key=YOUR_KEY_HERE "
    
    monitor = WeChatMonitor(webhook_url=WEBHOOK_URL, screenshot_interval=60)  # 每60秒截图一次
    monitor.start_monitoring(duration_minutes=120)  # 运行120分钟