2008-08-19 00:53:25 +02:00
|
|
|
#!/usr/bin/env python
|
|
|
|
"""Database.py
|
|
|
|
|
|
|
|
Create and manage the database objects.
|
|
|
|
"""
|
|
|
|
# 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
|
|
|
|
|
|
|
|
########################################################################
|
|
|
|
|
|
|
|
# postmaster -D /var/lib/pgsql/data
|
|
|
|
|
|
|
|
# Standard Library modules
|
2008-09-15 22:31:55 +02:00
|
|
|
import sys
|
2008-10-04 22:43:50 +02:00
|
|
|
import traceback
|
2009-05-28 00:34:10 +02:00
|
|
|
from datetime import datetime, date, time, timedelta
|
2009-08-01 01:06:07 +02:00
|
|
|
from time import time, strftime, sleep
|
2009-08-08 19:59:44 +02:00
|
|
|
from decimal import Decimal
|
2009-06-02 00:27:56 +02:00
|
|
|
import string
|
2009-07-29 00:58:10 +02:00
|
|
|
import re
|
|
|
|
import logging
|
2009-07-31 22:24:21 +02:00
|
|
|
import Queue
|
2008-08-19 00:53:25 +02:00
|
|
|
|
|
|
|
# pyGTK modules
|
|
|
|
|
|
|
|
# FreePokerTools modules
|
2009-06-18 00:03:43 +02:00
|
|
|
import fpdb_db
|
2009-06-19 22:15:52 +02:00
|
|
|
import fpdb_simple
|
2008-08-19 00:53:25 +02:00
|
|
|
import Configuration
|
|
|
|
import SQL
|
2009-05-21 05:26:00 +02:00
|
|
|
import Card
|
2008-08-19 00:53:25 +02:00
|
|
|
|
|
|
|
class Database:
|
2009-07-21 23:26:23 +02:00
|
|
|
|
|
|
|
MYSQL_INNODB = 2
|
|
|
|
PGSQL = 3
|
|
|
|
SQLITE = 4
|
|
|
|
|
2009-07-29 00:58:10 +02:00
|
|
|
# Data Structures for index and foreign key creation
|
|
|
|
# drop_code is an int with possible values: 0 - don't drop for bulk import
|
|
|
|
# 1 - drop during bulk import
|
|
|
|
# db differences:
|
|
|
|
# - note that mysql automatically creates indexes on constrained columns when
|
|
|
|
# foreign keys are created, while postgres does not. Hence the much longer list
|
|
|
|
# of indexes is required for postgres.
|
|
|
|
# all primary keys are left on all the time
|
|
|
|
#
|
|
|
|
# table column drop_code
|
|
|
|
|
|
|
|
indexes = [
|
|
|
|
[ ] # no db with index 0
|
|
|
|
, [ ] # no db with index 1
|
|
|
|
, [ # indexes for mysql (list index 2)
|
2009-07-29 08:57:18 +02:00
|
|
|
{'tab':'Players', 'col':'name', 'drop':0}
|
|
|
|
, {'tab':'Hands', 'col':'siteHandNo', 'drop':0}
|
|
|
|
, {'tab':'Hands', 'col':'gametypeId', 'drop':0} # mct 22/3/09
|
|
|
|
, {'tab':'HandsPlayers', 'col':'handId', 'drop':0} # not needed, handled by fk
|
|
|
|
, {'tab':'HandsPlayers', 'col':'playerId', 'drop':0} # not needed, handled by fk
|
2009-08-03 20:15:36 +02:00
|
|
|
, {'tab':'HandsPlayers', 'col':'tourneyTypeId', 'drop':0}
|
2009-07-29 08:57:18 +02:00
|
|
|
, {'tab':'HandsPlayers', 'col':'tourneysPlayersId', 'drop':0}
|
|
|
|
, {'tab':'Tourneys', 'col':'siteTourneyNo', 'drop':0}
|
2009-07-29 00:58:10 +02:00
|
|
|
]
|
|
|
|
, [ # indexes for postgres (list index 3)
|
2009-07-29 07:17:51 +02:00
|
|
|
{'tab':'Gametypes', 'col':'siteId', 'drop':0}
|
2009-07-29 00:58:10 +02:00
|
|
|
, {'tab':'Hands', 'col':'gametypeId', 'drop':0} # mct 22/3/09
|
|
|
|
, {'tab':'Hands', 'col':'siteHandNo', 'drop':0}
|
|
|
|
, {'tab':'HandsActions', 'col':'handsPlayerId', 'drop':0}
|
|
|
|
, {'tab':'HandsPlayers', 'col':'handId', 'drop':1}
|
|
|
|
, {'tab':'HandsPlayers', 'col':'playerId', 'drop':1}
|
|
|
|
, {'tab':'HandsPlayers', 'col':'tourneysPlayersId', 'drop':0}
|
|
|
|
, {'tab':'HudCache', 'col':'gametypeId', 'drop':1}
|
|
|
|
, {'tab':'HudCache', 'col':'playerId', 'drop':0}
|
|
|
|
, {'tab':'HudCache', 'col':'tourneyTypeId', 'drop':0}
|
|
|
|
, {'tab':'Players', 'col':'siteId', 'drop':1}
|
|
|
|
, {'tab':'Players', 'col':'name', 'drop':0}
|
|
|
|
, {'tab':'Tourneys', 'col':'tourneyTypeId', 'drop':1}
|
|
|
|
, {'tab':'Tourneys', 'col':'siteTourneyNo', 'drop':0}
|
|
|
|
, {'tab':'TourneysPlayers', 'col':'playerId', 'drop':0}
|
|
|
|
, {'tab':'TourneysPlayers', 'col':'tourneyId', 'drop':0}
|
|
|
|
, {'tab':'TourneyTypes', 'col':'siteId', 'drop':0}
|
|
|
|
]
|
|
|
|
, [ # indexes for sqlite (list index 4)
|
|
|
|
]
|
|
|
|
]
|
|
|
|
|
|
|
|
foreignKeys = [
|
|
|
|
[ ] # no db with index 0
|
|
|
|
, [ ] # no db with index 1
|
2009-08-06 21:31:46 +02:00
|
|
|
, [ # foreign keys for mysql (index 2)
|
2009-07-29 00:58:10 +02:00
|
|
|
{'fktab':'Hands', 'fkcol':'gametypeId', 'rtab':'Gametypes', 'rcol':'id', 'drop':1}
|
|
|
|
, {'fktab':'HandsPlayers', 'fkcol':'handId', 'rtab':'Hands', 'rcol':'id', 'drop':1}
|
|
|
|
, {'fktab':'HandsPlayers', 'fkcol':'playerId', 'rtab':'Players', 'rcol':'id', 'drop':1}
|
|
|
|
, {'fktab':'HandsActions', 'fkcol':'handsPlayerId', 'rtab':'HandsPlayers', 'rcol':'id', 'drop':1}
|
|
|
|
, {'fktab':'HudCache', 'fkcol':'gametypeId', 'rtab':'Gametypes', 'rcol':'id', 'drop':1}
|
|
|
|
, {'fktab':'HudCache', 'fkcol':'playerId', 'rtab':'Players', 'rcol':'id', 'drop':0}
|
|
|
|
, {'fktab':'HudCache', 'fkcol':'tourneyTypeId', 'rtab':'TourneyTypes', 'rcol':'id', 'drop':1}
|
|
|
|
]
|
2009-08-06 21:31:46 +02:00
|
|
|
, [ # foreign keys for postgres (index 3)
|
2009-07-29 00:58:10 +02:00
|
|
|
{'fktab':'Hands', 'fkcol':'gametypeId', 'rtab':'Gametypes', 'rcol':'id', 'drop':1}
|
|
|
|
, {'fktab':'HandsPlayers', 'fkcol':'handId', 'rtab':'Hands', 'rcol':'id', 'drop':1}
|
|
|
|
, {'fktab':'HandsPlayers', 'fkcol':'playerId', 'rtab':'Players', 'rcol':'id', 'drop':1}
|
|
|
|
, {'fktab':'HandsActions', 'fkcol':'handsPlayerId', 'rtab':'HandsPlayers', 'rcol':'id', 'drop':1}
|
|
|
|
, {'fktab':'HudCache', 'fkcol':'gametypeId', 'rtab':'Gametypes', 'rcol':'id', 'drop':1}
|
|
|
|
, {'fktab':'HudCache', 'fkcol':'playerId', 'rtab':'Players', 'rcol':'id', 'drop':0}
|
|
|
|
, {'fktab':'HudCache', 'fkcol':'tourneyTypeId', 'rtab':'TourneyTypes', 'rcol':'id', 'drop':1}
|
|
|
|
]
|
2009-08-06 21:31:46 +02:00
|
|
|
, [ # no foreign keys in sqlite (index 4)
|
|
|
|
]
|
2009-07-29 00:58:10 +02:00
|
|
|
]
|
|
|
|
|
|
|
|
|
|
|
|
# MySQL Notes:
|
|
|
|
# "FOREIGN KEY (handId) REFERENCES Hands(id)" - requires index on Hands.id
|
|
|
|
# - creates index handId on <thistable>.handId
|
|
|
|
# alter table t drop foreign key fk
|
|
|
|
# alter table t add foreign key (fkcol) references tab(rcol)
|
|
|
|
# alter table t add constraint c foreign key (fkcol) references tab(rcol)
|
|
|
|
# (fkcol is used for foreigh key name)
|
|
|
|
|
|
|
|
# mysql to list indexes:
|
|
|
|
# SELECT table_name, index_name, non_unique, column_name
|
|
|
|
# FROM INFORMATION_SCHEMA.STATISTICS
|
|
|
|
# WHERE table_name = 'tbl_name'
|
|
|
|
# AND table_schema = 'db_name'
|
|
|
|
# ORDER BY table_name, index_name, seq_in_index
|
|
|
|
#
|
|
|
|
# ALTER TABLE Tourneys ADD INDEX siteTourneyNo(siteTourneyNo)
|
|
|
|
# ALTER TABLE tab DROP INDEX idx
|
|
|
|
|
|
|
|
# mysql to list fks:
|
|
|
|
# SELECT constraint_name, table_name, column_name, referenced_table_name, referenced_column_name
|
|
|
|
# FROM information_schema.KEY_COLUMN_USAGE
|
|
|
|
# WHERE REFERENCED_TABLE_SCHEMA = (your schema name here)
|
|
|
|
# AND REFERENCED_TABLE_NAME is not null
|
|
|
|
# ORDER BY TABLE_NAME, COLUMN_NAME;
|
|
|
|
|
|
|
|
# this may indicate missing object
|
|
|
|
# _mysql_exceptions.OperationalError: (1025, "Error on rename of '.\\fpdb\\hands' to '.\\fpdb\\#sql2-7f0-1b' (errno: 152)")
|
|
|
|
|
|
|
|
|
|
|
|
# PG notes:
|
|
|
|
|
|
|
|
# To add a foreign key constraint to a table:
|
|
|
|
# ALTER TABLE tab ADD CONSTRAINT c FOREIGN KEY (col) REFERENCES t2(col2) MATCH FULL;
|
|
|
|
# ALTER TABLE tab DROP CONSTRAINT zipchk
|
|
|
|
#
|
|
|
|
# Note: index names must be unique across a schema
|
|
|
|
# CREATE INDEX idx ON tab(col)
|
|
|
|
# DROP INDEX idx
|
|
|
|
|
2009-06-24 00:44:37 +02:00
|
|
|
def __init__(self, c, db_name = None, game = None, sql = None): # db_name and game not used any more
|
2009-06-30 23:00:55 +02:00
|
|
|
print "\ncreating Database instance, sql =", sql
|
2009-06-18 00:03:43 +02:00
|
|
|
self.fdb = fpdb_db.fpdb_db() # sets self.fdb.db self.fdb.cursor and self.fdb.sql
|
|
|
|
self.fdb.do_connect(c)
|
|
|
|
self.connection = self.fdb.db
|
2008-08-19 00:53:25 +02:00
|
|
|
|
2009-06-18 00:03:43 +02:00
|
|
|
db_params = c.get_db_parameters()
|
2009-06-19 23:33:33 +02:00
|
|
|
self.import_options = c.get_import_parameters()
|
2009-06-02 00:27:56 +02:00
|
|
|
self.type = db_params['db-type']
|
2009-06-19 23:33:33 +02:00
|
|
|
self.backend = db_params['db-backend']
|
2009-06-26 00:14:32 +02:00
|
|
|
self.db_server = db_params['db-server']
|
2009-07-22 21:19:41 +02:00
|
|
|
|
|
|
|
if self.backend == self.PGSQL:
|
|
|
|
from psycopg2.extensions import ISOLATION_LEVEL_AUTOCOMMIT, ISOLATION_LEVEL_READ_COMMITTED, ISOLATION_LEVEL_SERIALIZABLE
|
|
|
|
#ISOLATION_LEVEL_AUTOCOMMIT = 0
|
|
|
|
#ISOLATION_LEVEL_READ_COMMITTED = 1
|
|
|
|
#ISOLATION_LEVEL_SERIALIZABLE = 2
|
|
|
|
|
2009-06-24 00:44:37 +02:00
|
|
|
# where possible avoid creating new SQL instance by using the global one passed in
|
|
|
|
if sql == None:
|
|
|
|
self.sql = SQL.Sql(type = self.type, db_server = db_params['db-server'])
|
|
|
|
else:
|
|
|
|
self.sql = sql
|
2009-07-26 02:42:09 +02:00
|
|
|
|
2009-08-02 06:19:33 +02:00
|
|
|
self.pcache = None # PlayerId cache
|
|
|
|
self.cachemiss = 0 # Delete me later - using to count player cache misses
|
2009-08-04 12:22:29 +02:00
|
|
|
self.cachehit = 0 # Delete me later - using to count player cache hits
|
2009-08-02 06:19:33 +02:00
|
|
|
|
2009-07-26 02:42:09 +02:00
|
|
|
# config while trying out new hudcache mechanism
|
|
|
|
self.use_date_in_hudcache = True
|
|
|
|
|
2009-08-02 00:15:04 +02:00
|
|
|
#self.hud_hero_style = 'T' # Duplicate set of vars just for hero - not used yet.
|
|
|
|
#self.hud_hero_hands = 2000 # Idea is that you might want all-time stats for others
|
|
|
|
#self.hud_hero_days = 30 # but last T days or last H hands for yourself
|
2009-07-26 02:42:09 +02:00
|
|
|
|
2009-08-02 00:15:04 +02:00
|
|
|
# vars for hand ids or dates fetched according to above config:
|
|
|
|
self.hand_1day_ago = 0 # max hand id more than 24 hrs earlier than now
|
|
|
|
self.date_ndays_ago = 'd000000' # date N days ago ('d' + YYMMDD)
|
|
|
|
self.date_nhands_ago = {} # dates N hands ago per player - not used yet
|
2009-07-18 19:29:06 +02:00
|
|
|
|
2009-08-02 00:15:04 +02:00
|
|
|
self.cursor = self.fdb.cursor
|
2009-07-21 23:26:23 +02:00
|
|
|
|
2009-06-19 23:33:33 +02:00
|
|
|
self.saveActions = False if self.import_options['saveActions'] == False else True
|
|
|
|
|
2009-07-21 23:26:23 +02:00
|
|
|
self.connection.rollback() # make sure any locks taken so far are released
|
2009-07-29 00:58:10 +02:00
|
|
|
#end def __init__
|
2009-07-21 23:26:23 +02:00
|
|
|
|
2009-07-18 19:29:06 +02:00
|
|
|
# could be used by hud to change hud style
|
|
|
|
def set_hud_style(self, style):
|
|
|
|
self.hud_style = style
|
|
|
|
|
2009-06-30 23:00:55 +02:00
|
|
|
def do_connect(self, c):
|
|
|
|
self.fdb.do_connect(c)
|
2009-05-28 00:34:10 +02:00
|
|
|
|
2009-06-19 22:15:52 +02:00
|
|
|
def commit(self):
|
|
|
|
self.fdb.db.commit()
|
2009-05-28 00:34:10 +02:00
|
|
|
|
2009-07-19 19:28:13 +02:00
|
|
|
def rollback(self):
|
|
|
|
self.fdb.db.rollback()
|
|
|
|
|
2009-07-21 23:26:23 +02:00
|
|
|
def get_cursor(self):
|
|
|
|
return self.connection.cursor()
|
|
|
|
|
2008-09-15 22:31:55 +02:00
|
|
|
def close_connection(self):
|
|
|
|
self.connection.close()
|
2009-06-30 23:00:55 +02:00
|
|
|
|
|
|
|
def disconnect(self, due_to_error=False):
|
|
|
|
"""Disconnects the DB (rolls back if param is true, otherwise commits"""
|
|
|
|
self.fdb.disconnect(due_to_error)
|
|
|
|
|
|
|
|
def reconnect(self, due_to_error=False):
|
|
|
|
"""Reconnects the DB"""
|
|
|
|
self.fdb.reconnect(due_to_error=False)
|
|
|
|
|
|
|
|
def get_backend_name(self):
|
2009-08-06 21:31:46 +02:00
|
|
|
"""Returns the name of the currently used backend"""
|
|
|
|
if self.backend==2:
|
|
|
|
return "MySQL InnoDB"
|
|
|
|
elif self.backend==3:
|
|
|
|
return "PostgreSQL"
|
|
|
|
elif self.backend==4:
|
|
|
|
return "SQLite"
|
|
|
|
else:
|
|
|
|
raise fpdb_simple.FpdbError("invalid backend")
|
2009-06-30 23:00:55 +02:00
|
|
|
|
2008-08-19 00:53:25 +02:00
|
|
|
def get_table_name(self, hand_id):
|
|
|
|
c = self.connection.cursor()
|
2008-09-15 22:31:55 +02:00
|
|
|
c.execute(self.sql.query['get_table_name'], (hand_id, ))
|
2008-08-19 00:53:25 +02:00
|
|
|
row = c.fetchone()
|
|
|
|
return row
|
|
|
|
|
|
|
|
def get_last_hand(self):
|
|
|
|
c = self.connection.cursor()
|
|
|
|
c.execute(self.sql.query['get_last_hand'])
|
|
|
|
row = c.fetchone()
|
|
|
|
return row[0]
|
|
|
|
|
|
|
|
def get_xml(self, hand_id):
|
|
|
|
c = self.connection.cursor()
|
|
|
|
c.execute(self.sql.query['get_xml'], (hand_id))
|
|
|
|
row = c.fetchone()
|
|
|
|
return row[0]
|
|
|
|
|
|
|
|
def get_recent_hands(self, last_hand):
|
|
|
|
c = self.connection.cursor()
|
|
|
|
c.execute(self.sql.query['get_recent_hands'], {'last_hand': last_hand})
|
|
|
|
return c.fetchall()
|
|
|
|
|
|
|
|
def get_hand_info(self, new_hand_id):
|
|
|
|
c = self.connection.cursor()
|
|
|
|
c.execute(self.sql.query['get_hand_info'], new_hand_id)
|
|
|
|
return c.fetchall()
|
|
|
|
|
2008-10-10 02:50:12 +02:00
|
|
|
def get_actual_seat(self, hand_id, name):
|
|
|
|
c = self.connection.cursor()
|
|
|
|
c.execute(self.sql.query['get_actual_seat'], (hand_id, name))
|
|
|
|
row = c.fetchone()
|
|
|
|
return row[0]
|
|
|
|
|
2008-08-19 00:53:25 +02:00
|
|
|
def get_cards(self, hand):
|
2009-02-26 05:35:15 +01:00
|
|
|
"""Get and return the cards for each player in the hand."""
|
2009-06-17 05:00:46 +02:00
|
|
|
cards = {} # dict of cards, the key is the seat number,
|
|
|
|
# the value is a tuple of the players cards
|
|
|
|
# example: {1: (0, 0, 20, 21, 22, 0 , 0)}
|
2008-08-19 00:53:25 +02:00
|
|
|
c = self.connection.cursor()
|
2009-05-02 01:28:53 +02:00
|
|
|
c.execute(self.sql.query['get_cards'], [hand])
|
2008-08-19 00:53:25 +02:00
|
|
|
for row in c.fetchall():
|
2009-06-17 05:00:46 +02:00
|
|
|
cards[row[0]] = row[1:]
|
2009-02-26 05:35:15 +01:00
|
|
|
return cards
|
|
|
|
|
2009-03-10 16:14:23 +01:00
|
|
|
def get_common_cards(self, hand):
|
|
|
|
"""Get and return the community cards for the specified hand."""
|
2008-08-19 00:53:25 +02:00
|
|
|
cards = {}
|
2009-03-10 16:14:23 +01:00
|
|
|
c = self.connection.cursor()
|
2009-05-02 01:28:53 +02:00
|
|
|
c.execute(self.sql.query['get_common_cards'], [hand])
|
2009-06-19 21:51:56 +02:00
|
|
|
# row = c.fetchone()
|
|
|
|
cards['common'] = c.fetchone()
|
2009-03-10 16:14:23 +01:00
|
|
|
return cards
|
|
|
|
|
2009-02-26 05:35:15 +01:00
|
|
|
def convert_cards(self, d):
|
|
|
|
ranks = ('', '', '2', '3', '4', '5', '6', '7', '8', '9', 'T', 'J', 'Q', 'K', 'A')
|
|
|
|
cards = ""
|
2009-03-09 10:06:36 +01:00
|
|
|
for i in xrange(1, 8):
|
2009-03-12 03:46:28 +01:00
|
|
|
# key = 'card' + str(i) + 'Value'
|
|
|
|
# if not d.has_key(key): continue
|
|
|
|
# if d[key] == None:
|
|
|
|
# break
|
|
|
|
# elif d[key] == 0:
|
|
|
|
# cards += "xx"
|
|
|
|
# else:
|
|
|
|
# cards += ranks[d['card' + str(i) + 'Value']] + d['card' +str(i) + 'Suit']
|
2009-06-01 03:25:36 +02:00
|
|
|
cv = "card%dvalue" % i
|
2009-03-11 11:38:17 +01:00
|
|
|
if cv not in d or d[cv] == None:
|
2009-02-26 05:35:15 +01:00
|
|
|
break
|
2009-03-11 11:38:17 +01:00
|
|
|
elif d[cv] == 0:
|
2009-03-10 16:14:23 +01:00
|
|
|
cards += "xx"
|
2009-02-26 05:35:15 +01:00
|
|
|
else:
|
2009-06-01 03:25:36 +02:00
|
|
|
cs = "card%dsuit" % i
|
2009-03-11 11:31:47 +01:00
|
|
|
cards = "%s%s%s" % (cards, ranks[d[cv]], d[cs])
|
2009-02-26 05:35:15 +01:00
|
|
|
return cards
|
2008-09-15 22:31:55 +02:00
|
|
|
|
2008-11-13 04:45:09 +01:00
|
|
|
def get_action_from_hand(self, hand_no):
|
|
|
|
action = [ [], [], [], [], [] ]
|
|
|
|
c = self.connection.cursor()
|
2009-06-18 00:03:43 +02:00
|
|
|
c.execute(self.sql.query['get_action_from_hand'], (hand_no,))
|
2008-11-13 04:45:09 +01:00
|
|
|
for row in c.fetchall():
|
|
|
|
street = row[0]
|
|
|
|
act = row[1:]
|
|
|
|
action[street].append(act)
|
|
|
|
return action
|
2008-12-08 20:10:45 +01:00
|
|
|
|
|
|
|
def get_winners_from_hand(self, hand):
|
|
|
|
"""Returns a hash of winners:amount won, given a hand number."""
|
|
|
|
winners = {}
|
|
|
|
c = self.connection.cursor()
|
2009-06-18 00:03:43 +02:00
|
|
|
c.execute(self.sql.query['get_winners_from_hand'], (hand,))
|
2008-12-08 20:10:45 +01:00
|
|
|
for row in c.fetchall():
|
|
|
|
winners[row[0]] = row[1]
|
|
|
|
return winners
|
|
|
|
|
2009-08-02 14:00:55 +02:00
|
|
|
def init_hud_stat_vars(self, hud_days):
|
2009-08-02 00:15:04 +02:00
|
|
|
"""Initialise variables used by Hud to fetch stats."""
|
|
|
|
|
|
|
|
try:
|
|
|
|
# self.hand_1day_ago used to fetch stats for current session (i.e. if hud_style = 'S')
|
|
|
|
self.hand_1day_ago = 1
|
|
|
|
c = self.get_cursor()
|
|
|
|
c.execute(self.sql.query['get_hand_1day_ago'])
|
|
|
|
row = c.fetchone()
|
|
|
|
if row and row[0]:
|
|
|
|
self.hand_1day_ago = row[0]
|
|
|
|
#print "hand 1day ago =", self.hand_1day_ago
|
|
|
|
|
|
|
|
# self.date_ndays_ago used if hud_style = 'T'
|
2009-08-02 14:00:55 +02:00
|
|
|
d = timedelta(days=hud_days)
|
2009-08-02 00:15:04 +02:00
|
|
|
now = datetime.utcnow() - d
|
|
|
|
self.date_ndays_ago = "d%02d%02d%02d" % (now.year-2000, now.month, now.day)
|
|
|
|
except:
|
|
|
|
err = traceback.extract_tb(sys.exc_info()[2])[-1]
|
|
|
|
print "***Error: "+err[2]+"("+str(err[1])+"): "+str(sys.exc_info()[1])
|
|
|
|
|
|
|
|
def init_player_hud_stat_vars(self, playerid):
|
|
|
|
# not sure if this is workable, to be continued ...
|
|
|
|
try:
|
|
|
|
# self.date_nhands_ago is used for fetching stats for last n hands (hud_style = 'H')
|
|
|
|
# This option not used yet - needs to be called for each player :-(
|
|
|
|
self.date_nhands_ago[str(playerid)] = 'd000000'
|
|
|
|
|
|
|
|
# should use aggregated version of query if appropriate
|
|
|
|
c.execute(self.sql.query['get_date_nhands_ago'], (self.hud_hands, playerid))
|
|
|
|
row = c.fetchone()
|
|
|
|
if row and row[0]:
|
|
|
|
self.date_nhands_ago[str(playerid)] = row[0]
|
|
|
|
c.close()
|
|
|
|
print "date n hands ago = " + self.date_nhands_ago[str(playerid)] + "(playerid "+str(playerid)+")"
|
|
|
|
except:
|
|
|
|
err = traceback.extract_tb(sys.exc_info()[2])[-1]
|
|
|
|
print "***Error: "+err[2]+"("+str(err[1])+"): "+str(sys.exc_info()[1])
|
|
|
|
|
2009-08-02 14:00:55 +02:00
|
|
|
def get_stats_from_hand(self, hand, aggregate = False, hud_style = 'A', agg_bb_mult = 100):
|
|
|
|
if hud_style == 'S':
|
|
|
|
|
2009-05-28 00:34:10 +02:00
|
|
|
return( self.get_stats_from_hand_session(hand) )
|
2009-08-02 14:00:55 +02:00
|
|
|
|
|
|
|
else: # hud_style == A
|
|
|
|
|
|
|
|
if hud_style == 'T':
|
|
|
|
stylekey = self.date_ndays_ago
|
|
|
|
#elif hud_style == 'H':
|
|
|
|
# stylekey = date_nhands_ago needs array by player here ...
|
|
|
|
else: # assume A (all-time)
|
|
|
|
stylekey = '0000000' # all stylekey values should be higher than this
|
|
|
|
|
2009-05-28 00:34:10 +02:00
|
|
|
if aggregate:
|
|
|
|
query = 'get_stats_from_hand_aggregated'
|
2009-08-02 14:00:55 +02:00
|
|
|
subs = (hand, stylekey, agg_bb_mult, agg_bb_mult)
|
2009-05-28 00:34:10 +02:00
|
|
|
else:
|
|
|
|
query = 'get_stats_from_hand'
|
2009-08-02 14:00:55 +02:00
|
|
|
subs = (hand, stylekey)
|
|
|
|
|
|
|
|
#print "get stats: hud style =", hud_style, "query =", query, "subs =", subs
|
2009-05-28 00:34:10 +02:00
|
|
|
c = self.connection.cursor()
|
2008-08-19 00:53:25 +02:00
|
|
|
|
2009-05-28 00:34:10 +02:00
|
|
|
# now get the stats
|
2008-11-11 15:50:20 +01:00
|
|
|
c.execute(self.sql.query[query], subs)
|
2008-08-19 00:53:25 +02:00
|
|
|
colnames = [desc[0] for desc in c.description]
|
|
|
|
stat_dict = {}
|
|
|
|
for row in c.fetchall():
|
|
|
|
t_dict = {}
|
|
|
|
for name, val in zip(colnames, row):
|
2009-04-27 22:29:02 +02:00
|
|
|
t_dict[name.lower()] = val
|
2009-04-29 18:16:39 +02:00
|
|
|
# print t_dict
|
2008-08-19 00:53:25 +02:00
|
|
|
stat_dict[t_dict['player_id']] = t_dict
|
2009-05-28 00:34:10 +02:00
|
|
|
|
|
|
|
return stat_dict
|
|
|
|
|
|
|
|
# uses query on handsplayers instead of hudcache to get stats on just this session
|
|
|
|
def get_stats_from_hand_session(self, hand):
|
|
|
|
|
2009-08-02 14:00:55 +02:00
|
|
|
query = self.sql.query['get_stats_from_hand_session']
|
|
|
|
if self.db_server == 'mysql':
|
|
|
|
query = query.replace("<signed>", 'signed ')
|
|
|
|
else:
|
|
|
|
query = query.replace("<signed>", '')
|
2009-05-28 00:34:10 +02:00
|
|
|
|
|
|
|
subs = (self.hand_1day_ago, hand)
|
2009-08-02 00:15:04 +02:00
|
|
|
c = self.get_cursor()
|
2009-05-28 00:34:10 +02:00
|
|
|
|
|
|
|
# now get the stats
|
|
|
|
#print "sess_stats: subs =", subs, "subs[0] =", subs[0]
|
|
|
|
c.execute(query, subs)
|
|
|
|
colnames = [desc[0] for desc in c.description]
|
|
|
|
n,stat_dict = 0,{}
|
|
|
|
row = c.fetchone()
|
|
|
|
while row:
|
|
|
|
if colnames[0].lower() == 'player_id':
|
|
|
|
playerid = row[0]
|
|
|
|
else:
|
|
|
|
print "ERROR: query %s result does not have player_id as first column" % (query,)
|
|
|
|
break
|
|
|
|
|
|
|
|
for name, val in zip(colnames, row):
|
|
|
|
if not playerid in stat_dict:
|
|
|
|
stat_dict[playerid] = {}
|
|
|
|
stat_dict[playerid][name.lower()] = val
|
|
|
|
elif not name.lower() in stat_dict[playerid]:
|
|
|
|
stat_dict[playerid][name.lower()] = val
|
|
|
|
elif name.lower() not in ('hand_id', 'player_id', 'seat', 'screen_name', 'seats'):
|
|
|
|
stat_dict[playerid][name.lower()] += val
|
|
|
|
n += 1
|
|
|
|
if n >= 4000: break # todo: don't think this is needed so set nice and high
|
|
|
|
# for now - comment out or remove?
|
|
|
|
row = c.fetchone()
|
|
|
|
#print " %d rows fetched, len(stat_dict) = %d" % (n, len(stat_dict))
|
|
|
|
|
|
|
|
#print "session stat_dict =", stat_dict
|
2008-08-19 00:53:25 +02:00
|
|
|
return stat_dict
|
|
|
|
|
|
|
|
def get_player_id(self, config, site, player_name):
|
|
|
|
c = self.connection.cursor()
|
2009-08-05 00:17:31 +02:00
|
|
|
c.execute(self.sql.query['get_player_id'], (player_name, site))
|
2008-08-19 00:53:25 +02:00
|
|
|
row = c.fetchone()
|
2009-05-21 22:27:44 +02:00
|
|
|
if row:
|
|
|
|
return row[0]
|
|
|
|
else:
|
|
|
|
return None
|
2009-07-29 00:58:10 +02:00
|
|
|
|
|
|
|
def get_site_id(self, site):
|
|
|
|
c = self.get_cursor()
|
|
|
|
c.execute(self.sql.query['getSiteId'], (site,))
|
|
|
|
result = c.fetchall()
|
|
|
|
return result
|
2008-08-19 00:53:25 +02:00
|
|
|
|
2009-08-06 21:31:46 +02:00
|
|
|
def get_last_insert_id(self, cursor=None):
|
|
|
|
ret = None
|
2009-07-26 02:42:09 +02:00
|
|
|
try:
|
2009-08-06 21:31:46 +02:00
|
|
|
if self.backend == self.MYSQL_INNODB:
|
|
|
|
ret = self.connection.insert_id()
|
|
|
|
if ret < 1 or ret > 999999999:
|
|
|
|
print "getLastInsertId(): problem fetching insert_id? ret=", ret
|
|
|
|
ret = -1
|
|
|
|
elif self.backend == self.PGSQL:
|
|
|
|
# some options:
|
|
|
|
# currval(hands_id_seq) - use name of implicit seq here
|
|
|
|
# lastval() - still needs sequences set up?
|
|
|
|
# insert ... returning is useful syntax (but postgres specific?)
|
|
|
|
# see rules (fancy trigger type things)
|
|
|
|
c = self.get_cursor()
|
|
|
|
ret = c.execute ("SELECT lastval()")
|
|
|
|
row = c.fetchone()
|
|
|
|
if not row:
|
|
|
|
print "getLastInsertId(%s): problem fetching lastval? row=" % seq, row
|
|
|
|
ret = -1
|
|
|
|
else:
|
|
|
|
ret = row[0]
|
|
|
|
elif self.backend == self.SQLITE:
|
|
|
|
ret = cursor.lastrowid
|
|
|
|
else:
|
|
|
|
print "getLastInsertId(): unknown backend ", self.backend
|
|
|
|
ret = -1
|
2009-07-26 02:42:09 +02:00
|
|
|
except:
|
2009-08-06 21:31:46 +02:00
|
|
|
ret = -1
|
|
|
|
err = traceback.extract_tb(sys.exc_info()[2])
|
|
|
|
print "***get_last_insert_id error: " + str(sys.exc_info()[1])
|
|
|
|
print "\n".join( [e[0]+':'+str(e[1])+" "+e[2] for e in err] )
|
|
|
|
raise
|
2009-07-26 02:42:09 +02:00
|
|
|
return ret
|
2009-06-20 00:22:57 +02:00
|
|
|
|
2009-06-19 22:15:52 +02:00
|
|
|
|
|
|
|
#stores a stud/razz hand into the database
|
2009-07-26 02:42:09 +02:00
|
|
|
def ring_stud(self, config, settings, base, category, site_hand_no, gametype_id, hand_start_time
|
2009-06-19 22:15:52 +02:00
|
|
|
,names, player_ids, start_cashes, antes, card_values, card_suits, winnings, rakes
|
|
|
|
,action_types, allIns, action_amounts, actionNos, hudImportData, maxSeats, tableName
|
|
|
|
,seatNos):
|
|
|
|
|
2009-07-31 22:24:21 +02:00
|
|
|
fpdb_simple.fillCardArrays(len(names), base, category, card_values, card_suits)
|
2009-06-19 22:15:52 +02:00
|
|
|
|
2009-07-31 22:24:21 +02:00
|
|
|
hands_id = self.storeHands(self.backend, site_hand_no, gametype_id
|
|
|
|
,hand_start_time, names, tableName, maxSeats, hudImportData
|
|
|
|
,(None, None, None, None, None), (None, None, None, None, None))
|
2009-06-19 22:15:52 +02:00
|
|
|
|
2009-07-31 22:24:21 +02:00
|
|
|
#print "before calling store_hands_players_stud, antes:", antes
|
|
|
|
hands_players_ids = self.store_hands_players_stud(self.backend, hands_id, player_ids
|
|
|
|
,start_cashes, antes, card_values
|
|
|
|
,card_suits, winnings, rakes, seatNos)
|
2009-06-19 22:15:52 +02:00
|
|
|
|
2009-07-31 22:24:21 +02:00
|
|
|
if 'dropHudCache' not in settings or settings['dropHudCache'] != 'drop':
|
|
|
|
self.storeHudCache(self.backend, base, category, gametype_id, hand_start_time, player_ids, hudImportData)
|
2009-06-19 22:15:52 +02:00
|
|
|
|
|
|
|
return hands_id
|
|
|
|
#end def ring_stud
|
|
|
|
|
2009-07-26 02:42:09 +02:00
|
|
|
def ring_holdem_omaha(self, config, settings, base, category, site_hand_no, gametype_id
|
2009-06-19 22:15:52 +02:00
|
|
|
,hand_start_time, names, player_ids, start_cashes, positions, card_values
|
|
|
|
,card_suits, board_values, board_suits, winnings, rakes, action_types, allIns
|
|
|
|
,action_amounts, actionNos, hudImportData, maxSeats, tableName, seatNos):
|
|
|
|
"""stores a holdem/omaha hand into the database"""
|
|
|
|
|
2009-07-31 22:24:21 +02:00
|
|
|
t0 = time()
|
|
|
|
#print "in ring_holdem_omaha"
|
|
|
|
fpdb_simple.fillCardArrays(len(names), base, category, card_values, card_suits)
|
|
|
|
t1 = time()
|
|
|
|
fpdb_simple.fill_board_cards(board_values, board_suits)
|
|
|
|
t2 = time()
|
|
|
|
|
|
|
|
hands_id = self.storeHands(self.backend, site_hand_no, gametype_id
|
|
|
|
,hand_start_time, names, tableName, maxSeats
|
|
|
|
,hudImportData, board_values, board_suits)
|
|
|
|
#TEMPORARY CALL! - Just until all functions are migrated
|
|
|
|
t3 = time()
|
|
|
|
hands_players_ids = self.store_hands_players_holdem_omaha(
|
|
|
|
self.backend, category, hands_id, player_ids, start_cashes
|
|
|
|
, positions, card_values, card_suits, winnings, rakes, seatNos, hudImportData)
|
|
|
|
t4 = time()
|
|
|
|
if 'dropHudCache' not in settings or settings['dropHudCache'] != 'drop':
|
|
|
|
self.storeHudCache(self.backend, base, category, gametype_id, hand_start_time, player_ids, hudImportData)
|
|
|
|
t5 = time()
|
|
|
|
#print "fills=(%4.3f) saves=(%4.3f,%4.3f,%4.3f)" % (t2-t0, t3-t2, t4-t3, t5-t4)
|
2009-06-19 22:15:52 +02:00
|
|
|
return hands_id
|
|
|
|
#end def ring_holdem_omaha
|
|
|
|
|
2009-07-26 02:42:09 +02:00
|
|
|
def tourney_holdem_omaha(self, config, settings, base, category, siteTourneyNo, buyin, fee, knockout
|
2009-06-19 22:15:52 +02:00
|
|
|
,entries, prizepool, tourney_start, payin_amounts, ranks, tourneyTypeId
|
|
|
|
,siteId #end of tourney specific params
|
|
|
|
,site_hand_no, gametype_id, hand_start_time, names, player_ids
|
|
|
|
,start_cashes, positions, card_values, card_suits, board_values
|
|
|
|
,board_suits, winnings, rakes, action_types, allIns, action_amounts
|
|
|
|
,actionNos, hudImportData, maxSeats, tableName, seatNos):
|
|
|
|
"""stores a tourney holdem/omaha hand into the database"""
|
|
|
|
|
2009-07-31 22:24:21 +02:00
|
|
|
fpdb_simple.fillCardArrays(len(names), base, category, card_values, card_suits)
|
|
|
|
fpdb_simple.fill_board_cards(board_values, board_suits)
|
2009-06-19 22:15:52 +02:00
|
|
|
|
2009-07-31 22:24:21 +02:00
|
|
|
tourney_id = self.store_tourneys(tourneyTypeId, siteTourneyNo, entries, prizepool, tourney_start)
|
|
|
|
tourneys_players_ids = self.store_tourneys_players(tourney_id, player_ids, payin_amounts, ranks, winnings)
|
2009-06-19 22:15:52 +02:00
|
|
|
|
2009-07-31 22:24:21 +02:00
|
|
|
hands_id = self.storeHands(self.backend, site_hand_no, gametype_id
|
|
|
|
,hand_start_time, names, tableName, maxSeats
|
|
|
|
,hudImportData, board_values, board_suits)
|
2009-06-19 22:15:52 +02:00
|
|
|
|
2009-07-31 22:24:21 +02:00
|
|
|
hands_players_ids = self.store_hands_players_holdem_omaha_tourney(
|
|
|
|
self.backend, category, hands_id, player_ids, start_cashes, positions
|
|
|
|
, card_values, card_suits, winnings, rakes, seatNos, tourneys_players_ids
|
|
|
|
, hudImportData)
|
2009-06-19 22:15:52 +02:00
|
|
|
|
2009-07-31 22:24:21 +02:00
|
|
|
#print "tourney holdem, backend=%d" % backend
|
|
|
|
if 'dropHudCache' not in settings or settings['dropHudCache'] != 'drop':
|
|
|
|
self.storeHudCache(self.backend, base, category, gametype_id, hand_start_time, player_ids, hudImportData)
|
2009-06-19 22:15:52 +02:00
|
|
|
|
|
|
|
return hands_id
|
|
|
|
#end def tourney_holdem_omaha
|
|
|
|
|
2009-07-26 02:42:09 +02:00
|
|
|
def tourney_stud(self, config, settings, base, category, siteTourneyNo, buyin, fee, knockout, entries
|
2009-06-19 22:15:52 +02:00
|
|
|
,prizepool, tourneyStartTime, payin_amounts, ranks, tourneyTypeId, siteId
|
|
|
|
,siteHandNo, gametypeId, handStartTime, names, playerIds, startCashes, antes
|
|
|
|
,cardValues, cardSuits, winnings, rakes, actionTypes, allIns, actionAmounts
|
|
|
|
,actionNos, hudImportData, maxSeats, tableName, seatNos):
|
|
|
|
#stores a tourney stud/razz hand into the database
|
|
|
|
|
2009-07-31 22:24:21 +02:00
|
|
|
fpdb_simple.fillCardArrays(len(names), base, category, cardValues, cardSuits)
|
2009-06-19 22:15:52 +02:00
|
|
|
|
2009-07-31 22:24:21 +02:00
|
|
|
tourney_id = self.store_tourneys(tourneyTypeId, siteTourneyNo, entries, prizepool, tourneyStartTime)
|
2009-06-19 22:15:52 +02:00
|
|
|
|
2009-07-31 22:24:21 +02:00
|
|
|
tourneys_players_ids = self.store_tourneys_players(tourney_id, playerIds, payin_amounts, ranks, winnings)
|
2009-06-19 22:15:52 +02:00
|
|
|
|
2009-07-31 22:24:21 +02:00
|
|
|
hands_id = self.storeHands( self.backend, siteHandNo, gametypeId
|
|
|
|
, handStartTime, names, tableName, maxSeats
|
|
|
|
, hudImportData, board_values, board_suits )
|
2009-06-19 22:15:52 +02:00
|
|
|
|
2009-07-31 22:24:21 +02:00
|
|
|
hands_players_ids = self.store_hands_players_stud_tourney(self.backend, hands_id
|
|
|
|
, playerIds, startCashes, antes, cardValues, cardSuits
|
|
|
|
, winnings, rakes, seatNos, tourneys_players_ids)
|
2009-06-19 22:15:52 +02:00
|
|
|
|
2009-07-31 22:24:21 +02:00
|
|
|
if 'dropHudCache' not in settings or settings['dropHudCache'] != 'drop':
|
|
|
|
self.storeHudCache(self.backend, base, category, gametypeId, hand_start_time, playerIds, hudImportData)
|
2009-06-19 22:15:52 +02:00
|
|
|
|
|
|
|
return hands_id
|
|
|
|
#end def tourney_stud
|
|
|
|
|
2009-07-29 00:58:10 +02:00
|
|
|
def prepareBulkImport(self):
|
|
|
|
"""Drop some indexes/foreign keys to prepare for bulk import.
|
|
|
|
Currently keeping the standalone indexes as needed to import quickly"""
|
|
|
|
stime = time()
|
2009-07-29 07:17:51 +02:00
|
|
|
c = self.get_cursor()
|
2009-07-29 08:57:18 +02:00
|
|
|
# sc: don't think autocommit=0 is needed, should already be in that mode
|
2009-07-29 00:58:10 +02:00
|
|
|
if self.backend == self.MYSQL_INNODB:
|
2009-07-29 07:17:51 +02:00
|
|
|
c.execute("SET foreign_key_checks=0")
|
|
|
|
c.execute("SET autocommit=0")
|
2009-07-29 00:58:10 +02:00
|
|
|
return
|
|
|
|
if self.backend == self.PGSQL:
|
|
|
|
self.connection.set_isolation_level(0) # allow table/index operations to work
|
|
|
|
for fk in self.foreignKeys[self.backend]:
|
|
|
|
if fk['drop'] == 1:
|
|
|
|
if self.backend == self.MYSQL_INNODB:
|
|
|
|
c.execute("SELECT constraint_name " +
|
|
|
|
"FROM information_schema.KEY_COLUMN_USAGE " +
|
|
|
|
#"WHERE REFERENCED_TABLE_SCHEMA = 'fpdb'
|
|
|
|
"WHERE 1=1 " +
|
|
|
|
"AND table_name = %s AND column_name = %s " +
|
|
|
|
"AND referenced_table_name = %s " +
|
|
|
|
"AND referenced_column_name = %s ",
|
|
|
|
(fk['fktab'], fk['fkcol'], fk['rtab'], fk['rcol']) )
|
|
|
|
cons = c.fetchone()
|
2009-07-31 22:24:21 +02:00
|
|
|
#print "preparebulk find fk: cons=", cons
|
2009-07-29 00:58:10 +02:00
|
|
|
if cons:
|
|
|
|
print "dropping mysql fk", cons[0], fk['fktab'], fk['fkcol']
|
|
|
|
try:
|
|
|
|
c.execute("alter table " + fk['fktab'] + " drop foreign key " + cons[0])
|
|
|
|
except:
|
2009-07-29 08:57:18 +02:00
|
|
|
print " drop failed: " + str(sys.exc_info())
|
2009-07-29 00:58:10 +02:00
|
|
|
elif self.backend == self.PGSQL:
|
|
|
|
# DON'T FORGET TO RECREATE THEM!!
|
|
|
|
print "dropping pg fk", fk['fktab'], fk['fkcol']
|
|
|
|
try:
|
|
|
|
# try to lock table to see if index drop will work:
|
|
|
|
# hmmm, tested by commenting out rollback in grapher. lock seems to work but
|
|
|
|
# then drop still hangs :-( does work in some tests though??
|
|
|
|
# will leave code here for now pending further tests/enhancement ...
|
|
|
|
c.execute( "lock table %s in exclusive mode nowait" % (fk['fktab'],) )
|
|
|
|
#print "after lock, status:", c.statusmessage
|
|
|
|
#print "alter table %s drop constraint %s_%s_fkey" % (fk['fktab'], fk['fktab'], fk['fkcol'])
|
|
|
|
try:
|
|
|
|
c.execute("alter table %s drop constraint %s_%s_fkey" % (fk['fktab'], fk['fktab'], fk['fkcol']))
|
|
|
|
print "dropped pg fk pg fk %s_%s_fkey, continuing ..." % (fk['fktab'], fk['fkcol'])
|
|
|
|
except:
|
|
|
|
if "does not exist" not in str(sys.exc_value):
|
|
|
|
print "warning: drop pg fk %s_%s_fkey failed: %s, continuing ..." \
|
|
|
|
% (fk['fktab'], fk['fkcol'], str(sys.exc_value).rstrip('\n') )
|
|
|
|
except:
|
|
|
|
print "warning: constraint %s_%s_fkey not dropped: %s, continuing ..." \
|
|
|
|
% (fk['fktab'],fk['fkcol'], str(sys.exc_value).rstrip('\n'))
|
|
|
|
else:
|
|
|
|
print "Only MySQL and Postgres supported so far"
|
|
|
|
return -1
|
|
|
|
|
2009-07-29 07:17:51 +02:00
|
|
|
for idx in self.indexes[self.backend]:
|
2009-07-29 00:58:10 +02:00
|
|
|
if idx['drop'] == 1:
|
|
|
|
if self.backend == self.MYSQL_INNODB:
|
|
|
|
print "dropping mysql index ", idx['tab'], idx['col']
|
|
|
|
try:
|
2009-07-29 08:57:18 +02:00
|
|
|
# apparently nowait is not implemented in mysql so this just hangs if there are locks
|
2009-07-29 00:58:10 +02:00
|
|
|
# preventing the index drop :-(
|
2009-07-29 08:57:18 +02:00
|
|
|
c.execute( "alter table %s drop index %s;", (idx['tab'],idx['col']) )
|
2009-07-29 00:58:10 +02:00
|
|
|
except:
|
2009-07-29 08:57:18 +02:00
|
|
|
print " drop index failed: " + str(sys.exc_info())
|
|
|
|
# ALTER TABLE `fpdb`.`handsplayers` DROP INDEX `playerId`;
|
|
|
|
# using: 'HandsPlayers' drop index 'playerId'
|
2009-07-29 00:58:10 +02:00
|
|
|
elif self.backend == self.PGSQL:
|
|
|
|
# DON'T FORGET TO RECREATE THEM!!
|
|
|
|
print "dropping pg index ", idx['tab'], idx['col']
|
|
|
|
try:
|
|
|
|
# try to lock table to see if index drop will work:
|
|
|
|
c.execute( "lock table %s in exclusive mode nowait" % (idx['tab'],) )
|
|
|
|
#print "after lock, status:", c.statusmessage
|
|
|
|
try:
|
|
|
|
# table locked ok so index drop should work:
|
|
|
|
#print "drop index %s_%s_idx" % (idx['tab'],idx['col'])
|
|
|
|
c.execute( "drop index if exists %s_%s_idx" % (idx['tab'],idx['col']) )
|
|
|
|
#print "dropped pg index ", idx['tab'], idx['col']
|
|
|
|
except:
|
|
|
|
if "does not exist" not in str(sys.exc_value):
|
|
|
|
print "warning: drop index %s_%s_idx failed: %s, continuing ..." \
|
|
|
|
% (idx['tab'],idx['col'], str(sys.exc_value).rstrip('\n'))
|
|
|
|
except:
|
|
|
|
print "warning: index %s_%s_idx not dropped %s, continuing ..." \
|
|
|
|
% (idx['tab'],idx['col'], str(sys.exc_value).rstrip('\n'))
|
|
|
|
else:
|
|
|
|
print "Error: Only MySQL and Postgres supported so far"
|
|
|
|
return -1
|
2009-07-21 23:26:23 +02:00
|
|
|
|
2009-07-29 00:58:10 +02:00
|
|
|
if self.backend == self.PGSQL:
|
|
|
|
self.connection.set_isolation_level(1) # go back to normal isolation level
|
|
|
|
self.commit() # seems to clear up errors if there were any in postgres
|
|
|
|
ptime = time() - stime
|
|
|
|
print "prepare import took", ptime, "seconds"
|
|
|
|
#end def prepareBulkImport
|
|
|
|
|
|
|
|
def afterBulkImport(self):
|
|
|
|
"""Re-create any dropped indexes/foreign keys after bulk import"""
|
2009-07-21 23:26:23 +02:00
|
|
|
stime = time()
|
2009-07-29 00:58:10 +02:00
|
|
|
|
2009-07-29 07:17:51 +02:00
|
|
|
c = self.get_cursor()
|
2009-07-29 00:58:10 +02:00
|
|
|
if self.backend == self.MYSQL_INNODB:
|
|
|
|
c.execute("SET foreign_key_checks=1")
|
|
|
|
c.execute("SET autocommit=1")
|
|
|
|
return
|
|
|
|
|
|
|
|
if self.backend == self.PGSQL:
|
|
|
|
self.connection.set_isolation_level(0) # allow table/index operations to work
|
|
|
|
for fk in self.foreignKeys[self.backend]:
|
|
|
|
if fk['drop'] == 1:
|
|
|
|
if self.backend == self.MYSQL_INNODB:
|
|
|
|
c.execute("SELECT constraint_name " +
|
|
|
|
"FROM information_schema.KEY_COLUMN_USAGE " +
|
|
|
|
#"WHERE REFERENCED_TABLE_SCHEMA = 'fpdb'
|
|
|
|
"WHERE 1=1 " +
|
|
|
|
"AND table_name = %s AND column_name = %s " +
|
|
|
|
"AND referenced_table_name = %s " +
|
|
|
|
"AND referenced_column_name = %s ",
|
|
|
|
(fk['fktab'], fk['fkcol'], fk['rtab'], fk['rcol']) )
|
|
|
|
cons = c.fetchone()
|
|
|
|
#print "afterbulk: cons=", cons
|
|
|
|
if cons:
|
|
|
|
pass
|
|
|
|
else:
|
|
|
|
print "creating fk ", fk['fktab'], fk['fkcol'], "->", fk['rtab'], fk['rcol']
|
|
|
|
try:
|
|
|
|
c.execute("alter table " + fk['fktab'] + " add foreign key ("
|
|
|
|
+ fk['fkcol'] + ") references " + fk['rtab'] + "("
|
|
|
|
+ fk['rcol'] + ")")
|
|
|
|
except:
|
2009-07-29 08:57:18 +02:00
|
|
|
print " create fk failed: " + str(sys.exc_info())
|
2009-07-29 00:58:10 +02:00
|
|
|
elif self.backend == self.PGSQL:
|
|
|
|
print "creating fk ", fk['fktab'], fk['fkcol'], "->", fk['rtab'], fk['rcol']
|
|
|
|
try:
|
|
|
|
c.execute("alter table " + fk['fktab'] + " add constraint "
|
|
|
|
+ fk['fktab'] + '_' + fk['fkcol'] + '_fkey'
|
|
|
|
+ " foreign key (" + fk['fkcol']
|
|
|
|
+ ") references " + fk['rtab'] + "(" + fk['rcol'] + ")")
|
|
|
|
except:
|
2009-07-29 08:57:18 +02:00
|
|
|
print " create fk failed: " + str(sys.exc_info())
|
2009-07-29 00:58:10 +02:00
|
|
|
else:
|
|
|
|
print "Only MySQL and Postgres supported so far"
|
|
|
|
return -1
|
|
|
|
|
2009-07-29 07:17:51 +02:00
|
|
|
for idx in self.indexes[self.backend]:
|
2009-07-29 00:58:10 +02:00
|
|
|
if idx['drop'] == 1:
|
|
|
|
if self.backend == self.MYSQL_INNODB:
|
|
|
|
print "creating mysql index ", idx['tab'], idx['col']
|
|
|
|
try:
|
2009-08-03 02:30:51 +02:00
|
|
|
s = "alter table %s add index %s(%s)" % (idx['tab'],idx['col'],idx['col'])
|
|
|
|
c.execute(s)
|
2009-07-29 00:58:10 +02:00
|
|
|
except:
|
2009-07-29 08:57:18 +02:00
|
|
|
print " create fk failed: " + str(sys.exc_info())
|
2009-07-29 00:58:10 +02:00
|
|
|
elif self.backend == self.PGSQL:
|
|
|
|
# pass
|
|
|
|
# mod to use tab_col for index name?
|
|
|
|
print "creating pg index ", idx['tab'], idx['col']
|
|
|
|
try:
|
2009-08-03 02:30:51 +02:00
|
|
|
s = "create index %s_%s_idx on %s(%s)" % (idx['tab'], idx['col'], idx['tab'], idx['col'])
|
|
|
|
c.execute(s)
|
2009-07-29 00:58:10 +02:00
|
|
|
except:
|
2009-07-29 08:57:18 +02:00
|
|
|
print " create index failed: " + str(sys.exc_info())
|
2009-07-29 00:58:10 +02:00
|
|
|
else:
|
|
|
|
print "Only MySQL and Postgres supported so far"
|
|
|
|
return -1
|
|
|
|
|
|
|
|
if self.backend == self.PGSQL:
|
|
|
|
self.connection.set_isolation_level(1) # go back to normal isolation level
|
|
|
|
self.commit() # seems to clear up errors if there were any in postgres
|
|
|
|
atime = time() - stime
|
|
|
|
print "After import took", atime, "seconds"
|
|
|
|
#end def afterBulkImport
|
|
|
|
|
|
|
|
def drop_referential_integrity(self):
|
|
|
|
"""Update all tables to remove foreign keys"""
|
|
|
|
|
|
|
|
c = self.get_cursor()
|
|
|
|
c.execute(self.sql.query['list_tables'])
|
|
|
|
result = c.fetchall()
|
|
|
|
|
|
|
|
for i in range(len(result)):
|
|
|
|
c.execute("SHOW CREATE TABLE " + result[i][0])
|
|
|
|
inner = c.fetchall()
|
|
|
|
|
|
|
|
for j in range(len(inner)):
|
|
|
|
# result[i][0] - Table name
|
|
|
|
# result[i][1] - CREATE TABLE parameters
|
|
|
|
#Searching for CONSTRAINT `tablename_ibfk_1`
|
|
|
|
for m in re.finditer('(ibfk_[0-9]+)', inner[j][1]):
|
|
|
|
key = "`" + inner[j][0] + "_" + m.group() + "`"
|
|
|
|
c.execute("ALTER TABLE " + inner[j][0] + " DROP FOREIGN KEY " + key)
|
|
|
|
self.commit()
|
|
|
|
#end drop_referential_inegrity
|
|
|
|
|
|
|
|
def recreate_tables(self):
|
|
|
|
"""(Re-)creates the tables of the current DB"""
|
|
|
|
|
|
|
|
self.drop_tables()
|
|
|
|
self.create_tables()
|
|
|
|
self.createAllIndexes()
|
2009-07-21 23:26:23 +02:00
|
|
|
self.commit()
|
2009-07-29 00:58:10 +02:00
|
|
|
print "Finished recreating tables"
|
|
|
|
#end def recreate_tables
|
|
|
|
|
|
|
|
def create_tables(self):
|
|
|
|
#todo: should detect and fail gracefully if tables already exist.
|
|
|
|
try:
|
|
|
|
logging.debug(self.sql.query['createSettingsTable'])
|
|
|
|
c = self.get_cursor()
|
|
|
|
c.execute(self.sql.query['createSettingsTable'])
|
|
|
|
logging.debug(self.sql.query['createSitesTable'])
|
|
|
|
c.execute(self.sql.query['createSitesTable'])
|
|
|
|
c.execute(self.sql.query['createGametypesTable'])
|
|
|
|
c.execute(self.sql.query['createPlayersTable'])
|
|
|
|
c.execute(self.sql.query['createAutoratesTable'])
|
|
|
|
c.execute(self.sql.query['createHandsTable'])
|
|
|
|
c.execute(self.sql.query['createTourneyTypesTable'])
|
|
|
|
c.execute(self.sql.query['createTourneysTable'])
|
|
|
|
c.execute(self.sql.query['createTourneysPlayersTable'])
|
|
|
|
c.execute(self.sql.query['createHandsPlayersTable'])
|
|
|
|
c.execute(self.sql.query['createHandsActionsTable'])
|
|
|
|
c.execute(self.sql.query['createHudCacheTable'])
|
|
|
|
#c.execute(self.sql.query['addTourneyIndex'])
|
|
|
|
#c.execute(self.sql.query['addHandsIndex'])
|
|
|
|
#c.execute(self.sql.query['addPlayersIndex'])
|
|
|
|
self.fillDefaultData()
|
|
|
|
self.commit()
|
|
|
|
except:
|
2009-08-04 23:06:03 +02:00
|
|
|
#print "Error creating tables: ", str(sys.exc_value)
|
|
|
|
err = traceback.extract_tb(sys.exc_info()[2])[-1]
|
|
|
|
print "***Error creating tables: "+err[2]+"("+str(err[1])+"): "+str(sys.exc_info()[1])
|
2009-07-29 00:58:10 +02:00
|
|
|
self.rollback()
|
2009-08-04 23:06:03 +02:00
|
|
|
raise
|
2009-07-29 00:58:10 +02:00
|
|
|
#end def disconnect
|
|
|
|
|
|
|
|
def drop_tables(self):
|
|
|
|
"""Drops the fpdb tables from the current db"""
|
|
|
|
|
|
|
|
try:
|
2009-07-29 07:17:51 +02:00
|
|
|
c = self.get_cursor()
|
2009-07-29 00:58:10 +02:00
|
|
|
if(self.get_backend_name() == 'MySQL InnoDB'):
|
|
|
|
#Databases with FOREIGN KEY support need this switched of before you can drop tables
|
|
|
|
self.drop_referential_integrity()
|
|
|
|
|
|
|
|
# Query the DB to see what tables exist
|
|
|
|
c.execute(self.sql.query['list_tables'])
|
|
|
|
for table in c:
|
|
|
|
c.execute(self.sql.query['drop_table'] + table[0])
|
|
|
|
elif(self.get_backend_name() == 'PostgreSQL'):
|
|
|
|
self.commit()# I have no idea why this makes the query work--REB 07OCT2008
|
|
|
|
c.execute(self.sql.query['list_tables'])
|
|
|
|
tables = c.fetchall()
|
|
|
|
for table in tables:
|
|
|
|
c.execute(self.sql.query['drop_table'] + table[0] + ' cascade')
|
|
|
|
elif(self.get_backend_name() == 'SQLite'):
|
|
|
|
c.execute(self.sql.query['list_tables'])
|
|
|
|
for table in c.fetchall():
|
|
|
|
logging.debug(self.sql.query['drop_table'] + table[0])
|
|
|
|
c.execute(self.sql.query['drop_table'] + table[0])
|
|
|
|
|
|
|
|
self.commit()
|
|
|
|
except:
|
2009-08-06 21:31:46 +02:00
|
|
|
err = traceback.extract_tb(sys.exc_info()[2])[-1]
|
2009-08-04 23:06:03 +02:00
|
|
|
print "***Error dropping tables: "+err[2]+"("+str(err[1])+"): "+str(sys.exc_info()[1])
|
|
|
|
self.rollback()
|
|
|
|
raise
|
2009-07-29 00:58:10 +02:00
|
|
|
#end def drop_tables
|
|
|
|
|
|
|
|
def createAllIndexes(self):
|
|
|
|
"""Create new indexes"""
|
|
|
|
|
|
|
|
try:
|
|
|
|
if self.backend == self.PGSQL:
|
|
|
|
self.connection.set_isolation_level(0) # allow table/index operations to work
|
|
|
|
for idx in self.indexes[self.backend]:
|
|
|
|
if self.backend == self.MYSQL_INNODB:
|
|
|
|
print "creating mysql index ", idx['tab'], idx['col']
|
|
|
|
try:
|
2009-08-03 02:30:51 +02:00
|
|
|
s = "create index %s on %s(%s)" % (idx['col'],idx['tab'],idx['col'])
|
|
|
|
self.get_cursor().execute(s)
|
2009-07-29 00:58:10 +02:00
|
|
|
except:
|
2009-08-03 02:30:51 +02:00
|
|
|
print " create idx failed: " + str(sys.exc_info())
|
2009-07-29 00:58:10 +02:00
|
|
|
elif self.backend == self.PGSQL:
|
|
|
|
# mod to use tab_col for index name?
|
|
|
|
print "creating pg index ", idx['tab'], idx['col']
|
|
|
|
try:
|
2009-08-03 02:30:51 +02:00
|
|
|
s = "create index %s_%s_idx on %s(%s)" % (idx['tab'], idx['col'], idx['tab'], idx['col'])
|
|
|
|
self.get_cursor().execute(s)
|
2009-07-29 00:58:10 +02:00
|
|
|
except:
|
2009-08-03 02:30:51 +02:00
|
|
|
print " create idx failed: " + str(sys.exc_info())
|
2009-07-29 00:58:10 +02:00
|
|
|
else:
|
|
|
|
print "Only MySQL and Postgres supported so far"
|
|
|
|
return -1
|
|
|
|
if self.backend == self.PGSQL:
|
|
|
|
self.connection.set_isolation_level(1) # go back to normal isolation level
|
|
|
|
except:
|
|
|
|
print "Error creating indexes: " + str(sys.exc_value)
|
|
|
|
raise fpdb_simple.FpdbError( "Error creating indexes " + str(sys.exc_value) )
|
|
|
|
#end def createAllIndexes
|
|
|
|
|
|
|
|
def dropAllIndexes(self):
|
|
|
|
"""Drop all standalone indexes (i.e. not including primary keys or foreign keys)
|
|
|
|
using list of indexes in indexes data structure"""
|
|
|
|
# maybe upgrade to use data dictionary?? (but take care to exclude PK and FK)
|
|
|
|
if self.backend == self.PGSQL:
|
|
|
|
self.connection.set_isolation_level(0) # allow table/index operations to work
|
|
|
|
for idx in self.indexes[self.backend]:
|
|
|
|
if self.backend == self.MYSQL_INNODB:
|
|
|
|
print "dropping mysql index ", idx['tab'], idx['col']
|
|
|
|
try:
|
|
|
|
self.get_cursor().execute( "alter table %s drop index %s"
|
|
|
|
, (idx['tab'],idx['col']) )
|
|
|
|
except:
|
|
|
|
pass
|
|
|
|
elif self.backend == self.PGSQL:
|
|
|
|
print "dropping pg index ", idx['tab'], idx['col']
|
|
|
|
# mod to use tab_col for index name?
|
|
|
|
try:
|
|
|
|
self.get_cursor().execute( "drop index %s_%s_idx"
|
|
|
|
% (idx['tab'],idx['col']) )
|
|
|
|
except:
|
|
|
|
pass
|
|
|
|
else:
|
|
|
|
print "Only MySQL and Postgres supported so far"
|
|
|
|
return -1
|
|
|
|
if self.backend == self.PGSQL:
|
|
|
|
self.connection.set_isolation_level(1) # go back to normal isolation level
|
|
|
|
#end def dropAllIndexes
|
|
|
|
|
|
|
|
def fillDefaultData(self):
|
2009-08-05 18:58:25 +02:00
|
|
|
c = self.get_cursor()
|
2009-07-29 00:58:10 +02:00
|
|
|
c.execute("INSERT INTO Settings (version) VALUES (118);")
|
|
|
|
c.execute("INSERT INTO Sites (name,currency) VALUES ('Full Tilt Poker', 'USD')")
|
|
|
|
c.execute("INSERT INTO Sites (name,currency) VALUES ('PokerStars', 'USD')")
|
|
|
|
c.execute("INSERT INTO Sites (name,currency) VALUES ('Everleaf', 'USD')")
|
|
|
|
c.execute("INSERT INTO Sites (name,currency) VALUES ('Win2day', 'USD')")
|
2009-08-06 04:18:48 +02:00
|
|
|
c.execute("INSERT INTO Sites (name,currency) VALUES ('OnGame', 'USD')")
|
|
|
|
c.execute("INSERT INTO Sites (name,currency) VALUES ('UltimateBet', 'USD')")
|
|
|
|
c.execute("INSERT INTO Sites (name,currency) VALUES ('Betfair', 'USD')")
|
2009-08-06 03:07:16 +02:00
|
|
|
c.execute("INSERT INTO Sites (name,currency) VALUES ('Absolute', 'USD')")
|
2009-08-06 14:54:39 +02:00
|
|
|
c.execute("INSERT INTO Sites (name,currency) VALUES ('PartyPoker', 'USD')")
|
2009-08-04 23:06:03 +02:00
|
|
|
if self.backend == self.SQLITE:
|
|
|
|
c.execute("INSERT INTO TourneyTypes VALUES (NULL, 1, 0, 0, 0, 0);")
|
|
|
|
else:
|
|
|
|
c.execute("INSERT INTO TourneyTypes VALUES (DEFAULT, 1, 0, 0, 0, False);")
|
2009-07-29 00:58:10 +02:00
|
|
|
#c.execute("""INSERT INTO TourneyTypes
|
|
|
|
# (siteId,buyin,fee,knockout,rebuyOrAddon) VALUES
|
|
|
|
# (1,0,0,0,?)""",(False,) )
|
|
|
|
#end def fillDefaultData
|
|
|
|
|
|
|
|
def rebuild_hudcache(self):
|
|
|
|
"""clears hudcache and rebuilds from the individual handsplayers records"""
|
|
|
|
|
|
|
|
try:
|
|
|
|
stime = time()
|
2009-08-07 01:04:44 +02:00
|
|
|
self.get_cursor().execute(self.sql.query['clearHudCache'])
|
|
|
|
self.get_cursor().execute(self.sql.query['rebuildHudCache'])
|
2009-07-29 00:58:10 +02:00
|
|
|
self.commit()
|
|
|
|
print "Rebuild hudcache took %.1f seconds" % (time() - stime,)
|
|
|
|
except:
|
2009-08-07 01:04:44 +02:00
|
|
|
err = traceback.extract_tb(sys.exc_info()[2])[-1]
|
2009-07-29 00:58:10 +02:00
|
|
|
print "Error rebuilding hudcache:", str(sys.exc_value)
|
2009-08-07 01:04:44 +02:00
|
|
|
print err
|
2009-07-21 23:26:23 +02:00
|
|
|
#end def rebuild_hudcache
|
|
|
|
|
|
|
|
|
|
|
|
def analyzeDB(self):
|
|
|
|
"""Do whatever the DB can offer to update index/table statistics"""
|
|
|
|
stime = time()
|
|
|
|
if self.backend == self.MYSQL_INNODB:
|
|
|
|
try:
|
2009-07-26 02:42:09 +02:00
|
|
|
self.get_cursor().execute(self.sql.query['analyze'])
|
2009-07-21 23:26:23 +02:00
|
|
|
except:
|
2009-07-26 02:42:09 +02:00
|
|
|
print "Error during analyze:", str(sys.exc_value)
|
2009-07-21 23:26:23 +02:00
|
|
|
elif self.backend == self.PGSQL:
|
|
|
|
self.connection.set_isolation_level(0) # allow vacuum to work
|
|
|
|
try:
|
2009-07-26 02:42:09 +02:00
|
|
|
self.get_cursor().execute(self.sql.query['analyze'])
|
2009-07-21 23:26:23 +02:00
|
|
|
except:
|
|
|
|
print "Error during analyze:", str(sys.exc_value)
|
|
|
|
self.connection.set_isolation_level(1) # go back to normal isolation level
|
|
|
|
self.commit()
|
|
|
|
atime = time() - stime
|
|
|
|
print "Analyze took %.1f seconds" % (atime,)
|
|
|
|
#end def analyzeDB
|
|
|
|
|
2009-07-26 02:42:09 +02:00
|
|
|
|
|
|
|
# Start of Hand Writing routines. Idea is to provide a mixture of routines to store Hand data
|
|
|
|
# however the calling prog requires. Main aims:
|
|
|
|
# - existing static routines from fpdb_simple just modified
|
2009-07-28 22:19:31 +02:00
|
|
|
|
|
|
|
def lock_for_insert(self):
|
|
|
|
"""Lock tables in MySQL to try to speed inserts up"""
|
|
|
|
try:
|
|
|
|
self.get_cursor().execute(self.sql.query['lockForInsert'])
|
|
|
|
except:
|
|
|
|
print "Error during fdb.lock_for_insert:", str(sys.exc_value)
|
|
|
|
#end def lock_for_insert
|
2009-07-29 07:17:51 +02:00
|
|
|
|
2009-08-02 06:19:33 +02:00
|
|
|
def getSqlPlayerIDs(self, pnames, siteid):
|
|
|
|
result = {}
|
|
|
|
if(self.pcache == None):
|
|
|
|
self.pcache = LambdaDict(lambda key:self.insertPlayer(key, siteid))
|
|
|
|
|
|
|
|
for player in pnames:
|
|
|
|
result[player] = self.pcache[player]
|
2009-08-04 12:22:29 +02:00
|
|
|
# NOTE: Using the LambdaDict does the same thing as:
|
|
|
|
#if player in self.pcache:
|
|
|
|
# #print "DEBUG: cachehit"
|
|
|
|
# pass
|
|
|
|
#else:
|
|
|
|
# self.pcache[player] = self.insertPlayer(player, siteid)
|
|
|
|
#result[player] = self.pcache[player]
|
2009-08-02 06:19:33 +02:00
|
|
|
|
|
|
|
return result
|
|
|
|
|
|
|
|
def insertPlayer(self, name, site_id):
|
|
|
|
result = None
|
|
|
|
c = self.get_cursor()
|
2009-08-08 08:59:23 +02:00
|
|
|
q = "SELECT name, id FROM Players WHERE siteid=%s and name=%s"
|
|
|
|
q = q.replace('%s', self.sql.query['placeholder'])
|
|
|
|
|
2009-08-08 10:15:36 +02:00
|
|
|
#print "DEBUG: name: %s site: %s" %(name, site_id)
|
2009-08-08 08:59:23 +02:00
|
|
|
|
|
|
|
c.execute (q, (site_id, name))
|
|
|
|
|
2009-08-06 21:31:46 +02:00
|
|
|
tmp = c.fetchone()
|
2009-08-08 06:33:25 +02:00
|
|
|
if (tmp == None): #new player
|
2009-08-06 21:31:46 +02:00
|
|
|
c.execute ("INSERT INTO Players (name, siteId) VALUES (%s, %s)".replace('%s',self.sql.query['placeholder'])
|
|
|
|
,(name, site_id))
|
2009-08-04 12:22:29 +02:00
|
|
|
#Get last id might be faster here.
|
2009-08-06 21:31:46 +02:00
|
|
|
#c.execute ("SELECT id FROM Players WHERE name=%s", (name,))
|
|
|
|
tmp = [self.get_last_insert_id(c)]
|
|
|
|
return tmp[0]
|
2009-08-02 06:19:33 +02:00
|
|
|
|
2009-08-06 21:31:46 +02:00
|
|
|
def insertGameTypes(self, row):
|
|
|
|
c = self.get_cursor()
|
|
|
|
c.execute( self.sql.query['insertGameTypes'], row )
|
|
|
|
return [self.get_last_insert_id(c)]
|
2009-07-29 07:17:51 +02:00
|
|
|
|
|
|
|
def store_the_hand(self, h):
|
|
|
|
"""Take a HandToWrite object and store it in the db"""
|
|
|
|
|
|
|
|
# Following code writes hands to database and commits (or rolls back if there is an error)
|
|
|
|
try:
|
|
|
|
result = None
|
|
|
|
if h.isTourney:
|
|
|
|
ranks = map(lambda x: 0, h.names) # create an array of 0's equal to the length of names
|
|
|
|
payin_amounts = fpdb_simple.calcPayin(len(h.names), h.buyin, h.fee)
|
|
|
|
|
|
|
|
if h.base == "hold":
|
|
|
|
result = self.tourney_holdem_omaha(
|
|
|
|
h.config, h.settings, h.base, h.category, h.siteTourneyNo, h.buyin
|
|
|
|
, h.fee, h.knockout, h.entries, h.prizepool, h.tourneyStartTime
|
2009-07-31 22:24:21 +02:00
|
|
|
, payin_amounts, ranks, h.tourneyTypeId, h.siteID, h.siteHandNo
|
2009-07-29 07:17:51 +02:00
|
|
|
, h.gametypeID, h.handStartTime, h.names, h.playerIDs, h.startCashes
|
|
|
|
, h.positions, h.cardValues, h.cardSuits, h.boardValues, h.boardSuits
|
|
|
|
, h.winnings, h.rakes, h.actionTypes, h.allIns, h.actionAmounts
|
|
|
|
, h.actionNos, h.hudImportData, h.maxSeats, h.tableName, h.seatNos)
|
|
|
|
elif h.base == "stud":
|
|
|
|
result = self.tourney_stud(
|
|
|
|
h.config, h.settings, h.base, h.category, h.siteTourneyNo
|
|
|
|
, h.buyin, h.fee, h.knockout, h.entries, h.prizepool, h.tourneyStartTime
|
2009-07-31 22:24:21 +02:00
|
|
|
, payin_amounts, ranks, h.tourneyTypeId, h.siteID, h.siteHandNo
|
2009-07-29 07:17:51 +02:00
|
|
|
, h.gametypeID, h.handStartTime, h.names, h.playerIDs, h.startCashes
|
|
|
|
, h.antes, h.cardValues, h.cardSuits, h.winnings, h.rakes, h.actionTypes
|
|
|
|
, h.allIns, h.actionAmounts, h.actionNos, h.hudImportData, h.maxSeats
|
|
|
|
, h.tableName, h.seatNos)
|
|
|
|
else:
|
2009-07-31 22:24:21 +02:00
|
|
|
raise fpdb_simple.FpdbError("unrecognised category")
|
2009-07-29 07:17:51 +02:00
|
|
|
else:
|
|
|
|
if h.base == "hold":
|
|
|
|
result = self.ring_holdem_omaha(
|
|
|
|
h.config, h.settings, h.base, h.category, h.siteHandNo
|
|
|
|
, h.gametypeID, h.handStartTime, h.names, h.playerIDs
|
|
|
|
, h.startCashes, h.positions, h.cardValues, h.cardSuits
|
|
|
|
, h.boardValues, h.boardSuits, h.winnings, h.rakes
|
|
|
|
, h.actionTypes, h.allIns, h.actionAmounts, h.actionNos
|
|
|
|
, h.hudImportData, h.maxSeats, h.tableName, h.seatNos)
|
|
|
|
elif h.base == "stud":
|
|
|
|
result = self.ring_stud(
|
|
|
|
h.config, h.settings, h.base, h.category, h.siteHandNo, h.gametypeID
|
|
|
|
, h.handStartTime, h.names, h.playerIDs, h.startCashes, h.antes
|
|
|
|
, h.cardValues, h.cardSuits, h.winnings, h.rakes, h.actionTypes, h.allIns
|
|
|
|
, h.actionAmounts, h.actionNos, h.hudImportData, h.maxSeats, h.tableName
|
|
|
|
, h.seatNos)
|
|
|
|
else:
|
2009-07-31 22:24:21 +02:00
|
|
|
raise fpdb_simple.FpdbError("unrecognised category")
|
2009-07-29 07:17:51 +02:00
|
|
|
except:
|
|
|
|
print "Error storing hand: " + str(sys.exc_value)
|
|
|
|
self.rollback()
|
2009-07-31 22:24:21 +02:00
|
|
|
# re-raise the exception so that the calling routine can decide what to do:
|
|
|
|
# (e.g. a write thread might try again)
|
|
|
|
raise
|
2009-07-29 07:17:51 +02:00
|
|
|
|
|
|
|
return result
|
|
|
|
#end def store_the_hand
|
|
|
|
|
2009-08-06 21:31:46 +02:00
|
|
|
def storeHand(self, p):
|
|
|
|
#stores into table hands:
|
2009-08-08 10:15:36 +02:00
|
|
|
q = """INSERT INTO Hands (
|
2009-08-06 21:31:46 +02:00
|
|
|
tablename,
|
|
|
|
gametypeid,
|
2009-08-08 10:15:36 +02:00
|
|
|
sitehandno,
|
2009-08-06 21:31:46 +02:00
|
|
|
handstart,
|
|
|
|
importtime,
|
2009-08-08 19:59:44 +02:00
|
|
|
seats,
|
2009-08-06 21:31:46 +02:00
|
|
|
maxseats,
|
|
|
|
boardcard1,
|
|
|
|
boardcard2,
|
|
|
|
boardcard3,
|
|
|
|
boardcard4,
|
2009-08-08 10:15:36 +02:00
|
|
|
boardcard5
|
2009-08-06 21:31:46 +02:00
|
|
|
)
|
|
|
|
VALUES
|
2009-08-08 19:59:44 +02:00
|
|
|
(%s, %s, %s, %s, %s, %s, %s, %s, %s, %s,
|
|
|
|
%s, %s)"""
|
2009-08-08 10:15:36 +02:00
|
|
|
#--- texture,
|
|
|
|
#-- playersVpi,
|
|
|
|
#-- playersAtStreet1,
|
|
|
|
#-- playersAtStreet2,
|
|
|
|
#-- playersAtStreet3,
|
|
|
|
#-- playersAtStreet4,
|
|
|
|
#-- playersAtShowdown,
|
|
|
|
#-- street0Raises,
|
|
|
|
#-- street1Raises,
|
|
|
|
#-- street2Raises,
|
|
|
|
#-- street3Raises,
|
|
|
|
#-- street4Raises,
|
|
|
|
#-- street1Pot,
|
|
|
|
#-- street2Pot,
|
|
|
|
#-- street3Pot,
|
|
|
|
#-- street4Pot,
|
|
|
|
#-- showdownPot
|
|
|
|
#-- seats,
|
|
|
|
|
|
|
|
q = q.replace('%s', self.sql.query['placeholder'])
|
2009-08-08 19:59:44 +02:00
|
|
|
print "DEBUG: p: %s" %p
|
|
|
|
print "DEBUG: gtid: %s" % p['gameTypeId']
|
2009-08-08 10:15:36 +02:00
|
|
|
self.cursor.execute(q, (
|
|
|
|
p['tableName'],
|
2009-08-08 19:59:44 +02:00
|
|
|
p['gameTypeId'],
|
2009-08-08 10:15:36 +02:00
|
|
|
p['siteHandNo'],
|
2009-08-06 21:31:46 +02:00
|
|
|
p['handStart'],
|
2009-08-08 10:15:36 +02:00
|
|
|
datetime.today(), #importtime
|
|
|
|
# len(p['names']), #seats
|
2009-08-06 21:31:46 +02:00
|
|
|
p['maxSeats'],
|
2009-08-08 19:59:44 +02:00
|
|
|
p['seats'],
|
2009-08-06 21:31:46 +02:00
|
|
|
p['boardcard1'],
|
|
|
|
p['boardcard2'],
|
|
|
|
p['boardcard3'],
|
|
|
|
p['boardcard4'],
|
2009-08-08 10:15:36 +02:00
|
|
|
p['boardcard5'])
|
|
|
|
# hudCache['playersVpi'],
|
|
|
|
# hudCache['playersAtStreet1'],
|
|
|
|
# hudCache['playersAtStreet2'],
|
|
|
|
# hudCache['playersAtStreet3'],
|
|
|
|
# hudCache['playersAtStreet4'],
|
|
|
|
# hudCache['playersAtShowdown'],
|
|
|
|
# hudCache['street0Raises'],
|
|
|
|
# hudCache['street1Raises'],
|
|
|
|
# hudCache['street2Raises'],
|
|
|
|
# hudCache['street3Raises'],
|
|
|
|
# hudCache['street4Raises'],
|
|
|
|
# hudCache['street1Pot'],
|
|
|
|
# hudCache['street2Pot'],
|
|
|
|
# hudCache['street3Pot'],
|
|
|
|
# hudCache['street4Pot'],
|
|
|
|
# hudCache['showdownPot']
|
2009-08-06 21:31:46 +02:00
|
|
|
)
|
|
|
|
#return getLastInsertId(backend, conn, cursor)
|
|
|
|
# def storeHand
|
|
|
|
|
2009-07-26 02:42:09 +02:00
|
|
|
def storeHands(self, backend, site_hand_no, gametype_id
|
|
|
|
,hand_start_time, names, tableName, maxSeats, hudCache
|
|
|
|
,board_values, board_suits):
|
|
|
|
|
|
|
|
cards = [Card.cardFromValueSuit(v,s) for v,s in zip(board_values,board_suits)]
|
|
|
|
#stores into table hands:
|
|
|
|
try:
|
2009-08-06 21:31:46 +02:00
|
|
|
c = self.get_cursor()
|
|
|
|
c.execute ("""INSERT INTO Hands
|
|
|
|
(siteHandNo, gametypeId, handStart, seats, tableName, importTime, maxSeats
|
|
|
|
,boardcard1,boardcard2,boardcard3,boardcard4,boardcard5
|
|
|
|
,playersVpi, playersAtStreet1, playersAtStreet2
|
|
|
|
,playersAtStreet3, playersAtStreet4, playersAtShowdown
|
|
|
|
,street0Raises, street1Raises, street2Raises
|
|
|
|
,street3Raises, street4Raises, street1Pot
|
|
|
|
,street2Pot, street3Pot, street4Pot
|
|
|
|
,showdownPot
|
|
|
|
)
|
|
|
|
VALUES
|
|
|
|
(%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s,
|
|
|
|
%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
|
|
|
|
""".replace('%s', self.sql.query['placeholder'])
|
|
|
|
, (site_hand_no, gametype_id, hand_start_time, len(names), tableName, datetime.today(), maxSeats
|
|
|
|
,cards[0], cards[1], cards[2], cards[3], cards[4]
|
|
|
|
,hudCache['playersVpi'], hudCache['playersAtStreet1'], hudCache['playersAtStreet2']
|
|
|
|
,hudCache['playersAtStreet3'], hudCache['playersAtStreet4'], hudCache['playersAtShowdown']
|
|
|
|
,hudCache['street0Raises'], hudCache['street1Raises'], hudCache['street2Raises']
|
|
|
|
,hudCache['street3Raises'], hudCache['street4Raises'], hudCache['street1Pot']
|
|
|
|
,hudCache['street2Pot'], hudCache['street3Pot'], hudCache['street4Pot']
|
|
|
|
,hudCache['showdownPot']
|
|
|
|
))
|
|
|
|
ret = self.get_last_insert_id(c)
|
2009-07-26 02:42:09 +02:00
|
|
|
except:
|
|
|
|
ret = -1
|
|
|
|
raise fpdb_simple.FpdbError( "storeHands error: " + str(sys.exc_value) )
|
|
|
|
|
|
|
|
return ret
|
|
|
|
#end def storeHands
|
|
|
|
|
|
|
|
def store_hands_players_holdem_omaha(self, backend, category, hands_id, player_ids, start_cashes
|
|
|
|
,positions, card_values, card_suits, winnings, rakes, seatNos, hudCache):
|
|
|
|
result=[]
|
|
|
|
|
|
|
|
# postgres (and others?) needs the booleans converted to ints before saving:
|
|
|
|
# (or we could just save them as boolean ... but then we can't sum them so easily in sql ???)
|
|
|
|
# NO - storing booleans for now so don't need this
|
|
|
|
#hudCacheInt = {}
|
|
|
|
#for k,v in hudCache.iteritems():
|
|
|
|
# if k in ('wonWhenSeenStreet1', 'wonAtSD', 'totalProfit'):
|
|
|
|
# hudCacheInt[k] = v
|
|
|
|
# else:
|
|
|
|
# hudCacheInt[k] = map(lambda x: 1 if x else 0, v)
|
|
|
|
|
|
|
|
try:
|
|
|
|
inserts = []
|
|
|
|
for i in xrange(len(player_ids)):
|
|
|
|
card1 = Card.cardFromValueSuit(card_values[i][0], card_suits[i][0])
|
|
|
|
card2 = Card.cardFromValueSuit(card_values[i][1], card_suits[i][1])
|
|
|
|
|
|
|
|
if (category=="holdem"):
|
|
|
|
startCards = Card.twoStartCards(card_values[i][0], card_suits[i][0], card_values[i][1], card_suits[i][1])
|
|
|
|
card3 = None
|
|
|
|
card4 = None
|
|
|
|
elif (category=="omahahi" or category=="omahahilo"):
|
|
|
|
startCards = Card.fourStartCards(card_values[i][0], card_suits[i][0], card_values[i][1], card_suits[i][1]
|
|
|
|
,card_values[i][2], card_suits[i][2], card_values[i][3], card_suits[i][3])
|
|
|
|
card3 = Card.cardFromValueSuit(card_values[i][2], card_suits[i][2])
|
|
|
|
card4 = Card.cardFromValueSuit(card_values[i][3], card_suits[i][3])
|
|
|
|
else:
|
|
|
|
raise fpdb_simple.FpdbError("invalid category")
|
|
|
|
|
|
|
|
inserts.append( (
|
|
|
|
hands_id, player_ids[i], start_cashes[i], positions[i], 1, # tourneytypeid
|
|
|
|
card1, card2, card3, card4, startCards,
|
|
|
|
winnings[i], rakes[i], seatNos[i], hudCache['totalProfit'][i],
|
|
|
|
hudCache['street0VPI'][i], hudCache['street0Aggr'][i],
|
|
|
|
hudCache['street0_3BChance'][i], hudCache['street0_3BDone'][i],
|
|
|
|
hudCache['street1Seen'][i], hudCache['street2Seen'][i], hudCache['street3Seen'][i],
|
|
|
|
hudCache['street4Seen'][i], hudCache['sawShowdown'][i],
|
|
|
|
hudCache['street1Aggr'][i], hudCache['street2Aggr'][i], hudCache['street3Aggr'][i], hudCache['street4Aggr'][i],
|
|
|
|
hudCache['otherRaisedStreet1'][i], hudCache['otherRaisedStreet2'][i],
|
|
|
|
hudCache['otherRaisedStreet3'][i], hudCache['otherRaisedStreet4'][i],
|
|
|
|
hudCache['foldToOtherRaisedStreet1'][i], hudCache['foldToOtherRaisedStreet2'][i],
|
|
|
|
hudCache['foldToOtherRaisedStreet3'][i], hudCache['foldToOtherRaisedStreet4'][i],
|
|
|
|
hudCache['wonWhenSeenStreet1'][i], hudCache['wonAtSD'][i],
|
|
|
|
hudCache['stealAttemptChance'][i], hudCache['stealAttempted'][i], hudCache['foldBbToStealChance'][i],
|
|
|
|
hudCache['foldedBbToSteal'][i], hudCache['foldSbToStealChance'][i], hudCache['foldedSbToSteal'][i],
|
|
|
|
hudCache['street1CBChance'][i], hudCache['street1CBDone'][i], hudCache['street2CBChance'][i], hudCache['street2CBDone'][i],
|
|
|
|
hudCache['street3CBChance'][i], hudCache['street3CBDone'][i], hudCache['street4CBChance'][i], hudCache['street4CBDone'][i],
|
|
|
|
hudCache['foldToStreet1CBChance'][i], hudCache['foldToStreet1CBDone'][i],
|
|
|
|
hudCache['foldToStreet2CBChance'][i], hudCache['foldToStreet2CBDone'][i],
|
|
|
|
hudCache['foldToStreet3CBChance'][i], hudCache['foldToStreet3CBDone'][i],
|
|
|
|
hudCache['foldToStreet4CBChance'][i], hudCache['foldToStreet4CBDone'][i],
|
|
|
|
hudCache['street1CheckCallRaiseChance'][i], hudCache['street1CheckCallRaiseDone'][i],
|
|
|
|
hudCache['street2CheckCallRaiseChance'][i], hudCache['street2CheckCallRaiseDone'][i],
|
|
|
|
hudCache['street3CheckCallRaiseChance'][i], hudCache['street3CheckCallRaiseDone'][i],
|
|
|
|
hudCache['street4CheckCallRaiseChance'][i], hudCache['street4CheckCallRaiseDone'][i],
|
|
|
|
hudCache['street0Calls'][i], hudCache['street1Calls'][i], hudCache['street2Calls'][i], hudCache['street3Calls'][i], hudCache['street4Calls'][i],
|
|
|
|
hudCache['street0Bets'][i], hudCache['street1Bets'][i], hudCache['street2Bets'][i], hudCache['street3Bets'][i], hudCache['street4Bets'][i]
|
|
|
|
) )
|
2009-08-06 21:31:46 +02:00
|
|
|
c = self.get_cursor()
|
|
|
|
c.executemany ("""
|
2009-07-26 02:42:09 +02:00
|
|
|
INSERT INTO HandsPlayers
|
|
|
|
(handId, playerId, startCash, position, tourneyTypeId,
|
|
|
|
card1, card2, card3, card4, startCards, winnings, rake, seatNo, totalProfit,
|
|
|
|
street0VPI, street0Aggr, street0_3BChance, street0_3BDone,
|
|
|
|
street1Seen, street2Seen, street3Seen, street4Seen, sawShowdown,
|
|
|
|
street1Aggr, street2Aggr, street3Aggr, street4Aggr,
|
|
|
|
otherRaisedStreet1, otherRaisedStreet2, otherRaisedStreet3, otherRaisedStreet4,
|
|
|
|
foldToOtherRaisedStreet1, foldToOtherRaisedStreet2, foldToOtherRaisedStreet3, foldToOtherRaisedStreet4,
|
|
|
|
wonWhenSeenStreet1, wonAtSD,
|
|
|
|
stealAttemptChance, stealAttempted, foldBbToStealChance, foldedBbToSteal, foldSbToStealChance, foldedSbToSteal,
|
|
|
|
street1CBChance, street1CBDone, street2CBChance, street2CBDone,
|
|
|
|
street3CBChance, street3CBDone, street4CBChance, street4CBDone,
|
|
|
|
foldToStreet1CBChance, foldToStreet1CBDone, foldToStreet2CBChance, foldToStreet2CBDone,
|
|
|
|
foldToStreet3CBChance, foldToStreet3CBDone, foldToStreet4CBChance, foldToStreet4CBDone,
|
|
|
|
street1CheckCallRaiseChance, street1CheckCallRaiseDone, street2CheckCallRaiseChance, street2CheckCallRaiseDone,
|
|
|
|
street3CheckCallRaiseChance, street3CheckCallRaiseDone, street4CheckCallRaiseChance, street4CheckCallRaiseDone,
|
|
|
|
street0Calls, street1Calls, street2Calls, street3Calls, street4Calls,
|
|
|
|
street0Bets, street1Bets, street2Bets, street3Bets, street4Bets
|
|
|
|
)
|
|
|
|
VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s,
|
|
|
|
%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s,
|
|
|
|
%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s,
|
2009-08-06 21:31:46 +02:00
|
|
|
%s, %s, %s, %s, %s, %s, %s, %s, %s, %s)""".replace('%s', self.sql.query['placeholder'])
|
2009-07-26 02:42:09 +02:00
|
|
|
,inserts )
|
2009-08-06 21:31:46 +02:00
|
|
|
result.append( self.get_last_insert_id(c) ) # wrong? not used currently
|
2009-07-26 02:42:09 +02:00
|
|
|
except:
|
|
|
|
raise fpdb_simple.FpdbError( "store_hands_players_holdem_omaha error: " + str(sys.exc_value) )
|
|
|
|
|
|
|
|
return result
|
|
|
|
#end def store_hands_players_holdem_omaha
|
|
|
|
|
|
|
|
def store_hands_players_stud(self, backend, hands_id, player_ids, start_cashes, antes,
|
|
|
|
card_values, card_suits, winnings, rakes, seatNos):
|
|
|
|
#stores hands_players rows for stud/razz games. returns an array of the resulting IDs
|
|
|
|
|
|
|
|
try:
|
|
|
|
result=[]
|
|
|
|
#print "before inserts in store_hands_players_stud, antes:", antes
|
|
|
|
for i in xrange(len(player_ids)):
|
|
|
|
card1 = Card.cardFromValueSuit(card_values[i][0], card_suits[i][0])
|
|
|
|
card2 = Card.cardFromValueSuit(card_values[i][1], card_suits[i][1])
|
|
|
|
card3 = Card.cardFromValueSuit(card_values[i][2], card_suits[i][2])
|
|
|
|
card4 = Card.cardFromValueSuit(card_values[i][3], card_suits[i][3])
|
|
|
|
card5 = Card.cardFromValueSuit(card_values[i][4], card_suits[i][4])
|
|
|
|
card6 = Card.cardFromValueSuit(card_values[i][5], card_suits[i][5])
|
|
|
|
card7 = Card.cardFromValueSuit(card_values[i][6], card_suits[i][6])
|
|
|
|
|
2009-08-06 21:31:46 +02:00
|
|
|
c = self.get_cursor()
|
|
|
|
c.execute ("""INSERT INTO HandsPlayers
|
2009-07-26 02:42:09 +02:00
|
|
|
(handId, playerId, startCash, ante, tourneyTypeId,
|
|
|
|
card1, card2,
|
|
|
|
card3, card4,
|
|
|
|
card5, card6,
|
|
|
|
card7, winnings, rake, seatNo)
|
2009-08-06 21:31:46 +02:00
|
|
|
VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)""".replace('%s', self.sql.query['placeholder']),
|
2009-07-26 02:42:09 +02:00
|
|
|
(hands_id, player_ids[i], start_cashes[i], antes[i], 1,
|
|
|
|
card1, card2,
|
|
|
|
card3, card4,
|
|
|
|
card5, card6,
|
|
|
|
card7, winnings[i], rakes[i], seatNos[i]))
|
|
|
|
#cursor.execute("SELECT id FROM HandsPlayers WHERE handId=%s AND playerId+0=%s", (hands_id, player_ids[i]))
|
|
|
|
#result.append(cursor.fetchall()[0][0])
|
2009-08-06 21:31:46 +02:00
|
|
|
result.append( self.get_last_insert_id(c) )
|
2009-07-26 02:42:09 +02:00
|
|
|
except:
|
|
|
|
raise fpdb_simple.FpdbError( "store_hands_players_stud error: " + str(sys.exc_value) )
|
|
|
|
|
|
|
|
return result
|
|
|
|
#end def store_hands_players_stud
|
|
|
|
|
|
|
|
def store_hands_players_holdem_omaha_tourney(self, backend, category, hands_id, player_ids
|
|
|
|
,start_cashes, positions, card_values, card_suits
|
|
|
|
,winnings, rakes, seatNos, tourneys_players_ids
|
|
|
|
,hudCache):
|
|
|
|
#stores hands_players for tourney holdem/omaha hands
|
|
|
|
|
|
|
|
try:
|
|
|
|
result=[]
|
|
|
|
inserts = []
|
|
|
|
for i in xrange(len(player_ids)):
|
|
|
|
card1 = Card.cardFromValueSuit(card_values[i][0], card_suits[i][0])
|
|
|
|
card2 = Card.cardFromValueSuit(card_values[i][1], card_suits[i][1])
|
|
|
|
|
|
|
|
if len(card_values[0])==2:
|
|
|
|
startCards = Card.twoStartCards(card_values[i][0], card_suits[i][0], card_values[i][1], card_suits[i][1])
|
|
|
|
card3 = None
|
|
|
|
card4 = None
|
|
|
|
elif len(card_values[0])==4:
|
|
|
|
startCards = Card.fourStartCards(card_values[i][0], card_suits[i][0], card_values[i][1], card_suits[i][1]
|
|
|
|
,card_values[i][2], card_suits[i][2], card_values[i][3], card_suits[i][3])
|
|
|
|
card3 = Card.cardFromValueSuit(card_values[i][2], card_suits[i][2])
|
|
|
|
card4 = Card.cardFromValueSuit(card_values[i][3], card_suits[i][3])
|
|
|
|
else:
|
|
|
|
raise FpdbError ("invalid card_values length:"+str(len(card_values[0])))
|
|
|
|
|
|
|
|
inserts.append( (hands_id, player_ids[i], start_cashes[i], positions[i], 1, # tourneytypeid
|
|
|
|
card1, card2, card3, card4, startCards,
|
|
|
|
winnings[i], rakes[i], tourneys_players_ids[i], seatNos[i], hudCache['totalProfit'][i],
|
|
|
|
hudCache['street0VPI'][i], hudCache['street0Aggr'][i],
|
|
|
|
hudCache['street0_3BChance'][i], hudCache['street0_3BDone'][i],
|
|
|
|
hudCache['street1Seen'][i], hudCache['street2Seen'][i], hudCache['street3Seen'][i],
|
|
|
|
hudCache['street4Seen'][i], hudCache['sawShowdown'][i],
|
|
|
|
hudCache['street1Aggr'][i], hudCache['street2Aggr'][i], hudCache['street3Aggr'][i], hudCache['street4Aggr'][i],
|
|
|
|
hudCache['otherRaisedStreet1'][i], hudCache['otherRaisedStreet2'][i],
|
|
|
|
hudCache['otherRaisedStreet3'][i], hudCache['otherRaisedStreet4'][i],
|
|
|
|
hudCache['foldToOtherRaisedStreet1'][i], hudCache['foldToOtherRaisedStreet2'][i],
|
|
|
|
hudCache['foldToOtherRaisedStreet3'][i], hudCache['foldToOtherRaisedStreet4'][i],
|
|
|
|
hudCache['wonWhenSeenStreet1'][i], hudCache['wonAtSD'][i],
|
|
|
|
hudCache['stealAttemptChance'][i], hudCache['stealAttempted'][i], hudCache['foldBbToStealChance'][i],
|
|
|
|
hudCache['foldedBbToSteal'][i], hudCache['foldSbToStealChance'][i], hudCache['foldedSbToSteal'][i],
|
|
|
|
hudCache['street1CBChance'][i], hudCache['street1CBDone'][i], hudCache['street2CBChance'][i], hudCache['street2CBDone'][i],
|
|
|
|
hudCache['street3CBChance'][i], hudCache['street3CBDone'][i], hudCache['street4CBChance'][i], hudCache['street4CBDone'][i],
|
|
|
|
hudCache['foldToStreet1CBChance'][i], hudCache['foldToStreet1CBDone'][i],
|
|
|
|
hudCache['foldToStreet2CBChance'][i], hudCache['foldToStreet2CBDone'][i],
|
|
|
|
hudCache['foldToStreet3CBChance'][i], hudCache['foldToStreet3CBDone'][i],
|
|
|
|
hudCache['foldToStreet4CBChance'][i], hudCache['foldToStreet4CBDone'][i],
|
|
|
|
hudCache['street1CheckCallRaiseChance'][i], hudCache['street1CheckCallRaiseDone'][i],
|
|
|
|
hudCache['street2CheckCallRaiseChance'][i], hudCache['street2CheckCallRaiseDone'][i],
|
|
|
|
hudCache['street3CheckCallRaiseChance'][i], hudCache['street3CheckCallRaiseDone'][i],
|
|
|
|
hudCache['street4CheckCallRaiseChance'][i], hudCache['street4CheckCallRaiseDone'][i],
|
|
|
|
hudCache['street0Calls'][i], hudCache['street1Calls'][i], hudCache['street2Calls'][i],
|
|
|
|
hudCache['street3Calls'][i], hudCache['street4Calls'][i],
|
|
|
|
hudCache['street0Bets'][i], hudCache['street1Bets'][i], hudCache['street2Bets'][i],
|
|
|
|
hudCache['street3Bets'][i], hudCache['street4Bets'][i]
|
|
|
|
) )
|
|
|
|
|
2009-08-06 21:31:46 +02:00
|
|
|
c = self.get_cursor()
|
|
|
|
c.executemany ("""
|
2009-07-26 02:42:09 +02:00
|
|
|
INSERT INTO HandsPlayers
|
|
|
|
(handId, playerId, startCash, position, tourneyTypeId,
|
|
|
|
card1, card2, card3, card4, startCards, winnings, rake, tourneysPlayersId, seatNo, totalProfit,
|
|
|
|
street0VPI, street0Aggr, street0_3BChance, street0_3BDone,
|
|
|
|
street1Seen, street2Seen, street3Seen, street4Seen, sawShowdown,
|
|
|
|
street1Aggr, street2Aggr, street3Aggr, street4Aggr,
|
|
|
|
otherRaisedStreet1, otherRaisedStreet2, otherRaisedStreet3, otherRaisedStreet4,
|
|
|
|
foldToOtherRaisedStreet1, foldToOtherRaisedStreet2, foldToOtherRaisedStreet3, foldToOtherRaisedStreet4,
|
|
|
|
wonWhenSeenStreet1, wonAtSD,
|
|
|
|
stealAttemptChance, stealAttempted, foldBbToStealChance, foldedBbToSteal, foldSbToStealChance, foldedSbToSteal,
|
|
|
|
street1CBChance, street1CBDone, street2CBChance, street2CBDone,
|
|
|
|
street3CBChance, street3CBDone, street4CBChance, street4CBDone,
|
|
|
|
foldToStreet1CBChance, foldToStreet1CBDone, foldToStreet2CBChance, foldToStreet2CBDone,
|
|
|
|
foldToStreet3CBChance, foldToStreet3CBDone, foldToStreet4CBChance, foldToStreet4CBDone,
|
|
|
|
street1CheckCallRaiseChance, street1CheckCallRaiseDone, street2CheckCallRaiseChance, street2CheckCallRaiseDone,
|
|
|
|
street3CheckCallRaiseChance, street3CheckCallRaiseDone, street4CheckCallRaiseChance, street4CheckCallRaiseDone,
|
|
|
|
street0Calls, street1Calls, street2Calls, street3Calls, street4Calls,
|
|
|
|
street0Bets, street1Bets, street2Bets, street3Bets, street4Bets
|
|
|
|
)
|
|
|
|
VALUES
|
|
|
|
(%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s,
|
|
|
|
%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s,
|
|
|
|
%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s,
|
2009-08-06 21:31:46 +02:00
|
|
|
%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)""".replace('%s', self.sql.query['placeholder'])
|
2009-07-26 02:42:09 +02:00
|
|
|
,inserts )
|
|
|
|
|
2009-08-06 21:31:46 +02:00
|
|
|
result.append( self.get_last_insert_id(c) )
|
2009-07-26 02:42:09 +02:00
|
|
|
#cursor.execute("SELECT id FROM HandsPlayers WHERE handId=%s AND playerId+0=%s", (hands_id, player_ids[i]))
|
|
|
|
#result.append(cursor.fetchall()[0][0])
|
|
|
|
except:
|
|
|
|
raise fpdb_simple.FpdbError( "store_hands_players_holdem_omaha_tourney error: " + str(sys.exc_value) )
|
|
|
|
|
|
|
|
return result
|
|
|
|
#end def store_hands_players_holdem_omaha_tourney
|
|
|
|
|
|
|
|
def store_hands_players_stud_tourney(self, backend, hands_id, player_ids, start_cashes,
|
|
|
|
antes, card_values, card_suits, winnings, rakes, seatNos, tourneys_players_ids):
|
|
|
|
#stores hands_players for tourney stud/razz hands
|
|
|
|
|
|
|
|
try:
|
|
|
|
result=[]
|
|
|
|
for i in xrange(len(player_ids)):
|
2009-08-06 21:31:46 +02:00
|
|
|
c = self.get_cursor()
|
|
|
|
c.execute ("""INSERT INTO HandsPlayers
|
2009-07-26 02:42:09 +02:00
|
|
|
(handId, playerId, startCash, ante,
|
|
|
|
card1Value, card1Suit, card2Value, card2Suit,
|
|
|
|
card3Value, card3Suit, card4Value, card4Suit,
|
|
|
|
card5Value, card5Suit, card6Value, card6Suit,
|
|
|
|
card7Value, card7Suit, winnings, rake, tourneysPlayersId, seatNo)
|
|
|
|
VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s,
|
2009-08-06 21:31:46 +02:00
|
|
|
%s, %s, %s, %s, %s, %s)""".replace('%s', self.sql.query['placeholder']),
|
2009-07-26 02:42:09 +02:00
|
|
|
(hands_id, player_ids[i], start_cashes[i], antes[i],
|
|
|
|
card_values[i][0], card_suits[i][0], card_values[i][1], card_suits[i][1],
|
|
|
|
card_values[i][2], card_suits[i][2], card_values[i][3], card_suits[i][3],
|
|
|
|
card_values[i][4], card_suits[i][4], card_values[i][5], card_suits[i][5],
|
|
|
|
card_values[i][6], card_suits[i][6], winnings[i], rakes[i], tourneys_players_ids[i], seatNos[i]))
|
|
|
|
#cursor.execute("SELECT id FROM HandsPlayers WHERE handId=%s AND playerId+0=%s", (hands_id, player_ids[i]))
|
|
|
|
#result.append(cursor.fetchall()[0][0])
|
2009-08-06 21:31:46 +02:00
|
|
|
result.append( self.get_last_insert_id(c) )
|
2009-07-26 02:42:09 +02:00
|
|
|
except:
|
|
|
|
raise fpdb_simple.FpdbError( "store_hands_players_stud_tourney error: " + str(sys.exc_value) )
|
|
|
|
|
|
|
|
return result
|
|
|
|
#end def store_hands_players_stud_tourney
|
|
|
|
|
|
|
|
def storeHudCache(self, backend, base, category, gametypeId, hand_start_time, playerIds, hudImportData):
|
|
|
|
"""Update cached statistics. If update fails because no record exists, do an insert.
|
|
|
|
Can't use array updates here (not easily anyway) because we need to insert any rows
|
|
|
|
that don't get updated."""
|
|
|
|
|
|
|
|
# if (category=="holdem" or category=="omahahi" or category=="omahahilo"):
|
|
|
|
try:
|
|
|
|
if self.use_date_in_hudcache:
|
|
|
|
#print "key =", "d%02d%02d%02d " % (hand_start_time.year-2000, hand_start_time.month, hand_start_time.day)
|
|
|
|
styleKey = "d%02d%02d%02d" % (hand_start_time.year-2000, hand_start_time.month, hand_start_time.day)
|
|
|
|
else:
|
|
|
|
# hard-code styleKey as 'A000000' (all-time cache, no key) for now
|
|
|
|
styleKey = 'A000000'
|
|
|
|
|
|
|
|
#print "storeHudCache, len(playerIds)=", len(playerIds), " len(vpip)=" \
|
|
|
|
#, len(hudImportData['street0VPI']), " len(totprof)=", len(hudImportData['totalProfit'])
|
|
|
|
for player in xrange(len(playerIds)):
|
|
|
|
|
|
|
|
# Set up a clean row
|
|
|
|
row=[]
|
|
|
|
row.append(0)#blank for id
|
|
|
|
row.append(gametypeId)
|
|
|
|
row.append(playerIds[player])
|
|
|
|
row.append(len(playerIds))#seats
|
|
|
|
for i in xrange(len(hudImportData)+2):
|
|
|
|
row.append(0)
|
|
|
|
|
|
|
|
if base=="hold":
|
|
|
|
row[4]=hudImportData['position'][player]
|
|
|
|
else:
|
|
|
|
row[4]=0
|
|
|
|
row[5]=1 #tourneysGametypeId
|
|
|
|
row[6]+=1 #HDs
|
|
|
|
if hudImportData['street0VPI'][player]: row[7]+=1
|
|
|
|
if hudImportData['street0Aggr'][player]: row[8]+=1
|
|
|
|
if hudImportData['street0_3BChance'][player]: row[9]+=1
|
|
|
|
if hudImportData['street0_3BDone'][player]: row[10]+=1
|
|
|
|
if hudImportData['street1Seen'][player]: row[11]+=1
|
|
|
|
if hudImportData['street2Seen'][player]: row[12]+=1
|
|
|
|
if hudImportData['street3Seen'][player]: row[13]+=1
|
|
|
|
if hudImportData['street4Seen'][player]: row[14]+=1
|
|
|
|
if hudImportData['sawShowdown'][player]: row[15]+=1
|
|
|
|
if hudImportData['street1Aggr'][player]: row[16]+=1
|
|
|
|
if hudImportData['street2Aggr'][player]: row[17]+=1
|
|
|
|
if hudImportData['street3Aggr'][player]: row[18]+=1
|
|
|
|
if hudImportData['street4Aggr'][player]: row[19]+=1
|
|
|
|
if hudImportData['otherRaisedStreet1'][player]: row[20]+=1
|
|
|
|
if hudImportData['otherRaisedStreet2'][player]: row[21]+=1
|
|
|
|
if hudImportData['otherRaisedStreet3'][player]: row[22]+=1
|
|
|
|
if hudImportData['otherRaisedStreet4'][player]: row[23]+=1
|
|
|
|
if hudImportData['foldToOtherRaisedStreet1'][player]: row[24]+=1
|
|
|
|
if hudImportData['foldToOtherRaisedStreet2'][player]: row[25]+=1
|
|
|
|
if hudImportData['foldToOtherRaisedStreet3'][player]: row[26]+=1
|
|
|
|
if hudImportData['foldToOtherRaisedStreet4'][player]: row[27]+=1
|
|
|
|
if hudImportData['wonWhenSeenStreet1'][player]!=0.0: row[28]+=hudImportData['wonWhenSeenStreet1'][player]
|
|
|
|
if hudImportData['wonAtSD'][player]!=0.0: row[29]+=hudImportData['wonAtSD'][player]
|
|
|
|
if hudImportData['stealAttemptChance'][player]: row[30]+=1
|
|
|
|
if hudImportData['stealAttempted'][player]: row[31]+=1
|
|
|
|
if hudImportData['foldBbToStealChance'][player]: row[32]+=1
|
|
|
|
if hudImportData['foldedBbToSteal'][player]: row[33]+=1
|
|
|
|
if hudImportData['foldSbToStealChance'][player]: row[34]+=1
|
|
|
|
if hudImportData['foldedSbToSteal'][player]: row[35]+=1
|
|
|
|
|
|
|
|
if hudImportData['street1CBChance'][player]: row[36]+=1
|
|
|
|
if hudImportData['street1CBDone'][player]: row[37]+=1
|
|
|
|
if hudImportData['street2CBChance'][player]: row[38]+=1
|
|
|
|
if hudImportData['street2CBDone'][player]: row[39]+=1
|
|
|
|
if hudImportData['street3CBChance'][player]: row[40]+=1
|
|
|
|
if hudImportData['street3CBDone'][player]: row[41]+=1
|
|
|
|
if hudImportData['street4CBChance'][player]: row[42]+=1
|
|
|
|
if hudImportData['street4CBDone'][player]: row[43]+=1
|
|
|
|
|
|
|
|
if hudImportData['foldToStreet1CBChance'][player]: row[44]+=1
|
|
|
|
if hudImportData['foldToStreet1CBDone'][player]: row[45]+=1
|
|
|
|
if hudImportData['foldToStreet2CBChance'][player]: row[46]+=1
|
|
|
|
if hudImportData['foldToStreet2CBDone'][player]: row[47]+=1
|
|
|
|
if hudImportData['foldToStreet3CBChance'][player]: row[48]+=1
|
|
|
|
if hudImportData['foldToStreet3CBDone'][player]: row[49]+=1
|
|
|
|
if hudImportData['foldToStreet4CBChance'][player]: row[50]+=1
|
|
|
|
if hudImportData['foldToStreet4CBDone'][player]: row[51]+=1
|
|
|
|
|
|
|
|
#print "player=", player
|
|
|
|
#print "len(totalProfit)=", len(hudImportData['totalProfit'])
|
|
|
|
if hudImportData['totalProfit'][player]:
|
|
|
|
row[52]+=hudImportData['totalProfit'][player]
|
|
|
|
|
|
|
|
if hudImportData['street1CheckCallRaiseChance'][player]: row[53]+=1
|
|
|
|
if hudImportData['street1CheckCallRaiseDone'][player]: row[54]+=1
|
|
|
|
if hudImportData['street2CheckCallRaiseChance'][player]: row[55]+=1
|
|
|
|
if hudImportData['street2CheckCallRaiseDone'][player]: row[56]+=1
|
|
|
|
if hudImportData['street3CheckCallRaiseChance'][player]: row[57]+=1
|
|
|
|
if hudImportData['street3CheckCallRaiseDone'][player]: row[58]+=1
|
|
|
|
if hudImportData['street4CheckCallRaiseChance'][player]: row[59]+=1
|
|
|
|
if hudImportData['street4CheckCallRaiseDone'][player]: row[60]+=1
|
|
|
|
|
|
|
|
# Try to do the update first:
|
|
|
|
cursor = self.get_cursor()
|
|
|
|
num = cursor.execute("""UPDATE HudCache
|
|
|
|
SET HDs=HDs+%s, street0VPI=street0VPI+%s, street0Aggr=street0Aggr+%s,
|
|
|
|
street0_3BChance=street0_3BChance+%s, street0_3BDone=street0_3BDone+%s,
|
|
|
|
street1Seen=street1Seen+%s, street2Seen=street2Seen+%s, street3Seen=street3Seen+%s,
|
|
|
|
street4Seen=street4Seen+%s, sawShowdown=sawShowdown+%s,
|
|
|
|
street1Aggr=street1Aggr+%s, street2Aggr=street2Aggr+%s, street3Aggr=street3Aggr+%s,
|
|
|
|
street4Aggr=street4Aggr+%s, otherRaisedStreet1=otherRaisedStreet1+%s,
|
|
|
|
otherRaisedStreet2=otherRaisedStreet2+%s, otherRaisedStreet3=otherRaisedStreet3+%s,
|
|
|
|
otherRaisedStreet4=otherRaisedStreet4+%s,
|
|
|
|
foldToOtherRaisedStreet1=foldToOtherRaisedStreet1+%s, foldToOtherRaisedStreet2=foldToOtherRaisedStreet2+%s,
|
|
|
|
foldToOtherRaisedStreet3=foldToOtherRaisedStreet3+%s, foldToOtherRaisedStreet4=foldToOtherRaisedStreet4+%s,
|
|
|
|
wonWhenSeenStreet1=wonWhenSeenStreet1+%s, wonAtSD=wonAtSD+%s, stealAttemptChance=stealAttemptChance+%s,
|
|
|
|
stealAttempted=stealAttempted+%s, foldBbToStealChance=foldBbToStealChance+%s,
|
|
|
|
foldedBbToSteal=foldedBbToSteal+%s,
|
|
|
|
foldSbToStealChance=foldSbToStealChance+%s, foldedSbToSteal=foldedSbToSteal+%s,
|
|
|
|
street1CBChance=street1CBChance+%s, street1CBDone=street1CBDone+%s, street2CBChance=street2CBChance+%s,
|
|
|
|
street2CBDone=street2CBDone+%s, street3CBChance=street3CBChance+%s,
|
|
|
|
street3CBDone=street3CBDone+%s, street4CBChance=street4CBChance+%s, street4CBDone=street4CBDone+%s,
|
|
|
|
foldToStreet1CBChance=foldToStreet1CBChance+%s, foldToStreet1CBDone=foldToStreet1CBDone+%s,
|
|
|
|
foldToStreet2CBChance=foldToStreet2CBChance+%s, foldToStreet2CBDone=foldToStreet2CBDone+%s,
|
|
|
|
foldToStreet3CBChance=foldToStreet3CBChance+%s,
|
|
|
|
foldToStreet3CBDone=foldToStreet3CBDone+%s, foldToStreet4CBChance=foldToStreet4CBChance+%s,
|
|
|
|
foldToStreet4CBDone=foldToStreet4CBDone+%s, totalProfit=totalProfit+%s,
|
|
|
|
street1CheckCallRaiseChance=street1CheckCallRaiseChance+%s,
|
|
|
|
street1CheckCallRaiseDone=street1CheckCallRaiseDone+%s, street2CheckCallRaiseChance=street2CheckCallRaiseChance+%s,
|
|
|
|
street2CheckCallRaiseDone=street2CheckCallRaiseDone+%s, street3CheckCallRaiseChance=street3CheckCallRaiseChance+%s,
|
|
|
|
street3CheckCallRaiseDone=street3CheckCallRaiseDone+%s, street4CheckCallRaiseChance=street4CheckCallRaiseChance+%s,
|
|
|
|
street4CheckCallRaiseDone=street4CheckCallRaiseDone+%s
|
|
|
|
WHERE gametypeId+0=%s
|
|
|
|
AND playerId=%s
|
|
|
|
AND activeSeats=%s
|
|
|
|
AND position=%s
|
|
|
|
AND tourneyTypeId+0=%s
|
|
|
|
AND styleKey=%s
|
2009-08-06 21:31:46 +02:00
|
|
|
""".replace('%s', self.sql.query['placeholder'])
|
|
|
|
,(row[6], row[7], row[8], row[9], row[10],
|
|
|
|
row[11], row[12], row[13], row[14], row[15],
|
|
|
|
row[16], row[17], row[18], row[19], row[20],
|
|
|
|
row[21], row[22], row[23], row[24], row[25],
|
|
|
|
row[26], row[27], row[28], row[29], row[30],
|
|
|
|
row[31], row[32], row[33], row[34], row[35],
|
|
|
|
row[36], row[37], row[38], row[39], row[40],
|
|
|
|
row[41], row[42], row[43], row[44], row[45],
|
|
|
|
row[46], row[47], row[48], row[49], row[50],
|
|
|
|
row[51], row[52], row[53], row[54], row[55],
|
|
|
|
row[56], row[57], row[58], row[59], row[60],
|
|
|
|
row[1], row[2], row[3], str(row[4]), row[5], styleKey))
|
2009-07-26 02:42:09 +02:00
|
|
|
# Test statusmessage to see if update worked, do insert if not
|
2009-08-07 01:04:44 +02:00
|
|
|
#print "storehud2, upd num =", num.rowcount
|
|
|
|
# num is a cursor in sqlite
|
2009-07-26 02:42:09 +02:00
|
|
|
if ( (backend == self.PGSQL and cursor.statusmessage != "UPDATE 1")
|
2009-08-07 01:04:44 +02:00
|
|
|
or (backend == self.MYSQL_INNODB and num == 0)
|
|
|
|
or (backend == self.SQLITE and num.rowcount == 0)
|
|
|
|
):
|
2009-07-26 02:42:09 +02:00
|
|
|
#print "playerid before insert:",row[2]," num = ", num
|
2009-08-07 01:04:44 +02:00
|
|
|
num = cursor.execute("""INSERT INTO HudCache
|
2009-07-26 02:42:09 +02:00
|
|
|
(gametypeId, playerId, activeSeats, position, tourneyTypeId, styleKey,
|
|
|
|
HDs, street0VPI, street0Aggr, street0_3BChance, street0_3BDone,
|
|
|
|
street1Seen, street2Seen, street3Seen, street4Seen, sawShowdown,
|
|
|
|
street1Aggr, street2Aggr, street3Aggr, street4Aggr, otherRaisedStreet1,
|
|
|
|
otherRaisedStreet2, otherRaisedStreet3, otherRaisedStreet4, foldToOtherRaisedStreet1, foldToOtherRaisedStreet2,
|
|
|
|
foldToOtherRaisedStreet3, foldToOtherRaisedStreet4, wonWhenSeenStreet1, wonAtSD, stealAttemptChance,
|
|
|
|
stealAttempted, foldBbToStealChance, foldedBbToSteal, foldSbToStealChance, foldedSbToSteal,
|
|
|
|
street1CBChance, street1CBDone, street2CBChance, street2CBDone, street3CBChance,
|
|
|
|
street3CBDone, street4CBChance, street4CBDone, foldToStreet1CBChance, foldToStreet1CBDone,
|
|
|
|
foldToStreet2CBChance, foldToStreet2CBDone, foldToStreet3CBChance, foldToStreet3CBDone, foldToStreet4CBChance,
|
|
|
|
foldToStreet4CBDone, totalProfit, street1CheckCallRaiseChance, street1CheckCallRaiseDone, street2CheckCallRaiseChance,
|
|
|
|
street2CheckCallRaiseDone, street3CheckCallRaiseChance, street3CheckCallRaiseDone, street4CheckCallRaiseChance, street4CheckCallRaiseDone)
|
|
|
|
VALUES (%s, %s, %s, %s, %s, %s,
|
|
|
|
%s, %s, %s, %s, %s,
|
|
|
|
%s, %s, %s, %s, %s,
|
|
|
|
%s, %s, %s, %s, %s,
|
|
|
|
%s, %s, %s, %s, %s,
|
|
|
|
%s, %s, %s, %s, %s,
|
|
|
|
%s, %s, %s, %s, %s,
|
|
|
|
%s, %s, %s, %s, %s,
|
|
|
|
%s, %s, %s, %s, %s,
|
|
|
|
%s, %s, %s, %s, %s,
|
|
|
|
%s, %s, %s, %s, %s,
|
2009-08-06 21:31:46 +02:00
|
|
|
%s, %s, %s, %s, %s)""".replace('%s', self.sql.query['placeholder'])
|
2009-07-26 02:42:09 +02:00
|
|
|
, (row[1], row[2], row[3], row[4], row[5], styleKey, row[6], row[7], row[8], row[9], row[10]
|
|
|
|
,row[11], row[12], row[13], row[14], row[15], row[16], row[17], row[18], row[19], row[20]
|
|
|
|
,row[21], row[22], row[23], row[24], row[25], row[26], row[27], row[28], row[29], row[30]
|
|
|
|
,row[31], row[32], row[33], row[34], row[35], row[36], row[37], row[38], row[39], row[40]
|
|
|
|
,row[41], row[42], row[43], row[44], row[45], row[46], row[47], row[48], row[49], row[50]
|
|
|
|
,row[51], row[52], row[53], row[54], row[55], row[56], row[57], row[58], row[59], row[60]) )
|
2009-08-07 01:04:44 +02:00
|
|
|
#print "hopefully inserted hud data line: ", cursor.rowcount
|
2009-07-26 02:42:09 +02:00
|
|
|
# message seems to be "INSERT 0 1"
|
|
|
|
else:
|
|
|
|
#print "updated(2) hud data line"
|
|
|
|
pass
|
|
|
|
# else:
|
|
|
|
# print "todo: implement storeHudCache for stud base"
|
|
|
|
|
|
|
|
except:
|
|
|
|
raise fpdb_simple.FpdbError( "storeHudCache error: " + str(sys.exc_value) )
|
|
|
|
|
|
|
|
#end def storeHudCache
|
|
|
|
|
|
|
|
def store_tourneys(self, tourneyTypeId, siteTourneyNo, entries, prizepool, startTime):
|
|
|
|
try:
|
|
|
|
cursor = self.get_cursor()
|
2009-08-06 21:31:46 +02:00
|
|
|
cursor.execute("SELECT id FROM Tourneys WHERE siteTourneyNo=%s AND tourneyTypeId+0=%s".replace('%s', self.sql.query['placeholder'])
|
|
|
|
, (siteTourneyNo, tourneyTypeId))
|
2009-07-26 02:42:09 +02:00
|
|
|
tmp=cursor.fetchone()
|
|
|
|
#print "tried SELECTing tourneys.id, result:",tmp
|
|
|
|
|
|
|
|
try:
|
|
|
|
len(tmp)
|
|
|
|
except TypeError:#means we have to create new one
|
|
|
|
cursor.execute("""INSERT INTO Tourneys
|
|
|
|
(tourneyTypeId, siteTourneyNo, entries, prizepool, startTime)
|
2009-08-06 21:31:46 +02:00
|
|
|
VALUES (%s, %s, %s, %s, %s)""".replace('%s', self.sql.query['placeholder'])
|
|
|
|
,(tourneyTypeId, siteTourneyNo, entries, prizepool, startTime))
|
2009-07-26 02:42:09 +02:00
|
|
|
cursor.execute("SELECT id FROM Tourneys WHERE siteTourneyNo=%s AND tourneyTypeId+0=%s", (siteTourneyNo, tourneyTypeId))
|
|
|
|
tmp=cursor.fetchone()
|
|
|
|
#print "created new tourneys.id:",tmp
|
|
|
|
except:
|
|
|
|
raise fpdb_simple.FpdbError( "store_tourneys error: " + str(sys.exc_value) )
|
|
|
|
|
|
|
|
return tmp[0]
|
|
|
|
#end def store_tourneys
|
|
|
|
|
|
|
|
def store_tourneys_players(self, tourney_id, player_ids, payin_amounts, ranks, winnings):
|
|
|
|
try:
|
|
|
|
result=[]
|
|
|
|
cursor = self.get_cursor()
|
|
|
|
#print "in store_tourneys_players. tourney_id:",tourney_id
|
|
|
|
#print "player_ids:",player_ids
|
|
|
|
#print "payin_amounts:",payin_amounts
|
|
|
|
#print "ranks:",ranks
|
|
|
|
#print "winnings:",winnings
|
|
|
|
for i in xrange(len(player_ids)):
|
2009-08-06 21:31:46 +02:00
|
|
|
cursor.execute("SELECT id FROM TourneysPlayers WHERE tourneyId=%s AND playerId+0=%s".replace('%s', self.sql.query['placeholder'])
|
|
|
|
,(tourney_id, player_ids[i]))
|
2009-07-26 02:42:09 +02:00
|
|
|
tmp=cursor.fetchone()
|
|
|
|
#print "tried SELECTing tourneys_players.id:",tmp
|
|
|
|
|
|
|
|
try:
|
|
|
|
len(tmp)
|
|
|
|
except TypeError:
|
|
|
|
cursor.execute("""INSERT INTO TourneysPlayers
|
2009-08-06 21:31:46 +02:00
|
|
|
(tourneyId, playerId, payinAmount, rank, winnings) VALUES (%s, %s, %s, %s, %s)""".replace('%s', self.sql.query['placeholder']),
|
2009-07-26 02:42:09 +02:00
|
|
|
(tourney_id, player_ids[i], payin_amounts[i], ranks[i], winnings[i]))
|
|
|
|
|
2009-08-06 21:31:46 +02:00
|
|
|
cursor.execute("SELECT id FROM TourneysPlayers WHERE tourneyId=%s AND playerId+0=%s".replace('%s', self.sql.query['placeholder']),
|
2009-07-26 02:42:09 +02:00
|
|
|
(tourney_id, player_ids[i]))
|
|
|
|
tmp=cursor.fetchone()
|
|
|
|
#print "created new tourneys_players.id:",tmp
|
|
|
|
result.append(tmp[0])
|
|
|
|
except:
|
|
|
|
raise fpdb_simple.FpdbError( "store_tourneys_players error: " + str(sys.exc_value) )
|
|
|
|
|
|
|
|
return result
|
|
|
|
#end def store_tourneys_players
|
|
|
|
|
|
|
|
|
2009-07-31 22:24:21 +02:00
|
|
|
# read HandToWrite objects from q and insert into database
|
|
|
|
def insert_queue_hands(self, q, maxwait=10, commitEachHand=True):
|
|
|
|
n,fails,maxTries,firstWait = 0,0,4,0.1
|
|
|
|
sendFinal = False
|
|
|
|
t0 = time()
|
|
|
|
while True:
|
|
|
|
try:
|
|
|
|
h = q.get(True) # (True,maxWait) has probs if 1st part of import is all dups
|
|
|
|
except Queue.Empty:
|
|
|
|
# Queue.Empty exception thrown if q was empty for
|
|
|
|
# if q.empty() also possible - no point if testing for Queue.Empty exception
|
|
|
|
# maybe increment a counter and only break after a few times?
|
|
|
|
# could also test threading.active_count() or look through threading.enumerate()
|
|
|
|
# so break immediately if no threads, but count up to X exceptions if a writer
|
|
|
|
# thread is still alive???
|
|
|
|
print "queue empty too long - writer stopping ..."
|
|
|
|
break
|
|
|
|
except:
|
|
|
|
print "writer stopping, error reading queue: " + str(sys.exc_info())
|
|
|
|
break
|
|
|
|
#print "got hand", str(h.get_finished())
|
|
|
|
|
|
|
|
tries,wait,again = 0,firstWait,True
|
|
|
|
while again:
|
|
|
|
try:
|
|
|
|
again = False # set this immediately to avoid infinite loops!
|
|
|
|
if h.get_finished():
|
|
|
|
# all items on queue processed
|
|
|
|
sendFinal = True
|
|
|
|
else:
|
|
|
|
self.store_the_hand(h)
|
|
|
|
# optional commit, could be every hand / every N hands / every time a
|
|
|
|
# commit message received?? mark flag to indicate if commits outstanding
|
|
|
|
if commitEachHand:
|
|
|
|
self.commit()
|
|
|
|
n = n + 1
|
|
|
|
except:
|
|
|
|
#print "iqh store error", sys.exc_value # debug
|
|
|
|
self.rollback()
|
|
|
|
if re.search('deadlock', str(sys.exc_info()[1]), re.I):
|
|
|
|
# deadlocks only a problem if hudcache is being updated
|
|
|
|
tries = tries + 1
|
|
|
|
if tries < maxTries and wait < 5: # wait < 5 just to make sure
|
|
|
|
print "deadlock detected - trying again ..."
|
2009-08-01 01:06:07 +02:00
|
|
|
sleep(wait)
|
2009-07-31 22:24:21 +02:00
|
|
|
wait = wait + wait
|
|
|
|
again = True
|
|
|
|
else:
|
|
|
|
print "too many deadlocks - failed to store hand " + h.get_siteHandNo()
|
|
|
|
if not again:
|
|
|
|
fails = fails + 1
|
|
|
|
err = traceback.extract_tb(sys.exc_info()[2])[-1]
|
|
|
|
print "***Error storing hand: "+err[2]+"("+str(err[1])+"): "+str(sys.exc_info()[1])
|
|
|
|
# finished trying to store hand
|
|
|
|
|
|
|
|
# always reduce q count, whether or not this hand was saved ok
|
|
|
|
q.task_done()
|
|
|
|
# while True loop
|
|
|
|
|
|
|
|
self.commit()
|
|
|
|
if sendFinal:
|
|
|
|
q.task_done()
|
|
|
|
print "db writer finished: stored %d hands (%d fails) in %.1f seconds" % (n, fails, time()-t0)
|
|
|
|
# end def insert_queue_hands():
|
|
|
|
|
|
|
|
|
|
|
|
def send_finish_msg(self, q):
|
|
|
|
try:
|
|
|
|
h = HandToWrite(True)
|
|
|
|
q.put(h)
|
|
|
|
except:
|
|
|
|
err = traceback.extract_tb(sys.exc_info()[2])[-1]
|
|
|
|
print "***Error sending finish: "+err[2]+"("+str(err[1])+"): "+str(sys.exc_info()[1])
|
|
|
|
# end def send_finish_msg():
|
|
|
|
|
|
|
|
|
2009-07-29 07:17:51 +02:00
|
|
|
# Class used to hold all the data needed to write a hand to the db
|
|
|
|
# mainParser() in fpdb_parse_logic.py creates one of these and then passes it to
|
2009-07-31 22:24:21 +02:00
|
|
|
# self.insert_queue_hands()
|
2009-07-29 07:17:51 +02:00
|
|
|
|
|
|
|
class HandToWrite:
|
|
|
|
|
|
|
|
def __init__(self, finished = False): # db_name and game not used any more
|
|
|
|
try:
|
|
|
|
self.finished = finished
|
|
|
|
self.config = None
|
|
|
|
self.settings = None
|
|
|
|
self.base = None
|
|
|
|
self.category = None
|
|
|
|
self.siteTourneyNo = None
|
|
|
|
self.buyin = None
|
|
|
|
self.fee = None
|
|
|
|
self.knockout = None
|
|
|
|
self.entries = None
|
|
|
|
self.prizepool = None
|
|
|
|
self.tourneyStartTime = None
|
|
|
|
self.isTourney = None
|
|
|
|
self.tourneyTypeId = None
|
|
|
|
self.siteID = None
|
|
|
|
self.siteHandNo = None
|
|
|
|
self.gametypeID = None
|
|
|
|
self.handStartTime = None
|
|
|
|
self.names = None
|
|
|
|
self.playerIDs = None
|
|
|
|
self.startCashes = None
|
|
|
|
self.positions = None
|
|
|
|
self.antes = None
|
|
|
|
self.cardValues = None
|
|
|
|
self.cardSuits = None
|
|
|
|
self.boardValues = None
|
|
|
|
self.boardSuits = None
|
|
|
|
self.winnings = None
|
|
|
|
self.rakes = None
|
|
|
|
self.actionTypes = None
|
|
|
|
self.allIns = None
|
|
|
|
self.actionAmounts = None
|
|
|
|
self.actionNos = None
|
|
|
|
self.hudImportData = None
|
|
|
|
self.maxSeats = None
|
|
|
|
self.tableName = None
|
|
|
|
self.seatNos = None
|
2009-07-31 03:50:08 +02:00
|
|
|
self.payin_amounts = None # tourney import was complaining mightily about this missing
|
2009-07-29 07:17:51 +02:00
|
|
|
except:
|
2009-07-29 08:57:18 +02:00
|
|
|
print "htw.init error: " + str(sys.exc_info())
|
2009-07-29 07:17:51 +02:00
|
|
|
raise
|
|
|
|
# end def __init__
|
|
|
|
|
|
|
|
def set_all( self, config, settings, base, category, siteTourneyNo, buyin
|
|
|
|
, fee, knockout, entries, prizepool, tourneyStartTime
|
|
|
|
, isTourney, tourneyTypeId, siteID, siteHandNo
|
|
|
|
, gametypeID, handStartTime, names, playerIDs, startCashes
|
|
|
|
, positions, antes, cardValues, cardSuits, boardValues, boardSuits
|
|
|
|
, winnings, rakes, actionTypes, allIns, actionAmounts
|
|
|
|
, actionNos, hudImportData, maxSeats, tableName, seatNos):
|
|
|
|
|
|
|
|
try:
|
|
|
|
self.config = config
|
|
|
|
self.settings = settings
|
|
|
|
self.base = base
|
|
|
|
self.category = category
|
|
|
|
self.siteTourneyNo = siteTourneyNo
|
|
|
|
self.buyin = buyin
|
|
|
|
self.fee = fee
|
|
|
|
self.knockout = knockout
|
|
|
|
self.entries = entries
|
|
|
|
self.prizepool = prizepool
|
|
|
|
self.tourneyStartTime = tourneyStartTime
|
|
|
|
self.isTourney = isTourney
|
|
|
|
self.tourneyTypeId = tourneyTypeId
|
|
|
|
self.siteID = siteID
|
|
|
|
self.siteHandNo = siteHandNo
|
|
|
|
self.gametypeID = gametypeID
|
|
|
|
self.handStartTime = handStartTime
|
|
|
|
self.names = names
|
|
|
|
self.playerIDs = playerIDs
|
|
|
|
self.startCashes = startCashes
|
|
|
|
self.positions = positions
|
|
|
|
self.antes = antes
|
|
|
|
self.cardValues = cardValues
|
|
|
|
self.cardSuits = cardSuits
|
|
|
|
self.boardValues = boardValues
|
|
|
|
self.boardSuits = boardSuits
|
|
|
|
self.winnings = winnings
|
|
|
|
self.rakes = rakes
|
|
|
|
self.actionTypes = actionTypes
|
|
|
|
self.allIns = allIns
|
|
|
|
self.actionAmounts = actionAmounts
|
|
|
|
self.actionNos = actionNos
|
|
|
|
self.hudImportData = hudImportData
|
|
|
|
self.maxSeats = maxSeats
|
|
|
|
self.tableName = tableName
|
|
|
|
self.seatNos = seatNos
|
|
|
|
except:
|
2009-07-29 08:57:18 +02:00
|
|
|
print "htw.set_all error: " + str(sys.exc_info())
|
2009-07-29 07:17:51 +02:00
|
|
|
raise
|
|
|
|
# end def set_hand
|
|
|
|
|
|
|
|
def get_finished(self):
|
|
|
|
return( self.finished )
|
|
|
|
# end def get_finished
|
|
|
|
|
|
|
|
def get_siteHandNo(self):
|
|
|
|
return( self.siteHandNo )
|
|
|
|
# end def get_siteHandNo
|
|
|
|
|
|
|
|
|
2008-08-19 00:53:25 +02:00
|
|
|
if __name__=="__main__":
|
|
|
|
c = Configuration.Config()
|
|
|
|
|
2008-10-04 22:43:50 +02:00
|
|
|
db_connection = Database(c, 'fpdb', 'holdem') # mysql fpdb holdem
|
|
|
|
# db_connection = Database(c, 'fpdb-p', 'test') # mysql fpdb holdem
|
2008-08-19 00:53:25 +02:00
|
|
|
# db_connection = Database(c, 'PTrackSv2', 'razz') # mysql razz
|
|
|
|
# db_connection = Database(c, 'ptracks', 'razz') # postgres
|
|
|
|
print "database connection object = ", db_connection.connection
|
|
|
|
print "database type = ", db_connection.type
|
|
|
|
|
2009-08-04 23:06:03 +02:00
|
|
|
db_connection.recreate_tables()
|
|
|
|
|
2008-08-19 00:53:25 +02:00
|
|
|
h = db_connection.get_last_hand()
|
|
|
|
print "last hand = ", h
|
|
|
|
|
|
|
|
hero = db_connection.get_player_id(c, 'PokerStars', 'nutOmatic')
|
2009-05-21 22:27:44 +02:00
|
|
|
if hero:
|
|
|
|
print "nutOmatic is id_player = %d" % hero
|
2008-08-19 00:53:25 +02:00
|
|
|
|
2008-09-15 22:31:55 +02:00
|
|
|
stat_dict = db_connection.get_stats_from_hand(h)
|
|
|
|
for p in stat_dict.keys():
|
|
|
|
print p, " ", stat_dict[p]
|
|
|
|
|
2009-06-18 00:03:43 +02:00
|
|
|
#print "nutOmatics stats:"
|
|
|
|
#stat_dict = db_connection.get_stats_from_hand(h, hero)
|
|
|
|
#for p in stat_dict.keys():
|
|
|
|
# print p, " ", stat_dict[p]
|
2008-09-15 22:31:55 +02:00
|
|
|
|
2009-06-01 03:25:36 +02:00
|
|
|
print "cards =", db_connection.get_cards(u'1')
|
2008-09-15 22:31:55 +02:00
|
|
|
db_connection.close_connection
|
|
|
|
|
|
|
|
print "press enter to continue"
|
|
|
|
sys.stdin.readline()
|
2009-08-02 06:19:33 +02:00
|
|
|
|
|
|
|
|
|
|
|
#Code borrowed from http://push.cx/2008/caching-dictionaries-in-python-vs-ruby
|
|
|
|
class LambdaDict(dict):
|
|
|
|
def __init__(self, l):
|
|
|
|
super(LambdaDict, self).__init__()
|
|
|
|
self.l = l
|
|
|
|
|
|
|
|
def __getitem__(self, key):
|
|
|
|
if key in self:
|
|
|
|
return self.get(key)
|
|
|
|
else:
|
|
|
|
self.__setitem__(key, self.l(key))
|
|
|
|
return self.get(key)
|