python_toolbox/python_toolbox/monkeypatch_envelopes.py at master · sphill007/python_toolbox

29 lines (21 loc) · 1.17 KB

# This program is distributed under the MIT license.

'''Module for monkeypatching our own copy of `envelopes`.'''

### Monkeypatching envelopes: #################################################

# #

from python_toolbox.third_party import envelopes

from python_toolbox import monkeypatching_tools

@monkeypatching_tools.monkeypatch(envelopes.Envelope)

def add_attachment_from_string(self, file_data, file_name,

mimetype='application/octet-stream'):

from python_toolbox.third_party.envelopes.envelope import \

MIMEBase, email_encoders, os

type_maj, type_min = mimetype.split('/')

part = MIMEBase(type_maj, type_min)

part.set_payload(file_data)

email_encoders.encode_base64(part)

part.add_header('Content-Disposition',

f'attachment; filename="{file_name}"')

self._parts.append((mimetype, part))

# #

### Finished monkeypatching envelopes. ########################################

Provide feedback