mirror of https://github.com/python/cpython
113 lines
2.5 KiB
Python
113 lines
2.5 KiB
Python
# Various tools used by MIME-reading or MIME-writing programs.
|
|
|
|
|
|
import string
|
|
import rfc822
|
|
|
|
|
|
# A derived class of rfc822.Message that knows about MIME headers and
|
|
# contains some hooks for decoding encoded and multipart messages.
|
|
|
|
class Message(rfc822.Message):
|
|
|
|
def __init__(self, fp):
|
|
rfc822.Message.__init__(self, fp)
|
|
self.encodingheader = \
|
|
self.getheader('content-transfer-encoding')
|
|
self.typeheader = \
|
|
self.getheader('content-type')
|
|
self.parsetype()
|
|
self.parseplist()
|
|
|
|
def parsetype(self):
|
|
str = self.typeheader
|
|
if str == None:
|
|
str = 'text/plain'
|
|
if ';' in str:
|
|
i = string.index(str, ';')
|
|
self.plisttext = str[i:]
|
|
str = str[:i]
|
|
else:
|
|
self.plisttext = ''
|
|
fields = string.splitfields(str, '/')
|
|
for i in range(len(fields)):
|
|
fields[i] = string.lower(string.strip(fields[i]))
|
|
self.type = string.joinfields(fields, '/')
|
|
self.maintype = fields[0]
|
|
self.subtype = string.joinfields(fields[1:], '/')
|
|
|
|
def parseplist(self):
|
|
str = self.plisttext
|
|
self.plist = []
|
|
while str[:1] == ';':
|
|
str = str[1:]
|
|
if ';' in str:
|
|
# XXX Should parse quotes!
|
|
end = string.index(str, ';')
|
|
else:
|
|
end = len(str)
|
|
f = str[:end]
|
|
if '=' in f:
|
|
i = string.index(f, '=')
|
|
f = string.lower(string.strip(f[:i])) + \
|
|
'=' + string.strip(f[i+1:])
|
|
self.plist.append(string.strip(f))
|
|
|
|
def getplist(self):
|
|
return self.plist
|
|
|
|
def getparam(self, name):
|
|
name = string.lower(name) + '='
|
|
n = len(name)
|
|
for p in self.plist:
|
|
if p[:n] == name:
|
|
return rfc822.unquote(p[n:])
|
|
return None
|
|
|
|
def getencoding(self):
|
|
if self.encodingheader == None:
|
|
return '7bit'
|
|
return self.encodingheader
|
|
|
|
def gettype(self):
|
|
return self.type
|
|
|
|
def getmaintype(self):
|
|
return self.maintype
|
|
|
|
def getsubtype(self):
|
|
return self.subtype
|
|
|
|
|
|
|
|
|
|
# Utility functions
|
|
# -----------------
|
|
|
|
|
|
# Return a random string usable as a multipart boundary.
|
|
# The method used is so that it is *very* unlikely that the same
|
|
# string of characters will every occur again in the Universe,
|
|
# so the caller needn't check the data it is packing for the
|
|
# occurrence of the boundary.
|
|
#
|
|
# The boundary contains dots so you have to quote it in the header.
|
|
|
|
_prefix = None
|
|
|
|
def choose_boundary():
|
|
global _generation, _prefix, _timestamp
|
|
import time
|
|
import rand
|
|
if _prefix == None:
|
|
import socket
|
|
import os
|
|
hostid = socket.gethostbyname(socket.gethostname())
|
|
uid = `os.getuid()`
|
|
pid = `os.getpid()`
|
|
seed = `rand.rand()`
|
|
_prefix = hostid + '.' + uid + '.' + pid
|
|
timestamp = `int(time.time())`
|
|
seed = `rand.rand()`
|
|
return _prefix + '.' + timestamp + '.' + seed
|