fpdb/pyfpdb/SQL.py

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

Set up all of the SQL statements for a given game and database type.
"""
#    Copyright 2008, 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

#    pyGTK modules

#    FreePokerTools modules

class Sql:
    
    def __init__(self, game = 'holdem', type = 'PT3'):
        self.query = {}

############################################################################
#
#    Support for the ptracks database, a cut down PT2 stud database.
#    You can safely ignore this unless you are me.
#
        if game == 'razz' and type == 'ptracks':
            
            self.query['get_table_name'] = "select table_name from game where game_id = %s"

            self.query['get_last_hand'] = "select max(game_id) from game"

            self.query['get_recent_hands'] = "select game_id from game where game_id > %(last_hand)d"
            
            self.query['get_xml'] = "select xml from hand_history where game_id = %s"

            self.query['get_player_id'] = """
                    select player_id from players 
                    where screen_name = %(player)s 
                """
                
            self.query['get_hand_info'] = """
                    SELECT 
                        game_id, 
                        CONCAT(hole_card_1, hole_card_2, hole_card_3, hole_card_4, hole_card_5, hole_card_6, hole_card_7) AS hand,  
                        total_won-total_bet AS net
                    FROM game_players 
                    WHERE game_id = %s AND player_id = 3
                """

            self.query['get_cards'] = """
                    select 
                        seat_number, 
                        screen_name, 
                        hole_card_1, 
                        hole_card_2, 
                        hole_card_3, 
                        hole_card_4, 
                        hole_card_5, 
                        hole_card_6, 
                        hole_card_7 
                    from game_players, players 
                    where game_id = %s and game_players.player_id = players.player_id 
                    order by seat_number
                """
        
            self.query['get_stats_from_hand'] = """
                    SELECT player_id, 
                        count(*)                    AS n,
                        sum(pre_fourth_raise_n)     AS pfr,
                        sum(fourth_raise_n)         AS raise_n_2,
                        sum(fourth_ck_raise_n)      AS cr_n_2,
                        sum(fifth_bet_raise_n)      AS br_n_3,
                        sum(fifth_bet_ck_raise_n)   AS cr_n_3,
                        sum(sixth_bet_raise_n)      AS br_n_4,
                        sum(sixth_bet_ck_raise_n)   AS cr_n_4,
                        sum(river_bet_raise_n)      AS br_n_5,
                        sum(river_bet_ck_raise_n)   AS cr_n_5,
                        sum(went_to_showdown_n)     AS sd,
                        sum(saw_fourth_n)           AS saw_f,
                        sum(raised_first_pf)        AS first_pfr,
                        sum(vol_put_money_in_pot)   AS vpip,
                        sum(limp_with_prev_callers) AS limp_w_callers,

                        sum(ppossible_actions)      AS poss_a_pf, 
                        sum(pfold)                  AS fold_pf, 
                        sum(pcheck)                 AS check_pf, 
                        sum(praise)                 AS raise_pf, 
                        sum(pcall)                  AS raise_pf, 
                        sum(limp_call_reraise_pf)   AS limp_call_pf,

                        sum(pfr_check)              AS check_after_raise, 
                        sum(pfr_call)               AS call_after_raise, 
                        sum(pfr_fold)               AS fold_after_raise, 
                        sum(pfr_bet)                AS bet_after_raise, 
                        sum(pfr_raise)              AS raise_after_raise, 
                        sum(folded_to_river_bet)    AS fold_to_r_bet, 

                        sum(fpossible_actions)      AS poss_a_2, 
                        sum(ffold)                  AS fold_2,
                        sum(fcheck)                 AS check_2, 
                        sum(fbet)                   AS bet_2, 
                        sum(fraise)                 AS raise_2, 
                        sum(fcall)                  AS raise_2,

                        sum(fifpossible_actions)    AS poss_a_3, 
                        sum(fiffold)                AS fold_3,
                        sum(fifcheck)               AS check_3, 
                        sum(fifbet)                 AS bet_3, 
                        sum(fifraise)               AS raise_3, 
                        sum(fifcall)                AS call_3,

                        sum(spossible_actions)      AS poss_a_4, 
                        sum(sfold)                  AS fold_4,
                        sum(scheck)                 AS check_4, 
                        sum(sbet)                   AS bet_4, 
                        sum(sraise)                 AS raise_4, 
                        sum(scall)                  AS call_4,

                        sum(rpossible_actions)      AS poss_a_5, 
                        sum(rfold)                  AS fold_5,
                        sum(rcheck)                 AS check_5, 
                        sum(rbet)                   AS bet_5, 
                        sum(rraise)                 AS raise_5, 
                        sum(rcall)                  AS call_5,

                        sum(cold_call_pf)           AS cc_pf,
                        sum(saw_fifth_n)            AS saw_3,
                        sum(saw_sixth_n)            AS saw_4,
                        sum(saw_river_n)            AS saw_5
                    FROM game_players
                    WHERE player_id in 
                        (SELECT player_id FROM game_players 
                        WHERE game_id = %s AND NOT player_id = %s)
                    GROUP BY player_id
                """
# alternate form of WHERE for above
#                        WHERE game_id = %(hand)d AND NOT player_id = %(hero)d)
#                        WHERE game_id = %s AND NOT player_id = %s)

            self.query['get_players_from_hand'] = """
                    SELECT game_players.player_id, seat_number, screen_name
                    FROM game_players INNER JOIN players ON (game_players.player_id = players.player_id)
                    WHERE game_id = %s
                """

###############################################################################3
#    Support for the Free Poker DataBase = fpdb   http://fpdb.sourceforge.net/
#
        if type == 'fpdb':
   
            self.query['get_last_hand'] = "select max(id) from Hands"

            self.query['get_player_id'] = """
                    select Players.id AS player_id from Players, Sites 
                    where Players.name = %(player)s 
                    and Sites.name = %(site)s 
                    and Players.SiteId = Sites.id
                """

            self.query['get_stats_from_hand'] = """
                    SELECT HudCache.playerId             AS player_id, 
                        sum(HDs)                         AS n,
                        sum(street0VPI)                  AS vpip,
                        sum(street0Aggr)                 AS pfr,
                        sum(street0_3B4BChance)          AS TB_opp_0,
                        sum(street0_3B4BDone)            AS TB_0,
                        sum(street1Seen)                 AS saw_f,
                        sum(street1Seen)                 AS saw_1,
                        sum(street2Seen)                 AS saw_2,
                        sum(street3Seen)                 AS saw_3,
                        sum(street4Seen)                 AS saw_4,
                        sum(sawShowdown)                 AS sd,
                        sum(street1Aggr)                 AS aggr_1,
                        sum(street2Aggr)                 AS aggr_2,
                        sum(street3Aggr)                 AS aggr_3,
                        sum(street4Aggr)                 AS aggr_4,
                        sum(otherRaisedStreet1)          AS was_raised_1,
                        sum(otherRaisedStreet2)          AS was_raised_2,
                        sum(otherRaisedStreet3)          AS was_raised_3,
                        sum(otherRaisedStreet4)          AS was_raised_4,
                        sum(foldToOtherRaisedStreet1)    AS f_freq_1,
                        sum(foldToOtherRaisedStreet2)    AS f_freq_2,
                        sum(foldToOtherRaisedStreet3)    AS f_freq_3,
                        sum(foldToOtherRaisedStreet4)    AS f_freq_4,
                        sum(wonWhenSeenStreet1)          AS w_w_s_1,
                        sum(wonAtSD)                     AS wmsd,
                        sum(stealAttemptChance)          AS steal_opp,
                        sum(stealAttempted)              AS steal,
                        sum(foldSbToStealChance)         AS SBstolen,
                        sum(foldedSbToSteal)             AS SBnotDef,
                        sum(foldBbToStealChance)         AS BBstolen,
                        sum(foldedBbToSteal)             AS BBnotDef,
                        sum(street1CBChance)             AS CB_opp_1,
                        sum(street1CBDone)               AS CB_1,
                        sum(street2CBChance)             AS CB_opp_2,
                        sum(street2CBDone)               AS CB_2,
                        sum(street3CBChance)             AS CB_opp_3,
                        sum(street3CBDone)               AS CB_3,
                        sum(street4CBChance)             AS CB_opp_4,
                        sum(street4CBDone)               AS CB_4,
                        sum(foldToStreet1CBChance)       AS f_cb_opp_1,
                        sum(foldToStreet1CBDone)         AS f_cb_1,
                        sum(foldToStreet2CBChance)       AS f_cb_opp_2,
                        sum(foldToStreet2CBDone)         AS f_cb_2,
                        sum(foldToStreet3CBChance)       AS f_cb_opp_3,
                        sum(foldToStreet3CBDone)         AS f_cb_3,
                        sum(foldToStreet4CBChance)       AS f_cb_opp_4,
                        sum(foldToStreet4CBDone)         AS f_cb_4,
                        sum(totalProfit)                 AS net,
                        sum(street1CheckCallRaiseChance) AS ccr_opp_1,
                        sum(street1CheckCallRaiseDone)   AS ccr_1,
                        sum(street2CheckCallRaiseChance) AS ccr_opp_2,
                        sum(street2CheckCallRaiseDone)   AS ccr_2,
                        sum(street3CheckCallRaiseChance) AS ccr_opp_3,
                        sum(street3CheckCallRaiseDone)   AS ccr_3,
                        sum(street4CheckCallRaiseChance) AS ccr_opp_4,
                        sum(street4CheckCallRaiseDone)   AS ccr_4
                    FROM Hands
                         INNER JOIN HandsPlayers ON (HandsPlayers.handId = %s)
                         INNER JOIN HudCache ON (    HudCache.PlayerId = HandsPlayers.PlayerId+0
                                                 AND HudCache.gametypeId+0 = Hands.gametypeId+0)
                    WHERE Hands.id = %s
                    GROUP BY HudCache.PlayerId
                """

#    same as above except stats are aggregated for all blind/limit levels
            self.query['get_stats_from_hand_aggregated'] = """
                    SELECT HudCache.playerId             AS player_id, 
                        sum(HDs)                         AS n,
                        sum(street0VPI)                  AS vpip,
                        sum(street0Aggr)                 AS pfr,
                        sum(street0_3B4BChance)          AS TB_opp_0,
                        sum(street0_3B4BDone)            AS TB_0,
                        sum(street1Seen)                 AS saw_f,
                        sum(street1Seen)                 AS saw_1,
                        sum(street2Seen)                 AS saw_2,
                        sum(street3Seen)                 AS saw_3,
                        sum(street4Seen)                 AS saw_4,
                        sum(sawShowdown)                 AS sd,
                        sum(street1Aggr)                 AS aggr_1,
                        sum(street2Aggr)                 AS aggr_2,
                        sum(street3Aggr)                 AS aggr_3,
                        sum(street4Aggr)                 AS aggr_4,
                        sum(otherRaisedStreet1)          AS was_raised_1,
                        sum(otherRaisedStreet2)          AS was_raised_2,
                        sum(otherRaisedStreet3)          AS was_raised_3,
                        sum(otherRaisedStreet4)          AS was_raised_4,
                        sum(foldToOtherRaisedStreet1)    AS f_freq_1,
                        sum(foldToOtherRaisedStreet2)    AS f_freq_2,
                        sum(foldToOtherRaisedStreet3)    AS f_freq_3,
                        sum(foldToOtherRaisedStreet4)    AS f_freq_4,
                        sum(wonWhenSeenStreet1)          AS w_w_s_1,
                        sum(wonAtSD)                     AS wmsd,
                        sum(stealAttemptChance)          AS steal_opp,
                        sum(stealAttempted)              AS steal,
                        sum(foldSbToStealChance)         AS SBstolen,
                        sum(foldedSbToSteal)             AS SBnotDef,
                        sum(foldBbToStealChance)         AS BBstolen,
                        sum(foldedBbToSteal)             AS BBnotDef,
                        sum(street1CBChance)             AS CB_opp_1,
                        sum(street1CBDone)               AS CB_1,
                        sum(street2CBChance)             AS CB_opp_2,
                        sum(street2CBDone)               AS CB_2,
                        sum(street3CBChance)             AS CB_opp_3,
                        sum(street3CBDone)               AS CB_3,
                        sum(street4CBChance)             AS CB_opp_4,
                        sum(street4CBDone)               AS CB_4,
                        sum(foldToStreet1CBChance)       AS f_cb_opp_1,
                        sum(foldToStreet1CBDone)         AS f_cb_1,
                        sum(foldToStreet2CBChance)       AS f_cb_opp_2,
                        sum(foldToStreet2CBDone)         AS f_cb_2,
                        sum(foldToStreet3CBChance)       AS f_cb_opp_3,
                        sum(foldToStreet3CBDone)         AS f_cb_3,
                        sum(foldToStreet4CBChance)       AS f_cb_opp_4,
                        sum(foldToStreet4CBDone)         AS f_cb_4,
                        sum(totalProfit)                 AS net,
                        sum(street1CheckCallRaiseChance) AS ccr_opp_1,
                        sum(street1CheckCallRaiseDone)   AS ccr_1,
                        sum(street2CheckCallRaiseChance) AS ccr_opp_2,
                        sum(street2CheckCallRaiseDone)   AS ccr_2,
                        sum(street3CheckCallRaiseChance) AS ccr_opp_3,
                        sum(street3CheckCallRaiseDone)   AS ccr_3,
                        sum(street4CheckCallRaiseChance) AS ccr_opp_4,
                        sum(street4CheckCallRaiseDone)   AS ccr_4
                    FROM HudCache, Hands
                    WHERE HudCache.PlayerId in 
                        (SELECT PlayerId FROM HandsPlayers 
                        WHERE handId = %s)
                    AND  Hands.id = %s
                    AND  HudCache.gametypeId in
                        (SELECT gt1.id from Gametypes gt1, Gametypes gt2, Hands
                        WHERE  gt1.siteid = gt2.siteid
                        AND    gt1.type = gt2.type
                        AND    gt1.category = gt2.category
                        AND    gt1.limittype = gt2.limittype
                        AND    gt2.id = Hands.gametypeId
                        AND    Hands.id = %s)
                    GROUP BY HudCache.PlayerId
                """
         
            self.query['get_players_from_hand'] = """
                    SELECT HandsPlayers.playerId, seatNo, name
                    FROM  HandsPlayers INNER JOIN Players ON (HandsPlayers.playerId = Players.id)
                    WHERE handId = %s
                """
#                    WHERE handId = %s AND Players.id LIKE %s

            self.query['get_winners_from_hand'] = """
                    SELECT name, winnings
                    FROM HandsPlayers, Players
                    WHERE winnings > 0
                        AND Players.id = HandsPlayers.playerId
                        AND handId = %s;
                """

            self.query['get_table_name'] = """
                    select tableName, maxSeats, category 
                    from Hands,Gametypes 
                    where Hands.id = %s
                    and Gametypes.id = Hands.gametypeId
                """

            self.query['get_actual_seat'] = """
                    select seatNo
                    from HandsPlayers
                    where HandsPlayers.handId = %s
                    and   HandsPlayers.playerId  = (select Players.id from Players
                                                    where Players.name = %s)
                """

            self.query['get_cards'] = """
                    select 
                        seatNo     AS seat_number, 
                        name       AS screen_name, 
                        card1Value, card1Suit,
                        card2Value, card2Suit,
                        card3Value, card3Suit,
                        card4Value, card4Suit,
                        card5Value, card5Suit,
                        card6Value, card6Suit,
                        card7Value, card7Suit
                    from HandsPlayers, Players 
                    where handID = %s and HandsPlayers.playerId = Players.id 
                    order by seatNo
                """

            self.query['get_action_from_hand'] = """
                SELECT street, Players.name, HandsActions.action, HandsActions.amount, actionno
                FROM Players, HandsActions, HandsPlayers
                WHERE HandsPlayers.handid = %s
                AND HandsPlayers.playerid = Players.id
                AND HandsActions.handPlayerId = HandsPlayers.id
                ORDER BY street, actionno
            """
if __name__== "__main__":
#    just print the default queries and exit
    s = Sql(game = 'razz', type = 'ptracks')
    for key in s.query:
        print "For query " + key + ", sql ="
        print s.query[key]