1""" Python 'base64_codec' Codec - base64 content transfer encoding
2
3    Unlike most of the other codecs which target Unicode, this codec
4    will return Python string objects for both encode and decode.
5
6    Written by Marc-Andre Lemburg (mal@lemburg.com).
7
8"""
9import codecs, base64
10
11### Codec APIs
12
13def base64_encode(input,errors='strict'):
14
15    """ Encodes the object input and returns a tuple (output
16        object, length consumed).
17
18        errors defines the error handling to apply. It defaults to
19        'strict' handling which is the only currently supported
20        error handling for this codec.
21
22    """
23    assert errors == 'strict'
24    output = base64.encodestring(input)
25    return (output, len(input))
26
27def base64_decode(input,errors='strict'):
28
29    """ Decodes the object input and returns a tuple (output
30        object, length consumed).
31
32        input must be an object which provides the bf_getreadbuf
33        buffer slot. Python strings, buffer objects and memory
34        mapped files are examples of objects providing this slot.
35
36        errors defines the error handling to apply. It defaults to
37        'strict' handling which is the only currently supported
38        error handling for this codec.
39
40    """
41    assert errors == 'strict'
42    output = base64.decodestring(input)
43    return (output, len(input))
44
45class Codec(codecs.Codec):
46
47    def encode(self, input,errors='strict'):
48        return base64_encode(input,errors)
49    def decode(self, input,errors='strict'):
50        return base64_decode(input,errors)
51
52class IncrementalEncoder(codecs.IncrementalEncoder):
53    def encode(self, input, final=False):
54        assert self.errors == 'strict'
55        return base64.encodestring(input)
56
57class IncrementalDecoder(codecs.IncrementalDecoder):
58    def decode(self, input, final=False):
59        assert self.errors == 'strict'
60        return base64.decodestring(input)
61
62class StreamWriter(Codec,codecs.StreamWriter):
63    pass
64
65class StreamReader(Codec,codecs.StreamReader):
66    pass
67
68### encodings module API
69
70def getregentry():
71    return codecs.CodecInfo(
72        name='base64',
73        encode=base64_encode,
74        decode=base64_decode,
75        incrementalencoder=IncrementalEncoder,
76        incrementaldecoder=IncrementalDecoder,
77        streamwriter=StreamWriter,
78        streamreader=StreamReader,
79    )
80