def __init__(self,remote,db_name,exclude_pattern=None):
# Keep remote address:
self._remote = remote
# Keep remote db name:
self._db_name = db_name
# A thread executor. Allows only one task to be run every time.
self._te = ThreadExecutor()
# A regexp pattern that identifies functions that are not named, and
# should be ignored.
self._exclude_pattern = exclude_pattern
# A thread safe print function. I am not sure if this is rquired. It is
# done to be one the safe side:
self._print = print
def calculate(self, wait=False, end_line=None):
_debug("Calculating")
self.__freeze_changes()
parent = None
have_error = False
executor = None
for chunk in self.iterate_chunks(end_line=end_line):
if isinstance(chunk, StatementChunk):
if chunk.needs_compile or chunk.needs_execute:
if not executor:
executor = ThreadExecutor(parent)
if executor:
statement = chunk.get_clean_statement(self)
executor.add_statement(statement)
parent = chunk.statement
# See if there are any more statements after the ones we are executing
more_statements = (end_line is not None) and \
any(isinstance(chunk, StatementChunk) for chunk in self.iterate_chunks(start_line=end_line))
if executor:
if wait:
loop = executor.event_loop
def on_statement_execution_state_changed(executor, statement):
if (statement.state == Statement.COMPILE_ERROR
or statement.state == Statement.EXECUTE_ERROR
or statement.state == Statement.INTERRUPTED):
self.__executor_error = True
statement.chunk.update_statement()
if self.__freeze_changes_count == 0:
self.__freeze_changes()
self.__chunk_changed(statement.chunk)
self.__thaw_changes()
else:
self.__chunk_changed(statement.chunk)
def on_complete(executor):
self.__executor.destroy()
self.__executor = None
if self.__executor_error:
self.__set_state(NotebookFile.ERROR)
elif more_statements:
self.__set_state(NotebookFile.NEEDS_EXECUTE)
else:
self.__set_state(NotebookFile.EXECUTE_SUCCESS)
if wait:
loop.quit()
self.__executor = executor
self.__executor_error = False
self.__set_state(NotebookFile.EXECUTING)
executor.sig_statement_executing.connect(
on_statement_execution_state_changed)
executor.sig_statement_complete.connect(
on_statement_execution_state_changed)
executor.sig_complete.connect(on_complete)
if executor.compile():
executor.execute()
if wait:
loop.run()
else:
# Nothing to execute, we could have been in a non-success state if statements were deleted
# at the end of the file.
if not more_statements:
self.__set_state(NotebookFile.EXECUTE_SUCCESS)
self.__thaw_changes()
