在Python控制台窗口中获取鼠标点击

Question

有没有什么方法可以在Python中获取控制台窗口的鼠标点击事件？我知道这是可以做到的，因为如果你在Windows系统上，可以打开命令提示符（cmd）然后输入“edit”。那么在Python中要怎么实现这个呢？谢谢。

Answer 1

是的，不过这需要花费不少精力。

http://sourceforge.net/projects/pywin32/files/

它的工作方式就像用C语言和Win32API对话一样。

如果你在网上搜索函数名称，微软的文档其实写得挺好的。

pywin32里面有一个叫做win32console_demo.py的文件。

要启用鼠标输入，你只需在创建conin之后添加以下一行代码。

conin.SetConsoleMode(ENABLE_WINDOW_INPUT | ENABLE_MOUSE_INPUT)

win32console_demo.py已经修改以启用鼠标输入。

import win32con
import win32file
from win32console import *
import traceback, time


virtual_keys={}
for k,v in win32con.__dict__.items():
    if k.startswith('VK_'):
        virtual_keys[v]=k 

free_console=True
try:
    AllocConsole()
except error, exc:
    if exc.winerror!=5:
        raise
    ## only free console if one was created successfully
    free_console=False

stdout=GetStdHandle(STD_OUTPUT_HANDLE)

conin=PyConsoleScreenBufferType( win32file.CreateFile( "CONIN$", win32con.GENERIC_READ|win32con.GENERIC_WRITE, win32con.FILE_SHARE_READ, None, win32con.OPEN_EXISTING, 0, 0))

conin.SetConsoleMode(ENABLE_WINDOW_INPUT | ENABLE_MOUSE_INPUT)

newbuffer=CreateConsoleScreenBuffer()
newbuffer.SetConsoleActiveScreenBuffer()
newbuffer.SetConsoleTextAttribute(FOREGROUND_RED|FOREGROUND_INTENSITY
        |BACKGROUND_GREEN|BACKGROUND_INTENSITY)
newbuffer.WriteConsole('This is a new screen buffer\n')

newbuffer.SetConsoleTextAttribute(FOREGROUND_RED|FOREGROUND_INTENSITY
        |BACKGROUND_GREEN|BACKGROUND_INTENSITY)
newbuffer.WriteConsole('Press some keys, click some characters with the mouse\n')

newbuffer.SetConsoleTextAttribute(FOREGROUND_BLUE|FOREGROUND_INTENSITY
        |BACKGROUND_RED|BACKGROUND_INTENSITY)
newbuffer.WriteConsole('Hit "Esc" key to quit\n')


breakout=False
while not breakout:
    input_records=conin.ReadConsoleInput(10)
    for input_record in input_records:
        if input_record.EventType==KEY_EVENT:
            if input_record.KeyDown:
                if input_record.Char=='\0':
                    newbuffer.WriteConsole(virtual_keys.get(input_record.VirtualKeyCode, 'VirtualKeyCode: %s' %input_record.VirtualKeyCode))
                else:
                    newbuffer.WriteConsole(input_record.Char)
                if input_record.VirtualKeyCode==win32con.VK_ESCAPE:
                    breakout=True
                    break
        elif input_record.EventType==MOUSE_EVENT:
            if input_record.EventFlags==0:  ## 0 indicates a button event
                if input_record.ButtonState!=0:   ## exclude button releases
                    pos=input_record.MousePosition
                    # switch the foreground and background colors of the character that was clicked
                    attr=newbuffer.ReadConsoleOutputAttribute(Length=1, ReadCoord=pos)[0]
                    new_attr=attr
                    if attr&FOREGROUND_BLUE:
                        new_attr=(new_attr&~FOREGROUND_BLUE)|BACKGROUND_BLUE
                    if attr&FOREGROUND_RED:
                        new_attr=(new_attr&~FOREGROUND_RED)|BACKGROUND_RED
                    if attr&FOREGROUND_GREEN:
                        new_attr=(new_attr&~FOREGROUND_GREEN)|BACKGROUND_GREEN

                    if attr&BACKGROUND_BLUE:
                        new_attr=(new_attr&~BACKGROUND_BLUE)|FOREGROUND_BLUE
                    if attr&BACKGROUND_RED:
                        new_attr=(new_attr&~BACKGROUND_RED)|FOREGROUND_RED
                    if attr&BACKGROUND_GREEN:
                        new_attr=(new_attr&~BACKGROUND_GREEN)|FOREGROUND_GREEN
                    newbuffer.WriteConsoleOutputAttribute((new_attr,),pos)
        else:
            newbuffer.WriteConsole(str(input_record))
    time.sleep(0.1)

newbuffer.Close()

if free_console:
     FreeConsole()

Answer 2

之前的两个回答都是对的：在Windows上你可以用pywin32来处理鼠标输入，而在Linux上可以用curses。不过，这两个工具在对方的平台上是不能用的。

如果你想要一个可以同时在两个平台上使用，或者是更简单的接口，你可以试试我最近写的一个包——asciimatics。这个包里的Screen类可以跨平台处理控制台的鼠标输入，它把上面提到的两种解决方案结合在一起了。

Answer 3

我个人不太喜欢上一个回答提到的PyWin32，因为它只在第一次导入时有效，之后就无法再导入它的主要模块了。
而且它也不支持跨平台。

curses是一个更好的选择，虽然它和python27的兼容性不是很好，但我已经在Wine中成功运行了它：

import _curses # _curses.pyd supplied locally for python27 win32
import curses 

screen = curses.initscr() 
#curses.noecho() 
curses.curs_set(0) 
screen.keypad(1) 
curses.mousemask(curses.ALL_MOUSE_EVENTS)

screen.addstr("This is a Sample Curses Script\n\n") 

key=0
while key!=27: # Esc to close
    key = screen.getch() 
    #screen.erase()
    if key == curses.KEY_MOUSE:
        _, mx, my, _, _ = curses.getmouse()
        y, x = screen.getyx()
        screen.addstr('mx, my = %i,%i                \r'%(mx,my))
    screen.refresh()

curses.endwin()

在这个例子中，你需要点击、拖动并释放才能让事件被识别，但我觉得有办法可以解决这个问题。

我现在只是自己在玩这个（我不想和PyQt或PyGLFW搞得太复杂）

编辑：
我已经成功让它按预期工作，并更新了代码。你不再需要拖动才能让事件被识别了。