ffmpeg_streaming._media_streams

Complexity

Total Complexity

12

Size/Duplication

Total Lines	47
Duplicated Lines	0 %

Importance

Changes

0

9 Methods

Rating	Name	Duplication	Size	Complexity
A	Streams._get_streams()	0	4	3
A	Streams.video()	0	2	1
A	Streams.all()	0	2	1
A	Streams.first_stream()	0	2	1
A	Streams._get_stream()	0	5	2
A	Streams.videos()	0	2	1
A	Streams.__init__()	0	2	1
A	Streams.audio()	0	2	1
A	Streams.audios()	0	2	1

"""
ffmpeg_streaming.streams
~~~~~~~~~~~~

Parse streams that is the output of the FFProbe


:copyright: (c) 2020 by Amin Yazdanpanah.
:website: https://www.aminyazdanpanah.com
:email: contact@aminyazdanpanah.com
:license: MIT, see LICENSE for more details.
"""


class Streams:
    def __init__(self, streams):
        self.streams = streams

    def video(self):
        return self._get_stream('video')

    def audio(self):
        return self._get_stream('audio')

    def first_stream(self):
        return self.streams[0]

    def all(self):
        return self.streams

    def videos(self):
        return self._get_streams('video')

    def audios(self):
        return self._get_streams('audio')

    def _get_stream(self, media):
        media_attr = next((stream for stream in self.streams if stream['codec_type'] == media), None)

The variable stream does not seem to be defined.

        if media_attr is None:
            raise ValueError('No ' + str(media) + ' stream found')
        return media_attr

    def _get_streams(self, media):
        for stream in self.streams:
            if stream['codec_type'] == media:
                yield stream