def restoreGeometry(self, geometry):
"""
Restores the geometry of this subwindow
:param geometry: the saved state as a QByteArray instance
:return:
"""
if geometry.size() < 4:
return False
stream = QDataStream(geometry)
if stream.readUInt32() != 0x1D9D0CB:
return False
if stream.readUInt16() != 1:
return False
stream.readUInt16() # minorVersion is ignored.
x = stream.readInt64()
y = stream.readInt64()
width = stream.readInt64()
height = stream.readInt64()
restoredFrameGeometry = QRect(x, y, width, height)
x = stream.readInt64()
y = stream.readInt64()
width = stream.readInt64()
height = stream.readInt64()
restoredNormalGeometry = QRect(x, y, width, height)
maximized = stream.readUInt32()
fullScreen = stream.readUInt32()
frameHeight = 20
if not restoredFrameGeometry.isValid():
restoredFrameGeometry = QRect(QPoint(0, 0), self.sizeHint())
if not restoredNormalGeometry.isValid():
restoredNormalGeometry = QRect(QPoint(0, frameHeight),
self.sizeHint())
restoredFrameGeometry.moveTop(max(restoredFrameGeometry.top(), 0))
restoredNormalGeometry.moveTop(
max(restoredNormalGeometry.top(), 0 + frameHeight))
if maximized or fullScreen:
self.setGeometry(restoredNormalGeometry)
ws = self.windowState()
if maximized:
ws |= Qt.WindowMaximized
if fullScreen:
ws |= Qt.WindowFullScreen
self.setWindowState(ws)
else:
offset = QPoint()
self.setWindowState(self.windowState()
& ~(Qt.WindowMaximized | Qt.WindowFullScreen))
self.move(restoredFrameGeometry.topLeft() + offset)
self.resize(restoredNormalGeometry.size())
return True
def start(self):
self._sender = self._socket.peerAddress()
stream = QDataStream(self._socket)
if (self._file_size == 0):
if (self._socket.bytesAvailable() < getsizeof(int)):
return
self._file_size = stream.readUInt32()
if (self._socket.bytesAvailable() < self._file_size):
return
self._file_name = stream.readString()
line = QByteArray()
line = self._socket.readAll()
directory = Path(Setup().get_download_dir())
if not (directory.exists() and directory.is_dir()):
directory.mkdir()
self._receiving_file = QFile(Setup().get_download_dir() +
str(self._file_name))
print(Setup().get_download_dir() + str(self._file_name))
if not (self._receiving_file.open(QIODevice.WriteOnly)):
print("can't open file")
return
self._receiving_file.write(line)
self._receiving_file.close()
self._socket.disconnectFromHost()
print("finished")
return self._receiving_file
class QDataStreamWrite(unittest.TestCase):
'''Test case for QDatastream write* functions'''
def setUp(self):
self.ba = QByteArray()
self.read = QDataStream(self.ba, QIODevice.ReadOnly)
self.write = QDataStream(self.ba, QIODevice.WriteOnly)
def testWriteUInt8(self):
'''QDataStream.writeUInt8 (accepting str of size 1)'''
x = 0xFF
self.write.writeUInt8(x)
y = self.read.readUInt8()
self.assertEqual(x, y)
self.assertRaises(TypeError, self.write.writeUInt8, 'aaaaa')
def testWriteInt8(self):
'''QDataStream.writeInt8 (accepting str of size 1)'''
x = 65
self.write.writeInt8(chr(x))
y = self.read.readInt8()
self.assertEqual(x, y)
self.assertRaises(TypeError, self.write.writeInt8, 'aaaaa')
def testWriteUInt8Int(self):
'''QDataStream.writeUInt8 (accepting integer)'''
x = 0xFF
self.write.writeUInt8(x)
y = self.read.readUInt8()
self.assertEqual(x, y)
def testWriteInt8Int(self):
'''QDataStream.writeInt8 (accepting integer)'''
x = 65
self.write.writeInt8(x)
y = self.read.readInt8()
self.assertEqual(x, y)
def testWriteUInt16(self):
'''QDataStream.writeUInt16'''
x = 0x4423
self.write.writeUInt16(x)
y = int(self.read.readUInt16())
self.assertEqual(x, y)
def testWriteUInt32(self):
'''QDataStream.writeUInt32'''
x = 0xdeadbeef
self.write.writeUInt32(x)
y = int(self.read.readUInt32())
self.assertEqual(x, y)
class QDataStreamWrite(unittest.TestCase):
'''Test case for QDatastream write* functions'''
def setUp(self):
self.ba = QByteArray()
self.read = QDataStream(self.ba, QIODevice.ReadOnly)
self.write = QDataStream(self.ba, QIODevice.WriteOnly)
def testWriteUInt8(self):
'''QDataStream.writeUInt8 (accepting str of size 1)'''
x = 0xFF
self.write.writeUInt8(x)
y = self.read.readUInt8()
self.assertEqual(x, y)
self.assertRaises(TypeError, self.write.writeUInt8, 'aaaaa')
def testWriteInt8(self):
'''QDataStream.writeInt8 (accepting str of size 1)'''
x = 65
self.write.writeInt8(chr(x))
y = self.read.readInt8()
self.assertEqual(x, y)
self.assertRaises(TypeError, self.write.writeInt8, 'aaaaa')
def testWriteUInt8Int(self):
'''QDataStream.writeUInt8 (accepting integer)'''
x = 0xFF
self.write.writeUInt8(x)
y = self.read.readUInt8()
self.assertEqual(x, y)
def testWriteInt8Int(self):
'''QDataStream.writeInt8 (accepting integer)'''
x = 65
self.write.writeInt8(x)
y = self.read.readInt8()
self.assertEqual(x, y)
def testWriteUInt16(self):
'''QDataStream.writeUInt16'''
x = 0x4423
self.write.writeUInt16(x)
y = int(self.read.readUInt16())
self.assertEqual(x, y)
def testWriteUInt32(self):
'''QDataStream.writeUInt32'''
x = 0xdeadbeef
self.write.writeUInt32(x)
y = int(self.read.readUInt32())
self.assertEqual(x, y)
