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 #from ipypbx import models
  19 from PyQt4 import QtCore, QtGui, QtSql
  20
  21
  22 class BaseController(QtCore.QObject):
  23     """
  24     Base class for other controllers.
  25
  26     Doesn't do anything useful on its own.
  27     """
  28     fields = ()
  29     view_list_fields = ()
  30     view_display_fields = ()
  31
  32     def __init__(self, model=None, view_list=None, view_display=None, parent=None, views=None):
  33         super(BaseController, self).__init__(parent=parent)
  34
  35         self.views = views
  36
  37         # Find out base name.
  38         classname = self.__class__.__name__
  39         self.basename = (
  40             classname[:-10] if classname.endswith('Controller')
  41             else classname)
  42         self.basename = self.basename[0].lower() + self.basename[1:]
  43
  44         if model:
  45             # We're given an existing model.
  46             self.model = model
  47         else:
  48             # Initialize a new model.
  49             self.model = QtSql.QSqlTableModel(parent)
  50             self.model.setTable(self.basename + 's')
  51
  52             # Create model header from fields list.
  53             for i, field in enumerate(self.fields):
  54                 self.model.setHeaderData(
  55                     i, QtCore.Qt.Horizontal, QtCore.QVariant(field))
  56
  57             # Fetch model data.
  58             self.model.select()
  59
  60         if view_list:
  61             # We're given an existing view list.
  62             self.view_list = view_list
  63         else:
  64             # Get view list from the parent.
  65             self.view_list = getattr(views, self.basename + 'ViewList')
  66             self.view_list.setModel(self.model)
  67
  68             # Hide fields not meant for display.
  69             for i, field in enumerate(self.fields):
  70                 if field not in self.view_list_fields:
  71                     self.view_list.setColumnHidden(i, True)
  72
  73             # Stretch headers to fill all available width.
  74             self.view_list.setSelectionMode(QtGui.QTableView.SingleSelection)
  75             self.view_list.setSelectionBehavior(QtGui.QTableView.SelectRows)
  76             self.view_list.resizeColumnsToContents()
  77             self.view_list.resizeRowsToContents()
  78             self.view_list.horizontalHeader().setStretchLastSection(True)
  79
  80         if view_display:
  81             # We're given an existing view display.
  82             self.view_display = view_display
  83         else:
  84             # Get view display from the parent.
  85             self.view_display = QtGui.QDataWidgetMapper(parent)
  86             #getattr(parent, self.basename + 'ViewDisplay')
  87             self.view_display.setModel(self.model)
  88
  89             # If view_display_fields is not send, display all fields except
  90             # the first one that is usually the ID.
  91             display_fields = self.view_display_fields or self.fields[1:]
  92             for i, field in enumerate(self.fields):
  93                 if field in display_fields:
  94                     field_name = self.basename + ''.join(
  95                         word.capitalize() for word in field.split(' '))
  96                     self.view_display.addMapping(
  97                         getattr(views, field_name), i)
  98
  99         # Register signals for this controller.
 100         for sender, signal, receiver in self.getSignalsData():
 101             QtCore.QObject.connect(sender, QtCore.SIGNAL(signal), receiver)
 102
 103     def getSignalsData(self):
 104         """
 105         Default signals built from controller's base name.
 106         """
 107         return (
 108             (getattr(self.views, self.basename + 'Add'), 'clicked()', self.add),
 109             (getattr(self.views, self.basename + 'ViewList'),
 110              'currentRowChanged(int)', self.select),
 111             (getattr(self.views, self.basename + 'Save'), 'clicked()',
 112              self.save),
 113             )
 114
 115     def add(self):
 116         """
 117         Add new connection.
 118         """
 119         num_rows = self.model.rowCount()
 120         self.model.insertRows(num_rows, 1)
 121         self.view_list.selectRow(num_rows)
 122         getattr(self.views, self.basename + 'Add').setEnabled(False)
 123
 124         #self.parent.ui.connectionName.setText('New connection')
 125         #self.parent.ui.connectionName.setFocus()
 126         #self.parent.ui.connectionName.selectAll()
 127         #self.parent.ui.connectionLocalIpAddress.clear()
 128         #self.parent.ui.connectionLocalPort.clear()
 129         #self.parent.ui.connectionFreeswitchIpAddress.clear()
 130         #self.parent.ui.connectionFreeswitchPort.clear()
 131
 132     def save(self):
 133         """
 134         TODO: Default implementation.
 135         """
 136         return NotImplemented
 137
 138
 139 class ConnectionController(BaseController):
 140     """
 141     Connections handler.
 142     """
 143     fields = (
 144         'ID', 'Name', 'Local IP Address', 'Local Port',
 145         'Freeswitch IP Address', 'Freeswitch Port')
 146     view_list_fields = ('Name', 'Freeswitch IP Address', 'Freeswitch Port')
 147
 148     def select(self, row):
 149         """
 150         Select another connection as current.
 151         """
 152         self.currentConnection = self.connections[row]
 153
 154         # Fill in form based on selection.
 155         self.parent.ui.connectionName.setText(self.currentConnection.name)
 156         self.parent.ui.connectionLocalIpAddress.setText(
 157             self.currentConnection.local_ip_address)
 158         self.parent.ui.connectionLocalPort.setText(
 159             unicode(self.currentConnection.local_port))
 160         self.parent.ui.connectionFreeswitchIpAddress.setText(
 161             self.currentConnection.freeswitch_ip_address)
 162         self.parent.ui.connectionFreeswitchPort.setText(
 163             unicode(self.currentConnection.freeswitch_port))
 164
 165     def clone(self):
 166         """
 167         TODO: Clone an existing connection.
 168
 169         This creates a new connection with bound data copied from another one.
 170         """
 171
 172     def save(self):
 173         """
 174         Save new or existing connection.
 175         """
 176         name = unicode(self.parent.ui.connectionName.text())
 177
 178         # Add to connection list if we've created it.
 179         if self.currentConnection is None:
 180             #self.currentConnection = models.Connection(store=state.store)
 181             self.connections.append(self.currentConnection)
 182             self.parent.ui.connectionList.addItem(name)
 183
 184         self.currentConnection.name = name
 185         self.currentConnection.local_ip_address = unicode(
 186             self.parent.ui.connectionLocalIpAddress.text())
 187         self.currentConnection.local_port = int(
 188             self.parent.ui.connectionLocalPort.text())
 189         self.currentConnection.freeswitch_ip_address = unicode(
 190             self.parent.ui.connectionFreeswitchIpAddress.text())
 191         self.currentConnection.freeswitch_port = int(
 192             self.parent.ui.connectionFreeswitchPort.text())
 193
 194         self.currentConnection.checkpoint()