fpdb/pyfpdb/fpdb.py

#!/usr/bin/python

#Copyright 2008 Steffen Jobbagy-Felso
#This program is free software: you can redistribute it and/or modify
#it under the terms of the GNU Affero General Public License as published by
#the Free Software Foundation, version 3 of the License.
#
#This program is distributed in the hope that it will be useful,
#but WITHOUT ANY WARRANTY; without even the implied warranty of
#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
#GNU General Public License for more details.
#
#You should have received a copy of the GNU Affero General Public License
#along with this program. If not, see <http://www.gnu.org/licenses/>.
#In the "official" distribution you can find the license in
#agpl-3.0.txt in the docs folder of the package.

import os
import sys

print "Note: error output is being diverted to fpdb-error-log.txt and HUD-error.txt. Any major error will be reported there _only_."
errorFile = open('fpdb-error-log.txt', 'w', 0)
sys.stderr = errorFile

import pygtk
pygtk.require('2.0')
import gtk

import fpdb_db
import fpdb_simple
import GuiBulkImport
import GuiTableViewer
import GuiAutoImport
import GuiGraphViewer
import FpdbSQLQueries

class fpdb:
	def tab_clicked(self, widget, tab_name):
		"""called when a tab button is clicked to activate that tab"""
		#print "start of tab_clicked"
		self.display_tab(tab_name)
	#end def tab_clicked
		
	def add_and_display_tab(self, new_tab, new_tab_name):
		"""just calls the component methods"""
		self.add_tab(new_tab, new_tab_name)
		self.display_tab(new_tab_name)
	#end def add_and_display_tab
	
	def add_tab(self, new_tab, new_tab_name):
		"""adds a tab, namely creates the button and displays it and appends all the relevant arrays"""
		#print "start of add_tab"
		for i in self.tab_names: #todo: check this is valid
			if i==new_tab_name:
				raise fpdb_simple.FpdbError("duplicate tab_name not permitted")
		
		self.tabs.append(new_tab)
		self.tab_names.append(new_tab_name)
		
		new_tab_sel_button=gtk.ToggleButton(new_tab_name)
		new_tab_sel_button.connect("clicked", self.tab_clicked, new_tab_name)
		self.tab_box.add(new_tab_sel_button)
		new_tab_sel_button.show()
		self.tab_buttons.append(new_tab_sel_button)
	#end def add_tab
	
	def display_tab(self, new_tab_name):
		"""displays the indicated tab"""
		#print "start of display_tab, len(self.tab_names):",len(self.tab_names)
		tab_no=-1
		#if len(self.tab_names)>1:
		for i in range(len(self.tab_names)):
			#print "display_tab, new_tab_name:",new_tab_name,"  self.tab_names[i]:", self.tab_names[i]
			if (new_tab_name==self.tab_names[i]):
				tab_no=i
				#self.tab_buttons[i].set_active(False)
		#else:
		#	tab_no=0
		
		#current_tab_no=-1
		for i in range(len(self.tab_names)):
			if self.current_tab==self.tabs[i]:
				#self.tab_buttons[i].set_active(False)
				pass
		
		if tab_no==-1:
			raise fpdb_simple.FpdbError("invalid tab_no")
		else:
			self.main_vbox.remove(self.current_tab)
			#self.current_tab.destroy()
			self.current_tab=self.tabs[tab_no]
			self.main_vbox.add(self.current_tab)
			self.tab_buttons[tab_no].set_active(True)
			self.current_tab.show()
	#end def display_tab
	
	def delete_event(self, widget, event, data=None):
		return False
	#end def delete_event

	def destroy(self, widget, data=None):
		self.quit(widget, data)
	#end def destroy
	
	def dia_about(self, widget, data):
		print "todo: implement dia_about"
	#end def dia_about
	
	def dia_create_del_database(self, widget, data):
		print "todo: implement dia_create_del_database"
		obtain_global_lock()
	#end def dia_create_del_database
	
	def dia_create_del_user(self, widget, data):
		print "todo: implement dia_create_del_user"
		obtain_global_lock()
	#end def dia_create_del_user
	
	def dia_database_stats(self, widget, data):
		print "todo: implement dia_database_stats"
		#string=fpdb_db.getDbStats(db, cursor)
	#end def dia_database_stats
	
	def dia_delete_db_parts(self, widget, data):
		print "todo: implement dia_delete_db_parts"
		obtain_global_lock()
	#end def dia_delete_db_parts
	
	def dia_edit_profile(self, widget=None, data=None, create_default=False, path=None):
		print "todo: implement dia_edit_profile"
		obtain_global_lock()
	#end def dia_edit_profile

	def dia_export_db(self, widget, data):
		print "todo: implement dia_export_db"
		obtain_global_lock()
	#end def dia_export_db
	
	def dia_get_db_root_credentials(self):
		"""obtains db root credentials from user"""
		print "todo: implement dia_get_db_root_credentials"
#		user, pw=None, None
#		
#		dialog=gtk.Dialog(title="DB Credentials needed", parent=None, flags=0,
#				buttons=(gtk.STOCK_CANCEL,gtk.RESPONSE_CANCEL,"Connect and recreate",gtk.RESPONSE_OK))
#		
#		label_warning1=gtk.Label("Please enter credentials for a database user for "+self.host+" that has permissions to create a database.")
#		
#		
#		label_user=gtk.Label("Username")
#		dialog.vbox.add(label_user)
#		label_user.show()
#		
#		response=dialog.run()
#		dialog.destroy()
#		return (user, pw, response)
	#end def dia_get_db_root_credentials
	
	def dia_import_db(self, widget, data):
		print "todo: implement dia_import_db"
		obtain_global_lock()
	#end def dia_import_db
	
	def dia_licensing(self, widget, data):
		print "todo: implement dia_licensing"
	#end def dia_licensing
	
	def dia_load_profile(self, widget, data):
		"""Dialogue to select a file to load a profile from"""
		self.obtain_global_lock()
		chooser = gtk.FileChooserDialog(title="Please select a profile file to load",
				action=gtk.FILE_CHOOSER_ACTION_OPEN,
				buttons=(gtk.STOCK_CANCEL,gtk.RESPONSE_CANCEL,gtk.STOCK_OPEN,gtk.RESPONSE_OK))
		chooser.set_filename(self.profile)
		
		response = chooser.run()
		chooser.destroy()	
		if response == gtk.RESPONSE_OK:
			self.load_profile(chooser.get_filename())
		elif response == gtk.RESPONSE_CANCEL:
			print 'User cancelled loading profile'
	#end def dia_load_profile
	
	def dia_recreate_tables(self, widget, data):
		"""Dialogue that asks user to confirm that he wants to delete and recreate the tables"""
		self.obtain_global_lock()
		dia_confirm=gtk.MessageDialog(parent=None, flags=0, type=gtk.MESSAGE_WARNING,
				buttons=(gtk.BUTTONS_YES_NO), message_format="Confirm deleting and recreating tables")
		diastring=("Please confirm that you want to (re-)create the tables. If there already are tables in the database "+self.db.database+" on "+self.db.host+" they will be deleted.")
		dia_confirm.format_secondary_text(diastring)#todo: make above string with bold for db, host and deleted

		response=dia_confirm.run()
		dia_confirm.destroy()
		if response == gtk.RESPONSE_YES:
			self.db.recreate_tables()
		elif response == gtk.RESPONSE_NO:
			print 'User cancelled recreating tables'
	#end def dia_recreate_tables
	
	def dia_regression_test(self, widget, data):
		print "todo: implement dia_regression_test"
		self.obtain_global_lock()
	#end def dia_regression_test
	
	def dia_save_profile(self, widget, data):
		print "todo: implement dia_save_profile"
	#end def dia_save_profile
	
	def diaSetupWizard(self, path):
		print "todo: implement setup wizard"
		print "setup wizard not implemented - please create the default configuration file:", path	
		diaSetupWizard = gtk.Dialog(title="Fatal Error - Config File Missing", parent=None, flags=0, buttons=(gtk.STOCK_QUIT,gtk.RESPONSE_OK))

		label = gtk.Label("Please copy the config file from the docs folder to:")
		diaSetupWizard.vbox.add(label)
		label.show()
		
		label = gtk.Label(path)
		diaSetupWizard.vbox.add(label)
		label.show()
		
		label = gtk.Label("and edit it according to the install documentation at http://fpdb.sourceforge.net")
		diaSetupWizard.vbox.add(label)
		label.show()

		response = diaSetupWizard.run()
		sys.exit(1)
	#end def diaSetupWizard
	
	def get_menu(self, window):
		"""returns the menu for this program"""
		accel_group = gtk.AccelGroup()
		self.item_factory = gtk.ItemFactory(gtk.MenuBar, "<main>", accel_group)
		self.item_factory.create_items(self.menu_items)
		window.add_accel_group(accel_group)
		return self.item_factory.get_widget("<main>")
	#end def get_menu
	
	def load_default_profile(self):
		"""Loads the defaut profile"""
		defaultpath=os.path.expanduser("~")
		if not defaultpath.endswith(os.sep):#todo: check if this is needed in *nix, if not delete it
			defaultpath+=(os.sep)
		
		if (os.sep=="\\"):#ie. if Windows use application data folder
			defaultpath=os.environ["APPDATA"]+os.sep
		else:#ie. if POSIX OS prefix fpdb with a .
			defaultpath+="."
		defaultpath+=("fpdb"+os.sep+"default.conf")
		
		if os.path.exists(defaultpath):
			self.load_profile(defaultpath)
		else:
			self.diaSetupWizard(path=defaultpath)
	#end def load_default_profile
	
	def load_profile(self, filename):
		"""Loads profile from the provided path name. also see load_default_profile"""
		self.obtain_global_lock()
		file=open(filename, "rU")
		lines=file.readlines()
		print "Opened and read profile file", filename
		self.profile=filename
		
		self.settings={'db-host':"localhost", 'db-backend':2, 'db-databaseName':"fpdb", 'db-user':"fpdb"}
		if (os.sep=="/"):
			self.settings['os']="linuxmac"
		else:
			self.settings['os']="windows"
		self.settings['tv-combinedStealFold']=True
		self.settings['tv-combined2B3B']=True
		self.settings['imp-callFpdbHud']=True
		
		if self.settings['os']=="windows":
			self.settings['bulkImport-defaultPath']="C:\\Program Files\\PokerStars\\HandHistory\\filename.txt"
			self.settings['hud-defaultPath']="C:\\Program Files\\PokerStars\\HandHistory\\"
		else:
			self.settings['bulkImport-defaultPath'] = os.path.expanduser("~") + "/.wine/drive_c/Program Files/PokerStars/HandHistory/filename.txt"
			self.settings['hud-defaultPath'] = os.path.expanduser("~")+"/.wine/drive_c/Program Files/PokerStars/HandHistory/"
			
		self.settings['hud-defaultInterval']=10
		
		for i in range(len(lines)):
			if lines[i].startswith("db-backend="):
				self.settings['db-backend']=int(lines[i][11:-1])
			elif lines[i].startswith("db-host="):
				self.settings['db-host']=lines[i][8:-1]
			elif lines[i].startswith("db-databaseName="):
				self.settings['db-databaseName']=lines[i][16:-1]
			elif lines[i].startswith("db-user="):
				self.settings['db-user']=lines[i][8:-1]
			elif lines[i].startswith("db-password="):
				self.settings['db-password']=lines[i][12:-1]
			elif lines[i].startswith("imp-callFpdbHud="):
				if lines[i].find("True")!=-1:
					self.settings['imp-callFpdbHud']=True
				else:
					self.settings['imp-callFpdbHud']=False
			elif lines[i].startswith("tv-combinedPostflop="):
				if lines[i].find("True")!=-1:
					self.settings['tv-combinedPostflop']=True
				else:
					self.settings['tv-combinedPostflop']=False
			elif lines[i].startswith("tv-combinedStealFold="):
				if lines[i].find("True")!=-1:
					self.settings['tv-combinedStealFold']=True
				else:
					self.settings['tv-combinedStealFold']=False
			elif lines[i].startswith("tv-combined2B3B="):
				if lines[i].find("True")!=-1:
					self.settings['tv-combined2B3B']=True
				else:
					self.settings['tv-combined2B3B']=False
			elif lines[i].startswith("bulkImport-defaultPath="):
				if lines[i][23:-1]!="default":
					self.settings['bulkImport-defaultPath']=lines[i][23:-1]
			elif lines[i].startswith("hud-defaultPath="):
				if lines[i][15:-1]!="default":
					self.settings['hud-defaultPath']=lines[i][16:-1]
			elif lines[i].startswith("#"):
				pass #comment - dont parse
			else:
				raise fpdb_simple.FpdbError("invalid line in profile file: "+lines[i]+"   if you don't know what to do just remove it from "+filename)
		
		if self.db!=None:
			self.db.disconnect()
		
		self.db = fpdb_db.fpdb_db()
		#print "end of fpdb.load_profile, databaseName:",self.settings['db-databaseName']
		self.db.connect(self.settings['db-backend'], self.settings['db-host'], self.settings['db-databaseName'], self.settings['db-user'], self.settings['db-password'])
		if self.db.wrongDbVersion:
			diaDbVersionWarning = gtk.Dialog(title="Strong Warning - Invalid database version", parent=None, flags=0, buttons=(gtk.STOCK_OK,gtk.RESPONSE_OK))

			label = gtk.Label("An invalid DB version or missing tables have been detected.")
			diaDbVersionWarning.vbox.add(label)
			label.show()
		
			label = gtk.Label("This error is not necessarily fatal but it is strongly recommended that you recreate the tables by using the Database menu.")
			diaDbVersionWarning.vbox.add(label)
			label.show()
		
			label = gtk.Label("Not doing this will likely lead to misbehaviour including fpdb crashes, corrupt data etc.")
			diaDbVersionWarning.vbox.add(label)
			label.show()

			response = diaDbVersionWarning.run()
			diaDbVersionWarning.destroy()

		# Database connected to successfully, load queries to pass on to other classes
                self.querydict = FpdbSQLQueries.FpdbSQLQueries(self.db.get_backend_name())
	#end def load_profile
	
	def not_implemented(self):
		print "todo: called unimplemented menu entry (users: pls ignore this)"#remove this once more entries are implemented
	#end def not_implemented
	
	def obtain_global_lock(self):
		#print "todo: implement obtain_global_lock (users: pls ignore this)"
		pass
	#end def obtain_global_lock
	
	def quit(self, widget, data):
		print "Quitting normally"
		#check if current settings differ from profile, if so offer to save or abort
		self.db.disconnect()
		gtk.main_quit()
	#end def quit_cliecked
	
	def release_global_lock(self):
		print "todo: implement release_global_lock"
	#end def release_global_lock
	
	def tab_abbreviations(self, widget, data):
		print "todo: implement tab_abbreviations"
	#end def tab_abbreviations
	
	def tab_auto_import(self, widget, data):
		"""opens the auto import tab"""
		new_aimp_thread=GuiAutoImport.GuiAutoImport(self.settings)
		self.threads.append(new_aimp_thread)
		aimp_tab=new_aimp_thread.get_vbox()
		self.add_and_display_tab(aimp_tab, "Auto Import")
	#end def tab_auto_import

	def tab_bulk_import(self, widget, data):
		"""opens a tab for bulk importing"""
		#print "start of tab_bulk_import"
		new_import_thread=GuiBulkImport.GuiBulkImport(self.db, self.settings)
		self.threads.append(new_import_thread)
		bulk_tab=new_import_thread.get_vbox()
		self.add_and_display_tab(bulk_tab, "Bulk Import")
	#end def tab_bulk_import

	def tab_main_help(self, widget, data):
		"""Displays a tab with the main fpdb help screen"""
		#print "start of tab_main_help"
		mh_tab=gtk.Label("""Welcome to Fpdb!
For documentation please visit our website at http://fpdb.sourceforge.net/ or check the docs directory in the fpdb folder.
This program is licensed under the AGPL3, see docs"""+os.sep+"agpl-3.0.txt")
		self.add_and_display_tab(mh_tab, "Help")
	#end def tab_main_help
	
	def tab_table_viewer(self, widget, data):
		"""opens a table viewer tab"""
		#print "start of tab_table_viewer"
		new_tv_thread=GuiTableViewer.GuiTableViewer(self.db, self.settings)
		self.threads.append(new_tv_thread)
		tv_tab=new_tv_thread.get_vbox()
		self.add_and_display_tab(tv_tab, "Table Viewer")
	#end def tab_table_viewer

	def tabGraphViewer(self, widget, data):
		"""opens a graph viewer tab"""
		#print "start of tabGraphViewer"
		new_gv_thread=GuiGraphViewer.GuiGraphViewer(self.db, self.settings,self.querydict)
		self.threads.append(new_gv_thread)
		gv_tab=new_gv_thread.get_vbox()
		self.add_and_display_tab(gv_tab, "Graphs")
	#end def tabGraphViewer

	def __init__(self):
		self.threads=[]
		self.db=None
		self.load_default_profile()
		
		self.window = gtk.Window(gtk.WINDOW_TOPLEVEL)
		self.window.connect("delete_event", self.delete_event)
		self.window.connect("destroy", self.destroy)
		self.window.set_title("Free Poker DB - version: alpha7, p136")
		self.window.set_border_width(1)
		self.window.set_size_request(1020,400)
		self.window.set_resizable(True)
		
		self.menu_items = (
				( "/_Main",                                 None,         None, 0, "<Branch>" ),
				( "/Main/_Load Profile (broken)",                    "<control>L", self.dia_load_profile, 0, None ),
				( "/Main/_Edit Profile (todo)",                    "<control>E", self.dia_edit_profile, 0, None ),
				( "/Main/_Save Profile (todo)",                    None,         self.dia_save_profile, 0, None ),
				("/Main/sep1", None, None, 0, "<Separator>" ),
				("/Main/_Quit", "<control>Q", self.quit, 0, None ),
				("/_Import",                               None,         None, 0, "<Branch>" ),
				("/Import/_Bulk Import",  "<control>B", self.tab_bulk_import, 0, None ),
				("/Import/_Auto Import and HUD", "<control>A", self.tab_auto_import, 0, None ),
				("/Import/Auto _Rating (todo)",                   "<control>R", self.not_implemented, 0, None ),
				("/_Viewers", None, None, 0, "<Branch>" ),
				("/_Viewers/_Auto Import and HUD", "<control>A", self.tab_auto_import, 0, None ),
				("/Viewers/_Graphs", "<control>G", self.tabGraphViewer, 0, None ),
				("/Viewers/Hand _Replayer (todo)", None, self.not_implemented, 0, None ),
				("/Viewers/Player _Details (todo)", None, self.not_implemented, 0, None ),
				("/Viewers/_Player Stats (tabulated view) (todo)", None, self.not_implemented, 0, None ),
				("/Viewers/Starting _Hands (todo)", None, self.not_implemented, 0, None ),
				("/Viewers/_Session Replayer (todo)", None, self.not_implemented, 0, None ),
				("/Viewers/Poker_table Viewer (mostly obselete)", "<control>T", self.tab_table_viewer, 0, None ),
				#( "/Viewers/Tourney Replayer
				( "/_Database",                             None,         None, 0, "<Branch>" ),
				( "/Database/Create or Delete _Database (todo)",   None,         self.dia_create_del_database, 0, None ),
				( "/Database/Create or Delete _User (todo)",       None,         self.dia_create_del_user, 0, None ),
				( "/Database/Create or Recreate _Tables",   None,         self.dia_recreate_tables, 0, None ),
				( "/Database/_Statistics (todo)",                  None,         self.dia_database_stats, 0, None ),
				( "/D_ebugging",                            None,         None, 0, "<Branch>" ),
				( "/Debugging/_Delete Parts of Database (todo)",   None,         self.dia_delete_db_parts, 0, None ),
				( "/Debugging/_Export DB (todo)",                  None,         self.dia_export_db, 0, None ),
				( "/Debugging/_Import DB (todo)",                  None,         self.dia_import_db, 0, None ),
				( "/Debugging/_Regression test (todo)",            None,         self.dia_regression_test, 0, None ),
				( "/_Help",                                 None,         None, 0, "<LastBranch>" ),
				( "/Help/_Main Help",                       "<control>H", self.tab_main_help, 0, None ),
				( "/Help/_Abbrevations (todo)",                    None,         self.tab_abbreviations, 0, None ),
				( "/Help/sep1",                             None,         None, 0, "<Separator>" ),
				( "/Help/A_bout (todo)",                           None,         self.dia_about, 0, None ),
				( "/Help/_License and Copying (todo)",             None,         self.dia_licensing, 0, None )
		)
		
		self.main_vbox = gtk.VBox(False, 1)
		self.main_vbox.set_border_width(1)
		self.window.add(self.main_vbox)
		self.main_vbox.show()
		
		menubar = self.get_menu(self.window)
		self.main_vbox.pack_start(menubar, False, True, 0)
		menubar.show()
		#done menubar
		
		self.tabs=[]
		self.tab_names=[]
		self.tab_buttons=[]
		self.tab_box = gtk.HBox(False,1)
		self.main_vbox.pack_start(self.tab_box, False, True, 0)
		self.tab_box.show()
		#done tab bar
		
		self.current_tab = gtk.VBox(False,1)
		self.current_tab.set_border_width(1)
		self.main_vbox.add(self.current_tab)
		self.current_tab.show()
		
		self.tab_main_help(None, None)
		
		self.status_bar = gtk.Label("Status: Connected to "+self.db.get_backend_name()+" database named "+self.db.database+" on host "+self.db.host)
		self.main_vbox.pack_end(self.status_bar, False, True, 0)
		self.status_bar.show()
		
		self.window.show()
	#end def __init__

	def main(self):
		gtk.main()
		return 0
	#end def main

if __name__ == "__main__":
	me = fpdb()
	me.main()