fpdb/pyfpdb/XTables.py

#!/usr/bin/env python
"""XTables.py

XWindows specific methods for TableWindows Class.
"""
#    Copyright 2008 - 2010, 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 gtk

#    Other Library modules
import Xlib.display
from Xlib import Xatom

#    FreePokerTools modules
from TableWindow import Table_Window

#    We might as well do this once and make them globals
disp = Xlib.display.Display()
root = disp.screen().root
name_atom = disp.get_atom("WM_NAME", 1)
icon_atom = disp.get_atom("WM_ICON_NAME", 1)

class Table(Table_Window):

    def find_table_parameters(self, search_string):
        self.window = None
        done_looping = False
        for outside in root.query_tree().children:
            for inside in outside.query_tree().children:
                if done_looping: break
#                if inside.get_wm_name() and re.search(search_string, inside.get_wm_name()):
                if inside.get_wm_name() and re.search(search_string, inside.get_wm_icon_name()):
                    print "inside = ", inside
                    print "\n".join(dir(inside))
                    for atom in inside.list_properties(): print atom, disp.get_atom_name(atom), inside.get_full_property(atom, Xatom.STRING).value
                    if hasattr(inside, "window"): print "window = ", str(inside.window)
                    if self.check_bad_words(inside.get_wm_name()): continue
                    self.window = inside
                    self.parent = outside
                    done_looping = True
                    break

        if self.window == None or self.parent == None:
            print "Window %s not found. Skipping." % search_string
            return None

        my_geo = self.window.get_geometry()
        pa_geo = self.parent.get_geometry()

        self.x      = pa_geo.x + my_geo.x
        self.y      = pa_geo.y + my_geo.y
        self.width  = my_geo.width
        self.height = my_geo.height
        self.exe    = self.window.get_wm_class()[0]
        self.title  = self.window.get_wm_name()
        self.site   = ""
        self.hud    = None
        self.number = self.get_xid()
        
    def get_xid(self):
        mo = re.match('Xlib\.display\.Window\(([\dxabcdef]+)', str(self.window))
        if not mo:
            return None
        else:
            return int( mo.group(1), 0)

    def get_geometry(self):
        try:
            my_geo = self.window.get_geometry()
            pa_geo = self.parent.get_geometry()
            return {'x'      : pa_geo.x + my_geo.x,
                    'y'      : pa_geo.y + my_geo.y,
                    'width'  : my_geo.width,
                    'height' : my_geo.height
                   }
        except:
            return None

    def get_window_title(self):
        return self.window.get_wm_name()
    
    def topify(self, hud):
        hud.main_window.gdkhandle = gtk.gdk.window_foreign_new(hud.main_window.window.xid)
        hud.main_window.gdkhandle.set_transient_for(self.gdk_handle)