import os
import socket
import tkinter as tk
from tkinter import filedialog, messagebox, ttk
from PIL import Image, ImageTk
import json


class RemoteWallpaperClient:
    def __init__(self, root):
        self.root = root
        self.root.title("遠(yuǎn)程壁紙?jiān)O(shè)置客戶(hù)端")
        self.root.geometry("500x450")
        self.root.resizable(False, False)

        # 默認(rèn)服務(wù)器端口
        self.default_port = 12345

        # 當(dāng)前選擇的圖片路徑
        self.current_image_path = None

        # 創(chuàng)建界面
        self.create_widgets()

    def create_widgets(self):
        # 主框架
        main_frame = ttk.Frame(self.root, padding="20")
        main_frame.pack(fill=tk.BOTH, expand=True)

        # 標(biāo)題
        title_label = ttk.Label(main_frame, text="遠(yuǎn)程壁紙?jiān)O(shè)置客戶(hù)端",
                                font=("Arial", 16, "bold"))
        title_label.pack(pady=10)

        # 目標(biāo)計(jì)算機(jī)設(shè)置區(qū)域
        target_frame = ttk.LabelFrame(main_frame, text="目標(biāo)計(jì)算機(jī)設(shè)置", padding="10")
        target_frame.pack(fill=tk.X, pady=10)

        # IP地址輸入
        ip_frame = ttk.Frame(target_frame)
        ip_frame.pack(fill=tk.X, pady=5)

        ttk.Label(ip_frame, text="目標(biāo)IP地址:").pack(side=tk.LEFT)
        self.ip_entry = ttk.Entry(ip_frame, width=15)
        self.ip_entry.pack(side=tk.LEFT, padx=5)
        self.ip_entry.insert(0, "192.168.1.")

        # 端口輸入
        port_frame = ttk.Frame(target_frame)
        port_frame.pack(fill=tk.X, pady=5)

        ttk.Label(port_frame, text="端口:").pack(side=tk.LEFT)
        self.port_entry = ttk.Entry(port_frame, width=10)
        self.port_entry.pack(side=tk.LEFT, padx=5)
        self.port_entry.insert(0, str(self.default_port))

        # 壁紙選擇區(qū)域
        wallpaper_frame = ttk.LabelFrame(main_frame, text="壁紙?jiān)O(shè)置", padding="10")
        wallpaper_frame.pack(fill=tk.X, pady=10)

        # 選擇壁紙按鈕
        select_btn = ttk.Button(wallpaper_frame, text="選擇壁紙圖片",
                                command=self.select_wallpaper)
        select_btn.pack(pady=5)

        # 壁紙預(yù)覽區(qū)域
        preview_frame = ttk.Frame(wallpaper_frame)
        preview_frame.pack(fill=tk.X, pady=5)

        self.preview_label = tk.Label(preview_frame, text="未選擇圖片",
                                      relief=tk.SUNKEN, bg='white',
                                      width=50, height=12)
        self.preview_label.pack(fill=tk.BOTH, expand=True)

        # 按鈕框架
        btn_frame = ttk.Frame(main_frame)
        btn_frame.pack(pady=10)

        # 發(fā)送壁紙按鈕
        send_btn = ttk.Button(btn_frame, text="發(fā)送壁紙",
                              command=self.send_wallpaper)
        send_btn.pack(side=tk.LEFT, padx=5)

        # 測(cè)試連接按鈕
        test_btn = ttk.Button(btn_frame, text="測(cè)試連接",
                              command=self.test_connection)
        test_btn.pack(side=tk.LEFT, padx=5)

        # 狀態(tài)欄
        self.status_var = tk.StringVar()
        self.status_var.set("就緒")
        status_bar = ttk.Label(main_frame, textvariable=self.status_var,
                               relief=tk.SUNKEN, anchor=tk.W)
        status_bar.pack(fill=tk.X, side=tk.BOTTOM)

    def select_wallpaper(self):
        """選擇壁紙圖片"""
        file_types = [
            ("圖片文件", "*.jpg *.jpeg *.png *.bmp"),
            ("所有文件", "*.*")
        ]

        file_path = filedialog.askopenfilename(
            title="選擇壁紙圖片",
            filetypes=file_types
        )

        if file_path:
            self.current_image_path = file_path
            self.preview_wallpaper(file_path)
            self.status_var.set(f"已選擇: {os.path.basename(file_path)}")

    def preview_wallpaper(self, image_path):
        """預(yù)覽壁紙"""
        try:
            # 打開(kāi)圖片
            image = Image.open(image_path)

            # 調(diào)整圖片大小以適應(yīng)預(yù)覽區(qū)域
            preview_width = 400
            preview_height = 200
            image.thumbnail((preview_width, preview_height), Image.Resampling.LANCZOS)

            # 轉(zhuǎn)換為T(mén)kinter可用的格式
            photo = ImageTk.PhotoImage(image)

            # 更新預(yù)覽標(biāo)簽
            self.preview_label.configure(image=photo, text="")
            self.preview_label.image = photo  # 保持引用

        except Exception as e:
            self.preview_label.configure(image=None, text="預(yù)覽加載失敗")
            print(f"預(yù)覽圖片失敗: {e}")

    def test_connection(self):
        """測(cè)試與目標(biāo)計(jì)算機(jī)的連接"""
        ip = self.ip_entry.get().strip()
        port = self.port_entry.get().strip()

        if not ip:
            messagebox.showerror("錯(cuò)誤", "請(qǐng)輸入目標(biāo)IP地址")
            return

        try:
            port = int(port)
        except ValueError:
            messagebox.showerror("錯(cuò)誤", "端口號(hào)必須是數(shù)字")
            return

        try:
            # 嘗試連接
            sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            sock.settimeout(5)  # 5秒超時(shí)
            result = sock.connect_ex((ip, port))
            sock.close()

            if result == 0:
                messagebox.showinfo("成功", f"成功連接到 {ip}:{port}")
                self.status_var.set(f"已連接到 {ip}:{port}")
            else:
                messagebox.showerror("錯(cuò)誤", f"無(wú)法連接到 {ip}:{port}，請(qǐng)檢查目標(biāo)計(jì)算機(jī)是否運(yùn)行了服務(wù)端程序")
                self.status_var.set("連接失敗")

        except Exception as e:
            messagebox.showerror("錯(cuò)誤", f"連接測(cè)試失敗: {str(e)}")
            self.status_var.set("連接測(cè)試失敗")

    def send_wallpaper(self):
        """發(fā)送壁紙到目標(biāo)計(jì)算機(jī)"""
        # 獲取目標(biāo)IP和端口
        ip = self.ip_entry.get().strip()
        port = self.port_entry.get().strip()

        if not ip:
            messagebox.showerror("錯(cuò)誤", "請(qǐng)輸入目標(biāo)IP地址")
            return

        if not self.current_image_path:
            messagebox.showerror("錯(cuò)誤", "請(qǐng)先選擇壁紙圖片")
            return

        try:
            port = int(port)
        except ValueError:
            messagebox.showerror("錯(cuò)誤", "端口號(hào)必須是數(shù)字")
            return

        try:
            # 讀取圖片文件
            with open(self.current_image_path, 'rb') as f:
                image_data = f.read()

            # 獲取文件名
            filename = os.path.basename(self.current_image_path)

            # 創(chuàng)建連接
            sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            sock.settimeout(10)  # 10秒超時(shí)

            self.status_var.set(f"正在連接到 {ip}:{port}...")
            sock.connect((ip, port))

            # 發(fā)送文件名和文件大小
            file_info = {
                'filename': filename,
                'filesize': len(image_data)
            }
            file_info_json = json.dumps(file_info).encode('utf-8')

            # 先發(fā)送文件信息長(zhǎng)度，然后發(fā)送文件信息
            sock.send(len(file_info_json).to_bytes(4, byteorder='big'))
            sock.send(file_info_json)

            # 發(fā)送圖片數(shù)據(jù)
            self.status_var.set("正在發(fā)送壁紙數(shù)據(jù)...")
            sock.sendall(image_data)

            # 接收響應(yīng)
            response = sock.recv(1024).decode('utf-8')
            sock.close()

            if response == "SUCCESS":
                messagebox.showinfo("成功", f"壁紙已成功發(fā)送到 {ip}")
                self.status_var.set(f"壁紙已發(fā)送到 {ip}")
            else:
                messagebox.showerror("錯(cuò)誤", f"設(shè)置壁紙失敗: {response}")
                self.status_var.set("設(shè)置壁紙失敗")

        except Exception as e:
            messagebox.showerror("錯(cuò)誤", f"發(fā)送壁紙失敗: {str(e)}")
            self.status_var.set("發(fā)送失敗")


def main():
    # 檢查依賴(lài)
    try:
        from PIL import Image
    except ImportError:
        print("請(qǐng)先安裝 Pillow 庫(kù): pip install Pillow")
        return

    # 創(chuàng)建主窗口
    root = tk.Tk()
    app = RemoteWallpaperClient(root)

    # 啟動(dòng)主循環(huán)
    root.mainloop()


if __name__ == "__main__":
    main()

import os
import socket
import threading
import ctypes
import json
import tempfile


class WallpaperServer:
    def __init__(self, port=12345):
        self.port = port
        self.host = '0.0.0.0'  # 監(jiān)聽(tīng)所有接口
        self.running = False
        self.server_socket = None

    def start(self):
        """啟動(dòng)服務(wù)器"""
        try:
            self.server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            self.server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
            self.server_socket.bind((self.host, self.port))
            self.server_socket.listen(5)
            self.running = True

            print(f"壁紙服務(wù)器已啟動(dòng)，監(jiān)聽(tīng)端口 {self.port}")
            print("等待客戶(hù)端連接...")

            while self.running:
                try:
                    client_socket, addr = self.server_socket.accept()
                    print(f"接收到來(lái)自 {addr} 的連接")

                    # 為每個(gè)客戶(hù)端創(chuàng)建新線(xiàn)程
                    client_thread = threading.Thread(
                        target=self.handle_client,
                        args=(client_socket, addr)
                    )
                    client_thread.daemon = True
                    client_thread.start()

                except Exception as e:
                    if self.running:
                        print(f"接受連接時(shí)出錯(cuò): {e}")

        except Exception as e:
            print(f"啟動(dòng)服務(wù)器失敗: {e}")

    def handle_client(self, client_socket, addr):
        """處理客戶(hù)端連接"""
        try:
            # 接收文件信息長(zhǎng)度
            file_info_len = int.from_bytes(client_socket.recv(4), byteorder='big')

            # 接收文件信息
            file_info_json = client_socket.recv(file_info_len).decode('utf-8')
            file_info = json.loads(file_info_json)

            filename = file_info['filename']
            filesize = file_info['filesize']

            print(f"接收文件: {filename}, 大小: {filesize} 字節(jié)")

            # 接收文件數(shù)據(jù)
            received_data = b''
            while len(received_data) < filesize:
                chunk = client_socket.recv(min(4096, filesize - len(received_data)))
                if not chunk:
                    break
                received_data += chunk

            # 保存文件到臨時(shí)目錄
            temp_dir = tempfile.gettempdir()
            temp_file_path = os.path.join(temp_dir, filename)

            with open(temp_file_path, 'wb') as f:
                f.write(received_data)

            print(f"文件已保存到: {temp_file_path}")

            # 設(shè)置壁紙
            if self.set_windows_wallpaper(temp_file_path):
                print("壁紙?jiān)O(shè)置成功")
                client_socket.send("SUCCESS".encode('utf-8'))
            else:
                print("壁紙?jiān)O(shè)置失敗")
                client_socket.send("FAILED: 無(wú)法設(shè)置壁紙".encode('utf-8'))

        except Exception as e:
            print(f"處理客戶(hù)端請(qǐng)求時(shí)出錯(cuò): {e}")
            try:
                client_socket.send(f"FAILED: {str(e)}".encode('utf-8'))
            except:
                pass
        finally:
            client_socket.close()

    def set_windows_wallpaper(self, image_path):
        """設(shè)置Windows壁紙"""
        try:
            # 轉(zhuǎn)換為絕對(duì)路徑
            abs_path = os.path.abspath(image_path)

            # Windows API 常量
            SPI_SETDESKWALLPAPER = 20
            SPIF_UPDATEINIFILE = 0x01
            SPIF_SENDWININICHANGE = 0x02

            # 設(shè)置壁紙
            result = ctypes.windll.user32.SystemParametersInfoW(
                SPI_SETDESKWALLPAPER,
                0,
                abs_path,
                SPIF_UPDATEINIFILE | SPIF_SENDWININICHANGE
            )

            return bool(result)
        except Exception as e:
            print(f"設(shè)置壁紙失敗: {e}")
            return False

    def stop(self):
        """停止服務(wù)器"""
        self.running = False
        if self.server_socket:
            self.server_socket.close()
        print("壁紙服務(wù)器已停止")


def main():
    # 檢查操作系統(tǒng)
    if os.name != 'nt':
        print("此程序僅支持Windows系統(tǒng)")
        return

    # 創(chuàng)建并啟動(dòng)服務(wù)器
    server = WallpaperServer(port=12345)

    try:
        server.start()
    except KeyboardInterrupt:
        print("\n正在停止服務(wù)器...")
        server.stop()


if __name__ == "__main__":
    main()

壁紙服務(wù)器已啟動(dòng)，監(jiān)聽(tīng)端口 12345
等待客戶(hù)端連接...
接收到來(lái)自 192.168.1.100 的連接
接收文件: wallpaper.jpg, 大小: 2048576 字節(jié)
文件已保存到: C:\Users\Username\AppData\Local\Temp\wallpaper.jpg
壁紙?jiān)O(shè)置成功

pip install Pillow