fpdb/pyfpdb/RushNotesMerge.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""RushNotesMerge.py

            EXPERIMENTAL - USE WITH CARE

Merge .queue file with hero's note to generate fresh .merge file

normal usage 
$> ./pyfpdb/RushNotesMerge.py "/home/steve/.wine/drive_c/Program Files/Full Tilt Poker/heroname.xml"

The generated file can then replace heroname.xml (if all is well).


"""
#    Copyright 2010,  "Gimick" of the FPDB project  fpdb.sourceforge.net
#
#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.

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


#    Standard Library modules
import os
import sys
from xml.dom import minidom 

statqueue=0
statupdated=0
statadded=0

def cleannote(textin):
    if textin.find("~fpdb~") == -1: return textin
    if textin.find("~ends~") == -1: return textin
    if textin.find("~fpdb~") > textin.find("~ends~"): return textin
    return textin[0:textin.find("~fpdb~")] + textin[textin.find("~ends~")+6:]
# get out now if parameter not passed
try: 
    sys.argv[1] <> ""
except: 
    print "A parameter is required, quitting now"
    quit()

if not os.path.isfile(sys.argv[1]):
    print "Hero notes file not found, quitting"
    quit()

if not os.path.isfile((sys.argv[1]+".queue")):
    print "Nothing queued, quitting"
    quit()

print "reading from: ", sys.argv[1]
print "merging with: ", sys.argv[1]+".queue"

#read queue and turn into a dict
queuedict = {}
xmlqueue = minidom.parse(sys.argv[1]+".queue")
notelist = xmlqueue.getElementsByTagName('NOTE')
        
for noteentry in notelist:                            
    noteplayer = noteentry.getAttribute("PlayerId")  
    notetext = noteentry.getAttribute("Text")
    queuedict[noteplayer] = notetext  
    statqueue = statqueue + 1

#read existing player note file

xmlnotefile = minidom.parse(sys.argv[1])
notelist = xmlnotefile.getElementsByTagName('NOTE')

#
#for existing players, empty out existing fpdbtext and refill
#        
for noteentry in notelist: 
    noteplayer = noteentry.getAttribute("PlayerId")
    if noteplayer in queuedict:        
        existingnote = noteentry.getAttribute("Text")
        newnote=cleannote(existingnote)
        newnote = newnote + queuedict[noteplayer]
        noteentry.setAttribute("Text",newnote)
        statupdated = statupdated + 1
        del queuedict[noteplayer]                  

#
#create entries for new players (those remaining in the dictionary)
#
if len(queuedict) > 0:
    playerdata=xmlnotefile.lastChild #move to the PLAYERDATA node (assume last one in the list)
    notesnode=playerdata.childNodes[1] #Find NOTES node 

for newplayer in queuedict:
    newentry = xmlnotefile.createElement("NOTE")
    newentry.setAttribute("PlayerId", newplayer)
    newentry.setAttribute("Text", queuedict[newplayer])              
    notesnode.insertBefore(newentry,None)
    newentry = xmlnotefile.createTextNode("\n")
    notesnode.insertBefore(newentry,None)
    statadded=statadded+1
                
#print xmlnotefile.toprettyxml()

mergednotes = open(sys.argv[1]+".merged", 'w')
xmlnotefile.writexml(mergednotes)
mergednotes.close()

xmlnotefile.unlink

print "new file has been written to: ", sys.argv[1]+".merged"
print "number in queue: ", statqueue
print "existing players updated: ", statupdated
print "new players added: ", statadded