from email.generator import Generator
from email.policy import default
# Get the filename of the draft
if len(sys.argv) == 1:
print("Which draft should be formatted?")
sys.exit(0)
draft = sys.argv[1]
# Load the draft
with open(draft, 'rb') as fi:
mail = BytesParser(policy=default).parse(fi)
# Add mime headers
if 'Mime-Version' not in mail:
mail.add_header('Mime-Version', '1.0')
#mail.add_header('Content-Type', 'text-plain')
mail.set_param('charset', 'utf-8')
mail.add_header('Content-Transfer-Encoding', 'quoted-printable')
# Rewrite some fields
fields = ('to', 'cc', 'bcc', 'subject')
for field in fields:
if field in mail:
value = mail[field]
#mail.replace_header(field.capitalize(), value)
del mail[field]
mail.add_header(field.capitalize(), value)
# Add the potential attachment
if len(sys.argv) == 3:
