import datetime
from .model_evaluator import train_model
from multiprocessing import Pool

class LRScheduler:
    def __init__(self):
        pass
    def evaluate(self, phen):
        from .model_evaluator import train_model
        pool = Pool(processes=1)
        results = pool.apply_async(train_model, [phen])
        pool.close()
        pool.join()
        value, other_info = results.get()
        #return 1 - 0.2, ''
        #value = train_model("0.0")
        return -value, other_info

def test_lr_scheduler():
    import core.grammar as grammar
    import core.sge
    class Test_LRScheduler:
        def __init__(self):
            pass
        def evaluate(self, phen):
            return (1, '')
    experience_name = "test_LR" + str(datetime.datetime.now()) + "/"
    grammar = grammar.Grammar("grammars/grammar_proposal.txt", 6, 17)
    evaluation_function = Test_LRScheduler()
    core.sge.evolutionary_algorithm(grammar = grammar, eval_func=evaluation_function, exp_name=experience_name)
    return True

if __name__ == "__main__":
    #print("Starting sanity check...")
    #test_lr_scheduler()
    #print("Finished sanity check.")
    import core.grammar as grammar
    import core.sge
    #remote = input("Remote? (reply with y)")
    remote = 'y'
    if remote == 'y':
        from resources.email_script import send_email
    experience_name = "LR" + str(datetime.datetime.now()) + "/"
    grammar = grammar.Grammar("grammars/grammar_proposal.txt", 6, 17)
    evaluation_function = LRScheduler()
    """
    try:
        core.sge.evolutionary_algorithm(grammar = grammar, eval_func=evaluation_function, exp_name=experience_name)
    except:
        if remote == 'y':
            from resources.email_script import send_email
            send_email('Experience crashed')
    else:
        if remote == 'y':
            from resources.email_script import send_email
            send_email('Experience finished successfully!')
            """
    core.sge.evolutionary_algorithm(grammar = grammar, eval_func=evaluation_function, exp_name=experience_name)