utils.py

import random
import logging

def save(states, filename):
    with open(filename, 'w') as f:
        
        for state in states:
            # print(state)
            out = [str(x) for x in state]
            out = ','.join(out)
            f.write(out+'\n')
            
            out = [str(x) for x in states[state]]
            out = ','.join(out)
            f.write(out+'\n')


def load(filename):
    states = {}
    
    with open(filename, 'r') as f:
        lines = [x.strip() for x in f.readlines()]
        lines = [x.split(',') for x in lines]
        output  = []
        
        for line in lines:
            # print(line)
            temp = [int(x) for x in line]
            output.append(temp)
        
        for i in range(0, len(output), 2):
            states[tuple(output[i])] = output[i+1]

    return states


def strategy(actions, eps):

    a = list(set(actions))
    
    if len(a) == 1:
        return (a[0], eps)
    
    best_action = max(set(actions), key=actions.count)
    random_action = random.choice(a)
    
    if(random.random() > eps):
        return (best_action, eps)
    else:
        # eps *= 0.99
        return (random_action, eps)
        


def setup_logger(name, log_file, level=logging.INFO):
    """
    Setting up logger
    """
    formatter = logging.Formatter('%(asctime)s %(levelname)s %(message)s')
    handler = logging.FileHandler(log_file, mode = 'w')        
    handler.setFormatter(formatter)

    logger = logging.getLogger(name)
    logger.setLevel(level)
    logger.addHandler(handler)

    return logger

def ui(state):
    
    pos = list(state)
    
    for i in range(len(pos)):
        if pos[i] == 1: pos[i] = "X"
        if pos[i] == 2: pos[i] = "O"
        if pos[i] == 0: pos[i] = " "

    row1, row2, row3 = pos[0:3], pos[3:6], pos[6:9]

    print('\n')
    print("  ", row1[0], "  |  ", row1[1], "  |  ", row1[2])
    print('----------------------')
    print("  ", row2[0], "  |  ", row2[1], "  |  ", row2[2])
    print('----------------------')
    print("  ", row3[0], "  |  ", row3[1], "  |  ", row3[2])

    print("\nIndices\n")

    print("  ", "0", "  |  ", "1", "  |  ", "2")
    print('----------------------')
    print("  ", "3", "  |  ", "4", "  |  ", "5")
    print('----------------------')
    print("  ", "6", "  |  ", "7", "  |  ", "8")
    print('\n')