-class GotoViennaListModel(QAbstractListModel):
- def __init__(self, objects=None):
- QAbstractListModel.__init__(self)
- if objects is None:
- objects = []
- self._objects = objects
- self.setRoleNames({0: 'modelData'})
-
- def set_objects(self, objects):
- self._objects = objects
-
- def get_objects(self):
- return self._objects
-
- def get_object(self, index):
- return self._objects[index.row()]
-
- def rowCount(self, parent=QModelIndex()):
- return len(self._objects)
+class DepartureModel(QAbstractListModel):
+ LINE_ROLE = Qt.UserRole + 1
+ DIRECTION_ROLE = Qt.UserRole + 2
+ STATION_ROLE = Qt.UserRole + 3
+ TIME_ROLE = Qt.UserRole + 4
+ LOWFLOOR_ROLE = Qt.UserRole + 5
+
+ def __init__(self, parent=None):
+ super(DepartureModel, self).__init__(parent)
+ self._data = []
+
+ self.keys = {}
+ self.keys[DepartureModel.LINE_ROLE] = 'line'
+ self.keys[DepartureModel.DIRECTION_ROLE] = 'direction'
+ self.keys[DepartureModel.STATION_ROLE] = 'station'
+ self.keys[DepartureModel.TIME_ROLE] = 'time'
+ self.keys[DepartureModel.LOWFLOOR_ROLE] = 'lowfloor'
+ self.setRoleNames(self.keys)
+
+ def rowCount(self, index):
+ return len(self._data)