mirror of
https://github.com/python/cpython.git
synced 2025-10-31 13:41:24 +00:00
e7929cba16
Also inline necessary functionality from `sndhdr` into `email.mime.audio` for `MIMEAudio`. Co-authored-by: Hugo van Kemenade <hugovk@users.noreply.github.com>
98 lines
3 KiB
Python
98 lines
3 KiB
Python
# Copyright (C) 2001-2007 Python Software Foundation
|
|
# Author: Anthony Baxter
|
|
# Contact: email-sig@python.org
|
|
|
|
"""Class representing audio/* type MIME documents."""
|
|
|
|
__all__ = ['MIMEAudio']
|
|
|
|
from io import BytesIO
|
|
from email import encoders
|
|
from email.mime.nonmultipart import MIMENonMultipart
|
|
|
|
|
|
_tests = []
|
|
|
|
def _test_aifc_aiff(h, f):
|
|
if not h.startswith(b'FORM'):
|
|
return None
|
|
if h[8:12] in {b'AIFC', b'AIFF'}:
|
|
return 'x-aiff'
|
|
else:
|
|
return None
|
|
|
|
_tests.append(_test_aifc_aiff)
|
|
|
|
|
|
def _test_au(h, f):
|
|
if h.startswith(b'.snd'):
|
|
return 'basic'
|
|
else:
|
|
return None
|
|
|
|
_tests.append(_test_au)
|
|
|
|
|
|
def _test_wav(h, f):
|
|
import wave
|
|
# 'RIFF' <len> 'WAVE' 'fmt ' <len>
|
|
if not h.startswith(b'RIFF') or h[8:12] != b'WAVE' or h[12:16] != b'fmt ':
|
|
return None
|
|
else:
|
|
return "x-wav"
|
|
|
|
_tests.append(_test_wav)
|
|
|
|
|
|
# There are others in sndhdr that don't have MIME types. :(
|
|
# Additional ones to be added to sndhdr? midi, mp3, realaudio, wma??
|
|
def _whatsnd(data):
|
|
"""Try to identify a sound file type.
|
|
|
|
sndhdr.what() has a pretty cruddy interface, unfortunately. This is why
|
|
we re-do it here. It would be easier to reverse engineer the Unix 'file'
|
|
command and use the standard 'magic' file, as shipped with a modern Unix.
|
|
"""
|
|
hdr = data[:512]
|
|
fakefile = BytesIO(hdr)
|
|
for testfn in _tests:
|
|
res = testfn(hdr, fakefile)
|
|
if res is not None:
|
|
return res
|
|
else:
|
|
return None
|
|
|
|
|
|
class MIMEAudio(MIMENonMultipart):
|
|
"""Class for generating audio/* MIME documents."""
|
|
|
|
def __init__(self, _audiodata, _subtype=None,
|
|
_encoder=encoders.encode_base64, *, policy=None, **_params):
|
|
"""Create an audio/* type MIME document.
|
|
|
|
_audiodata is a string containing the raw audio data. If this data
|
|
can be decoded as au, wav, aiff, or aifc, then the
|
|
subtype will be automatically included in the Content-Type header.
|
|
Otherwise, you can specify the specific audio subtype via the
|
|
_subtype parameter. If _subtype is not given, and no subtype can be
|
|
guessed, a TypeError is raised.
|
|
|
|
_encoder is a function which will perform the actual encoding for
|
|
transport of the image data. It takes one argument, which is this
|
|
Image instance. It should use get_payload() and set_payload() to
|
|
change the payload to the encoded form. It should also add any
|
|
Content-Transfer-Encoding or other headers to the message as
|
|
necessary. The default encoding is Base64.
|
|
|
|
Any additional keyword arguments are passed to the base class
|
|
constructor, which turns them into parameters on the Content-Type
|
|
header.
|
|
"""
|
|
if _subtype is None:
|
|
_subtype = _whatsnd(_audiodata)
|
|
if _subtype is None:
|
|
raise TypeError('Could not find audio MIME subtype')
|
|
MIMENonMultipart.__init__(self, 'audio', _subtype, policy=policy,
|
|
**_params)
|
|
self.set_payload(_audiodata)
|
|
_encoder(self)
|