Commit | Line | Data |
---|---|---|
920dae64 AT |
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 | """ | |
9 | import codecs, base64 | |
10 | ||
11 | ### Codec APIs | |
12 | ||
13 | def 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 | ||
27 | def 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 | ||
45 | class 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 | ||
52 | class StreamWriter(Codec,codecs.StreamWriter): | |
53 | pass | |
54 | ||
55 | class StreamReader(Codec,codecs.StreamReader): | |
56 | pass | |
57 | ||
58 | ### encodings module API | |
59 | ||
60 | def getregentry(): | |
61 | ||
62 | return (base64_encode,base64_decode,StreamReader,StreamWriter) |