diff -r 2f08985625c0 -r b956c924cbbd dialogs/ConnectionDialog.py --- a/dialogs/ConnectionDialog.py Wed Jun 12 00:20:05 2013 +0200 +++ b/dialogs/ConnectionDialog.py Wed Jun 12 09:36:59 2013 +0200 @@ -32,6 +32,11 @@ # Set Connection Parameters Dialog #------------------------------------------------------------------------------- +""" +Class that implements a dialog for defining parameters of a connection graphic +element +""" + class ConnectionDialog(BlockPreviewDialog): def __init__(self, parent, controller, tagname, apply_button=False): @@ -123,6 +128,15 @@ # Connector radio button is default control having keyboard focus self.TypeRadioButtons[CONNECTOR].SetFocus() + def GetConnectionType(self): + """ + Return type selected for connection + @return: Type selected (CONNECTOR or CONTINUATION) + """ + return (CONNECTOR + if self.TypeRadioButtons[CONNECTOR].GetValue() + else CONTINUATION) + def SetValues(self, values): """ Set default connection parameters @@ -148,9 +162,7 @@ @return: {parameter_name: parameter_value,...} """ values = { - "type": (CONNECTOR - if self.TypeRadioButtons[CONNECTOR].GetValue() - else CONTINUATION), + "type": self.GetConnectionType(), "name": self.ConnectionName.GetValue()} values["width"], values["height"] = self.Element.GetSize() return values @@ -221,9 +233,7 @@ """ # Set graphic element displayed, creating a FBD connection element self.Element = FBD_Connector(self.Preview, - (CONNECTOR - if self.TypeRadioButtons[CONNECTOR].GetValue() - else CONTINUATION), + self.GetConnectionType(), self.ConnectionName.GetValue()) # Call BlockPreviewDialog function