MediaRecorderFixed30FPS.py

import fractions
from aiortc.contrib.media import MediaRecorder
from aiortc.mediastreams import MediaStreamError


class MediaRecorderFixed30FPS(MediaRecorder):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.video_frame_ctr = {}

    def addTrack(self, track):
        super().addTrack(track)
        if track.kind == "video":
            self.video_frame_ctr[track.id] = 0

    async def __run_track(self, track, context):
        while True:
            try:
                frame = await track.recv()
                if track.kind == 'video':
                    frame.pts = self.video_frame_ctr[track.id] 
                    frame.time_base = fractions.Fraction(1, 30) 
                    self.frame_ctr[track.id] += 1
            except MediaStreamError:
                return
            for packet in context.stream.encode(frame):
                self.__container.mux(packet)