| 
									
										
										
										
											2010-12-02 18:06:51 +00:00
										 |  |  | """Codec for quoted-printable encoding.
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2014-03-13 20:54:30 -04:00
										 |  |  | This codec de/encodes from bytes to bytes. | 
					
						
							| 
									
										
										
										
											2010-12-02 18:06:51 +00:00
										 |  |  | """
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | import codecs | 
					
						
							|  |  |  | import quopri | 
					
						
							|  |  |  | from io import BytesIO | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def quopri_encode(input, errors='strict'): | 
					
						
							|  |  |  |     assert errors == 'strict' | 
					
						
							|  |  |  |     f = BytesIO(input) | 
					
						
							|  |  |  |     g = BytesIO() | 
					
						
							| 
									
										
										
										
											2015-09-12 00:34:28 +00:00
										 |  |  |     quopri.encode(f, g, quotetabs=True) | 
					
						
							| 
									
										
										
										
											2010-12-02 18:06:51 +00:00
										 |  |  |     return (g.getvalue(), len(input)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def quopri_decode(input, errors='strict'): | 
					
						
							|  |  |  |     assert errors == 'strict' | 
					
						
							|  |  |  |     f = BytesIO(input) | 
					
						
							|  |  |  |     g = BytesIO() | 
					
						
							|  |  |  |     quopri.decode(f, g) | 
					
						
							|  |  |  |     return (g.getvalue(), len(input)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class Codec(codecs.Codec): | 
					
						
							|  |  |  |     def encode(self, input, errors='strict'): | 
					
						
							|  |  |  |         return quopri_encode(input, errors) | 
					
						
							|  |  |  |     def decode(self, input, errors='strict'): | 
					
						
							|  |  |  |         return quopri_decode(input, errors) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class IncrementalEncoder(codecs.IncrementalEncoder): | 
					
						
							|  |  |  |     def encode(self, input, final=False): | 
					
						
							|  |  |  |         return quopri_encode(input, self.errors)[0] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class IncrementalDecoder(codecs.IncrementalDecoder): | 
					
						
							|  |  |  |     def decode(self, input, final=False): | 
					
						
							|  |  |  |         return quopri_decode(input, self.errors)[0] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class StreamWriter(Codec, codecs.StreamWriter): | 
					
						
							|  |  |  |     charbuffertype = bytes | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class StreamReader(Codec, codecs.StreamReader): | 
					
						
							|  |  |  |     charbuffertype = bytes | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # encodings module API | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def getregentry(): | 
					
						
							|  |  |  |     return codecs.CodecInfo( | 
					
						
							|  |  |  |         name='quopri', | 
					
						
							|  |  |  |         encode=quopri_encode, | 
					
						
							|  |  |  |         decode=quopri_decode, | 
					
						
							|  |  |  |         incrementalencoder=IncrementalEncoder, | 
					
						
							|  |  |  |         incrementaldecoder=IncrementalDecoder, | 
					
						
							|  |  |  |         streamwriter=StreamWriter, | 
					
						
							|  |  |  |         streamreader=StreamReader, | 
					
						
							| 
									
										
										
										
											2014-02-24 14:43:03 +02:00
										 |  |  |         _is_text_encoding=False, | 
					
						
							| 
									
										
										
										
											2010-12-02 18:06:51 +00:00
										 |  |  |     ) |