47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
import random
|
|
from aiortc.contrib.media import MediaBlackhole, MediaPlayer, MediaRecorder, MediaRelay
|
|
from aiortc import MediaStreamTrack, VideoStreamTrack, AudioStreamTrack
|
|
|
|
class MyMediaPlayer(MediaPlayer):
|
|
pass
|
|
|
|
class MyTrackBase(MediaStreamTrack):
|
|
def __init__(self, source=None):
|
|
super().__init__()
|
|
self.reloaded = False
|
|
self.source = source
|
|
self.set_track()
|
|
print(f'{self.kind=}, {self.__class__.__name__}')
|
|
|
|
def set_track(self):
|
|
if self.kind == 'audio':
|
|
self.track = self.source.audio
|
|
else:
|
|
self.track = self.source.video
|
|
self.track.on('ended', self.reload)
|
|
|
|
def reload(self):
|
|
print(f'reload the source----{self.source._file_path}')
|
|
self.set_source(MyMediaPlayer(self.source._file_path))
|
|
self.reloaded = True
|
|
|
|
def set_source(self, source):
|
|
self.source = source
|
|
self.set_track()
|
|
|
|
async def recv(self):
|
|
print(f'{self.__class__.__name__}, {self.source.duration=}, {self.source.time=}')
|
|
if self.source is None:
|
|
return None
|
|
if self.track.readyState != 'live':
|
|
self.reload()
|
|
f = await self.track.recv()
|
|
return f
|
|
|
|
|
|
class MyAudioStreamTrack(MyTrackBase):
|
|
kind = 'audio'
|
|
|
|
class MyVideoStreamTrack(VideoStreamTrack, MyTrackBase):
|
|
kind = 'video'
|