def __init__(self, spec):
"""Create a basic MIME message according to 'spec'.
Each line of a spec has one content-type, which is optionally indented.
The indentation signifies how deep in the MIME hierarchy the
content-type is.
"""
parts = []
for line in spec.splitlines():
content_type = line.strip()
if not content_type:
continue
indent = self.getIndent(line)
if indent:
parts.append('\n--boundary-%s\n' % indent)
parts.append('Content-type: %s;\n' % content_type)
parts.append(self.table[content_type] % {'indent': indent + 1})
Message.__init__(self, StringIO(''.join(parts)))
def __init__(self, spec):
"""Create a basic MIME message according to 'spec'.
Each line of a spec has one content-type, which is optionally indented.
The indentation signifies how deep in the MIME hierarchy the
content-type is.
"""
parts = []
for line in spec.splitlines():
content_type = line.strip()
if not content_type:
continue
indent = self.getIndent(line)
if indent:
parts.append('\n--boundary-%s\n' % indent)
parts.append('Content-type: %s;\n' % content_type)
parts.append(self.table[content_type] % {'indent': indent + 1})
Message.__init__(self, StringIO(''.join(parts)))
def testMultipart(self):
m = Message(self.fp)
self.assert_(m is not None)
# skip the first bit
p = m.getpart()
self.assert_(p is not None)
self.assertEqual(p.fp.read(),
'This is a multipart message. Ignore this bit.\r\n')
# first text/plain
p = m.getpart()
self.assert_(p is not None)
self.assertEqual(p.gettype(), 'text/plain')
self.assertEqual(p.fp.read(),
'Hello, world!\r\n\r\nBlah blah\r\nfoo\r\n-foo\r\n')
# sub-multipart
p = m.getpart()
self.assert_(p is not None)
self.assertEqual(p.gettype(), 'multipart/alternative')
# sub-multipart text/plain
q = p.getpart()
self.assert_(q is not None)
q = p.getpart()
self.assert_(q is not None)
self.assertEqual(q.gettype(), 'text/plain')
self.assertEqual(q.fp.read(), 'Hello, world!\r\n\r\nBlah blah\r\n')
# sub-multipart text/html
q = p.getpart()
self.assert_(q is not None)
self.assertEqual(q.gettype(), 'text/html')
self.assertEqual(q.fp.read(), '<b>Hello, world!</b>\r\n')
# sub-multipart end
q = p.getpart()
self.assert_(q is None)
# final text/plain
p = m.getpart()
self.assert_(p is not None)
self.assertEqual(p.gettype(), 'text/plain')
self.assertEqual(p.fp.read(), 'Last bit\n')
# end
p = m.getpart()
self.assert_(p is None)
def testMultipart(self):
m = Message(self.fp)
self.assert_(m is not None)
# skip the first bit
p = m.getpart()
self.assert_(p is not None)
self.assertEqual(p.fp.read(),
'This is a multipart message. Ignore this bit.\r\n')
# first text/plain
p = m.getpart()
self.assert_(p is not None)
self.assertEqual(p.gettype(), 'text/plain')
self.assertEqual(p.fp.read(),
'Hello, world!\r\n\r\nBlah blah\r\nfoo\r\n-foo\r\n')
# sub-multipart
p = m.getpart()
self.assert_(p is not None)
self.assertEqual(p.gettype(), 'multipart/alternative')
# sub-multipart text/plain
q = p.getpart()
self.assert_(q is not None)
q = p.getpart()
self.assert_(q is not None)
self.assertEqual(q.gettype(), 'text/plain')
self.assertEqual(q.fp.read(), 'Hello, world!\r\n\r\nBlah blah\r\n')
# sub-multipart text/html
q = p.getpart()
self.assert_(q is not None)
self.assertEqual(q.gettype(), 'text/html')
self.assertEqual(q.fp.read(), '<b>Hello, world!</b>\r\n')
# sub-multipart end
q = p.getpart()
self.assert_(q is None)
# final text/plain
p = m.getpart()
self.assert_(p is not None)
self.assertEqual(p.gettype(), 'text/plain')
self.assertEqual(p.fp.read(),
'Last bit\n')
# end
p = m.getpart()
self.assert_(p is None)