48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
import random
|
|
from aiortc.contrib.media import MediaBlackhole, MediaPlayer, MediaRecorder, MediaRelay
|
|
from aiortc import VideoStreamTrack, AudioStreamTrack
|
|
|
|
class MyMediaPlayer(MediaPlayer):
|
|
pass
|
|
|
|
class MyTrackBasea:
|
|
def __init__(self, source=None):
|
|
self.source = source
|
|
self.track = self.set_track()
|
|
self.track.on('ended', self.reload)
|
|
|
|
def set_track(self):
|
|
if self.kind == 'audio':
|
|
self.track = self.source.audio
|
|
else:
|
|
self.track = self.source.video
|
|
|
|
def self.reload(self):
|
|
print(f'reload the source----{self.source._file_path}')
|
|
self.set_source(MyMediaPlayer(self.source._file_path))
|
|
|
|
def set_source(self, source):
|
|
self.source = source
|
|
self.set_track()
|
|
|
|
async def recv(self):
|
|
if self.source is None:
|
|
return None
|
|
f = await self.source.audio.recv()
|
|
while f is None:
|
|
print('MyAudioTrack::recv(): will reload source')
|
|
self.set_source(MyMediaPlayer(self.source._file_path))
|
|
f = await self.source.audio.recv()
|
|
return f
|
|
|
|
|
|
class MyAudioStreamTrack(AudioStreamTrack, MyTrackBasea):
|
|
def __init__(self, source=None):
|
|
AudioStreamTrack.__init__(self)
|
|
MyTrackBase.__init__(self, source=source)
|
|
|
|
class MyVideoStreamTrack(VideoStreamTrack, MyTrackBasea):
|
|
def __init__(self, source=None):
|
|
VideoStreamTrack.__init__(self)
|
|
MyTrackBase.__init__(self, source=source)
|