def __init__(self, vocabularyController, symbol):
self.vocabularyController = vocabularyController
self.log = logging.getLogger(__name__)
self._view = CustomTransformationFunctionView(self)
self.symbol = symbol
self.sourceCode = ""
self.reverseSourceCode = ""
self.functionName = ""
self.function = None
class CustomTransformationFunctionController(object):
"""Manage the creation of a custom transformation function"""
def __init__(self, vocabularyController, symbol):
self.vocabularyController = vocabularyController
self.log = logging.getLogger(__name__)
self._view = CustomTransformationFunctionView(self)
self.symbol = symbol
self.sourceCode = ""
self.reverseSourceCode = ""
self.functionName = ""
self.function = None
@property
def view(self):
return self._view
def run(self):
self.initSourceCode()
self.initReverseSourceCode()
self.updateMessages()
self._view.run()
def cancelButton_clicked_cb(self, widget):
"""Callback executed when the user clicks
on the cancel button"""
self._view.customTransformationFunctionDialog.destroy()
def applyButton_clicked_cb(self, widget):
"""Callback executed when the user clicks
on the apply button."""
self._view.customTransformationFunctionDialog.destroy()
if self.function is not None:
# Create the function add it to the workspace set of available functions
workspace = self.vocabularyController.getCurrentWorkspace()
workspace.addCustomTransformationFunction(self.function)
def testYourFunctionButton_clicked_cb(self, widget):
self.dataUpdated()
def sourceCodeIsTheSameForReverseCheckButton_toggled_cb(self, widget):
if self._view.sourceCodeIsTheSameForReverseCheckButton.get_active():
self._view.functionReverseTextView.set_editable(False)
else:
self._view.functionReverseTextView.set_editable(True)
def initSourceCode(self):
initialSource = """# Type below the Python source code of your function.
# The source code must edit the content of a 'message' variable. This variable
# contains an hexastring value (eg. '0b1c3489') you can 'function'.
# An example of a source code would be :
#
# ! This source code is for abstracting messages (decrypt, decompress received messages...)
#
# An example of a source code would be :
#
# message = '00'+message+'00'
"""
self._view.functionTextView.get_buffer().set_text(initialSource)
def initReverseSourceCode(self):
initialSource = """# Type below the Python source code of your function.
# The source code must edit the content of a 'message' variable. This variable
# contains an hexastring value (eg. '0b1c3489') you can 'function'.
#
# ! This source code is for specifying messages (encrypt, compress sent messages...)
#
# An example of a source code would be :
#
# message = '00'+message+'00'
"""
self._view.functionReverseTextView.get_buffer().set_text(initialSource)
def dataUpdated(self):
# retrieve the source code
self.sourceCode = self._view.functionTextView.get_buffer().get_text(self._view.functionTextView.get_buffer().get_start_iter(), self._view.functionTextView.get_buffer().get_end_iter(), True)
if self._view.sourceCodeIsTheSameForReverseCheckButton.get_active():
self.reverseSourceCode = self.sourceCode
else:
self.reverseSourceCode = self._view.functionReverseTextView.get_buffer().get_text(self._view.functionReverseTextView.get_buffer().get_start_iter(), self._view.functionReverseTextView.get_buffer().get_end_iter(), True)
# retrieve the name of the function
self.functionName = self._view.nameOfFunctionEntry.get_text()
if self.functionName is not None and len(self.functionName) > 0 and self.sourceCode is not None and len(self.sourceCode) > 0 and self.reverseSourceCode is not None and len(self.reverseSourceCode) > 0:
self.function = CustomFunction(self.functionName, self.sourceCode, self.reverseSourceCode)
errorMessage = self.function.compileSourceCode()
if errorMessage is None:
self._view.applyButton.set_sensitive(True)
self.updateMessages()
self._view.imageValid.show()
self._view.imageError.hide()
self._view.scrolledwindow3.show_all()
self._view.labelMessage.set_label(_("Verify below the transformed messages"))
self._view.labelMessage.show()
else:
self.function = None
self._view.applyButton.set_sensitive(False)
self.updateMessages()
self._view.imageValid.hide()
self._view.imageError.show()
self._view.scrolledwindow3.show_all()
self._view.labelMessage.set_label("{0}".format(errorMessage))
self._view.labelMessage.show()
return
errorMessage = self.function.compileReverseSourceCode()
if errorMessage is None:
self._view.applyButton.set_sensitive(True)
self._view.imageValid.show()
self._view.imageError.hide()
self._view.scrolledwindow3.show_all()
else:
self.function = None
self._view.applyButton.set_sensitive(False)
self.updateMessages()
self._view.imageValid.hide()
self._view.imageError.show()
self._view.scrolledwindow3.show_all()
self._view.labelMessage.set_label("Error with the reverse source code {0}".format(errorMessage))
self._view.labelMessage.show()
else:
self._view.applyButton.set_sensitive(False)
self._view.imageValid.hide()
self._view.imageError.show()
self._view.scrolledwindow3.show_all()
self._view.labelMessage.set_label(_("Specify the name and the source code of your Function"))
self._view.labelMessage.show()
def updateMessages(self):
self._view.messagesListStore.clear()
for message in self.symbol.getMessages():
original = message.getStringData()
if self.function is not None:
functioned = self.function.apply(original)
else:
functioned = ""
self.addMessage(original, functioned)
def addMessage(self, original, functioned):
i = self._view.messagesListStore.append()
self._view.messagesListStore.set(i, 0, original)
self._view.messagesListStore.set(i, 1, functioned)