#ub_mine.py v1.4 July 29th 2009
#written by Neko (2+2 handle)
#This code is released under the WTF licence. That means you can
#do W.hatever T.he F.uck you feel like with it.
#If you would like to make a small donation via UB or Absolute p.m. me on 2+2
#or email me neko2p2@gmail.com

import os,shutil,sys,string,time,traceback
from msvcrt import kbhit,getch

#maximum number of hands to store in one batch file
max_hands_per_file = 100

#if True converted .dat files will be deleted otherwise they will be moved
del_converted = True

root_dir=os.path.dirname(sys.argv[0])

#list of locations to look for AP instant replay data files
replay_dir_file = root_dir+"/replay_dirs.txt"

#move instant replay files to this directory after processing

print "Root directory:"+root_dir
converted_dir = root_dir+"/converted/"

#store the converted histories in this directory
hist_dir = root_dir+"/histories/"

#number of seconds to wait in between processing batches
sleep_time = 5



#===============================================================================
def main():
    replay_dir = get_replay_dir()
    if not replay_dir:
        die("Unable to find the instant replay directory")
    
    #make directory to move processed hands to if it doesn't already exist
    print "Looking for processed hand directory "+converted_dir
    if not os.path.isdir(converted_dir):
        "\tConverted directory not found. Trying to create now...",        
        try:
            os.mkdir(converted_dir)
        except:
            die("failed")
        else:
            print "success"

    else:
        print "\tAlready exists"
        
    #make directory to store converted hand history file
    if not os.path.isdir(hist_dir):
        try:
            os.mkdir(hist_dir)
        except:
            die("failed to make hand history directory")

    stop = False

    print "="*80
    print "Starting to convert hand histories"
    print "hit 'q' at any time to stop import"
    print "="*80

    ttl_hand_count = 0
    hand_counter=0
    out_file = None
    os.chdir(replay_dir)

    while not stop:

        if kbhit() and getch() == 'q': stop = True
        
        dirList=os.listdir(os.getcwd())

        for fname in dirList:
            if fname.find('.dat')!=-1:

                #get a new file to write to if the old one already contains
                #the max num of histories or an output file doesn't yet exist
#                if hand_counter ==  max_hands_per_file or not out_file:
#                    hand_counter = 0
#                    if isinstance(out_file,file):
#                        out_file.close()

#                    out_file = get_new_file_object()

                #extrat history and move/delete the .dat file
                if add_to_file(fname):
                    
                    handle_converted_file(fname)
        
                    hand_counter += 1
                    ttl_hand_count += 1
                    sys.stdout.write("\rTotal Hands Written: " + str(ttl_hand_count)) 
                    sys.stdout.flush()
           

        if not stop: time.sleep(sleep_time)

    if out_file:
        out_file.close()
    raw_input("\nSuccesfully Finished Execution. Press enter key to quit.")

    
#-------------------------------------------------------------------------------
#Extract the hh from the .dat file
def add_to_file(fname):
    try:
        infile = open(fname,'r')
        
        dat = infile.readlines()
        table = dat[1].partition("=")[2].strip()


        for i,line in enumerate(dat):
            dat[i] = dat[i].strip()
        infile.close()
    
        first = dat.index("[IHH]")
        dat = dat[first+1:len(dat)-1]
        hh = ''
        for line in dat:
            nxt_line = line.partition('=')[2]
            hh += nxt_line+'\n'
 
        f = open(get_output_file_name(table),'w')       
        f.write(hh)
        f.close()
    except:
        return False

    return True

#-------------------------------------------------------------------------------
#remove or delete the converted .dat files
def handle_converted_file(fname):
    try:
        if del_converted:
            os.remove(fname)
        else:
            # removing duplicate file if file exists
            pathToFile = converted_dir+fname
            if os.path.exists(pathToFile):
                os.remove(pathToFile)
            shutil.move(fname,converted_dir)
    except:
        pass
#-------------------------------------------------------------------------------
#iterate over the list of replay directories and return the first one that exists
def get_replay_dir():

    replayf = open(replay_dir_file,'r')
    dat = replayf.readlines()
    replayf.close()
    dirs = []
    for line in dat:
        dirs.append(line.strip())

    idx = 0
    print "Looking for Instant Replay directory...."
    for rdir in dirs:
        print "\tChecking "+rdir,
        if os.path.isdir(rdir):
            print "found"
            return rdir
        else:
            print "doesn't exist"

    return False
#-------------------------------------------------------------------------------
#generic exit point on errors
def die(msg = None):
    if msg: print msg
    raw_input("Press enter key to exit")
    os._exit(99)
#-------------------------------------------------------------------------------
#returns the next batch file name
def get_output_file_name(prefix):

    suf = get_file_suffix()
    num=suf.next()
    while os.path.exists(hist_dir+"IHH"+prefix+"_batch_"+str(num)+'.txt'):
        num  = suf.next()
    fname = hist_dir+"IHH"+prefix+"_batch_"+str(num)+'.txt'

    return fname

#-------------------------------------------------------------------------------
#batch file suffix generator
def get_file_suffix():
    k = 0
    while True:
        k+=1
        yield k
#-------------------------------------------------------------------------------
#returns a new file object to write a batch of histories to
def get_new_file_object():
           
    try:
        out_file = open(get_output_file_name(),'w')
        return out_file
    except:
        die("Unable to create new output file "+fname)

#===============================================================================
if __name__ == '__main__': 
    try:
        main()
    except:
        print ""
        print "An unhandled exception occured."
        traceback.print_exc()
        raw_input("press enter key to exit")

#ub_mine.py v1.4 Feb 16th 2009
#written by Neko (2+2 handle)
#This code is released under the WTF licence. That means you can
#do W.hatever T.he F.uck you feel like with it.
#If you would like to make a small donation via UB or Absolute p.m. me on 2+2
#or email me neko2p2@gmail.com

import os,shutil,sys,string,time,traceback
from msvcrt import kbhit,getch

#maximum number of hands to store in one batch file
max_hands_per_file = 100

#if True converted .dat files will be deleted otherwise they will be moved
del_converted = True

root_dir=os.path.dirname(sys.argv[0])

#list of locations to look for AP instant replay data files
replay_dir_file = root_dir+"/replay_dirs.txt"

#move instant replay files to this directory after processing

print "Root directory:"+root_dir
converted_dir = root_dir+"/converted/"

#store the converted histories in this directory
hist_dir = root_dir+"/histories/"

#number of seconds to wait in between processing batches
sleep_time = 5



#===============================================================================
def main():
    replay_dir = get_replay_dir()
    if not replay_dir:
        die("Unable to find the instant replay directory")
    
    #make directory to move processed hands to if it doesn't already exist
    print "Looking for processed hand directory "+converted_dir
    if not os.path.isdir(converted_dir):
        "\tConverted directory not found. Trying to create now...",        
        try:
            os.mkdir(converted_dir)
        except:
            die("failed")
        else:
            print "success"

    else:
        print "\tAlready exists"
        
    #make directory to store converted hand history file
    if not os.path.isdir(hist_dir):
        try:
            os.mkdir(hist_dir)
        except:
            die("failed to make hand history directory")

    stop = False

    print "="*80
    print "Starting to convert hand histories"
    print "hit 'q' at any time to stop import"
    print "="*80

    ttl_hand_count = 0
    hand_counter=0
    out_file = None
    os.chdir(replay_dir)

    while not stop:

        if kbhit() and getch() == 'q': stop = True
        
        dirList=os.listdir(os.getcwd())

        for fname in dirList:
            if fname.find('.dat')!=-1:

                #get a new file to write to if the old one already contains
                #the max num of histories or an output file doesn't yet exist
#                if hand_counter ==  max_hands_per_file or not out_file:
#                    hand_counter = 0
#                    if isinstance(out_file,file):
#                        out_file.close()

#                    out_file = get_new_file_object()

                #extrat history and move/delete the .dat file
                if add_to_file(fname):
                    
                    handle_converted_file(fname)
        
                    hand_counter += 1
                    ttl_hand_count += 1
                    sys.stdout.write("\rTotal Hands Written: " + str(ttl_hand_count)) 
                    sys.stdout.flush()
           

        if not stop: time.sleep(sleep_time)

    if out_file:
        out_file.close()
    raw_input("\nSuccesfully Finished Execution. Press enter key to quit.")

    
#-------------------------------------------------------------------------------
#Extract the hh from the .dat file
def add_to_file(fname):
    try:
        infile = open(fname,'r')
        
        dat = infile.readlines()
        table = dat[1].partition("=")[2].strip()


        for i,line in enumerate(dat):
            dat[i] = dat[i].strip()
        infile.close()
    
        first = dat.index("[IHH]")
        dat = dat[first+1:len(dat)-1]
        hh = ''
        for line in dat:
            nxt_line = line.partition('=')[2]
            hh += nxt_line+'\n'
 
        f = open(get_output_file_name(table),'w')       
        f.write(hh)
        f.close()
    except:
        return False

    return True

#-------------------------------------------------------------------------------
#remove or delete the converted .dat files
def handle_converted_file(fname):
    try:
        if del_converted:
            os.remove(fname)
        else:
            # removing duplicate file if file exists
            pathToFile = converted_dir+fname
            if os.path.exists(pathToFile):
                os.remove(pathToFile)
            shutil.move(fname,converted_dir)
    except:
        pass
#-------------------------------------------------------------------------------
#iterate over the list of replay directories and return the first one that exists
def get_replay_dir():

    replayf = open(replay_dir_file,'r')
    dat = replayf.readlines()
    replayf.close()
    dirs = []
    for line in dat:
        dirs.append(line.strip())

    idx = 0
    print "Looking for Instant Replay directory...."
    for rdir in dirs:
        print "\tChecking "+rdir,
        if os.path.isdir(rdir):
            print "found"
            return rdir
        else:
            print "doesn't exist"

    return False
#-------------------------------------------------------------------------------
#generic exit point on errors
def die(msg = None):
    if msg: print msg
    raw_input("Press enter key to exit")
    os._exit(99)
#-------------------------------------------------------------------------------
#returns the next batch file name
def get_output_file_name(table):

    suf = get_file_suffix()
    num=suf.next()
    while os.path.exists(hist_dir+"IHH"+str(num)+"_Table "+table+'.txt'):
        num  = suf.next()
    fname = hist_dir+"IHH"+str(num)+"_Table "+table+'.txt'
  
    return fname

#-------------------------------------------------------------------------------
#batch file suffix generator
def get_file_suffix():
    k = 0
    while True:
        k+=1
        yield k
#-------------------------------------------------------------------------------
#returns a new file object to write a batch of histories to
def get_new_file_object():
           
    try:
        out_file = open(get_output_file_name(),'w')
        return out_file
    except:
        die("Unable to create new output file "+fname)

#===============================================================================
if __name__ == '__main__': 
    try:
        main()
    except:
        print ""
        print "An unhandled exception occured."
        traceback.print_exc()
        raw_input("press enter key to exit")