Adding support for DB access from unit tests
[ipypbx] / src / ipypbx / controllers.py
index 132b7db..316e329 100644 (file)
 # You should have received a copy of the GNU General Public License
 # along with IPyPBX.  If not, see <http://www.gnu.org/licenses/>.
 
+"""
+GUI controllers.
+"""
+
+from ipypbx import http
 from PyQt4 import QtCore, QtGui, QtSql
 
 
@@ -33,7 +38,7 @@ class BaseController(QtCore.QObject):
     relations = ()
     delegate = None
     
-    def __init__(self, model=None, view_list=None, view_display=None, parent=None, views=None):
+    def __init__(self, parent=None, views=None):
         super(BaseController, self).__init__(parent=parent)
 
         self.views = views
@@ -45,65 +50,52 @@ class BaseController(QtCore.QObject):
             else classname)
         self.basename = self.basename[0].lower() + self.basename[1:]
 
-        # Are we given an existing model?
-        if model:
-            self.model = model
-        # Otherwise initialize a new model.
-        else:
-            self.model = QtSql.QSqlRelationalTableModel(parent)
-            self.model.setTable('ipypbxweb_%s' % self.basename.lower())
-            self.model.setEditStrategy(self.model.OnRowChange)
-
-            # Create model header from fields list.
-            for i, field in enumerate(self.fields):
-                self.model.setHeaderData(
-                    i, QtCore.Qt.Horizontal,
-                    QtCore.QVariant(QtGui.QApplication.translate(
-                        "MainWindow", field, None,
-                        QtGui.QApplication.UnicodeUTF8)))
-
-            # Fetch model data.
-            self.model.select()
-
-        # Are we given an existing view list?
-        if view_list:
-            self.view_list = view_list
+        # Initialize a new model.
+        self.model = QtSql.QSqlRelationalTableModel(parent)
+        self.model.setTable('ipypbxweb_%s' % self.basename.lower())
+        self.model.setEditStrategy(self.model.OnRowChange)
+
+        # Create model header from fields list.
+        for i, field in enumerate(self.fields):
+            self.model.setHeaderData(
+            i, QtCore.Qt.Horizontal,
+            QtCore.QVariant(QtGui.QApplication.translate(
+                "MainWindow", field, None,
+                QtGui.QApplication.UnicodeUTF8)))
+
+        # Fetch model data.
+        self.model.select()
+
         # Otherwise get view list from the parent.            
-        else:
-            self.view_list = getattr(views, self.basename + 'ViewList')
-            self.view_list.setModel(self.model)
-
-            # Hide fields not meant for display.
-            for i, field in enumerate(self.fields):
-                if field not in self.view_list_fields:
-                    self.view_list.hideColumn(i)
-
-            # Stretch headers to fill all available width.
-            self.view_list.setSelectionMode(QtGui.QTableView.SingleSelection)
-            self.view_list.setSelectionBehavior(QtGui.QTableView.SelectRows)
-            self.view_list.resizeColumnsToContents()
-            self.view_list.resizeRowsToContents()
-            self.view_list.horizontalHeader().setStretchLastSection(True)
+        self.view_list = getattr(views, self.basename + 'ViewList')
+        self.view_list.setModel(self.model)
+        self.view_list.setSelectionMode(self.view_list.SingleSelection)
+        
+        # Hide fields not meant for display.
+        for i, field in enumerate(self.fields):
+            if field not in self.view_list_fields:
+                self.view_list.hideColumn(i)
+
+        # Stretch headers to fill all available width.
+        self.view_list.setSelectionMode(QtGui.QTableView.SingleSelection)
+        self.view_list.setSelectionBehavior(QtGui.QTableView.SelectRows)
+        self.view_list.resizeColumnsToContents()
+        self.view_list.resizeRowsToContents()
+        self.view_list.horizontalHeader().setStretchLastSection(True)
 
         # Select first row.
         self.view_list.selectRow(0)
 
-        # Are we given an existing view display?
-        if view_display:
-            self.view_display = view_display
-        # Otherwise get view display from the parent.
-        else:
-            self.view_display = QtGui.QDataWidgetMapper(parent)
-            self.view_display.setModel(self.model)
-            if self.delegate:
-                self.view_display.setItemDelegate(QtSql.QSqlRelationalDelegate(self))
-
-            display_fields = self.getDisplayFields()
-            
-            for i, field in enumerate(self.fields):
-                if field in display_fields:
-                    field_widget = self.getFieldWidget(field)
-                    self.view_display.addMapping(field_widget, i)
+        # Get view display from the parent.
+        self.view_display = QtGui.QDataWidgetMapper(parent)
+        self.view_display.setModel(self.model)
+        
+        display_fields = self.getDisplayFields()
+        
+        for i, field in enumerate(self.fields):
+            if field in display_fields:
+                field_widget = self.getFieldWidget(field)
+                self.view_display.addMapping(field_widget, i)
 
         # Set relations for model & view display.
         if self.relations:
@@ -112,28 +104,48 @@ class BaseController(QtCore.QObject):
 
             for data in self.relations:
                 column, name, table, display = data                
-                column_id = self.model.fieldIndex(column)
+                column_index = self.model.fieldIndex(column)
 
                 # SetRelation screws table data filtering?
-#                self.model.setRelation(
-#                    column_id,
-#                    QtSql.QSqlRelation('ipypbxweb_%s' % table, 'id', display))
+                self.model.setRelation(
+                    column_index,
+                    QtSql.QSqlRelation('ipypbxweb_%s' % table, 'id', display))
+                #self.model.select()
 
-#                rel = self.model.relationModel(column_id)
+                rel = self.model.relationModel(column_index)
 
                 widget = self.getFieldWidget(name)
                 widget.setModel(self.parent().controllers[table].model)
-                widget.setItemDelegate(self.delegate)
+                widget.setModelColumn(rel.fieldIndex(display))
+                #widget.setItemDelegate(self.delegate)
+
 
         # Select first row in the view list.
         self.view_display.toFirst()
-        
-        # Register signals for this controller.
-        for data in self.getSignalsData():
+
+        # Signals for this controller.
+        signal_data = (
+            (getattr(self.views, self.basename + 'Add'), 'clicked()',
+             self.add),
+            (self.model, 'primeInsert(int,QSqlRecord&)', self.objectAdded),
+            (self.view_list.selectionModel(),
+             'currentRowChanged(QModelIndex,QModelIndex)',
+             self.view_display, 'setCurrentModelIndex(QModelIndex)'),
+            (self.parent().controllers.get('connection', self
+                                           ).view_list.selectionModel(),
+             'currentRowChanged(QModelIndex,QModelIndex)',
+             self.connectionChange),            
+            (getattr(self.views, self.basename + 'Save'), 'clicked()',
+             self.save))
+
+        # Connect all signals.
+        for data in signal_data:
             if len(data) == 3:
+                # Connect to python function.
                 sender, signal, receiver = data
                 QtCore.QObject.connect(sender, QtCore.SIGNAL(signal), receiver)
             elif len(data) == 4:
+                # Connect to Qt slot.
                 sender, signal, receiver, slot = data
                 QtCore.QObject.connect(
                     sender, QtCore.SIGNAL(signal), receiver, QtCore.SLOT(slot))
@@ -158,19 +170,6 @@ class BaseController(QtCore.QObject):
             field for field in self.fields
             if not field in self.view_display_fields_hidden]        
 
-    def getSignalsData(self):
-        """
-        Default signals built from controller's base name.
-        """
-        # Default signals handle row selection, Add and Save buttons.
-        return [
-            (getattr(self.views, self.basename + 'Add'), 'clicked()', self.add),
-            (self.view_list.selectionModel(),
-             'currentRowChanged(QModelIndex,QModelIndex)',
-             self.view_display, 'setCurrentModelIndex(QModelIndex)'),
-            (getattr(self.views, self.basename + 'Save'), 'clicked()',
-             self.save)]
-
     def add(self):
         """
         Add new object.
@@ -186,13 +185,19 @@ class BaseController(QtCore.QObject):
         # Focust to the first displayed field.
         self.getFieldWidget(self.getDisplayFields()[0]).setFocus()
 
-        # TODO: set default values?
+    def connectionChange(self, index, row):
+        """
+        Overload to handle connection change.
+        """
+        return NotImplemented
 
     def save(self):
         """
         Save to database.
         """
+        index = self.view_list.currentIndex()
         self.view_display.submit()
+        self.view_list.setCurrentIndex(index)
         self.getFieldWidget('Add').setEnabled(True)
 
 
@@ -208,28 +213,73 @@ class ConnectionController(BaseController):
         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Freeswitch IP Address'),
         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Freeswitch Port'))
     view_list_fields = 'Name', 'Freeswitch IP Address', 'Freeswitch Port'
+    servers = []
+
+    def __init__(self, parent=None, views=None):
+        super(ConnectionController, self).__init__(parent, views)
+
+        for row in range(self.model.rowCount()):
+            # Get local IP address and port from the table for each row.
+            server = http.FreeswitchConfigServer(self.parent(), )
+            server.setSocketData(*self.getSocketData(row))
+            server.startServer()
+            self.servers.append(server)
+
+    def getSocketData(self, row):
+        """
+        Return socket data for given row number.
+        """
+        record = self.model.record(row)
+        
+        # Local IP address.
+        local_ip_address = record.value('local_ip_address').toString()
+
+        # Local port.
+        local_port, _ok = record.value('local_port').toInt()
+        if not _ok:
+            local_port = None
+
+        # Connection ID.
+        connection_id, _ok = record.value('id').toInt()
+        if not _ok:
+            connection_id = None
+        
+        return local_ip_address, local_port, connection_id
+    
+    def connectionChange(self, index):
+        """
+        Restart config server on connection change if necessary.
+        """
+        current_row = index.row()
+        if current_row != -1:
+            # Select the new row.
+            connection_id, _ok = index.model().data(
+                index.sibling(current_row, 0)).toInt()
+
+            # Apply new socket location.
+            self.servers[current_row].setSocketData(
+                *self.getSocketData(current_row))
+        
+    def objectAdded(self, row, record):
+        """
+        New connection added.
+        """
+        self.addServer(*self.getSocketData(row))
+        
+    def addServer(self, host=None, port=None, connection_id=None):
+        """
+        Add a new config server.
+        """
+        server = http.FreeswitchConfigServer(self.model)
+        server.setSocketData(host, port, connection_id)
+        server.startServer()
+        self.servers.append(server)
 
 
 class ConnectionChangeListenerController(BaseController):
     """
     Mixin class for reacting on connection change.
     """
-    def getSignalsData(self):
-        """
-        Listen to connection change signal.
-        """
-        connection_controller = self.parent().controllers['connection']
-
-        signals = [
-            (connection_controller.view_list.selectionModel(),
-             'currentRowChanged(QModelIndex,QModelIndex)',
-             self.connectionChange),
-            (self.model, 'primeInsert(int,QSqlRecord&)',
-             self.setConnectionId)] 
-        signals.extend(super(
-            ConnectionChangeListenerController, self).getSignalsData())
-        return signals
-        
     def connectionChange(self, index):
         """
         Connection change handler.
@@ -237,14 +287,31 @@ class ConnectionChangeListenerController(BaseController):
         Filters table by a new connection ID and stores last connection ID
         locally.
         """
-        if index.row() != -1:
-            connection_id, ok = index.model().data(
+        index_row = index.row()
+        if index_row != -1:
+            # Get connection_id field value.
+            connection_id, _ok = index.model().data(
                 index.sibling(index.row(), 0)).toInt()
             self.connection_id = connection_id
-            self.model.setFilter('connection_id = %i' % connection_id)
-            self.model.index(0, 0)
 
-    def setConnectionId(self, row, record):
+            # Filter is customizable in order to allow ugly hacks :-)
+            self.model.setFilter(self.getFilter(connection_id))
+
+            # Select first row.
+            self.view_list.selectRow(0)
+
+            # Create a new object if none exist.
+            if not self.model.rowCount():
+                self.add()
+
+    def getFilter(self, connection_id):
+        return 'ipypbxweb_%s.connection_id = %i' % (
+            self.basename, connection_id)        
+
+    def objectAdded(self, row, record):
+        """
+        Set connection_id from currently selected connection.
+        """
         record.setValue('connection_id', self.connection_id)
 
         
@@ -278,7 +345,7 @@ class DomainController(ConnectionChangeListenerController):
         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Host Name'),
         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Is Active'))
     view_list_fields = 'SIP Profile ID', 'Host Name'
-    relations = (('sip_profile_id', 'SIP Profile ID', 'sipprofile', 'Name'),)
+    relations = (('sip_profile_id', 'SIP Profile ID', 'sipprofile', 'name'),)
     
 
 class GatewayController(ConnectionChangeListenerController):
@@ -294,12 +361,13 @@ class GatewayController(ConnectionChangeListenerController):
         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Password'),
         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Realm'),
         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'From Domain'),
+        QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Extension'),
         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Expire In Seconds'),
         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Retry In Seconds'),
         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Caller ID In From Field'),
         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Is Active'))
     view_list_fields = 'SIP Profile ID', 'Name'
-    relations = (('sip_profile_id', 'SIP Profile ID', 'sipprofile', 'Name'),)
+    relations = (('sip_profile_id', 'SIP Profile ID', 'sipprofile', 'name'),)
     
 
 class EndpointController(ConnectionChangeListenerController):
@@ -314,7 +382,7 @@ class EndpointController(ConnectionChangeListenerController):
         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Domain ID'),
         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Is Active'))
     view_list_fields = 'User ID', 'Domain ID'
-    relations = (('domain_id', 'Domain ID', 'domain', 'Host Name'),)
+    relations = (('domain_id', 'Domain ID', 'domain', 'host_name'),)
     
 
 class ExtensionController(ConnectionChangeListenerController):
@@ -331,5 +399,20 @@ class ExtensionController(ConnectionChangeListenerController):
         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Authenticate Calls'),
         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Is Active'))
     view_list_fields = 'Destination Match',
-    relations = (('domain_id', 'Domain ID', 'domain', 'Host Name'),)
+    view_display_fields_hidden = 'ID', 'Connection ID', 'Endpoint ID'
+    relations = (
+        ('domain_id', 'Domain ID', 'domain', 'host_name'),
+#        ('endpoint_id', 'Endpoint ID', 'endpoint', 'user_id'),
+        )
+        
+    def objectAdded(self, row, record):
+        record.setValue(
+            'xml_dialplan', '<action application="echo" data=""/>')
+        super(ExtensionController, self).objectAdded(row, record)
         
+    def getFilter(self, connection_id):
+        # Workaround for Qt bug:
+        # http://bugreports.qt.nokia.com/browse/QTBUG-8217 . Apparently they
+        # don't hurry to fix it.
+        return '1 = 1) or (ipypbxweb_%s.connection_id = %i' % (
+            self.basename, connection_id)