mirror of
				https://github.com/KevinMidboe/spotify-downloader.git
				synced 2025-10-29 18:00:15 +00:00 
			
		
		
		
	Bugfix: -o m4a would fail [Fixes #720]
In FFmpeg, a given encoding may not always point to the same format string.
This commit is contained in:
		| @@ -21,6 +21,13 @@ from spotdl.encode.exceptions import EncoderNotFoundError | |||||||
|  |  | ||||||
| """ | """ | ||||||
|  |  | ||||||
|  | _TARGET_FORMATS_FROM_ENCODING = { | ||||||
|  |     "m4a": "mp4", | ||||||
|  |     "mp3": "mp3", | ||||||
|  |     "opus": "opus", | ||||||
|  |     "flac": "flac" | ||||||
|  | } | ||||||
|  |  | ||||||
|  |  | ||||||
| class EncoderBase(ABC): | class EncoderBase(ABC): | ||||||
|     """ |     """ | ||||||
| @@ -44,6 +51,7 @@ class EncoderBase(ABC): | |||||||
|         self.encoder_path = encoder_path |         self.encoder_path = encoder_path | ||||||
|         self._loglevel = loglevel |         self._loglevel = loglevel | ||||||
|         self._additional_arguments = additional_arguments |         self._additional_arguments = additional_arguments | ||||||
|  |         self._target_formats_from_encoding = _TARGET_FORMATS_FROM_ENCODING | ||||||
|  |  | ||||||
|     def set_argument(self, argument): |     def set_argument(self, argument): | ||||||
|         """ |         """ | ||||||
| @@ -94,6 +102,14 @@ class EncoderBase(ABC): | |||||||
|         """ |         """ | ||||||
|         pass |         pass | ||||||
|  |  | ||||||
|  |     def target_format_from_encoding(self, encoding): | ||||||
|  |         """ | ||||||
|  |         This method generates the target stream format from given | ||||||
|  |         input encoding. | ||||||
|  |         """ | ||||||
|  |         target_format = self._target_formats_from_encoding[encoding] | ||||||
|  |         return target_format | ||||||
|  |  | ||||||
|     def re_encode_from_stdin(self, input_encoding, target_path): |     def re_encode_from_stdin(self, input_encoding, target_path): | ||||||
|         """ |         """ | ||||||
|         This method must invoke the encoder to encode stdin to a |         This method must invoke the encoder to encode stdin to a | ||||||
|   | |||||||
| @@ -76,7 +76,7 @@ class EncoderFFmpeg(EncoderBase): | |||||||
|             + ["-i", input_path] \ |             + ["-i", input_path] \ | ||||||
|             + arguments.split() \ |             + arguments.split() \ | ||||||
|             + self._additional_arguments \ |             + self._additional_arguments \ | ||||||
|             + ["-f", target_encoding] \ |             + ["-f", self.target_format_from_encoding(target_encoding)] \ | ||||||
|             + [target_file] |             + [target_file] | ||||||
|  |  | ||||||
|         return command |         return command | ||||||
|   | |||||||
| @@ -48,7 +48,7 @@ class TestEncodingDefaults: | |||||||
|             '-acodec', 'copy', |             '-acodec', 'copy', | ||||||
|             '-b:a', '192k', |             '-b:a', '192k', | ||||||
|             '-vn', |             '-vn', | ||||||
|             '-f', 'm4a', |             '-f', 'mp4', | ||||||
|              target_path |              target_path | ||||||
|         ] |         ] | ||||||
|         return command |         return command | ||||||
| @@ -112,7 +112,7 @@ class TestEncodingInDebugMode: | |||||||
|             '-acodec', 'copy', |             '-acodec', 'copy', | ||||||
|             '-b:a', '192k', |             '-b:a', '192k', | ||||||
|             '-vn', |             '-vn', | ||||||
|             '-f', 'm4a', |             '-f', 'mp4', | ||||||
|              target_path |              target_path | ||||||
|         ] |         ] | ||||||
|         return command |         return command | ||||||
| @@ -180,7 +180,7 @@ class TestEncodingAndTrimSilence: | |||||||
|             '-b:a', '192k', |             '-b:a', '192k', | ||||||
|             '-vn', |             '-vn', | ||||||
|             '-af', 'silenceremove=start_periods=1', |             '-af', 'silenceremove=start_periods=1', | ||||||
|             '-f', 'm4a', |             '-f', 'mp4', | ||||||
|             target_path |             target_path | ||||||
|         ] |         ] | ||||||
|         return command |         return command | ||||||
|   | |||||||
| @@ -86,3 +86,12 @@ class TestMethods: | |||||||
|     def test_encoder_not_found_error(self): |     def test_encoder_not_found_error(self): | ||||||
|         with pytest.raises(EncoderNotFoundError): |         with pytest.raises(EncoderNotFoundError): | ||||||
|             self.EncoderKid("/a/nonexistent/path", "0", []) |             self.EncoderKid("/a/nonexistent/path", "0", []) | ||||||
|  |  | ||||||
|  |     @pytest.mark.parametrize("encoding, target_format", [ | ||||||
|  |         ("m4a", "mp4"), | ||||||
|  |         ("mp3", "mp3"), | ||||||
|  |         ("opus", "opus"), | ||||||
|  |         ("flac", "flac"), | ||||||
|  |     ]) | ||||||
|  |     def test_target_format_from_encoding(self, encoderkid, encoding, target_format): | ||||||
|  |         assert encoderkid.target_format_from_encoding(encoding) == target_format | ||||||
|   | |||||||
| @@ -138,6 +138,7 @@ class YouTubeStreams(StreamsBase): | |||||||
|         self.all = [] |         self.all = [] | ||||||
|  |  | ||||||
|         for stream in audiostreams: |         for stream in audiostreams: | ||||||
|  |             encoding = "m4a" if "mp4a" in stream.audio_codec else stream.audio_codec | ||||||
|             standard_stream = { |             standard_stream = { | ||||||
|                 # Store only the integer part for bitrate. For example |                 # Store only the integer part for bitrate. For example | ||||||
|                 # the given bitrate would be "192kbps", we store only |                 # the given bitrate would be "192kbps", we store only | ||||||
| @@ -145,7 +146,7 @@ class YouTubeStreams(StreamsBase): | |||||||
|                 "bitrate": int(stream.abr[:-4]), |                 "bitrate": int(stream.abr[:-4]), | ||||||
|                 "connection": None, |                 "connection": None, | ||||||
|                 "download_url": stream.url, |                 "download_url": stream.url, | ||||||
|                 "encoding": stream.audio_codec, |                 "encoding": encoding, | ||||||
|                 "filesize": None, |                 "filesize": None, | ||||||
|             } |             } | ||||||
|             establish_connection = threading.Thread( |             establish_connection = threading.Thread( | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user