154 lines
6.3 KiB
Python
154 lines
6.3 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
"""Discover_TableWindow.py
|
||
|
|
|
||
|
|
Inspects the currently open windows and finds those of interest to us--that is
|
||
|
|
poker table windows from supported sites. Returns a list
|
||
|
|
of Table_Window objects representing the windows found.
|
||
|
|
"""
|
||
|
|
# 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 os
|
||
|
|
import sys
|
||
|
|
|
||
|
|
# pyGTK modules
|
||
|
|
import pygtk
|
||
|
|
import gtk
|
||
|
|
import gobject
|
||
|
|
|
||
|
|
# FreePokerTools modules
|
||
|
|
import Configuration
|
||
|
|
#if os.name == "posix":
|
||
|
|
# import XTables
|
||
|
|
#elif os.name == "nt":
|
||
|
|
# import WinTables
|
||
|
|
|
||
|
|
# Global used for figuring out the current game being played from the title
|
||
|
|
# The dict key is the fpdb name for the game
|
||
|
|
# The list is the names for those games used by the supported poker sites
|
||
|
|
# This is currently only used for HORSE, so it only needs to support those
|
||
|
|
# games on PokerStars and Full Tilt.
|
||
|
|
game_names = { #fpdb name Stars Name FTP Name
|
||
|
|
"holdem" : ("Hold\'em" , ),
|
||
|
|
"omahahilo" : ("Omaha H/L" , ),
|
||
|
|
"studhilo" : ("Stud H/L" , ),
|
||
|
|
"razz" : ("Razz" , ),
|
||
|
|
"studhi" : ("Stud" , "Stud Hi")
|
||
|
|
}
|
||
|
|
|
||
|
|
# A window title might have our table name + one of theses words/
|
||
|
|
# phrases. If it has this word in the title, it is not a table.
|
||
|
|
bad_words = ('History for table:', 'HUD:', 'Chat:')
|
||
|
|
|
||
|
|
# Here are the custom signals we define for allowing the 'client watcher'
|
||
|
|
# thread to communicate with the gui thread. Any time a poker client is
|
||
|
|
# is moved, resized, or closed on of these signals is emitted to the
|
||
|
|
# HUD main window.
|
||
|
|
gobject.signal_new("client_moved", gtk.Window,
|
||
|
|
gobject.SIGNAL_RUN_LAST,
|
||
|
|
gobject.TYPE_NONE,
|
||
|
|
(gobject.TYPE_PYOBJECT,))
|
||
|
|
|
||
|
|
gobject.signal_new("client_resized", gtk.Window,
|
||
|
|
gobject.SIGNAL_RUN_LAST,
|
||
|
|
gobject.TYPE_NONE,
|
||
|
|
(gobject.TYPE_PYOBJECT,))
|
||
|
|
|
||
|
|
gobject.signal_new("client_destroyed", gtk.Window,
|
||
|
|
gobject.SIGNAL_RUN_LAST,
|
||
|
|
gobject.TYPE_NONE,
|
||
|
|
(gobject.TYPE_PYOBJECT,))
|
||
|
|
|
||
|
|
# Each TableWindow object must have the following attributes correctly populated:
|
||
|
|
# tw.name = the table name from the title bar, which must to match the table name
|
||
|
|
# from the corresponding hand history.
|
||
|
|
# tw.site = the site name, e.g. PokerStars, FullTilt. This must match the site
|
||
|
|
# name specified in the config file.
|
||
|
|
# tw.number = This is the system id number for the client table window in the
|
||
|
|
# format that the system presents it. This is Xid in Xwindows and
|
||
|
|
# hwnd in Microsoft Windows.
|
||
|
|
# tw.title = The full title from the window title bar.
|
||
|
|
# tw.width, tw.height = The width and height of the window in pixels. This is
|
||
|
|
# the internal width and height, not including the title bar and
|
||
|
|
# window borders.
|
||
|
|
# tw.x, tw.y = The x, y (horizontal, vertical) location of the window relative
|
||
|
|
# to the top left of the display screen. This also does not include the
|
||
|
|
# title bar and window borders. To put it another way, this is the
|
||
|
|
# screen location of (0, 0) in the working window.
|
||
|
|
|
||
|
|
class Table_Window(object):
|
||
|
|
def __init__(self, table_name = None, tournament = None, table_number = None):
|
||
|
|
|
||
|
|
if table_name != None:
|
||
|
|
search_string = table_name
|
||
|
|
self.name = table_name
|
||
|
|
self.tournament = None
|
||
|
|
self.table = None
|
||
|
|
elif tournament != None and table_number != None:
|
||
|
|
print "tournament %s, table %s" % (tournament, table_number)
|
||
|
|
self.tournament = int(tournament)
|
||
|
|
self.table = int(table_number)
|
||
|
|
self.name = "%s - %s" % (self.tournament, self.table)
|
||
|
|
search_string = "%s.+Table\s%s" % (tournament, table_number)
|
||
|
|
else:
|
||
|
|
return None
|
||
|
|
|
||
|
|
self.find_table_parameters(search_string)
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
# __str__ method for testing
|
||
|
|
temp = 'TableWindow object\n'
|
||
|
|
temp = temp + " name = %s\n site = %s\n number = %s\n title = %s\n" % (self.name, self.site, self.number, self.title)
|
||
|
|
# temp = temp + " game = %s\n structure = %s\n max = %s\n" % (self.game, self.structure, self.max)
|
||
|
|
temp = temp + " width = %d\n height = %d\n x = %d\n y = %d\n" % (self.width, self.height, self.x, self.y)
|
||
|
|
if getattr(self, 'tournament', 0):
|
||
|
|
temp = temp + " tournament = %d\n table = %d" % (self.tournament, self.table)
|
||
|
|
return temp
|
||
|
|
|
||
|
|
def get_game(self):
|
||
|
|
title = self.get_window_title()
|
||
|
|
print title
|
||
|
|
for game, names in game_names.iteritems():
|
||
|
|
for name in names:
|
||
|
|
if name in title:
|
||
|
|
return game
|
||
|
|
return None
|
||
|
|
|
||
|
|
def check_geometry(self):
|
||
|
|
new_geo = self.get_geometry()
|
||
|
|
|
||
|
|
if new_geo == None: # window destroyed
|
||
|
|
return "client_destroyed"
|
||
|
|
|
||
|
|
elif self.x != new_geo['x'] or self.y != new_geo['y']: # window moved
|
||
|
|
self.x = new_geo['x']
|
||
|
|
self.y = new_geo['y']
|
||
|
|
return "client_moved"
|
||
|
|
|
||
|
|
elif self.width != new_geo['width'] or self.height != new_geo['height']: # window resized
|
||
|
|
self.width = new_geo['width']
|
||
|
|
self.height = new_geo['height']
|
||
|
|
return "client_resized"
|
||
|
|
|
||
|
|
else: return False # window not changed
|
||
|
|
|
||
|
|
def check_bad_words(self, title):
|
||
|
|
for word in bad_words:
|
||
|
|
if word in title: return True
|
||
|
|
return False
|