Networking glitchs sort of fixed.

This commit is contained in:
Christopher T. Johnson
2024-04-02 11:02:29 -04:00
parent 07ac20d940
commit 6926fffc22

View File

@@ -3,6 +3,7 @@ from typing import Optional, Self, Type, cast
from PyQt6.QtCore import ( from PyQt6.QtCore import (
QBuffer, QBuffer,
QByteArray, QByteArray,
QCryptographicHash,
QDir, QDir,
QObject, QObject,
QStandardPaths, QStandardPaths,
@@ -18,6 +19,7 @@ from PyQt6.QtMultimedia import (
from PyQt6.QtNetwork import ( from PyQt6.QtNetwork import (
QNetworkAccessManager, QNetworkAccessManager,
QNetworkDiskCache, QNetworkDiskCache,
QNetworkReply,
QNetworkRequest, QNetworkRequest,
) )
@@ -85,6 +87,7 @@ class SoundOff(QObject):
netCache.setCacheDirectory(cacheDir.path()) netCache.setCacheDirectory(cacheDir.path())
self.nam = QNetworkAccessManager() self.nam = QNetworkAccessManager()
self.nam.setCache(netCache) self.nam.setCache(netCache)
self.nam.finished.connect(self.finished)
return return
@pyqtSlot(QMediaPlayer.Error, str) @pyqtSlot(QMediaPlayer.Error, str)
@@ -95,7 +98,7 @@ class SoundOff(QObject):
@pyqtSlot(QMediaPlayer.MediaStatus) @pyqtSlot(QMediaPlayer.MediaStatus)
def mediaStatus(self, status: QMediaPlayer.MediaStatus) -> None: def mediaStatus(self, status: QMediaPlayer.MediaStatus) -> None:
print(f"mediaStatus: {status}") # print(f"mediaStatus: {status}")
if status == QMediaPlayer.MediaStatus.LoadedMedia: if status == QMediaPlayer.MediaStatus.LoadedMedia:
player: Optional[QMediaPlayer] = cast(QMediaPlayer, self.sender()) player: Optional[QMediaPlayer] = cast(QMediaPlayer, self.sender())
assert player is not None assert player is not None
@@ -104,7 +107,7 @@ class SoundOff(QObject):
@pyqtSlot(QMediaPlayer.PlaybackState) @pyqtSlot(QMediaPlayer.PlaybackState)
def playbackState(self, state: QMediaPlayer.PlaybackState) -> None: def playbackState(self, state: QMediaPlayer.PlaybackState) -> None:
print(f"playbackState: {state}") # print(f"playbackState: {state}")
return return
# #
@@ -121,10 +124,19 @@ class SoundOff(QObject):
url = QUrl(url) url = QUrl(url)
if not self.localPlayer.audioOutput(): if not self.localPlayer.audioOutput():
self.localPlayer.setAudioOutput(self.localOutput) self.localPlayer.setAudioOutput(self.localOutput)
if self.virtualPlayer and not self.virtualPlayer.audioOutput():
self.virtualPlayer.setAudioOutput(self.virtualOutput)
if url != self._lastUrl:
request = QNetworkRequest(url) request = QNetworkRequest(url)
self.reply = self.nam.get(request) self.nam.get(request)
assert self.reply is not None self._lastUrl = url
self.reply.finished.connect(self.finished) return
for player in [self.localPlayer, self.virtualPlayer]:
if not player:
continue
player.setPosition(0)
if player.mediaStatus() == QMediaPlayer.MediaStatus.LoadedMedia:
player.play()
return return
@pyqtSlot() @pyqtSlot()
@@ -135,33 +147,27 @@ class SoundOff(QObject):
# #
# Network slots # Network slots
# #
_buffer = QBuffer() _storage: dict[QMediaPlayer, QByteArray] = {}
_storage = QByteArray() _buffer: dict[QMediaPlayer, QBuffer] = {}
_lastUrl = QUrl()
@pyqtSlot() @pyqtSlot(QNetworkReply)
def finished(self) -> None: def finished(self, reply: QNetworkReply) -> None:
assert self.reply is not None storage = reply.readAll()
self._storage = self.reply.readAll()
if self.localPlayer.sourceDevice() != self._buffer: crypto = QCryptographicHash(QCryptographicHash.Algorithm.Sha256)
self._buffer.setBuffer(self._storage) for player in [self.localPlayer, self.virtualPlayer]:
self.localPlayer.setSourceDevice(self._buffer) if not player:
self.localPlayer.setPosition(0) continue
if ( self._storage[player] = QByteArray(storage)
self.localPlayer.mediaStatus() crypto.addData(self._storage[player])
== QMediaPlayer.MediaStatus.LoadedMedia print(player, crypto.result().toHex())
): crypto.reset()
self.localPlayer.play() self._buffer[player] = QBuffer(self._storage[player])
url = reply.request().url()
player.setSourceDevice(self._buffer[player], url)
player.setPosition(0)
if player.mediaStatus() == QMediaPlayer.MediaStatus.LoadedMedia:
player.play()
print("play") print("play")
if not self.virtualDevice:
return
if self.virtualPlayer.sourceDevice() != self._buffer:
self.virtualPlayer.setSourceDevice(self._buffer)
self.virtualPlayer.setPosition(0)
if not self.virtualPlayer.audioOutput():
self.virtualPlayer.setAudioOutput(self.virtualOutput)
if (
self.virtualPlayer.mediaStatus()
== QMediaPlayer.MediaStatus.LoadedMedia
):
self.virtualPlayer.play()
return return