144 lines
5.5 KiB
Python
144 lines
5.5 KiB
Python
#!/usr/bin/env python
|
|
"""WinTables.py
|
|
|
|
Routines for detecting and handling poker client windows for MS Windows.
|
|
"""
|
|
# Copyright 2008 - 2009, Ray E. Barker
|
|
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
|
|
########################################################################
|
|
|
|
# Standard Library modules
|
|
import re
|
|
|
|
# pyGTK modules
|
|
import pygtk
|
|
import gtk
|
|
|
|
# Other Library modules
|
|
import win32gui
|
|
import win32process
|
|
import win32api
|
|
import win32con
|
|
import win32security
|
|
|
|
# FreePokerTools modules
|
|
from TableWindow import Table_Window
|
|
|
|
# We don't know the border width or title bar height
|
|
# so we guess here. We can probably get these from a windows call.
|
|
b_width = 3
|
|
tb_height = 29
|
|
|
|
class Table(Table_Window):
|
|
|
|
def find_table_parameters(self, search_string):
|
|
"""Finds poker client window with the given table name."""
|
|
titles = {}
|
|
win32gui.EnumWindows(win_enum_handler, titles)
|
|
for hwnd in titles:
|
|
if re.search(search_string, titles[hwnd]):
|
|
if 'History for table:' in titles[hwnd]: continue # Everleaf Network HH viewer window
|
|
if 'HUD:' in titles[hwnd]: continue # FPDB HUD window
|
|
if 'Chat:' in titles[hwnd]: continue # Some sites (FTP? PS? Others?) have seperable or seperately constructed chat windows
|
|
self.window = hwnd
|
|
break
|
|
|
|
if self.window == None:
|
|
print "Window %s not found. Skipping." % search_string
|
|
return None
|
|
|
|
(x, y, width, height) = win32gui.GetWindowRect(hwnd)
|
|
print "x = %s y = %s width = %s height = %s" % (x, y, width, height)
|
|
self.x = int(x) + b_width
|
|
self.y = int(y) + tb_height
|
|
self.height = int(height) - b_width - tb_height
|
|
self.width = int(width) - 2*b_width
|
|
|
|
self.exe = self.get_nt_exe(hwnd)
|
|
self.title = titles[hwnd]
|
|
self.site = ""
|
|
self.hud = None
|
|
self.number = gtk.gdk.window_foreign_new(long(self.window))
|
|
|
|
def get_geometry(self):
|
|
|
|
if not win32gui.IsWindow(self.window): # window closed
|
|
return None
|
|
|
|
try:
|
|
(x, y, width, height) = win32gui.GetWindowRect(hwnd)
|
|
return {'x' : int(x) + b_width,
|
|
'y' : int(y) + tb_height,
|
|
'width' : int(height) - b_width - tb_height,
|
|
'height' : int(width) - 2*b_width
|
|
}
|
|
except:
|
|
return None
|
|
|
|
def get_window_title(self):
|
|
return win32gui.GetWindowText(self.window)
|
|
|
|
def get_nt_exe(self, hwnd):
|
|
"""Finds the name of the executable that the given window handle belongs to."""
|
|
|
|
# Request privileges to enable "debug process", so we can later use PROCESS_VM_READ, retardedly required to GetModuleFileNameEx()
|
|
priv_flags = win32security.TOKEN_ADJUST_PRIVILEGES | win32security.TOKEN_QUERY
|
|
hToken = win32security.OpenProcessToken (win32api.GetCurrentProcess(), priv_flags)
|
|
# enable "debug process"
|
|
privilege_id = win32security.LookupPrivilegeValue (None, win32security.SE_DEBUG_NAME)
|
|
old_privs = win32security.AdjustTokenPrivileges (hToken, 0, [(privilege_id, win32security.SE_PRIVILEGE_ENABLED)])
|
|
|
|
# Open the process, and query it's filename
|
|
processid = win32process.GetWindowThreadProcessId(hwnd)
|
|
pshandle = win32api.OpenProcess(win32con.PROCESS_QUERY_INFORMATION | win32con.PROCESS_VM_READ, False, processid[1])
|
|
exename = win32process.GetModuleFileNameEx(pshandle, 0)
|
|
|
|
# clean up
|
|
win32api.CloseHandle(pshandle)
|
|
win32api.CloseHandle(hToken)
|
|
|
|
return exename
|
|
def win_enum_handler(hwnd, titles):
|
|
titles[hwnd] = win32gui.GetWindowText(hwnd)
|
|
|
|
def topify_window(hud, window):
|
|
"""Set the specified gtk window to stayontop in MS Windows."""
|
|
|
|
def windowEnumerationHandler(hwnd, resultList):
|
|
'''Callback for win32gui.EnumWindows() to generate list of window handles.'''
|
|
resultList.append((hwnd, win32gui.GetWindowText(hwnd)))
|
|
|
|
unique_name = 'unique name for finding this window'
|
|
real_name = window.get_title()
|
|
window.set_title(unique_name)
|
|
tl_windows = []
|
|
win32gui.EnumWindows(windowEnumerationHandler, tl_windows)
|
|
|
|
for w in tl_windows:
|
|
if w[1] == unique_name:
|
|
hud.main_window.parentgdkhandle = gtk.gdk.window_foreign_new(long(hud.table.number))
|
|
hud.main_window.gdkhandle = gtk.gdk.window_foreign_new(w[0])
|
|
hud.main_window.gdkhandle.set_transient_for(hud.main_window.parentgdkhandle)
|
|
|
|
style = win32gui.GetWindowLong(self.table.number, win32con.GWL_EXSTYLE)
|
|
style |= win32con.WS_CLIPCHILDREN
|
|
win32gui.SetWindowLong(hud.table.number, win32con.GWL_EXSTYLE, style)
|
|
break
|
|
|
|
window.set_title(real_name)
|
|
|