Python-сообщество

class  srcTableModel(QtCore.QAbstractTableModel):
    
    def __init__(self, fileName = QtCore.QString()):
        super (srcTableModel, self).__init__()
        self.ships = []
        self.fileName = fileName 
        self.dirty = True


    def rowCount (self, index=QtCore.QModelIndex()):
        return len(self.ships)



    def columnCount (self, index=QtCore.QModelIndex()):
        return 11


    def headerData(self, section, orientation, role=QtCore.Qt.DisplayRole):
        if role != QtCore.Qt.DisplayRole:
            return
        if orientation == QtCore.Qt.Horizontal:
            if section == EQUIPMENT:
                return (QtGui.QApplication.translate("MainWindow", "Виробництво", None, QtGui.QApplication.UnicodeUTF8))
            elif section == SRCCODE:
                return (QtGui.QApplication.translate("MainWindow", "№ дж.", None, QtGui.QApplication.UnicodeUTF8))
            elif section == SRCNAME:
                return (QtGui.QApplication.translate("MainWindow", "Найменув. дж.", None, QtGui.QApplication.UnicodeUTF8))
            elif section == SRCX1:
                return (QtGui.QApplication.translate("MainWindow", "X,м", None, QtGui.QApplication.UnicodeUTF8))
            elif section == SRCY1:
                return (QtGui.QApplication.translate("MainWindow", "Y,м", None, QtGui.QApplication.UnicodeUTF8))
            elif section == SRCTYPE:
                return (QtGui.QApplication.translate("MainWindow", "Тип дж.", None, QtGui.QApplication.UnicodeUTF8))
            elif section == ZONESIZE:
                return (QtGui.QApplication.translate("MainWindow", "Розмір СЗЗ", None, QtGui.QApplication.UnicodeUTF8))
            elif section == SRCHEIGHT:
                return (QtGui.QApplication.translate("MainWindow", "Висота дж., м", None, QtGui.QApplication.UnicodeUTF8))
            elif section == SRCDIAMETER:
                return (QtGui.QApplication.translate("MainWindow", "Діаметер дж., м", None, QtGui.QApplication.UnicodeUTF8))
            elif section == GASDEPLETION:
                return (QtGui.QApplication.translate("MainWindow", "Витрата ПГВС, м3/с", None, QtGui.QApplication.UnicodeUTF8))
            elif section == GASTEMPERATURE:
                return (QtGui.QApplication.translate("MainWindow", "Темпер. ПГВС", None, QtGui.QApplication.UnicodeUTF8))

        return int(section + 1)


    def data (self, index, role=QtCore.Qt.DisplayRole):
        if index.isValid() and 0 <= index.row() < len(self.ships):
            ship = self.ships [index.row()]
            column = index.column()
            if role == QtCore.Qt.DisplayRole:
                if column == EQUIPMENT:
                    return QtCore.QVariant(ship.equipment)
                elif column == SRCCODE:
                    return QtCore.QVariant(ship.srcCode)
                elif column == SRCNAME:
                    return QtCore.QVariant(ship.srcName)
                elif column == SRCX1:
                    return QtCore.QVariant(ship.srcX1)
                elif column == SRCY1:
                    return QtCore.QVariant(ship.srcY1)
                elif column == SRCTYPE:
                    return QtCore.QVariant(ship.srcType)
                elif column == ZONESIZE:
                    return QtCore.QVariant(ship.zoneSize)
                elif column == SRCHEIGHT:
                    return QtCore.QVariant(ship.srcHeight)
                elif column == SRCDIAMETER:
                    return QtCore.QVariant(ship.srcDiameter)
                elif column == GASDEPLETION:
                    return QtCore.QVariant(ship.gasDepletion)
                elif column == GASTEMPERATURE:
                    return QtCore.QVariant(ship.gasTemperature)
        return

    
    def flags(self, index):
        if not index.isValid():
            return QtCore.Qt.ItemIsEnabled
        return QtCore.Qt.ItemFlags(QtCore.QAbstractTableModel.flags(self, index)|QtCore.Qt.ItemIsEditable)

      
    def setData (self, index, value, role=QtCore.Qt.DisplayRole):
        if index.isValid() and 0 <= index.row() < len (self.ships):
            ship = self.ships [index.row()]
            column = index.column()
            if column == EQUIPMENT:
                ship.equipment = value.toString()
            elif column == SRCCODE:
                value, ok =  value.toInt()
		if ok:
		    ship.srcCode = value
            elif column == SRCNAME:
                ship.srcName = value.toString()
	    elif column == SRCX1:
                value, ok =  value.toInt()
		if ok:
		    ship.srcX1 = value
	    elif column == SRCY1:
                value, ok =  value.toInt()
		if ok:
		    ship.srcY1 = value
            elif column == SRCTYPE:
                ship.srcType = value.toString()
            elif column == ZONESIZE:
                value, ok = value.toInt()
		if ok:
		    ship.zoneSize = value
	    elif column == SRCHEIGHT:
                value, ok = value.toInt()
		if ok:
		    ship.srcHeight = value
	    elif column == SRCDIAMETER:
                value, ok = value.toInt()
		if ok:
		    ship.srcDiameter = value	   
	    elif column == GASDEPLETION:
                value, ok = value.toInt()
		if ok:
		    ship.gasDepletion = value	    
	    elif column == GASTEMPERATURE:
                value, ok = value.toInt()
		if ok:
		    ship.gasTemperature = value
            self.dirty = True
            self.emit(QtCore.SIGNAL("dataChanged(QModelIndex, QModelIndex)"), index, index)
            return True
        return False     


    def insertRows (self, position, rows=1, index=QtCore.QModelIndex()):
        self.beginInsertRows(QtCore.QModelIndex(), position, position + rows - 1)
        for row in range(rows):
            self.ships.insert(position + row, srcShip("Unknown", 0, "Unknown", 0, 0, "Unknown", 0, 0, 0, 0, 0))
        self.endInsertRows()
        self.dirty = True
        return True


    def removeRows(self, position, rows=1, index=QtCore.QModelIndex()):
        self.beginRemoveRows(QtCore.QModelIndex(), position, position + rows - 1)
        self.ships = self.ships[:position] + self.ships [position + rows:]
        self.endRemoveRows()
        self.dirty = True
        return True

    def loadFile(self, fileName):
        exception = None
        fh = None
        self.fileName = QtCore.QString(fileName)
	try:
            if self.fileName.isEmpty():
                raise IOError, "no fileName specified for loading"
            fh = QtCore.QFile(self.fileName)
            if not fh.open(QtCore.QIODevice.ReadOnly):
                raise IOError, unicode(fh.errorString())
            stream = QtCore.QDataStream(fh)
            magic = stream.readInt32()
            if magic != MAGIC_NUMBER:
                raise IOError, "unrecognized file type"
            fileVersion = stream.readInt16()
            if fileVersion != FILE_VERSION:
                raise IOError, "unrecognized file type version"
            stream.setVersion(QtCore.QDataStream.Qt_4_5)
	    self.ships = []
            while not stream.atEnd():
                equipment = QtCore.QString()
                stream >> equipment 
                srcCode = stream.readInt32()
		srcName = QtCore.QString()
		stream >> srcName
		srcX1 = stream.readInt32()
		srcY1 = stream.readInt32()
		srcType = QtCore.QString()
		stream >> srcType
		zoneSize = stream.readInt32()
		srcHeight = stream.readInt32()
		srcDiameter = stream.readInt32()
		gasDepletion = stream.readInt32()
		gasTemperature = stream.readInt32()
		self.ships.append(srcShip(equipment, srcCode, srcName, srcX1, srcY1, srcType, zoneSize, srcHeight, srcDiameter, gasDepletion, gasTemperature))
            self.reset()
	    self.dirty = False
        except (IOError, OSError), e:
            exception = e
        finally:
            if fh is not None:
                fh.close()
            if exception is not None:
                raise exception


    def saveFile(self):
        exception = None
        fh = None
        try:
            if self.fileName.isEmpty():
		self.fileName = QtGui.QFileDialog.getSaveFileName(None, "Ships - Choose Save File", "", "Trianon project (*.tri)")
            fh = QtCore.QFile(self.fileName)
            if not fh.open(QtCore.QIODevice.WriteOnly):
                raise IOError, unicode(fh.errorString())
            stream = QtCore.QDataStream(fh)
            stream.writeInt32(MAGIC_NUMBER)
            stream.writeInt16(FILE_VERSION)
            stream.setVersion(QtCore.QDataStream.Qt_4_5)
            for ship in self.ships:
                stream << QtCore.QString(ship.equipment)  
                stream.writeInt32(ship.srcCode)
		stream << QtCore.QString(ship.srcName)
		stream.writeInt32(ship.srcX1)
		stream.writeInt32(ship.srcY1)
		stream << QtCore.QString(ship.srcType)
		stream.writeInt32(ship.zoneSize)
		stream.writeInt32(ship.srcHeight)
		stream.writeInt32(ship.srcDiameter)
		stream.writeInt32(ship.gasDepletion)
		stream.writeInt32(ship.gasTemperature)
            self.dirty = False
        except IOError, e:
            exception = e
        finally:
            if fh is not None:
                fh.close()
            if exception is not None:
                raise exception