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.parent(), )
 224             server.setSocketData(*self.getSocketData(row))
 225             server.startServer()
 226             self.servers.append(server)
 227
 228     def getSocketData(self, row):
 229         """
 230         Return socket data for given row number.
 231         """
 232         record = self.model.record(row)
 233
 234         # Local IP address.
 235         local_ip_address = record.value('local_ip_address').toString()
 236
 237         # Local port.
 238         local_port, _ok = record.value('local_port').toInt()
 239         if not _ok:
 240             local_port = None
 241
 242         # Connection ID.
 243         connection_id, _ok = record.value('id').toInt()
 244         if not _ok:
 245             connection_id = None
 246
 247         return local_ip_address, local_port, connection_id
 248
 249     def connectionChange(self, index):
 250         """
 251         Restart config server on connection change if necessary.
 252         """
 253         current_row = index.row()
 254         if current_row != -1:
 255             # Select the new row.
 256             connection_id, _ok = index.model().data(
 257                 index.sibling(current_row, 0)).toInt()
 258
 259             # Apply new socket location.
 260             self.servers[current_row].setSocketData(
 261                 *self.getSocketData(current_row))
 262
 263     def objectAdded(self, row, record):
 264         """
 265         New connection added.
 266         """
 267         self.addServer(*self.getSocketData(row))
 268
 269     def addServer(self, host=None, port=None, connection_id=None):
 270         """
 271         Add a new config server.
 272         """
 273         server = http.FreeswitchConfigServer(self.model)
 274         server.setSocketData(host, port, connection_id)
 275         server.startServer()
 276         self.servers.append(server)
 277
 278
 279 class ConnectionChangeListenerController(BaseController):
 280     """
 281     Mixin class for reacting on connection change.
 282     """
 283     def connectionChange(self, index):
 284         """
 285         Connection change handler.
 286
 287         Filters table by a new connection ID and stores last connection ID
 288         locally.
 289         """
 290         index_row = index.row()
 291         if index_row != -1:
 292             # Get connection_id field value.
 293             connection_id, _ok = index.model().data(
 294                 index.sibling(index.row(), 0)).toInt()
 295             self.connection_id = connection_id
 296
 297             # Filter is customizable in order to allow ugly hacks :-)
 298             self.model.setFilter(self.getFilter(connection_id))
 299
 300             # Select first row.
 301             self.view_list.selectRow(0)
 302
 303             # Create a new object if none exist.
 304             if not self.model.rowCount():
 305                 self.add()
 306
 307     def getFilter(self, connection_id):
 308         return 'ipypbxweb_%s.connection_id = %i' % (
 309             self.basename, connection_id)
 310
 311     def objectAdded(self, row, record):
 312         """
 313         Set connection_id from currently selected connection.
 314         """
 315         record.setValue('connection_id', self.connection_id)
 316
 317
 318 class SipProfileController(ConnectionChangeListenerController):
 319     """
 320     SIP Profile controller.
 321     """
 322     fields = (
 323         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'ID'),
 324         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Connection ID'),
 325         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Name'),
 326         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'External RTP IP'),
 327         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'External SIP IP'),
 328         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'RTP IP'),
 329         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'SIP IP'),
 330         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'SIP Port'),
 331         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Accept Blind Registration'),
 332         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Authenticate Calls'),
 333         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Is Active'))
 334     view_list_fields = 'Name', 'SIP IP', 'SIP Port'
 335
 336
 337 class DomainController(ConnectionChangeListenerController):
 338     """
 339     Domain controller.
 340     """
 341     fields = (
 342         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'ID'),
 343         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Connection ID'),
 344         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'SIP Profile ID'),
 345         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Host Name'),
 346         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Is Active'))
 347     view_list_fields = 'SIP Profile ID', 'Host Name'
 348     relations = (('sip_profile_id', 'SIP Profile ID', 'sipprofile', 'name'),)
 349
 350
 351 class GatewayController(ConnectionChangeListenerController):
 352     """
 353     Gateway controller.
 354     """
 355     fields = (
 356         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'ID'),
 357         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Connection ID'),
 358         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'SIP Profile ID'),
 359         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Name'),
 360         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Username'),
 361         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Password'),
 362         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Realm'),
 363         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'From Domain'),
 364         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Extension'),
 365         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Expire In Seconds'),
 366         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Retry In Seconds'),
 367         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Caller ID In From Field'),
 368         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Is Active'))
 369     view_list_fields = 'SIP Profile ID', 'Name'
 370     relations = (('sip_profile_id', 'SIP Profile ID', 'sipprofile', 'name'),)
 371
 372
 373 class EndpointController(ConnectionChangeListenerController):
 374     """
 375     Endpoint controller.
 376     """
 377     fields = (
 378         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'ID'),
 379         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Connection ID'),
 380         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'User ID'),
 381         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Password'),
 382         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Domain ID'),
 383         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Is Active'))
 384     view_list_fields = 'User ID', 'Domain ID'
 385     relations = (('domain_id', 'Domain ID', 'domain', 'host_name'),)
 386
 387
 388 class ExtensionController(ConnectionChangeListenerController):
 389     """
 390     Extension controller.
 391     """
 392     fields = (
 393         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'ID'),
 394         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Connection ID'),
 395         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Destination Match'),
 396         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'XML Dialplan'),
 397         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Domain ID'),
 398         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Endpoint ID'),
 399         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Authenticate Calls'),
 400         QtCore.QT_TRANSLATE_NOOP('MainWindow', 'Is Active'))
 401     view_list_fields = 'Destination Match',
 402     view_display_fields_hidden = 'ID', 'Connection ID', 'Endpoint ID'
 403     relations = (
 404         ('domain_id', 'Domain ID', 'domain', 'host_name'),
 405 #        ('endpoint_id', 'Endpoint ID', 'endpoint', 'user_id'),
 406         )
 407
 408     def objectAdded(self, row, record):
 409         record.setValue(
 410             'xml_dialplan', '<action application="echo" data=""/>')
 411         super(ExtensionController, self).objectAdded(row, record)
 412
 413     def getFilter(self, connection_id):
 414         # Workaround for Qt bug:
 415         # http://bugreports.qt.nokia.com/browse/QTBUG-8217 . Apparently they
 416         # don't hurry to fix it.
 417         return '1 = 1) or (ipypbxweb_%s.connection_id = %i' % (
 418             self.basename, connection_id)