2008-08-19 00:53:25 +02:00
#!/usr/bin/env python
2009-08-08 10:07:30 +02:00
# -*- coding: utf-8 -*-
2008-08-19 00:53:25 +02:00
""" Configuration.py
Handles HUD configuration files .
"""
2009-03-05 02:04:23 +01:00
# Copyright 2008, 2009, Ray E. Barker
2008-08-19 00:53:25 +02:00
#
# 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
2009-09-16 07:13:42 +02:00
from __future__ import with_statement
2008-09-27 01:21:38 +02:00
import os
import sys
2008-12-13 23:37:23 +01:00
import inspect
2008-10-05 02:22:53 +02:00
import string
2008-09-29 16:12:04 +02:00
import traceback
2008-09-15 22:31:55 +02:00
import shutil
2008-08-19 00:53:25 +02:00
import xml . dom . minidom
from xml . dom . minidom import Node
2009-08-12 02:46:39 +02:00
import logging , logging . config
2009-10-21 19:24:46 +02:00
import ConfigParser
2009-11-09 04:53:10 +01:00
##############################################################################
# Functions for finding config files and setting up logging
# Also used in other modules that use logging.
def get_default_config_path ( ) :
""" Returns the path where the fpdb config file _should_ be stored. """
if os . name == ' posix ' :
config_path = os . path . join ( os . path . expanduser ( " ~ " ) , ' .fpdb ' )
elif os . name == ' nt ' :
config_path = os . path . join ( os . environ [ " APPDATA " ] , ' fpdb ' )
else : config_path = False
return config_path
def get_exec_path ( ) :
""" Returns the path to the fpdb.(py|exe) file we are executing """
if hasattr ( sys , " frozen " ) : # compiled by py2exe
return os . path . dirname ( sys . executable )
else :
return os . path . dirname ( sys . path [ 0 ] )
def get_config ( file_name , fallback = True ) :
2009-11-24 01:29:56 +01:00
""" Looks in exec dir and in self.default_config_path for a config file. """
config_path = os . path . join ( DIR_SELF , file_name ) # look in exec dir
if os . path . exists ( config_path ) and os . path . isfile ( config_path ) :
return config_path # there is a file in the exec dir so we use it
else :
config_path = os . path . join ( DIR_CONFIG , file_name ) # look in config dir
if os . path . exists ( config_path ) and os . path . isfile ( config_path ) :
2009-11-09 04:53:10 +01:00
return config_path
# No file found
if not fallback :
return False
2009-11-24 01:29:56 +01:00
# OK, fall back to the .example file, should be in the exec dir
if os . path . exists ( os . path . join ( DIR_SELF , file_name + " .example " ) ) :
2009-11-09 04:53:10 +01:00
try :
2009-11-24 01:29:56 +01:00
shutil . copyfile ( os . path . join ( DIR_SELF , file_name + " .example " ) , os . path . join ( DIR_CONFIG , file_name ) )
2009-11-09 04:53:10 +01:00
print " No %s found, using %s .example. \n " % ( file_name , file_name )
2009-11-24 01:29:56 +01:00
print " A %s file has been created. You will probably have to edit it. " % os . path . join ( DIR_CONFIG , file_name )
log . error ( " No %s found, using %s .example. \n " % ( file_name , file_name ) )
2009-11-09 04:53:10 +01:00
except :
print " No %s found, cannot fall back. Exiting. \n " % file_name
2009-11-24 01:29:56 +01:00
log . critical ( " No %s found, cannot fall back. Exiting. \n " % file_name )
2009-11-09 04:53:10 +01:00
sys . exit ( )
return file_name
2009-11-10 01:30:23 +01:00
def get_logger ( file_name , config = " config " , fallback = False ) :
2009-11-09 04:53:10 +01:00
conf = get_config ( file_name , fallback = fallback )
if conf :
try :
logging . config . fileConfig ( conf )
2009-11-10 01:30:23 +01:00
log = logging . getLogger ( config )
2009-11-09 04:53:10 +01:00
return log
except :
pass
log = logging . basicConfig ( )
log = logging . getLogger ( )
2009-11-24 01:29:56 +01:00
log . error ( " basicConfig logger initialised " )
2009-11-09 04:53:10 +01:00
return log
2009-11-24 01:29:56 +01:00
def check_dir ( path , create = True ) :
""" Check if a dir exists, optionally creates if not. """
if os . path . exists ( path ) :
if os . path . isdir ( path ) :
return path
else :
return False
if create :
print " creating directory %s " % path
else :
return False
2009-10-21 19:24:46 +02:00
2009-11-02 10:06:54 +01:00
########################################################################
# application wide consts
APPLICATION_NAME_SHORT = ' fpdb '
APPLICATION_VERSION = ' xx.xx.xx '
2009-08-12 02:46:39 +02:00
2009-11-24 01:29:56 +01:00
DIR_SELF = get_exec_path ( )
DIR_CONFIG = check_dir ( get_default_config_path ( ) )
DIR_DATABASE = check_dir ( os . path . join ( DIR_CONFIG , ' database ' ) )
DIR_LOG = check_dir ( os . path . join ( DIR_CONFIG , ' log ' ) )
2009-11-03 10:50:13 +01:00
2009-11-02 10:06:54 +01:00
DATABASE_TYPE_POSTGRESQL = ' postgresql '
DATABASE_TYPE_SQLITE = ' sqlite '
DATABASE_TYPE_MYSQL = ' mysql '
2009-11-24 01:29:56 +01:00
#TODO: should this be a tuple or a dict
2009-11-02 10:06:54 +01:00
DATABASE_TYPES = (
DATABASE_TYPE_POSTGRESQL ,
DATABASE_TYPE_SQLITE ,
DATABASE_TYPE_MYSQL ,
)
2009-08-12 02:46:39 +02:00
2009-11-24 01:29:56 +01:00
# find a logging.conf file and set up logging
log = get_logger ( " logging.conf " , config = " config " )
log . debug ( " config logger initialised " )
# and then log our consts
log . info ( " DIR SELF = %s " % DIR_SELF )
log . info ( " DIR CONFIG = %s " % DIR_CONFIG )
log . info ( " DIR DATABASE = %s " % DIR_DATABASE )
log . info ( " DIR LOG = %s " % DIR_LOG )
2009-11-02 10:06:54 +01:00
########################################################################
2009-11-02 09:09:23 +01:00
def string_to_bool ( string , default = True ) :
""" converts a string representation of a boolean value to boolean True or False
@param string : ( str ) the string to convert
@param default : value to return if the string can not be converted to a boolean value
"""
string = string . lower ( )
if string in ( ' 1 ' , ' true ' , ' t ' ) :
2009-03-18 17:32:34 +01:00
return True
2009-11-02 09:09:23 +01:00
elif string in ( ' 0 ' , ' false ' , ' f ' ) :
2009-03-18 17:32:34 +01:00
return False
2009-03-21 16:38:17 +01:00
return default
2009-03-18 17:32:34 +01:00
2008-08-19 00:53:25 +02:00
class Layout :
2009-03-07 03:46:34 +01:00
def __init__ ( self , node ) :
2009-11-02 09:38:27 +01:00
self . max = int ( node . getAttribute ( ' max ' ) )
2009-03-07 03:46:34 +01:00
if node . hasAttribute ( ' fav_seat ' ) : self . fav_seat = int ( node . getAttribute ( ' fav_seat ' ) )
self . width = int ( node . getAttribute ( ' width ' ) )
self . height = int ( node . getAttribute ( ' height ' ) )
2008-08-19 00:53:25 +02:00
self . location = [ ]
2009-03-26 23:38:13 +01:00
self . location = map ( lambda x : None , range ( self . max + 1 ) ) # fill array with max seats+1 empty entries
2009-03-09 11:22:54 +01:00
2009-03-07 03:46:34 +01:00
for location_node in node . getElementsByTagName ( ' location ' ) :
if location_node . getAttribute ( ' seat ' ) != " " :
self . location [ int ( location_node . getAttribute ( ' seat ' ) ) ] = ( int ( location_node . getAttribute ( ' x ' ) ) , int ( location_node . getAttribute ( ' y ' ) ) )
elif location_node . getAttribute ( ' common ' ) != " " :
self . common = ( int ( location_node . getAttribute ( ' x ' ) ) , int ( location_node . getAttribute ( ' y ' ) ) )
2008-08-19 00:53:25 +02:00
def __str__ ( self ) :
2009-03-07 03:46:34 +01:00
temp = " Layout = %d max, width= %d , height = %d " % ( self . max , self . width , self . height )
if hasattr ( self , ' fav_seat ' ) : temp = temp + " , fav_seat = %d \n " % self . fav_seat
else : temp = temp + " \n "
if hasattr ( self , " common " ) :
temp = temp + " Common = ( %d , %d ) \n " % ( self . common [ 0 ] , self . common [ 1 ] )
2008-08-19 00:53:25 +02:00
temp = temp + " Locations = "
2009-03-09 11:40:38 +01:00
for i in range ( 1 , len ( self . location ) ) :
2008-08-19 00:53:25 +02:00
temp = temp + " ( %d , %d ) " % self . location [ i ]
2008-09-15 22:31:55 +02:00
return temp + " \n "
2008-08-19 00:53:25 +02:00
class Site :
def __init__ ( self , node ) :
2009-08-10 18:10:39 +02:00
def normalizePath ( path ) :
" Normalized existing pathes "
if os . path . exists ( path ) :
return os . path . abspath ( path )
return path
2008-08-19 00:53:25 +02:00
self . site_name = node . getAttribute ( " site_name " )
self . table_finder = node . getAttribute ( " table_finder " )
self . screen_name = node . getAttribute ( " screen_name " )
2009-08-10 18:10:39 +02:00
self . site_path = normalizePath ( node . getAttribute ( " site_path " ) )
2009-11-02 09:38:27 +01:00
self . HH_path = normalizePath ( node . getAttribute ( " HH_path " ) )
self . decoder = node . getAttribute ( " decoder " )
2008-10-27 12:12:04 +01:00
self . hudopacity = node . getAttribute ( " hudopacity " )
2008-10-27 11:29:39 +01:00
self . hudbgcolor = node . getAttribute ( " bgcolor " )
self . hudfgcolor = node . getAttribute ( " fgcolor " )
2008-11-04 10:40:03 +01:00
self . converter = node . getAttribute ( " converter " )
2008-11-16 23:53:31 +01:00
self . aux_window = node . getAttribute ( " aux_window " )
2009-11-02 09:38:27 +01:00
self . font = node . getAttribute ( " font " )
2008-12-18 03:57:05 +01:00
self . font_size = node . getAttribute ( " font_size " )
2009-06-20 15:43:05 +02:00
self . use_frames = node . getAttribute ( " use_frames " )
2009-11-02 09:38:27 +01:00
self . enabled = string_to_bool ( node . getAttribute ( " enabled " ) , default = True )
2009-06-20 15:43:05 +02:00
self . xpad = node . getAttribute ( " xpad " )
self . ypad = node . getAttribute ( " ypad " )
2008-08-19 00:53:25 +02:00
self . layout = { }
2009-11-04 07:44:25 +01:00
2009-11-03 19:49:16 +01:00
print " Loading site " , self . site_name
2009-03-07 03:46:34 +01:00
2008-08-19 00:53:25 +02:00
for layout_node in node . getElementsByTagName ( ' layout ' ) :
2009-03-07 03:46:34 +01:00
lo = Layout ( layout_node )
2008-08-19 00:53:25 +02:00
self . layout [ lo . max ] = lo
2009-06-20 15:43:05 +02:00
# Site defaults
2009-11-03 19:49:16 +01:00
self . xpad = 1 if self . xpad == " " else int ( self . xpad )
self . ypad = 0 if self . ypad == " " else int ( self . ypad )
self . font_size = 7 if self . font_size == " " else int ( self . font_size )
2009-11-03 20:43:12 +01:00
self . hudopacity = 1.0 if self . hudopacity == " " else float ( self . hudopacity )
2009-06-20 15:43:05 +02:00
if self . use_frames == " " : self . use_frames = False
if self . font == " " : self . font = " Sans "
2009-10-28 16:57:54 +01:00
if self . hudbgcolor == " " : self . hudbgcolor = " #000000 "
if self . hudfgcolor == " " : self . hudfgcolor = " #FFFFFF "
2009-06-20 15:43:05 +02:00
2008-08-19 00:53:25 +02:00
def __str__ ( self ) :
2009-03-09 11:40:38 +01:00
temp = " Site = " + self . site_name + " \n "
2008-08-19 00:53:25 +02:00
for key in dir ( self ) :
if key . startswith ( ' __ ' ) : continue
if key == ' layout ' : continue
value = getattr ( self , key )
if callable ( value ) : continue
2008-10-30 03:37:05 +01:00
temp = temp + ' ' + key + " = " + str ( value ) + " \n "
2008-08-19 00:53:25 +02:00
for layout in self . layout :
2009-03-09 11:40:38 +01:00
temp = temp + " %s " % self . layout [ layout ]
2008-08-19 00:53:25 +02:00
return temp
class Stat :
def __init__ ( self ) :
pass
def __str__ ( self ) :
2008-09-15 22:31:55 +02:00
temp = " stat_name = %s , row = %d , col = %d , tip = %s , click = %s , popup = %s \n " % ( self . stat_name , self . row , self . col , self . tip , self . click , self . popup )
2008-08-19 00:53:25 +02:00
return temp
class Game :
def __init__ ( self , node ) :
self . game_name = node . getAttribute ( " game_name " )
2009-11-02 09:38:27 +01:00
self . rows = int ( node . getAttribute ( " rows " ) )
self . cols = int ( node . getAttribute ( " cols " ) )
self . xpad = node . getAttribute ( " xpad " )
self . ypad = node . getAttribute ( " ypad " )
2009-06-20 15:43:05 +02:00
# Defaults
if self . xpad == " " : self . xpad = 1
else : self . xpad = int ( self . xpad )
if self . ypad == " " : self . ypad = 0
else : self . ypad = int ( self . ypad )
2009-03-05 02:04:23 +01:00
aux_text = node . getAttribute ( " aux " )
aux_list = aux_text . split ( ' , ' )
for i in range ( 0 , len ( aux_list ) ) :
aux_list [ i ] = aux_list [ i ] . strip ( )
self . aux = aux_list
2008-08-19 00:53:25 +02:00
2009-11-02 09:38:27 +01:00
self . stats = { }
2008-08-19 00:53:25 +02:00
for stat_node in node . getElementsByTagName ( ' stat ' ) :
stat = Stat ( )
stat . stat_name = stat_node . getAttribute ( " stat_name " )
2009-11-02 09:38:27 +01:00
stat . row = int ( stat_node . getAttribute ( " row " ) )
stat . col = int ( stat_node . getAttribute ( " col " ) )
stat . tip = stat_node . getAttribute ( " tip " )
stat . click = stat_node . getAttribute ( " click " )
stat . popup = stat_node . getAttribute ( " popup " )
2008-11-02 11:20:25 +01:00
stat . hudprefix = stat_node . getAttribute ( " hudprefix " )
stat . hudsuffix = stat_node . getAttribute ( " hudsuffix " )
2008-11-25 12:47:58 +01:00
stat . hudcolor = stat_node . getAttribute ( " hudcolor " )
2008-08-19 00:53:25 +02:00
self . stats [ stat . stat_name ] = stat
def __str__ ( self ) :
2009-03-09 11:40:38 +01:00
temp = " Game = " + self . game_name + " \n "
temp = temp + " rows = %d \n " % self . rows
temp = temp + " cols = %d \n " % self . cols
2009-06-20 15:43:05 +02:00
temp = temp + " xpad = %d \n " % self . xpad
temp = temp + " ypad = %d \n " % self . ypad
2009-03-09 11:40:38 +01:00
temp = temp + " aux = %s \n " % self . aux
2008-08-19 00:53:25 +02:00
for stat in self . stats . keys ( ) :
2009-03-09 11:40:38 +01:00
temp = temp + " %s " % self . stats [ stat ]
2008-08-19 00:53:25 +02:00
return temp
2009-11-02 09:38:27 +01:00
2008-08-19 00:53:25 +02:00
class Database :
def __init__ ( self , node ) :
self . db_name = node . getAttribute ( " db_name " )
2009-11-03 15:17:48 +01:00
self . db_server = node . getAttribute ( " db_server " ) . lower ( )
2009-11-02 09:38:27 +01:00
self . db_ip = node . getAttribute ( " db_ip " )
2008-08-19 00:53:25 +02:00
self . db_user = node . getAttribute ( " db_user " )
self . db_pass = node . getAttribute ( " db_pass " )
2009-11-02 09:16:18 +01:00
self . db_selected = string_to_bool ( node . getAttribute ( " default " ) , default = False )
2009-11-03 19:18:51 +01:00
log . debug ( " Database db_name: ' %(name)s ' db_server: ' %(server)s ' db_ip: ' %(ip)s ' db_user: ' %(user)s ' db_pass (not logged) selected: ' %(sel)s ' " \
% { ' name ' : self . db_name , ' server ' : self . db_server , ' ip ' : self . db_ip , ' user ' : self . db_user , ' sel ' : self . db_selected } )
2008-08-19 00:53:25 +02:00
def __str__ ( self ) :
temp = ' Database = ' + self . db_name + ' \n '
for key in dir ( self ) :
if key . startswith ( ' __ ' ) : continue
value = getattr ( self , key )
if callable ( value ) : continue
2009-08-15 23:36:14 +02:00
temp = temp + ' ' + key + " = " + repr ( value ) + " \n "
2008-08-19 00:53:25 +02:00
return temp
2008-11-16 23:53:31 +01:00
class Aux_window :
2008-08-19 00:53:25 +02:00
def __init__ ( self , node ) :
2008-11-16 23:53:31 +01:00
for ( name , value ) in node . attributes . items ( ) :
setattr ( self , name , value )
2009-03-07 03:46:34 +01:00
self . layout = { }
for layout_node in node . getElementsByTagName ( ' layout ' ) :
lo = Layout ( layout_node )
self . layout [ lo . max ] = lo
2008-08-19 00:53:25 +02:00
def __str__ ( self ) :
2008-11-16 23:53:31 +01:00
temp = ' Aux = ' + self . name + " \n "
2008-08-19 00:53:25 +02:00
for key in dir ( self ) :
if key . startswith ( ' __ ' ) : continue
2009-03-07 03:46:34 +01:00
if key == ' layout ' : continue
2008-08-19 00:53:25 +02:00
value = getattr ( self , key )
if callable ( value ) : continue
temp = temp + ' ' + key + " = " + value + " \n "
2009-03-07 03:46:34 +01:00
for layout in self . layout :
temp = temp + " %s " % self . layout [ layout ]
2008-08-19 00:53:25 +02:00
return temp
2009-03-24 14:58:45 +01:00
class HHC :
def __init__ ( self , node ) :
2009-11-02 09:38:27 +01:00
self . site = node . getAttribute ( " site " )
2009-03-24 14:58:45 +01:00
self . converter = node . getAttribute ( " converter " )
def __str__ ( self ) :
return " %s : \t %s " % ( self . site , self . converter )
2008-09-15 22:31:55 +02:00
class Popup :
def __init__ ( self , node ) :
self . name = node . getAttribute ( " pu_name " )
2009-11-02 09:38:27 +01:00
self . pu_stats = [ ]
2008-09-15 22:31:55 +02:00
for stat_node in node . getElementsByTagName ( ' pu_stat ' ) :
self . pu_stats . append ( stat_node . getAttribute ( " pu_stat_name " ) )
def __str__ ( self ) :
temp = " Popup = " + self . name + " \n "
for stat in self . pu_stats :
temp = temp + " " + stat
return temp + " \n "
2008-10-21 16:22:01 +02:00
class Import :
def __init__ ( self , node ) :
2009-09-10 03:21:27 +02:00
self . node = node
2009-11-02 09:38:27 +01:00
self . interval = node . getAttribute ( " interval " )
2008-11-08 14:01:07 +01:00
self . callFpdbHud = node . getAttribute ( " callFpdbHud " )
2009-02-19 13:15:12 +01:00
self . hhArchiveBase = node . getAttribute ( " hhArchiveBase " )
2009-11-02 09:14:57 +01:00
self . saveActions = string_to_bool ( node . getAttribute ( " saveActions " ) , default = True )
self . fastStoreHudCache = string_to_bool ( node . getAttribute ( " fastStoreHudCache " ) , default = False )
2008-10-21 16:22:01 +02:00
def __str__ ( self ) :
2009-03-15 20:54:22 +01:00
return " interval = %s \n callFpdbHud = %s \n hhArchiveBase = %s \n saveActions = %s \n fastStoreHudCache = %s \n " \
2009-11-02 09:38:27 +01:00
% ( self . interval , self . callFpdbHud , self . hhArchiveBase , self . saveActions , self . fastStoreHudCache )
2008-10-21 16:22:01 +02:00
2009-10-03 20:05:41 +02:00
class HudUI :
def __init__ ( self , node ) :
self . node = node
self . label = node . getAttribute ( ' label ' )
2009-10-14 15:04:09 +02:00
#
2009-11-02 09:09:23 +01:00
self . aggregate_ring = string_to_bool ( node . getAttribute ( ' aggregate_ring_game_stats ' ) )
self . aggregate_tour = string_to_bool ( node . getAttribute ( ' aggregate_tourney_stats ' ) )
2009-11-02 09:38:27 +01:00
self . hud_style = node . getAttribute ( ' stat_aggregation_range ' )
self . hud_days = node . getAttribute ( ' aggregation_days ' )
2009-10-14 15:04:09 +02:00
self . agg_bb_mult = node . getAttribute ( ' aggregation_level_multiplier ' )
#
2009-11-02 09:09:23 +01:00
self . h_aggregate_ring = string_to_bool ( node . getAttribute ( ' aggregate_hero_ring_game_stats ' ) )
self . h_aggregate_tour = string_to_bool ( node . getAttribute ( ' aggregate_hero_tourney_stats ' ) )
2009-10-14 15:04:09 +02:00
self . h_hud_style = node . getAttribute ( ' hero_stat_aggregation_range ' )
2009-11-02 09:38:27 +01:00
self . h_hud_days = node . getAttribute ( ' hero_aggregation_days ' )
self . h_agg_bb_mult = node . getAttribute ( ' hero_aggregation_level_multiplier ' )
2009-10-14 15:04:09 +02:00
2009-10-03 20:05:41 +02:00
def __str__ ( self ) :
return " label = %s \n " % self . label
2008-10-21 16:22:01 +02:00
class Tv :
def __init__ ( self , node ) :
2009-11-05 00:34:02 +01:00
self . combinedStealFold = string_to_bool ( node . getAttribute ( " combinedStealFold " ) , default = True )
self . combined2B3B = string_to_bool ( node . getAttribute ( " combined2B3B " ) , default = True )
self . combinedPostflop = string_to_bool ( node . getAttribute ( " combinedPostflop " ) , default = True )
2008-10-21 16:22:01 +02:00
def __str__ ( self ) :
return ( " combinedStealFold = %s \n combined2B3B = %s \n combinedPostflop = %s \n " %
( self . combinedStealFold , self . combined2B3B , self . combinedPostflop ) )
2008-08-19 00:53:25 +02:00
class Config :
2009-08-12 02:46:39 +02:00
def __init__ ( self , file = None , dbname = ' ' ) :
2008-09-27 01:21:38 +02:00
# "file" is a path to an xml file with the fpdb/HUD configuration
# we check the existence of "file" and try to recover if it doesn't exist
2009-11-09 04:53:10 +01:00
# self.default_config_path = self.get_default_config_path()
2009-11-03 20:30:52 +01:00
if file is not None : # config file path passed in
2009-06-15 05:14:53 +02:00
file = os . path . expanduser ( file )
2008-09-27 01:21:38 +02:00
if not os . path . exists ( file ) :
2009-11-24 01:29:56 +01:00
log . error ( " Specified configuration file %s not found. Using defaults. " % ( file ) )
2008-09-27 01:21:38 +02:00
file = None
2009-11-11 22:20:43 +01:00
if file is None : file = get_config ( " HUD_config.xml " )
2008-11-03 04:58:55 +01:00
# Parse even if there was no real config file found and we are using the example
# If using the example, we'll edit it later
2009-09-16 06:30:59 +02:00
log . info ( " Reading configuration file %s " % file )
2009-11-09 04:53:10 +01:00
print " \n Reading configuration file %s \n " % file
2008-09-29 16:12:04 +02:00
try :
doc = xml . dom . minidom . parse ( file )
2009-08-16 11:22:22 +02:00
except :
2009-08-12 02:46:39 +02:00
log . error ( " Error parsing %s . See error log file. " % ( file ) )
2008-09-29 16:12:04 +02:00
traceback . print_exc ( file = sys . stderr )
print " press enter to continue "
sys . stdin . readline ( )
sys . exit ( )
2008-08-19 00:53:25 +02:00
2008-09-15 22:31:55 +02:00
self . doc = doc
self . file = file
2008-08-19 00:53:25 +02:00
self . supported_sites = { }
self . supported_games = { }
2009-11-02 08:52:46 +01:00
self . supported_databases = { } # databaseName --> Database instance
2008-11-16 23:53:31 +01:00
self . aux_windows = { }
2009-03-24 14:58:45 +01:00
self . hhcs = { }
2008-09-15 22:31:55 +02:00
self . popup_windows = { }
2009-11-02 08:30:24 +01:00
self . db_selected = None # database the user would like to use
2009-11-05 00:34:02 +01:00
self . tv = None
2008-08-19 00:53:25 +02:00
# s_sites = doc.getElementsByTagName("supported_sites")
for site_node in doc . getElementsByTagName ( " site " ) :
site = Site ( node = site_node )
self . supported_sites [ site . site_name ] = site
2008-10-21 16:22:01 +02:00
# s_games = doc.getElementsByTagName("supported_games")
2008-08-19 00:53:25 +02:00
for game_node in doc . getElementsByTagName ( " game " ) :
game = Game ( node = game_node )
self . supported_games [ game . game_name ] = game
2009-11-02 09:29:58 +01:00
# parse databases defined by user in the <supported_databases> section
# the user may select the actual database to use via commandline or by setting the selected="bool"
# attribute of the tag. if no database is explicitely selected, we use the first one we come across
2008-10-21 16:22:01 +02:00
# s_dbs = doc.getElementsByTagName("supported_databases")
2009-11-02 08:40:34 +01:00
#TODO: do we want to take all <database> tags or all <database> tags contained in <supported_databases>
2009-11-02 09:38:27 +01:00
# ..this may break stuff for some users. so leave it unchanged for now untill there is a decission
2008-08-19 00:53:25 +02:00
for db_node in doc . getElementsByTagName ( " database " ) :
2009-11-02 08:48:07 +01:00
db = Database ( node = db_node )
if db . db_name in self . supported_databases :
2009-11-02 09:40:16 +01:00
raise ValueError ( " Database names must be unique " )
2009-11-02 09:29:58 +01:00
if self . db_selected is None or db . db_selected :
2009-11-02 08:48:07 +01:00
self . db_selected = db . db_name
self . supported_databases [ db . db_name ] = db
2009-11-02 09:54:16 +01:00
#TODO: if the user may passes '' (empty string) as database name via command line, his choice is ignored
2009-11-02 09:55:08 +01:00
# ..when we parse the xml we allow for ''. there has to be a decission if to allow '' or not
2009-09-04 23:12:35 +02:00
if dbname and dbname in self . supported_databases :
self . db_selected = dbname
2009-11-02 09:45:53 +01:00
#NOTE: fpdb can not handle the case when no database is defined in xml, so we throw an exception for now
if self . db_selected is None :
raise ValueError ( ' There must be at least one database defined ' )
2009-09-04 23:12:35 +02:00
2009-11-02 09:38:27 +01:00
# s_dbs = doc.getElementsByTagName("mucked_windows")
2008-11-16 23:53:31 +01:00
for aw_node in doc . getElementsByTagName ( " aw " ) :
aw = Aux_window ( node = aw_node )
self . aux_windows [ aw . name ] = aw
2008-08-19 00:53:25 +02:00
2009-11-02 09:38:27 +01:00
# s_dbs = doc.getElementsByTagName("mucked_windows")
2009-03-24 14:58:45 +01:00
for hhc_node in doc . getElementsByTagName ( " hhc " ) :
hhc = HHC ( node = hhc_node )
self . hhcs [ hhc . site ] = hhc
2008-10-21 16:22:01 +02:00
# s_dbs = doc.getElementsByTagName("popup_windows")
2008-09-15 22:31:55 +02:00
for pu_node in doc . getElementsByTagName ( " pu " ) :
pu = Popup ( node = pu_node )
self . popup_windows [ pu . name ] = pu
2008-10-21 16:22:01 +02:00
for imp_node in doc . getElementsByTagName ( " import " ) :
imp = Import ( node = imp_node )
self . imp = imp
2009-10-03 20:05:41 +02:00
for hui_node in doc . getElementsByTagName ( ' hud_ui ' ) :
hui = HudUI ( node = hui_node )
self . ui = hui
2008-10-21 16:22:01 +02:00
for tv_node in doc . getElementsByTagName ( " tv " ) :
2009-11-05 00:34:02 +01:00
self . tv = Tv ( node = tv_node )
2008-10-21 16:22:01 +02:00
2009-03-27 15:49:22 +01:00
db = self . get_db_parameters ( )
2008-11-03 04:58:55 +01:00
if db [ ' db-password ' ] == ' YOUR MYSQL PASSWORD ' :
df_file = self . find_default_conf ( )
2009-11-03 19:49:16 +01:00
if df_file is None : # this is bad
2008-11-03 04:58:55 +01:00
pass
else :
df_parms = self . read_default_conf ( df_file )
2008-11-05 11:49:05 +01:00
self . set_db_parameters ( db_name = ' fpdb ' , db_ip = df_parms [ ' db-host ' ] ,
2009-11-02 09:38:27 +01:00
db_user = df_parms [ ' db-user ' ] ,
db_pass = df_parms [ ' db-password ' ] )
2008-11-05 11:49:05 +01:00
self . save ( file = os . path . join ( self . default_config_path , " HUD_config.xml " ) )
2009-10-04 18:46:04 +02:00
print " "
2009-09-10 03:21:27 +02:00
def set_hhArchiveBase ( self , path ) :
self . imp . node . setAttribute ( " hhArchiveBase " , path )
2008-11-03 04:58:55 +01:00
def find_default_conf ( self ) :
if os . name == ' posix ' :
config_path = os . path . join ( os . path . expanduser ( " ~ " ) , ' .fpdb ' , ' default.conf ' )
elif os . name == ' nt ' :
config_path = os . path . join ( os . environ [ " APPDATA " ] , ' fpdb ' , ' default.conf ' )
else : config_path = False
if config_path and os . path . exists ( config_path ) :
file = config_path
else :
file = None
return file
2008-09-15 22:31:55 +02:00
def get_site_node ( self , site ) :
for site_node in self . doc . getElementsByTagName ( " site " ) :
if site_node . getAttribute ( " site_name " ) == site :
return site_node
2009-03-11 19:12:56 +01:00
def get_aux_node ( self , aux ) :
for aux_node in self . doc . getElementsByTagName ( " aw " ) :
if aux_node . getAttribute ( " name " ) == aux :
return aux_node
2008-11-05 11:49:05 +01:00
def get_db_node ( self , db_name ) :
for db_node in self . doc . getElementsByTagName ( " database " ) :
if db_node . getAttribute ( " db_name " ) == db_name :
return db_node
return None
2008-09-15 22:31:55 +02:00
def get_layout_node ( self , site_node , layout ) :
for layout_node in site_node . getElementsByTagName ( " layout " ) :
2009-11-03 20:30:52 +01:00
if layout_node . getAttribute ( " max " ) is None :
2008-10-04 22:43:50 +02:00
return None
2008-09-15 22:31:55 +02:00
if int ( layout_node . getAttribute ( " max " ) ) == int ( layout ) :
return layout_node
def get_location_node ( self , layout_node , seat ) :
2009-03-15 20:27:47 +01:00
if seat == " common " :
for location_node in layout_node . getElementsByTagName ( " location " ) :
if location_node . hasAttribute ( " common " ) :
return location_node
else :
for location_node in layout_node . getElementsByTagName ( " location " ) :
if int ( location_node . getAttribute ( " seat " ) ) == int ( seat ) :
return location_node
2008-09-15 22:31:55 +02:00
def save ( self , file = None ) :
2009-11-03 20:30:52 +01:00
if file is not None :
2009-09-16 07:13:42 +02:00
with open ( file , ' w ' ) as f :
self . doc . writexml ( f )
2008-09-15 22:31:55 +02:00
else :
shutil . move ( self . file , self . file + " .backup " )
2009-11-05 00:39:42 +01:00
with open ( file , ' w ' ) as f :
self . doc . writexml ( f )
2008-09-15 22:31:55 +02:00
def edit_layout ( self , site_name , max , width = None , height = None ,
fav_seat = None , locations = None ) :
site_node = self . get_site_node ( site_name )
layout_node = self . get_layout_node ( site_node , max )
2009-11-01 06:54:59 +01:00
# TODO: how do we support inserting new layouts?
2009-11-03 20:30:52 +01:00
if layout_node is None :
2009-11-01 06:54:59 +01:00
return
2009-03-09 11:40:38 +01:00
for i in range ( 1 , max + 1 ) :
2008-09-15 22:31:55 +02:00
location_node = self . get_location_node ( layout_node , i )
location_node . setAttribute ( " x " , str ( locations [ i - 1 ] [ 0 ] ) )
location_node . setAttribute ( " y " , str ( locations [ i - 1 ] [ 1 ] ) )
self . supported_sites [ site_name ] . layout [ max ] . location [ i ] = ( locations [ i - 1 ] [ 0 ] , locations [ i - 1 ] [ 1 ] )
2009-03-11 19:12:56 +01:00
def edit_aux_layout ( self , aux_name , max , width = None , height = None , locations = None ) :
aux_node = self . get_aux_node ( aux_name )
layout_node = self . get_layout_node ( aux_node , max )
2009-11-03 20:30:52 +01:00
if layout_node is None :
2009-03-15 20:27:47 +01:00
print " aux node not found "
return
print " editing locations = " , locations
for ( i , pos ) in locations . iteritems ( ) :
2009-03-11 19:12:56 +01:00
location_node = self . get_location_node ( layout_node , i )
2009-03-15 20:27:47 +01:00
location_node . setAttribute ( " x " , str ( locations [ i ] [ 0 ] ) )
location_node . setAttribute ( " y " , str ( locations [ i ] [ 1 ] ) )
if i == " common " :
self . aux_windows [ aux_name ] . layout [ max ] . common = ( locations [ i ] [ 0 ] , locations [ i ] [ 1 ] )
else :
self . aux_windows [ aux_name ] . layout [ max ] . location [ i ] = ( locations [ i ] [ 0 ] , locations [ i ] [ 1 ] )
2009-03-11 19:12:56 +01:00
2009-11-02 10:29:10 +01:00
#NOTE: we got a nice Database class, so why map it again here?
# user input validation should be done when initializing the Database class. this allows to give appropriate feddback when something goes wrong
# try ..except is evil here. it swallows all kinds of errors. dont do this
# naming database types 2, 3, 4 on the fly is no good idea. i see this all over the code. better use some globally defined consts (see DATABASE_TYPE_*)
# i would like to drop this method entirely and replace it by get_selected_database() or better get_active_database(), returning one of our Database instances
# thus we can drop self.db_selected (holding database name) entirely and replace it with self._active_database = Database, avoiding to define the same
# thing multiple times
2009-03-27 15:49:22 +01:00
def get_db_parameters ( self ) :
2008-10-05 02:22:53 +02:00
db = { }
2009-08-12 02:46:39 +02:00
name = self . db_selected
2009-09-16 06:30:59 +02:00
# TODO: What's up with all the exception handling here?!
2009-03-17 04:34:00 +01:00
try : db [ ' db-databaseName ' ] = name
except : pass
try : db [ ' db-host ' ] = self . supported_databases [ name ] . db_ip
except : pass
try : db [ ' db-user ' ] = self . supported_databases [ name ] . db_user
except : pass
try : db [ ' db-password ' ] = self . supported_databases [ name ] . db_pass
except : pass
try : db [ ' db-server ' ] = self . supported_databases [ name ] . db_server
except : pass
2009-11-03 15:17:48 +01:00
if self . supported_databases [ name ] . db_server == DATABASE_TYPE_MYSQL :
2009-03-17 04:34:00 +01:00
db [ ' db-backend ' ] = 2
2009-11-03 15:17:48 +01:00
elif self . supported_databases [ name ] . db_server == DATABASE_TYPE_POSTGRESQL :
2009-03-17 04:34:00 +01:00
db [ ' db-backend ' ] = 3
2009-11-03 15:17:48 +01:00
elif self . supported_databases [ name ] . db_server == DATABASE_TYPE_SQLITE :
2009-07-14 01:04:10 +02:00
db [ ' db-backend ' ] = 4
2009-11-03 15:17:48 +01:00
else :
raise ValueError ( ' Unsupported database backend: %s ' % self . supported_databases [ name ] . db_server )
2008-10-05 02:22:53 +02:00
return db
2008-11-05 11:49:05 +01:00
def set_db_parameters ( self , db_name = ' fpdb ' , db_ip = None , db_user = None ,
2009-11-03 19:18:51 +01:00
db_pass = None , db_server = None ) :
2008-11-05 11:49:05 +01:00
db_node = self . get_db_node ( db_name )
2009-03-26 19:57:40 +01:00
if db_node != None :
2009-11-03 20:30:52 +01:00
if db_ip is not None : db_node . setAttribute ( " db_ip " , db_ip )
if db_user is not None : db_node . setAttribute ( " db_user " , db_user )
if db_pass is not None : db_node . setAttribute ( " db_pass " , db_pass )
if db_server is not None : db_node . setAttribute ( " db_server " , db_server )
if db_type is not None : db_node . setAttribute ( " db_type " , db_type )
2008-11-05 11:49:05 +01:00
if self . supported_databases . has_key ( db_name ) :
2009-11-03 20:30:52 +01:00
if db_ip is not None : self . supported_databases [ db_name ] . dp_ip = db_ip
if db_user is not None : self . supported_databases [ db_name ] . dp_user = db_user
if db_pass is not None : self . supported_databases [ db_name ] . dp_pass = db_pass
if db_server is not None : self . supported_databases [ db_name ] . dp_server = db_server
if db_type is not None : self . supported_databases [ db_name ] . dp_type = db_type
2008-11-05 11:49:05 +01:00
return
2009-08-08 10:07:30 +02:00
def getDefaultSite ( self ) :
" Returns first enabled site or None "
for site_name , site in self . supported_sites . iteritems ( ) :
if site . enabled :
return site_name
return None
2008-11-05 11:49:05 +01:00
2008-10-21 16:22:01 +02:00
def get_tv_parameters ( self ) :
2009-11-05 00:34:02 +01:00
if self . tv is not None :
return {
' combinedStealFold ' : self . tv . combinedStealFold ,
' combined2B3B ' : self . tv . combined2B3B ,
' combinedPostflop ' : self . tv . combinedPostflop
}
return { }
2009-10-03 20:05:41 +02:00
# Allow to change the menu appearance
def get_hud_ui_parameters ( self ) :
hui = { }
2009-10-14 15:04:09 +02:00
2009-10-03 20:05:41 +02:00
default_text = ' FPDB Menu - Right click \n Left-Drag to Move '
try :
hui [ ' label ' ] = self . ui . label
2009-11-02 09:38:27 +01:00
if self . ui . label == ' ' : # Empty menu label is a big no-no
2009-10-03 20:05:41 +02:00
hui [ ' label ' ] = default_text
except :
hui [ ' label ' ] = default_text
2009-10-14 15:04:09 +02:00
try : hui [ ' aggregate_ring ' ] = self . ui . aggregate_ring
except : hui [ ' aggregate_ring ' ] = False
try : hui [ ' aggregate_tour ' ] = self . ui . aggregate_tour
except : hui [ ' aggregate_tour ' ] = True
try : hui [ ' hud_style ' ] = self . ui . hud_style
except : hui [ ' hud_style ' ] = ' A '
2009-11-02 09:38:27 +01:00
try : hui [ ' hud_days ' ] = int ( self . ui . hud_days )
except : hui [ ' hud_days ' ] = 90
2009-10-14 15:04:09 +02:00
2009-11-02 09:38:27 +01:00
try : hui [ ' agg_bb_mult ' ] = self . ui . agg_bb_mult
except : hui [ ' agg_bb_mult ' ] = 1
2009-10-14 15:04:09 +02:00
# Hero specific
try : hui [ ' h_aggregate_ring ' ] = self . ui . h_aggregate_ring
except : hui [ ' h_aggregate_ring ' ] = False
try : hui [ ' h_aggregate_tour ' ] = self . ui . h_aggregate_tour
except : hui [ ' h_aggregate_tour ' ] = True
2009-11-02 09:38:27 +01:00
try : hui [ ' h_hud_style ' ] = self . ui . h_hud_style
except : hui [ ' h_hud_style ' ] = ' S '
2009-10-14 15:04:09 +02:00
2009-11-02 09:38:27 +01:00
try : hui [ ' h_hud_days ' ] = int ( self . ui . h_hud_days )
except : hui [ ' h_hud_days ' ] = 30
2009-10-14 15:04:09 +02:00
try : hui [ ' h_agg_bb_mult ' ] = self . ui . h_agg_bb_mult
except : hui [ ' h_agg_bb_mult ' ] = 1
2009-10-03 20:05:41 +02:00
return hui
2008-10-21 16:22:01 +02:00
def get_import_parameters ( self ) :
imp = { }
2009-11-02 09:38:27 +01:00
try : imp [ ' callFpdbHud ' ] = self . imp . callFpdbHud
except : imp [ ' callFpdbHud ' ] = True
2009-03-17 04:34:00 +01:00
2009-11-02 09:38:27 +01:00
try : imp [ ' interval ' ] = self . imp . interval
except : imp [ ' interval ' ] = 10
2009-03-17 04:34:00 +01:00
2009-11-02 09:38:27 +01:00
try : imp [ ' hhArchiveBase ' ] = self . imp . hhArchiveBase
except : imp [ ' hhArchiveBase ' ] = " ~/.fpdb/HandHistories/ "
2009-03-17 04:34:00 +01:00
2009-11-02 09:38:27 +01:00
try : imp [ ' saveActions ' ] = self . imp . saveActions
except : imp [ ' saveActions ' ] = True
2009-03-17 04:34:00 +01:00
2009-11-02 09:38:27 +01:00
try : imp [ ' fastStoreHudCache ' ] = self . imp . fastStoreHudCache
2009-03-17 04:34:00 +01:00
except : imp [ ' fastStoreHudCache ' ] = True
2008-10-21 16:22:01 +02:00
return imp
2009-08-08 10:07:30 +02:00
def get_default_paths ( self , site = None ) :
if site is None : site = self . getDefaultSite ( )
2008-10-21 16:22:01 +02:00
paths = { }
try :
2009-08-08 10:07:30 +02:00
path = os . path . expanduser ( self . supported_sites [ site ] . HH_path )
assert ( os . path . isdir ( path ) or os . path . isfile ( path ) ) # maybe it should try another site?
paths [ ' hud-defaultPath ' ] = paths [ ' bulkImport-defaultPath ' ] = path
2009-08-16 11:22:22 +02:00
except AssertionError :
paths [ ' hud-defaultPath ' ] = paths [ ' bulkImport-defaultPath ' ] = " ** ERROR DEFAULT PATH IN CONFIG DOES NOT EXIST ** "
2008-10-21 16:22:01 +02:00
return paths
2009-02-25 21:25:58 +01:00
def get_frames ( self , site = " PokerStars " ) :
2009-08-06 05:57:26 +02:00
if site not in self . supported_sites : return False
2009-03-29 16:08:33 +02:00
return self . supported_sites [ site ] . use_frames == True
2008-10-21 16:22:01 +02:00
2008-10-30 03:37:05 +01:00
def get_default_colors ( self , site = " PokerStars " ) :
colors = { }
2009-08-06 05:57:26 +02:00
if site not in self . supported_sites or self . supported_sites [ site ] . hudopacity == " " :
2009-03-09 11:40:38 +01:00
colors [ ' hudopacity ' ] = 0.90
else :
colors [ ' hudopacity ' ] = float ( self . supported_sites [ site ] . hudopacity )
2009-08-06 05:57:26 +02:00
if site not in self . supported_sites or self . supported_sites [ site ] . hudbgcolor == " " :
2009-03-09 11:40:38 +01:00
colors [ ' hudbgcolor ' ] = " #FFFFFF "
else :
colors [ ' hudbgcolor ' ] = self . supported_sites [ site ] . hudbgcolor
2009-08-06 05:57:26 +02:00
if site not in self . supported_sites or self . supported_sites [ site ] . hudfgcolor == " " :
2009-03-09 11:40:38 +01:00
colors [ ' hudfgcolor ' ] = " #000000 "
else :
colors [ ' hudfgcolor ' ] = self . supported_sites [ site ] . hudfgcolor
2008-10-30 03:37:05 +01:00
return colors
2008-12-18 03:57:05 +01:00
2009-11-04 23:41:09 +01:00
def get_default_font ( self , site = ' PokerStars ' ) :
font = " Sans "
font_size = " 8 "
site = self . supported_sites . get ( site , None )
if site is not None :
if site . font :
font = site . font
if site . font_size :
font_size = site . font_size
return font , font_size
2008-10-30 03:37:05 +01:00
2009-11-05 01:21:32 +01:00
def get_locations ( self , site_name = " PokerStars " , max = 8 ) :
site = self . supported_sites . get ( site_name , None )
if site is not None :
location = site . layout . get ( max , None )
if location is not None :
return location . location
return (
( 0 , 0 ) , ( 684 , 61 ) , ( 689 , 239 ) , ( 692 , 346 ) ,
( 586 , 393 ) , ( 421 , 440 ) , ( 267 , 440 ) , ( 0 , 361 ) ,
( 0 , 280 ) , ( 121 , 280 ) , ( 46 , 30 )
)
2009-03-07 23:02:51 +01:00
def get_aux_locations ( self , aux = " mucked " , max = " 9 " ) :
2009-11-03 19:49:16 +01:00
2009-03-07 23:02:51 +01:00
try :
locations = self . aux_windows [ aux ] . layout [ max ] . location
except :
locations = ( ( 0 , 0 ) , ( 684 , 61 ) , ( 689 , 239 ) , ( 692 , 346 ) ,
2009-11-02 09:38:27 +01:00
( 586 , 393 ) , ( 421 , 440 ) , ( 267 , 440 ) , ( 0 , 361 ) ,
( 0 , 280 ) , ( 121 , 280 ) , ( 46 , 30 ) )
2009-03-07 23:02:51 +01:00
return locations
2009-11-05 00:11:43 +01:00
def get_supported_sites ( self , all = False ) :
2008-11-11 15:40:31 +01:00
""" Returns the list of supported sites. """
2009-11-05 00:11:43 +01:00
if all :
return self . supported_sites . keys ( )
else :
return [ site_name for ( site_name , site ) in self . supported_sites . items ( ) if site . enabled ]
2008-11-05 22:52:47 +01:00
def get_site_parameters ( self , site ) :
""" Returns a dict of the site parameters for the specified site """
parms = { }
parms [ " converter " ] = self . supported_sites [ site ] . converter
2009-11-02 09:38:27 +01:00
parms [ " decoder " ] = self . supported_sites [ site ] . decoder
2008-11-05 22:52:47 +01:00
parms [ " hudbgcolor " ] = self . supported_sites [ site ] . hudbgcolor
parms [ " hudfgcolor " ] = self . supported_sites [ site ] . hudfgcolor
parms [ " hudopacity " ] = self . supported_sites [ site ] . hudopacity
parms [ " screen_name " ] = self . supported_sites [ site ] . screen_name
parms [ " site_path " ] = self . supported_sites [ site ] . site_path
parms [ " table_finder " ] = self . supported_sites [ site ] . table_finder
2009-11-02 09:38:27 +01:00
parms [ " HH_path " ] = self . supported_sites [ site ] . HH_path
2008-11-11 15:40:31 +01:00
parms [ " site_name " ] = self . supported_sites [ site ] . site_name
2008-11-16 23:53:31 +01:00
parms [ " aux_window " ] = self . supported_sites [ site ] . aux_window
2009-11-02 09:38:27 +01:00
parms [ " font " ] = self . supported_sites [ site ] . font
2008-12-18 03:57:05 +01:00
parms [ " font_size " ] = self . supported_sites [ site ] . font_size
2009-11-02 09:38:27 +01:00
parms [ " enabled " ] = self . supported_sites [ site ] . enabled
parms [ " xpad " ] = self . supported_sites [ site ] . xpad
parms [ " ypad " ] = self . supported_sites [ site ] . ypad
2008-11-05 22:52:47 +01:00
return parms
def set_site_parameters ( self , site_name , converter = None , decoder = None ,
hudbgcolor = None , hudfgcolor = None ,
hudopacity = None , screen_name = None ,
site_path = None , table_finder = None ,
2008-12-18 03:57:05 +01:00
HH_path = None , enabled = None ,
font = None , font_size = None ) :
2008-11-05 22:52:47 +01:00
""" Sets the specified site parameters for the specified site. """
site_node = self . get_site_node ( site_name )
2009-11-03 20:30:52 +01:00
if db_node is not None :
if converter is not None : site_node . setAttribute ( " converter " , converter )
if decoder is not None : site_node . setAttribute ( " decoder " , decoder )
if hudbgcolor is not None : site_node . setAttribute ( " hudbgcolor " , hudbgcolor )
if hudfgcolor is not None : site_node . setAttribute ( " hudfgcolor " , hudfgcolor )
if hudopacity is not None : site_node . setAttribute ( " hudopacity " , hudopacity )
if screen_name is not None : site_node . setAttribute ( " screen_name " , screen_name )
if site_path is not None : site_node . setAttribute ( " site_path " , site_path )
if table_finder is not None : site_node . setAttribute ( " table_finder " , table_finder )
if HH_path is not None : site_node . setAttribute ( " HH_path " , HH_path )
if enabled is not None : site_node . setAttribute ( " enabled " , enabled )
if font is not None : site_node . setAttribute ( " font " , font )
if font_size is not None : site_node . setAttribute ( " font_size " , font_size )
2008-11-05 22:52:47 +01:00
return
2008-11-16 23:53:31 +01:00
def get_aux_windows ( self ) :
""" Gets the list of mucked window formats in the configuration. """
2009-11-04 23:58:48 +01:00
return self . aux_windows . keys ( )
2008-11-16 23:53:31 +01:00
def get_aux_parameters ( self , name ) :
""" Gets a dict of mucked window parameters from the named mw. """
param = { }
2009-03-09 11:40:38 +01:00
if self . aux_windows . has_key ( name ) :
2008-11-16 23:53:31 +01:00
for key in dir ( self . aux_windows [ name ] ) :
if key . startswith ( ' __ ' ) : continue
value = getattr ( self . aux_windows [ name ] , key )
if callable ( value ) : continue
param [ key ] = value
return param
return None
def get_game_parameters ( self , name ) :
""" Get the configuration parameters for the named game. """
param = { }
2009-03-09 11:40:38 +01:00
if self . supported_games . has_key ( name ) :
2008-11-16 23:53:31 +01:00
param [ ' game_name ' ] = self . supported_games [ name ] . game_name
2009-11-02 09:38:27 +01:00
param [ ' rows ' ] = self . supported_games [ name ] . rows
param [ ' cols ' ] = self . supported_games [ name ] . cols
param [ ' xpad ' ] = self . supported_games [ name ] . xpad
param [ ' ypad ' ] = self . supported_games [ name ] . ypad
param [ ' aux ' ] = self . supported_games [ name ] . aux
2008-11-16 23:53:31 +01:00
return param
def get_supported_games ( self ) :
""" Get the list of supported games. """
sg = [ ]
2009-03-09 11:40:38 +01:00
for game in c . supported_games . keys ( ) :
2008-11-16 23:53:31 +01:00
sg . append ( c . supported_games [ game ] . game_name )
return sg
2008-12-13 23:37:23 +01:00
def execution_path ( self , filename ) :
""" Join the fpdb path to filename. """
2009-03-05 02:12:15 +01:00
return os . path . join ( os . path . dirname ( inspect . getfile ( sys . _getframe ( 0 ) ) ) , filename )
2008-12-13 23:37:23 +01:00
2008-08-19 00:53:25 +02:00
if __name__ == " __main__ " :
c = Config ( )
print " \n ----------- SUPPORTED SITES ----------- "
for s in c . supported_sites . keys ( ) :
print c . supported_sites [ s ]
print " ----------- END SUPPORTED SITES ----------- "
print " \n ----------- SUPPORTED GAMES ----------- "
for game in c . supported_games . keys ( ) :
print c . supported_games [ game ]
print " ----------- END SUPPORTED GAMES ----------- "
print " \n ----------- SUPPORTED DATABASES ----------- "
for db in c . supported_databases . keys ( ) :
print c . supported_databases [ db ]
print " ----------- END SUPPORTED DATABASES ----------- "
2008-11-16 23:53:31 +01:00
print " \n ----------- AUX WINDOW FORMATS ----------- "
for w in c . aux_windows . keys ( ) :
print c . aux_windows [ w ]
print " ----------- END AUX WINDOW FORMATS ----------- "
2009-03-24 14:58:45 +01:00
print " \n ----------- HAND HISTORY CONVERTERS ----------- "
for w in c . hhcs . keys ( ) :
print c . hhcs [ w ]
print " ----------- END HAND HISTORY CONVERTERS ----------- "
2008-10-21 16:22:01 +02:00
2008-09-15 22:31:55 +02:00
print " \n ----------- POPUP WINDOW FORMATS ----------- "
for w in c . popup_windows . keys ( ) :
print c . popup_windows [ w ]
2008-11-16 23:53:31 +01:00
print " ----------- END POPUP WINDOW FORMATS ----------- "
2008-09-15 22:31:55 +02:00
2008-10-21 16:22:01 +02:00
print " \n ----------- IMPORT ----------- "
2008-12-03 18:01:37 +01:00
print c . imp
2008-10-21 16:22:01 +02:00
print " ----------- END IMPORT ----------- "
print " \n ----------- TABLE VIEW ----------- "
2008-10-30 03:37:05 +01:00
# print c.tv
2008-10-21 16:22:01 +02:00
print " ----------- END TABLE VIEW ----------- "
2008-09-15 22:31:55 +02:00
c . edit_layout ( " PokerStars " , 6 , locations = ( ( 1 , 1 ) , ( 2 , 2 ) , ( 3 , 3 ) , ( 4 , 4 ) , ( 5 , 5 ) , ( 6 , 6 ) ) )
2008-10-05 02:22:53 +02:00
c . save ( file = " testout.xml " )
2009-11-02 09:38:27 +01:00
print " db = " , c . get_db_parameters ( )
# print "tv = ", c.get_tv_parameters()
2008-10-30 03:37:05 +01:00
# print "imp = ", c.get_import_parameters()
print " paths = " , c . get_default_paths ( " PokerStars " )
print " colors = " , c . get_default_colors ( " PokerStars " )
print " locs = " , c . get_locations ( " PokerStars " , 8 )
2008-11-16 23:53:31 +01:00
for mw in c . get_aux_windows ( ) :
print c . get_aux_parameters ( mw )
2009-03-07 23:02:51 +01:00
print " mucked locations = " , c . get_aux_locations ( ' mucked ' , 9 )
2009-03-15 20:54:22 +01:00
# c.edit_aux_layout('mucked', 9, locations = [(487, 113), (555, 469), (572, 276), (522, 345),
# (333, 354), (217, 341), (150, 273), (150, 169), (230, 115)])
# print "mucked locations =", c.get_aux_locations('mucked', 9)
2009-03-11 19:12:56 +01:00
2008-11-05 22:52:47 +01:00
for site in c . supported_sites . keys ( ) :
print " site = " , site ,
2008-11-06 06:39:49 +01:00
print c . get_site_parameters ( site )
2008-12-18 03:57:05 +01:00
print c . get_default_font ( site )
2008-11-16 23:53:31 +01:00
for game in c . get_supported_games ( ) :
print c . get_game_parameters ( game )
2008-12-13 23:37:23 +01:00
2009-02-19 13:15:12 +01:00
print " start up path = " , c . execution_path ( " " )
2009-11-11 22:20:43 +01:00
2009-11-11 22:25:31 +01:00
from xml . dom . ext import PrettyPrint
for site_node in c . doc . getElementsByTagName ( " site " ) :
PrettyPrint ( site_node , stream = sys . stdout , encoding = " utf-8 " )