git.maemo.org Git - ipypbx/blob - src/ipypbx/controllers.py

   1 # Copyright (c) Stas Shtin, 2010
   2
   3 # This file is part of IPyPBX.
   4
   5 # IPyPBX is free software: you can redistribute it and/or modify
   6 # it under the terms of the GNU General Public License as published by
   7 # the Free Software Foundation, either version 3 of the License, or
   8 # (at your option) any later version.
   9
  10 # IPyPBX is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14
  15 # You should have received a copy of the GNU General Public License
  16 # along with IPyPBX.  If not, see <http://www.gnu.org/licenses/>.
  17
  18 """
  19 GUI controllers.
  20 """
  21
  22 from ipypbx import http
  23 from PyQt4 import QtCore, QtGui, QtSql
  24
  25
  26 class BaseController(QtCore.QObject):
  27     """
  28     Base class for other controllers.
  29
  30     Doesn't do anything useful on its own.
  31     """
  32     # TODO: possibly use a separate class for options and a meta-class.
  33     fields = ()
  34     view_list_fields = ()
  35     view_display_fields = ()
  36     view_display_fields_hidden = 'ID', 'Connection ID'
  37     is_bound_to_connection = True
  38     relations = ()
  39     delegate = None
  40
  41     def __init__(self, parent=None, views=None):
  42         super(BaseController, self).__init__(parent=parent)
  43
  44         self.views = views
  45
  46         # Find out base name.
  47         classname = self.__class__.__name__
  48         self.basename = (
  49             classname[:-10] if classname.endswith('Controller')
  50             else classname)
  51         self.basename = self.basename[0].lower() + self.basename[1:]
  52
  53         # Initialize a new model.
  54         self.model = QtSql.QSqlRelationalTableModel(parent)
  55         self.model.setTable('ipypbxweb_%s' % self.basename.lower())
  56         self.model.setEditStrategy(self.model.OnRowChange)
  57
  58         # Create model header from fields list.
  59         for i, field in enumerate(self.fields):
  60             self.model.setHeaderData(
  61             i, QtCore.Qt.Horizontal,
  62             QtCore.QVariant(QtGui.QApplication.translate(
  63                 "MainWindow", field, None,
  64                 QtGui.QApplication.UnicodeUTF8)))
  65
  66         # Fetch model data.
  67         self.model.select()
  68
  69         # Otherwise get view list from the parent.
  70         self.view_list = getattr(views, self.basename + 'ViewList')
  71         self.view_list.setModel(self.model)
  72         self.view_list.setSelectionMode(self.view_list.SingleSelection)
  73
  74         # Hide fields not meant for display.
  75         for i, field in enumerate(self.fields):
  76             if field not in self.view_list_fields:
  77                 self.view_list.hideColumn(i)
  78
  79         # Stretch headers to fill all available width.
  80         self.view_list.setSelectionMode(QtGui.QTableView.SingleSelection)
  81         self.view_list.setSelectionBehavior(QtGui.QTableView.SelectRows)
  82         self.view_list.resizeColumnsToContents()
  83         self.view_list.resizeRowsToContents()
  84         self.view_list.horizontalHeader().setStretchLastSection(True)
  85
  86         # Select first row.
  87         self.view_list.selectRow(0)
  88
  89         # Get view display from the parent.
  90         self.view_display = QtGui.QDataWidgetMapper(parent)
  91         self.view_display.setModel(self.model)
  92
  93         display_fields = self.getDisplayFields()
  94
  95         for i, field in enumerate(self.fields):
  96             if field in display_fields:
  97                 field_widget = self.getFieldWidget(field)
  98                 self.view_display.addMapping(field_widget, i)
  99
 100         # Set relations for model & view display.
 101         if self.relations:
 102             self.delegate = QtSql.QSqlRelationalDelegate(self)
 103             self.view_display.setItemDelegate(self.delegate)
 104
 105             for data in self.relations:
 106                 column, name, table, display = data
 107                 column_index = self.model.fieldIndex(column)
 108
 109                 # SetRelation screws table data filtering?
 110                 self.model.setRelation(
 111                     column_index,
 112                     QtSql.QSqlRelation('ipypbxweb_%s' % table, 'id', display))
 113                 #self.model.select()
 114
 115                 rel = self.model.relationModel(column_index)
 116
 117                 widget = self.getFieldWidget(name)
 118                 widget.setModel(self.parent().controllers[table].model)
 119                 widget.setModelColumn(rel.fieldIndex(display))
 120                 #widget.setItemDelegate(self.delegate)
 121
 122
 123         # Select first row in the view list.
 124         self.view_display.toFirst()
 125
 126         # Signals for this controller.
 127         signal_data = (
 128             (getattr(self.views, self.basename + 'Add'), 'clicked()',
 129              self.add),
 130             (self.model, 'primeInsert(int,QSqlRecord&)', self.objectAdded),
 131             (self.view_list.selectionModel(),
 132              'currentRowChanged(QModelIndex,QModelIndex)',
 133              self.view_display, 'setCurrentModelIndex(QModelIndex)'),
 134             (self.parent().controllers.get('connection', self
 135                                            ).view_list.selectionModel(),
 136              'currentRowChanged(QModelIndex,QModelIndex)',
 137              self.connectionChange),
 138             (getattr(self.views, self.basename + 'Save'), 'clicked()',
 139              self.save))
 140
 141         # Connect all signals.
 142         for data in signal_data:
 143             if len(data) == 3:
 144                 # Connect to python function.
 145                 sender, signal, receiver = data
 146                 QtCore.QObject.connect(sender, QtCore.SIGNAL(signal), receiver)
 147             elif len(data) == 4:
 148                 # Connect to Qt slot.
 149                 sender, signal, receiver, slot = data
 150                 QtCore.QObject.connect(
 151                     sender, QtCore.SIGNAL(signal), receiver, QtCore.SLOT(slot))
 152
 153     def getFieldWidget(self, field):
 154         """
 155         Return widget for given field name.
 156         """
 157         return getattr(
 158             self.views,
 159             self.basename + ''.join(word.capitalize()
 160                                     for word in field.split(' ')))
 161
 162     def getDisplayFields(self):
 163         """
 164         Return list of display fields.
 165
 166         If view_display_fields is not send, display all fields except
 167         the first one that is usually the ID.
 168         """
 169         return [
 170             field for field in self.fields
 171             if not field in self.view_display_fields_hidden]
 172
 173     def add(self):
 174         """
 175         Add new object.
 176         """
 177         # Add a new row to list view.
 178         num_rows = self.model.rowCount()
 179         self.model.insertRows(num_rows, 1)
 180         self.view_list.selectRow(num_rows)
 181
 182         # Disable adding more than one row.
 183         self.getFieldWidget('Add').setEnabled(False)
 184
 185         # Focust to the first displayed field.
 186         self.getFieldWidget(self.getDisplayFields()[0]).setFocus()
 187
 188     def connectionChange(self, index, row):
 189         """
 190         Overload to handle connection change.
 191         """
 192         return NotImplemented
 193
 194     def save(self):
 195         """
 196         Save to database.
 197         """
 198         index = self.view_list.currentIndex()
 199         self.view_display.submit()
 200         self.view_list.setCurrentIndex(index)
 201         self.getFieldWidget('Add').setEnabled(True)
 202
 203
 204 class ConnectionController(BaseController):
 205     """
 206     Connections controller.
 207     """
 208     fields = (
 209         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'ID'),
 210         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Name'),
 211         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Local IP Address'),
 212         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Local Port'),
 213         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Freeswitch IP Address'),
 214         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Freeswitch Port'))
 215     view_list_fields = 'Name', 'Freeswitch IP Address', 'Freeswitch Port'
 216     servers = []
 217
 218     def __init__(self, parent=None, views=None):
 219         super(ConnectionController, self).__init__(parent, views)
 220
 221         for row in range(self.model.rowCount()):
 222             # Get local IP address and port from the table for each row.
 223             server = http.FreeswitchConfigServer(self)
 224             server.setSocketData(*self.getSocketData(row))
 225             server.startServer()
 226             self.servers.append(server)
 227
 228     def getSocketData(self, row):
 229         local_ip_address = self.model.record(row).value(
 230             'local_ip_address').toString()
 231         local_port, _ok = self.model.record(row).value(
 232             'local_port').toInt()
 233         if not _ok:
 234             local_port = None
 235         return local_ip_address, local_port
 236
 237     def connectionChange(self, index):
 238         """
 239         Restart config server on connection change if necessary.
 240         """
 241         current_row = index.row()
 242         if current_row != -1:
 243             # Select the new row.
 244             connection_id, _ok = index.model().data(
 245                 index.sibling(current_row, 0)).toInt()
 246
 247             # Apply new socket location.
 248             self.servers[current_row].setSocketData(
 249                 *self.getSocketData(current_row))
 250
 251     def objectAdded(self, row, record):
 252         """
 253         New connection added.
 254         """
 255         self.addServer()
 256
 257     def addServer(self, host=None, port=None):
 258         """
 259         Add a new config server.
 260         """
 261         server = http.FreeswitchConfigServer(self)
 262         server.setSocketData(host, port)
 263         server.startServer()
 264         self.servers.append(server)
 265
 266
 267 class ConnectionChangeListenerController(BaseController):
 268     """
 269     Mixin class for reacting on connection change.
 270     """
 271     def connectionChange(self, index):
 272         """
 273         Connection change handler.
 274
 275         Filters table by a new connection ID and stores last connection ID
 276         locally.
 277         """
 278         index_row = index.row()
 279         if index_row != -1:
 280             # Get connection_id field value.
 281             connection_id, _ok = index.model().data(
 282                 index.sibling(index.row(), 0)).toInt()
 283             self.connection_id = connection_id
 284
 285             # Filter is customizable in order to allow ugly hacks :-)
 286             self.model.setFilter(self.getFilter(connection_id))
 287
 288             # Select first row.
 289             self.view_list.selectRow(0)
 290
 291             # Create a new object if none exist.
 292             if not self.model.rowCount():
 293                 self.add()
 294
 295     def getFilter(self, connection_id):
 296         return 'ipypbxweb_%s.connection_id = %i' % (
 297             self.basename, connection_id)
 298
 299     def objectAdded(self, row, record):
 300         """
 301         Set connection_id from currently selected connection.
 302         """
 303         record.setValue('connection_id', self.connection_id)
 304
 305
 306 class SipProfileController(ConnectionChangeListenerController):
 307     """
 308     SIP Profile controller.
 309     """
 310     fields = (
 311         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'ID'),
 312         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Connection ID'),
 313         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Name'),
 314         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'External RTP IP'),
 315         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'External SIP IP'),
 316         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'RTP IP'),
 317         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'SIP IP'),
 318         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'SIP Port'),
 319         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Accept Blind Registration'),
 320         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Authenticate Calls'),
 321         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Is Active'))
 322     view_list_fields = 'Name', 'SIP IP', 'SIP Port'
 323
 324
 325 class DomainController(ConnectionChangeListenerController):
 326     """
 327     Domain controller.
 328     """
 329     fields = (
 330         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'ID'),
 331         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Connection ID'),
 332         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'SIP Profile ID'),
 333         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Host Name'),
 334         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Is Active'))
 335     view_list_fields = 'SIP Profile ID', 'Host Name'
 336     relations = (('sip_profile_id', 'SIP Profile ID', 'sipprofile', 'name'),)
 337
 338
 339 class GatewayController(ConnectionChangeListenerController):
 340     """
 341     Gateway controller.
 342     """
 343     fields = (
 344         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'ID'),
 345         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Connection ID'),
 346         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'SIP Profile ID'),
 347         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Name'),
 348         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Username'),
 349         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Password'),
 350         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Realm'),
 351         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'From Domain'),
 352         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Extension'),
 353         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Expire In Seconds'),
 354         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Retry In Seconds'),
 355         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Caller ID In From Field'),
 356         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Is Active'))
 357     view_list_fields = 'SIP Profile ID', 'Name'
 358     relations = (('sip_profile_id', 'SIP Profile ID', 'sipprofile', 'name'),)
 359
 360
 361 class EndpointController(ConnectionChangeListenerController):
 362     """
 363     Endpoint controller.
 364     """
 365     fields = (
 366         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'ID'),
 367         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Connection ID'),
 368         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'User ID'),
 369         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Password'),
 370         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Domain ID'),
 371         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Is Active'))
 372     view_list_fields = 'User ID', 'Domain ID'
 373     relations = (('domain_id', 'Domain ID', 'domain', 'host_name'),)
 374
 375
 376 class ExtensionController(ConnectionChangeListenerController):
 377     """
 378     Extension controller.
 379     """
 380     fields = (
 381         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'ID'),
 382         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Connection ID'),
 383         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Destination Match'),
 384         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'XML Dialplan'),
 385         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Domain ID'),
 386         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Endpoint ID'),
 387         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Authenticate Calls'),
 388         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Is Active'))
 389     view_list_fields = 'Destination Match',
 390     view_display_fields_hidden = 'ID', 'Connection ID', 'Endpoint ID'
 391     relations = (
 392         ('domain_id', 'Domain ID', 'domain', 'host_name'),
 393 #        ('endpoint_id', 'Endpoint ID', 'endpoint', 'user_id'),
 394         )
 395
 396     def objectAdded(self, row, record):
 397         record.setValue(
 398             'xml_dialplan', '<action application="echo" data=""/>')
 399         super(ExtensionController, self).objectAdded(row, record)
 400
 401     def getFilter(self, connection_id):
 402         # Workaround for Qt bug:
 403         # http://bugreports.qt.nokia.com/browse/QTBUG-8217 . Apparently they
 404         # don't hurry to fix it.
 405         return '1 = 1) or (ipypbxweb_%s.connection_id = %i' % (
 406             self.basename, connection_id)