关于python写显示器输入源切换的问题

XinCb

最近有几台电脑需要共用一台显示器，每次手动osd里面改。比较麻烦
于是用gpt写了一个方便快捷切换。
目前遇到一个问题，就是只能从HDMI设备切换到另外俩个，无法从另两个输入源切换回HDMI。
大佬看看是什么问题。下是代码

[Python] 纯文本查看 复制代码

import sys
import customtkinter as ctk
from monitorcontrol import get_monitors

# 定义切换信号源的函数
def switch_input_source(source):
    monitors = get_monitors()
    for monitor in monitors:
        with monitor:
            monitor.set_input_source(source)

# 定义按钮点击事件
def on_click(source):
    # 直接切换输入源
    switch_input_source(source)
    # 更新标签显示当前输入源
    label.configure(text=f"当前输入源: {source}")

# 创建主窗口
class InputSwitcher(ctk.CTk):
    def __init__(self):
        super().__init__()

        self.initUI()
        self.protocol("WM_DELETE_WINDOW", self.on_closing)

    def initUI(self):
        self.title('输入源切换器')
        self.geometry('300x250')

        global label
        label = ctk.CTkLabel(self, text='当前输入源: 未知')
        label.pack(pady=20)

        hdmi_button = ctk.CTkButton(self, text='HDMI', command=lambda: on_click(1))
        hdmi_button.pack(pady=10)

        dp_button = ctk.CTkButton(self, text='DisplayPort', command=lambda: on_click(2))
        dp_button.pack(pady=10)

        typec_button = ctk.CTkButton(self, text='Type-C', command=lambda: on_click(3))
        typec_button.pack(pady=10)

    def on_closing(self):
        self.destroy()
        sys.exit()

# 主函数
if __name__ == '__main__':
    ctk.set_appearance_mode("System")
    ctk.set_default_color_theme("blue")
    app = InputSwitcher()
    app.mainloop()

XinCb

已解决

成品：https://www.52pojie.cn/thread-1967992-1-1.html
源码：

[Python] 纯文本查看 复制代码

import wmi
from monitorcontrol import get_monitors, InputSource
import customtkinter as ctk
import tkinter as tk
import json
import os
import threading
from PIL import Image, ImageTk
import pystray
from pystray import MenuItem as item
import keyboard  # 用于捕获全局热键
import sys

# 配置文件路径
CONFIG_FILE = "C:/HDMIfig.json"

# 保存输入源的编号
button_sources = {"HDMI": None, "DP": None, "USB-C": None}

# 托盘图标对象（全局变量，确保只创建一次）
tray_icon = None

# 图标路径
if getattr(sys, 'frozen', False):
    ICON_PATH = os.path.join(sys._MEIPASS, "hdmi", "hdmi.ico")
else:
    ICON_PATH = "hdmi/hdmi.ico"

# 获取显示器设备信息 (通过 WMI)
def get_display_info():
    wmi_service = wmi.WMI(namespace='wmi')
    displays = wmi_service.WmiMonitorID()
    return [display.InstanceName for display in displays]

# 列出所有可能的输入源，包括 DDC/CI 协议常用编号和厂商自定义源
def list_input_sources():
    input_sources = set()

    # 方式 1: 从 monitorcontrol 的 InputSource 枚举中获取输入源
    for source in InputSource:
        input_sources.add((source.value, source.name))

    # 方式 2: 常见输入源的硬编码列表，扩展到更多厂商常见编号
    common_sources = {
        1: "HDMI 1",
        2: "HDMI 2",
        3: "DisplayPort",
        4: "DVI",
        5: "VGA",
        6: "USB-C",
        7: "Thunderbolt",
        8: "Composite",
        9: "Component",
        10: "SCART",
        11: "S-Video",
        12: "Mini DisplayPort",
        13: "eDP",
        14: "MHL",
        15: "Wireless Display",
        16: "USB 3.0",
        17: "HDMI 3",
        18: "DisplayPort 2",
        19: "Thunderbolt 3",
    }

    for key, value in common_sources.items():
        input_sources.add((key, value))

    # 方式 3: 通过 WMI 获取设备信息
    display_info = get_display_info()
    for index, display in enumerate(display_info):
        input_sources.add((index + 100, f"Display {index + 1}: {display}"))

    return input_sources

# 切换显示器输入源
def switch_input_source(source_value):
    try:
        monitors = get_monitors()
        for monitor in monitors:
            with monitor:
                monitor.set_input_source(InputSource(source_value))
        output_textbox.insert(tk.END, f"成功切换到 {source_value} 输入源\n")
    except Exception as e:
        output_textbox.insert(tk.END, f"切换失败: {e}\n")

# 保存自定义按钮源编号并更新按钮名称
def save_source(button_name, source_value):
    button_sources[button_name] = source_value
    update_button_text(button_name)
    save_config()
    output_textbox.insert(tk.END, f"已保存 {button_name} 为输入源编号 {source_value}\n")

# 更新按钮文本
def update_button_text(button_name):
    if button_sources[button_name] is not None:
        if button_name == "HDMI":
            hdmi_button.configure(text=f"HDMI ({button_sources[button_name]})")
        elif button_name == "DP":
            dp_button.configure(text=f"DP ({button_sources[button_name]})")
        elif button_name == "USB-C":
            usbc_button.configure(text=f"USB-C ({button_sources[button_name]})")

# 获取输入框中的源编号，并在点击按钮时切换
def switch_and_save_source(button_name, entry_widget):
    # 获取当前输入框的值
    source_value = entry_widget.get()
    if source_value.isdigit():
        source_value = int(source_value)
        save_source(button_name, source_value)
        switch_input_source(source_value)
    elif button_sources[button_name] is not None:
        # 如果输入框没有手动值，则使用已保存的默认值
        switch_input_source(button_sources[button_name])
    else:
        output_textbox.insert(tk.END, "请输入有效的数字编号\n")

# 保存配置到文件
def save_config():
    with open(CONFIG_FILE, 'w') as f:
        json.dump(button_sources, f)

# 从文件加载配置
def load_config():
    global button_sources
    if os.path.exists(CONFIG_FILE):
        with open(CONFIG_FILE, 'r') as f:
            button_sources = json.load(f)
        output_textbox.insert(tk.END, "配置已加载\n")
        # 确保加载后更新按钮名称
        update_button_text("HDMI")
        update_button_text("DP")
        update_button_text("USB-C")
    else:
        output_textbox.insert(tk.END, "没有找到配置文件，使用默认设置\n")
        # 设置默认 HDMI 和 DP 编号
        button_sources["HDMI"] = 16
        button_sources["DP"] = 1
        button_sources["USB-C"] = None
        update_button_text("HDMI")
        update_button_text("DP")
        update_button_text("USB-C")

    # 读取当前显示器的输入源
    get_current_input_source()

# 读取当前显示器的输入源
def get_current_input_source():
    try:
        monitors = get_monitors()
        for monitor in monitors:
            with monitor:
                current_source = monitor.get_input_source()
                output_textbox.insert(tk.END, f"当前输入源编号为: {current_source.value} ({current_source.name})\n")
                return current_source.value
    except Exception as e:
        output_textbox.insert(tk.END, f"无法读取当前输入源: {e}\n")
        return None

# 恢复窗口
def restore_window(icon, item):
    global tray_icon
    root.after(0, root.deiconify)
    tray_icon.stop()

# 最小化到托盘功能
def minimize_to_tray():
    global tray_icon
    root.withdraw()  # 隐藏窗口
    if tray_icon is None:  # 确保托盘图标只创建一次
        image = Image.open(ICON_PATH)
        menu = (item('打开主窗口', restore_window), item('关闭程序', quit_application))
        tray_icon = pystray.Icon("Monitor Switcher", image, "Monitor Switcher", menu)
        threading.Thread(target=tray_icon.run, daemon=True).start()

# 退出程序
def quit_application(icon=None, item=None):
    global tray_icon
    if tray_icon:
        tray_icon.stop()
    root.quit()

# 处理窗口最小化事件
def on_minimize(event):
    minimize_to_tray()

# 切换到 HDMI 的快捷键处理
def switch_to_hdmi():
    if button_sources["HDMI"] is not None:
        switch_input_source(button_sources["HDMI"])

# 切换到 DP 的快捷键处理
def switch_to_dp():
    if button_sources["DP"] is not None:
        switch_input_source(button_sources["DP"])

# 切换到 USB-C 的快捷键处理
def switch_to_usbc():
    if button_sources["USB-C"] is not None:
        switch_input_source(button_sources["USB-C"])

# 注册全局热键
def register_hotkeys():
    keyboard.add_hotkey('alt+1', switch_to_hdmi)
    keyboard.add_hotkey('alt+2', switch_to_dp)
    keyboard.add_hotkey('alt+3', switch_to_usbc)

# GUI 界面
def create_gui():
    global root
    # 初始化主窗口
    root = ctk.CTk()
    root.title("显示器输入源切换器")
    root.geometry("300x250")

    # 设置窗口图标
    root.iconbitmap(ICON_PATH)

    # 捕捉窗口最小化事件
    root.protocol("WM_DELETE_WINDOW", quit_application)  # 点击关闭时退出程序
    root.bind("<Unmap>", lambda event: on_minimize(event) if root.state() == 'iconic' else None)  # 最小化事件

    # 创建按钮和对应的输入框
    global hdmi_button, dp_button, usbc_button

    # HDMI
    hdmi_frame = ctk.CTkFrame(root)
    hdmi_frame.pack(pady=2)

    hdmi_button = ctk.CTkButton(hdmi_frame, text="HDMI", command=lambda: switch_and_save_source("HDMI", hdmi_entry))
    hdmi_button.pack(side="left", padx=1)

    hdmi_entry = ctk.CTkEntry(hdmi_frame, width=30)
    hdmi_entry.pack(side="left")

    # DP
    dp_frame = ctk.CTkFrame(root)
    dp_frame.pack(pady=2)

    dp_button = ctk.CTkButton(dp_frame, text="DP", command=lambda: switch_and_save_source("DP", dp_entry))
    dp_button.pack(side="left", padx=1)

    dp_entry = ctk.CTkEntry(dp_frame, width=30)
    dp_entry.pack(side="left")

    # USB-C
    usbc_frame = ctk.CTkFrame(root)
    usbc_frame.pack(pady=2)

    usbc_button = ctk.CTkButton(usbc_frame, text="USB-C", command=lambda: switch_and_save_source("USB-C", usbc_entry))
    usbc_button.pack(side="left", padx=1)

    usbc_entry = ctk.CTkEntry(usbc_frame, width=30)
    usbc_entry.pack(side="left")

    # 输出信息框
    global output_textbox
    output_textbox = ctk.CTkTextbox(root, height=200, width=400)
    output_textbox.pack(pady=2)

    # 列出所有可能的输入源
    output_textbox.insert(tk.END, "可能的输入源:\n")
    input_sources = list_input_sources()
    for source in input_sources:
        output_textbox.insert(tk.END, f"{source[1]} (Code: {source[0]})\n")

    # 加载配置并更新按钮文本
    load_config()

    # 注册热键
    register_hotkeys()

    root.mainloop()

if __name__ == "__main__":
    create_gui()

xixicoco

ddc代码不支持非hdmi的运行，一般要自定义的，很多厂家不支持，都是私有代码

XinCb

搞定，由于不同显示器的输出源编号不同，只能是列出所有可能的，然后手动输入确定好后保存到按钮上，比如我的HDMI是16，1-15都是切到DP/tpyc的
虽然麻烦点，最终还是达到了我的需求。

hubindong

这个想法很好，正好也有类似需求。

bachelor66

大佬，有完整的码学习吗？                              

XinCb

bachelor66 发表于 2024-9-26 08:42
大佬，有完整的码学习吗？

还有一些小问题没完成

xmp788

程序在这电脑上运行，显示器切换到其他电脑去了，如何切换回来  不会用  

chenzhigang

插眼 后续学习一下

52bulesky

牛逼 牛逼 牛逼 牛逼

llyaomo

xmp788 发表于 2024-9-26 10:18
程序在这电脑上运行，显示器切换到其他电脑去了，如何切换回来  不会用

有道理，程序在A电脑上，把显示器切给了B，那怎么切回来？B电脑也放个软件吗？这样的话，我觉得，还是不如遥控器或者显示器直接切换，不算更麻烦的吧？