o ٷi('@sUddlmZddlZddlZddlZddlZddlmZmZm Z m Z z ddl m Z de d<Wn ey;ej ZYnwe ZGdddZGd d d eZGd d d eZGd ddeZ d"d#ddZd"d$ddZd%d d!ZdS)&) annotationsN)IOAnyMappingSequence)Popenztype[subprocess.Popen | Popen]popenc@sBeZdZdZ    ddd dZdddZ    d d!ddZdS)"FFmpegzhWrapper for various `FFmpeg `_ related applications (ffmpeg, ffprobe). ffmpegN executablestrglobal_optionsSequence[str] | str | Noneinputs/Mapping[str, Sequence[str] | str | None] | NoneoutputsreturnNonecCst||_|g|_|jt|dd7_|dur!|jt|dd7_|dur.|jt|7_t|j|_d|_dS)aJInitialize FFmpeg command line wrapper. Compiles FFmpeg command line from passed arguments (executable path, options, inputs and outputs). ``inputs`` and ``outputs`` are dictionares containing inputs/outputs as keys and their respective options as values. One dictionary value (set of options) must be either a single space separated string, or a list or strings without spaces (i.e. each part of the option is a separate item of the list, the result of calling ``split()`` on the options string). If the value is a list, it cannot be mixed, i.e. cannot contain items with spaces. An exception are complex FFmpeg command lines that contain quotes: the quoted part must be one string, even if it contains spaces (see *Examples* for more info). For more info about FFmpeg command line format see `here `_. :param str executable: path to ffmpeg executable; by default the ``ffmpeg`` command will be searched for in the ``PATH``, but can be overridden with an absolute path to ``ffmpeg`` executable :param iterable global_options: global options passed to ``ffmpeg`` executable (e.g. ``-y``, ``-v`` etc.); can be specified either as a list/tuple/set of strings, or one space-separated string; by default no global options are passed :param dict inputs: a dictionary specifying one or more input arguments as keys with their corresponding options (either as a list of strings or a single space separated string) as values :param dict outputs: a dictionary specifying one or more output arguments as keys with their corresponding options (either as a list of strings or a single space separated string) as values T) split_mixedN)add_minus_i_option)r _cmd_normalize_options_merge_args_opts subprocess list2cmdlinecmdprocess)selfr r rrr?/home/ubuntu/.local/lib/python3.10/site-packages/ffmpy/ffmpy.py__init__s! zFFmpeg.__init__cCsd|jjd|jdS)N< >) __class____name__r)rrrr__repr__FszFFmpeg.__repr__ input_data bytes | NonestdoutIO | int | NonestderrenvMapping[str, str] | Nonekwargsr!tuple[bytes | None, bytes | None]c Kszt|jftj|||d||_Wnty.}z|jtjkr)td|j dd}~ww|jj |d\}}|jj dkrHt |j |jj ||||fS)aExecute FFmpeg command line. ``input_data`` can contain input for FFmpeg in case ``pipe`` protocol is used for input. ``stdout`` and ``stderr`` specify where to redirect the ``stdout`` and ``stderr`` of the process. By default no redirection is done, which means all output goes to running shell (this mode should normally only be used for debugging purposes). If FFmpeg ``pipe`` protocol is used for output, ``stdout`` must be redirected to a pipe by passing `subprocess.PIPE` as ``stdout`` argument. You can pass custom environment to ffmpeg process with ``env``. Returns a 2-tuple containing ``stdout`` and ``stderr`` of the process. If there was no redirection or if the output was redirected to e.g. `os.devnull`, the value returned will be a tuple of two `None` values, otherwise it will contain the actual ``stdout`` and ``stderr`` data returned by ffmpeg process. More info about ``pipe`` protocol `here `_. :param str input_data: input data for FFmpeg to deal with (audio, video etc.) as bytes (e.g. the result of reading a file in binary mode) :param stdout: redirect FFmpeg ``stdout`` there (default is `None` which means no redirection) :param stderr: redirect FFmpeg ``stderr`` there (default is `None` which means no redirection) :param env: custom environment for ffmpeg process :param kwargs: any other keyword arguments to be forwarded to `subprocess.Popen `_ :return: a 2-tuple containing ``stdout`` and ``stderr`` of the process :rtype: tuple :raise: `FFRuntimeError` in case FFmpeg command exits with a non-zero code; `FFExecutableNotFoundError` in case the executable path passed was not valid )stdinr)r+r,z Executable 'z ' not foundN)inputr)rrrPIPErOSErrorerrnoENOENTFFExecutableNotFoundErrorr communicate returncodeFFRuntimeErrorr) rr'r)r+r,r.eo_stdouto_stderrrrrrunIs$&    z FFmpeg.run)r NNN) r r r rrrrrrr)rr )NNNN) r'r(r)r*r+r*r,r-r.rrr/)r% __module__ __qualname____doc__r r&r=rrrrr s .r cs*eZdZdZ   ddfd d ZZS)FFprobez=Wrapper for `ffprobe `_.ffprobeNr r r rrrrrcstj|||ddS)aCreate an instance of FFprobe. Compiles FFprobe command line from passed arguments (executable path, options, inputs). FFprobe executable by default is taken from ``PATH`` but can be overridden with an absolute path. For more info about FFprobe command line format see `here `_. :param str executable: absolute path to ffprobe executable :param iterable global_options: global options passed to ffmpeg executable; can be specified either as a list/tuple of strings or a space-separated string :param dict inputs: a dictionary specifying one or more inputs as keys with their corresponding options as values )r r rN)superr )rr r rr$rrr szFFprobe.__init__)rBNN)r r r rrrrrr%r>r?r@r __classcell__rrrDrrAs rAc@seZdZdZdS)r6z3Raise when FFmpeg/FFprobe executable was not found.N)r%r>r?r@rrrrr6sr6cs"eZdZdZd fd d ZZS)r9zRaise when FFmpeg/FFprobe command line execution returns a non-zero exit code. The resulting exception object will contain the attributes relates to command line execution: ``cmd``, ``exit_code``, ``stdout``, ``stderr``. rr exit_codeintr)bytes | str | Noner+rrcsJ||_||_||_||_t|}t|}d|j|||}t|dS)Nz2`{}` exited with status {} STDOUT: {} STDERR: {})rrGr)r+ _safe_decodeformatrCr )rrrGr)r+stdout_displaystderr_displaymessagerDrrr szFFRuntimeError.__init__) rr rGrHr)rIr+rIrrrErrrDrr9sr9Fargs_opts_dict(Mapping[str, Sequence[str] | str | None]rboolr list[str]cCsDg}|D]\}}|t|7}|sq|r|d||q|S)aMerge options with their corresponding arguments. Iterates over the dictionary holding arguments (keys) and options (values). Merges each options string with its corresponding argument. :param dict args_opts_dict: a dictionary of arguments and options :param dict kwargs: *input_option* - if specified prepends ``-i`` to input argument :return: merged list of strings with arguments and their corresponding options :rtype: list z-i)itemsrappend)rOrmergedargoptrrrrs   roptionsrrcCsD|durgSt|trt|S|rttjdd|DSt|S)a^Normalize options string or list of strings. Splits `options` into a list of strings. If `split_mixed` is `True`, splits (flattens) mixed options (i.e. list of strings with spaces) into separate items. :param options: options string or list of strings :param bool split_mixed: whether to split mixed options into separate items NcSsg|]}t|qSr)shlexsplit).0orrr sz&_normalize_options..) isinstancer rYrZlist itertoolschain)rXrrrrrs   r stream_datarIr cCs&|durdSt|tr|jddS|S)z1Convert FFmpeg output to text for error messages.Nreplace)errors)r^bytesdecode)rbrrrrJs   rJ)F)rOrPrrQrrR)rXrrrQrrR)rbrIrr ) __future__rr4r`rYrtypingrrrrpsutilr__annotations__ ImportErrorrr rA Exceptionr6r9rrrJrrrrs*    m"