fpdb/pyfpdb/GuiSessionViewer.py

#!/usr/bin/python

#Copyright 2008 Steffen Jobbagy-Felso
#This program is free software: you can redistribute it and/or modify
#it under the terms of the GNU Affero General Public License as published by
#the Free Software Foundation, version 3 of the License.
#
#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 Affero General Public License
#along with this program. If not, see <http://www.gnu.org/licenses/>.
#In the "official" distribution you can find the license in
#agpl-3.0.txt in the docs folder of the package.

import threading
import pygtk
pygtk.require('2.0')
import gtk
import os
from time import time, strftime, localtime
from numpy import diff, nonzero

import Card
import fpdb_import
import Database
import Filters
import FpdbSQLQueries

class GuiSessionViewer (threading.Thread):
    def __init__(self, config, querylist, debug=True):
        self.debug = debug
        self.conf = config
        self.sql = querylist
        self.MYSQL_INNODB   = 2
        self.PGSQL          = 3
        self.SQLITE         = 4
        
        # create new db connection to avoid conflicts with other threads
        self.db = Database.Database(self.conf, sql=self.sql)
        self.cursor = self.db.cursor

        settings = {}
        settings.update(self.conf.get_db_parameters())
        settings.update(self.conf.get_tv_parameters())
        settings.update(self.conf.get_import_parameters())
        settings.update(self.conf.get_default_paths())

        # text used on screen stored here so that it can be configured
        self.filterText = {'handhead':'Hand Breakdown for all levels listed above'
                          }

        filters_display = { "Heroes"   :  True,
                            "Sites"    :  True,
                            "Games"    :  False,
                            "Limits"   :  True,
                            "LimitSep" :  True,
                            "Seats"    :  True,
                            "SeatSep"  :  True,
                            "Dates"    :  False,
                            "Groups"   :  True,
                            "Button1"  :  True,
                            "Button2"  :  True
                          }

        self.filters = Filters.Filters(self.db, self.conf, self.sql, display = filters_display)
        self.filters.registerButton2Name("_Refresh")
        self.filters.registerButton2Callback(self.refreshStats)

        # ToDo: store in config
        # ToDo: create popup to adjust column config
        # columns to display, keys match column name returned by sql, values in tuple are:
        #     is column displayed, column heading, xalignment, formatting
        self.columns = [ ("game",     True,  "Game",     0.0, "%s")
                       , ("hand",     False, "Hand",     0.0, "%s")   # true not allowed for this line
                       , ("n",        True,  "Hds",      1.0, "%d")
                       , ("avgseats", True,  "Seats",    1.0, "%3.1f")
                       , ("vpip",     True,  "VPIP",     1.0, "%3.1f")
                       , ("pfr",      True,  "PFR",      1.0, "%3.1f")
                       , ("pf3",      True,  "PF3",      1.0, "%3.1f")
                       , ("steals",   True,  "Steals",   1.0, "%3.1f")
                       , ("saw_f",    True,  "Saw_F",    1.0, "%3.1f")
                       , ("sawsd",    True,  "SawSD",    1.0, "%3.1f")
                       , ("wtsdwsf",  True,  "WtSDwsF",  1.0, "%3.1f")
                       , ("wmsd",     True,  "W$SD",     1.0, "%3.1f")
                       , ("flafq",    True,  "FlAFq",    1.0, "%3.1f")
                       , ("tuafq",    True,  "TuAFq",    1.0, "%3.1f")
                       , ("rvafq",    True,  "RvAFq",    1.0, "%3.1f")
                       , ("pofafq",   False, "PoFAFq",   1.0, "%3.1f")
                       , ("net",      True,  "Net($)",   1.0, "%6.2f")
                       , ("bbper100", True,  "BB/100",   1.0, "%4.2f")
                       , ("rake",     True,  "Rake($)",  1.0, "%6.2f")
                       , ("variance", True,  "Variance", 1.0, "%5.2f")
                       ]

        self.stats_frame = None
        self.stats_vbox = None
        self.detailFilters = []   # the data used to enhance the sql select
        
        self.main_hbox = gtk.HBox(False, 0)
        self.main_hbox.show()

        self.stats_frame = gtk.Frame()
        self.stats_frame.show()

        self.stats_vbox = gtk.VBox(False, 0)
        self.stats_vbox.show()
        self.stats_frame.add(self.stats_vbox)
        self.fillStatsFrame(self.stats_vbox)

        self.main_hbox.pack_start(self.filters.get_vbox())
        self.main_hbox.pack_start(self.stats_frame, expand=True, fill=True)

################################


        # make sure Hand column is not displayed
        [x for x in self.columns if x[0] == 'hand'][0][1] == False

    def get_vbox(self):
        """returns the vbox of this thread"""
        return self.main_hbox

    def refreshStats(self, widget, data):
        try: self.stats_vbox.destroy()
        except AttributeError: pass
        self.stats_vbox = gtk.VBox(False, 0)
        self.stats_vbox.show()
        self.stats_frame.add(self.stats_vbox)
        self.fillStatsFrame(self.stats_vbox)

    def fillStatsFrame(self, vbox):
        sites = self.filters.getSites()
        heroes = self.filters.getHeroes()
        siteids = self.filters.getSiteIds()
        limits  = self.filters.getLimits()
        seats  = self.filters.getSeats()
        sitenos = []
        playerids = []

        # Which sites are selected?
        for site in sites:
            if sites[site] == True:
                sitenos.append(siteids[site])
                self.cursor.execute(self.sql.query['getPlayerId'], (heroes[site],))
                result = self.db.cursor.fetchall()
                if len(result) == 1:
                    playerids.append(result[0][0])

        if not sitenos:
            #Should probably pop up here.
            print "No sites selected - defaulting to PokerStars"
            sitenos = [2]
        if not playerids:
            print "No player ids found"
            return
        if not limits:
            print "No limits found"
            return

        self.createStatsTable(vbox, playerids, sitenos, limits, seats)

    def createStatsTable(self, vbox, playerids, sitenos, limits, seats):
        starttime = time()

        # Display summary table at top of page
        # 3rd parameter passes extra flags, currently includes:
        # holecards - whether to display card breakdown (True/False)
        flags = [False]
        self.addTable(vbox, 'playerDetailedStats', flags, playerids, sitenos, limits, seats)

        # Separator
        sep = gtk.HSeparator()
        vbox.pack_start(sep, expand=False, padding=3)
        sep.show_now()
        vbox.show_now()
        heading = gtk.Label(self.filterText['handhead'])
        heading.show()
        vbox.pack_start(heading, expand=False, padding=3)

        # Scrolled window for detailed table (display by hand)
        swin = gtk.ScrolledWindow(hadjustment=None, vadjustment=None)
        swin.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
        swin.show()
        vbox.pack_start(swin, expand=True, padding=3)

        vbox1 = gtk.VBox(False, 0)
        vbox1.show()
        swin.add_with_viewport(vbox1)

        # Detailed table
        flags = [True]
        self.addTable(vbox1, 'playerDetailedStats', flags, playerids, sitenos, limits, seats)

        self.db.rollback()
        print "Stats page displayed in %4.2f seconds" % (time() - starttime)
    #end def fillStatsFrame(self, vbox):

    def addTable(self, vbox, query, flags, playerids, sitenos, limits, seats):
        row = 0
        sqlrow = 0
        colalias,colshow,colheading,colxalign,colformat = 0,1,2,3,4
        if not flags:  holecards = False
        else:          holecards = flags[0]


        self.stats_table = gtk.Table(1, 1, False)
        self.stats_table.set_col_spacings(4)
        self.stats_table.show()
        
        self.db.cursor.execute("""select UNIX_TIMESTAMP(handStart) as time, id from Hands ORDER BY time""")
        THRESHOLD = 1800
        hands = self.db.cursor.fetchall()

        times = map(lambda x:long(x[0]), hands)
        handids = map(lambda x:int(x[1]), hands)
        print "DEBUG: len(times) %s" %(len(times))
        diffs = diff(times)
        print "DEBUG: len(diffs) %s" %(len(diffs))
        index = nonzero(diff(times) > THRESHOLD)
        print "DEBUG: len(index[0]) %s" %(len(index[0]))
        print "DEBUG: index %s" %(index)
        print "DEBUG: index[0][0] %s" %(index[0][0])

        total = 0

        last_idx = 0
        for i in range(len(index[0])):
            print "Hands in session %4s: %4s  Start: %s End: %s Total: %s" %(i, index[0][i] - last_idx, strftime("%d/%m/%Y %H:%M", localtime(times[last_idx])), strftime("%d/%m/%Y %H:%M", localtime(times[index[0][i]])), times[index[0][i]] - times[last_idx])
            total = total + (index[0][i] - last_idx)
            last_idx = index[0][i] + 1

        print "Total: ", total
#
#        colnames = [desc[0].lower() for desc in self.cursor.description]
#
#        # pre-fetch some constant values:
#        cols_to_show = [x for x in self.columns if x[colshow]]
#        hgametypeid_idx = colnames.index('hgametypeid')
#
#        liststore = gtk.ListStore(*([str] * len(cols_to_show)))
#        view = gtk.TreeView(model=liststore)
#        view.set_grid_lines(gtk.TREE_VIEW_GRID_LINES_BOTH)
#        vbox.pack_start(view, expand=False, padding=3)
#        textcell = gtk.CellRendererText()
#        numcell = gtk.CellRendererText()
#        numcell.set_property('xalign', 1.0)
#        listcols = []
#
#        # Create header row   eg column: ("game",     True, "Game",     0.0, "%s")
#        for col, column in enumerate(cols_to_show):
#            if column[colalias] == 'game' and holecards:
#                s = [x for x in self.columns if x[colalias] == 'hand'][0][colheading]
#            else:
#                s = column[colheading]
#            listcols.append(gtk.TreeViewColumn(s))
#            view.append_column(listcols[col])
#            if column[colformat] == '%s':
#                if col == 1 and holecards:
#                    listcols[col].pack_start(textcell, expand=True)
#                else:
#                    listcols[col].pack_start(textcell, expand=False)
#                listcols[col].add_attribute(textcell, 'text', col)
#            else:
#                listcols[col].pack_start(numcell, expand=False)
#                listcols[col].add_attribute(numcell, 'text', col)
#
#        rows = len(result) # +1 for title row
#
#        while sqlrow < rows:
#            treerow = []
#            if(row%2 == 0):
#                bgcolor = "white"
#            else:
#                bgcolor = "lightgrey"
#            for col,column in enumerate(cols_to_show):
#                if column[colalias] in colnames:
#                    value = result[sqlrow][colnames.index(column[colalias])]
#                else:
#                    if column[colalias] == 'game':
#                        if holecards:
#                            value = Card.twoStartCardString( result[sqlrow][hgametypeid_idx] )
#                        else:
#                            minbb = result[sqlrow][colnames.index('minbigblind')]
#                            maxbb = result[sqlrow][colnames.index('maxbigblind')]
#                            value = result[sqlrow][colnames.index('limittype')] + ' ' \
#                                    + result[sqlrow][colnames.index('category')].title() + ' ' \
#                                    + result[sqlrow][colnames.index('name')] + ' $'
#                            if 100 * int(minbb/100.0) != minbb:
#                                value += '%.2f' % (minbb/100.0)
#                            else:
#                                value += '%.0f' % (minbb/100.0)
#                            if minbb != maxbb:
#                                if 100 * int(maxbb/100.0) != maxbb:
#                                    value += ' - $' + '%.2f' % (maxbb/100.0)
#                                else:
#                                    value += ' - $' + '%.0f' % (maxbb/100.0)
#                    else:
#                        continue
#                if value and value != -999:
#                    treerow.append(column[colformat] % value)
#                else:
#                    treerow.append(' ')
#            iter = liststore.append(treerow)
#            sqlrow += 1
#            row += 1
        vbox.show_all()