import threading
import io
import sys
import os
import os.path
import traceback
from damaris.data import Resultable

class ResultHandling(threading.Thread):
    """
    runs the result script in sandbox
    """

    def __init__(self, script_data, result_iterator, data_pool):
        threading.Thread.__init__(self,name="result handler")
        self.script=script_data
        self.results=result_iterator
        self.data_space=data_pool
        self.quit_flag=self.results.quit_flag
        if self.data_space is not None:
            self.data_space["__recentresult"]=-1

    def run(self):
        # execute it
        dataspace={}
        data_classes = __import__('damaris.data', dataspace, dataspace, ['*'])
        for name in dir(data_classes):
            if name[:2]=="__" and name[-2:]=="__": continue
            dataspace[name]=data_classes.__dict__[name]
        del data_classes
        dataspace["results"]=self
        dataspace["data"]=self.data_space
        self.raised_exception=None
        self.location = None
        try:
            exec(self.script, dataspace)
        except Exception as e:
            self.raised_exception=e
            self.location=traceback.extract_tb(sys.exc_info()[2])[-1][1:3]
            traceback_file=io.StringIO()
            traceback.print_tb(sys.exc_info()[2], None, traceback_file)
            self.traceback=traceback_file.getvalue()
            traceback_file=None
            return
        if not "result" in dataspace:
            dataspace=None
            return
        try:
            dataspace["result"]()
        except Exception as e:
            self.raised_exception=e
            self.location=traceback.extract_tb(sys.exc_info()[2])[-1][1:3]
            traceback_file=io.StringIO()
            traceback.print_tb(sys.exc_info()[2], None, traceback_file)
            self.traceback=traceback_file.getvalue()
            traceback_file=None
            dataspace=None

    def __iter__(self):
        if self.quit_flag.isSet():
            self.results=None
            return
        for i in self.results:
            if hasattr(self.results, "in_advance"):
                self.data_space["__resultsinadvance"]=self.results.in_advance
            if self.quit_flag.isSet():
                self.results=None
                return
            if isinstance(i, Resultable.Resultable):
                if self.data_space is not None:
                    self.data_space["__recentresult"]=i.job_id+0
            yield i
            if self.quit_flag.isSet():
                self.results=None
                return

    def stop(self):
        self.quit_flag.set()