Home | History | Annotate | Download | only in unicode 
      1 import os, string
      2 
      3 codecs = {
      4     'cn': ('gb2312', 'gbk', 'gb18030', 'hz'),
      5     'tw': ('big5', 'cp950'),
      6     'hk': ('big5hkscs',),
      7     'jp': ('cp932', 'shift_jis', 'euc_jp', 'euc_jisx0213', 'shift_jisx0213',
      8            'euc_jis_2004', 'shift_jis_2004'),
      9     'kr': ('cp949', 'euc_kr', 'johab'),
     10     'iso2022': ('iso2022_jp', 'iso2022_jp_1', 'iso2022_jp_2',
     11                 'iso2022_jp_2004', 'iso2022_jp_3', 'iso2022_jp_ext',
     12                 'iso2022_kr'),
     13 }
     14 
     15 TEMPLATE = string.Template("""\
     16 #
     17 # $encoding.py: Python Unicode Codec for $ENCODING
     18 #
     19 # Written by Hye-Shik Chang <perky (at] FreeBSD.org>
     20 #
     21 
     22 import _codecs_$owner, codecs
     23 import _multibytecodec as mbc
     24 
     25 codec = _codecs_$owner.getcodec('$encoding')
     26 
     27 class Codec(codecs.Codec):
     28     encode = codec.encode
     29     decode = codec.decode
     30 
     31 class IncrementalEncoder(mbc.MultibyteIncrementalEncoder,
     32                          codecs.IncrementalEncoder):
     33     codec = codec
     34 
     35 class IncrementalDecoder(mbc.MultibyteIncrementalDecoder,
     36                          codecs.IncrementalDecoder):
     37     codec = codec
     38 
     39 class StreamReader(Codec, mbc.MultibyteStreamReader, codecs.StreamReader):
     40     codec = codec
     41 
     42 class StreamWriter(Codec, mbc.MultibyteStreamWriter, codecs.StreamWriter):
     43     codec = codec
     44 
     45 def getregentry():
     46     return codecs.CodecInfo(
     47         name='$encoding',
     48         encode=Codec().encode,
     49         decode=Codec().decode,
     50         incrementalencoder=IncrementalEncoder,
     51         incrementaldecoder=IncrementalDecoder,
     52         streamreader=StreamReader,
     53         streamwriter=StreamWriter,
     54     )
     55 """)
     56 
     57 def gencodecs(prefix):
     58     for loc, encodings in codecs.iteritems():
     59         for enc in encodings:
     60             code = TEMPLATE.substitute(ENCODING=enc.upper(),
     61                                        encoding=enc.lower(),
     62                                        owner=loc)
     63             codecpath = os.path.join(prefix, enc + '.py')
     64             open(codecpath, 'w').write(code)
     65 
     66 if __name__ == '__main__':
     67     import sys
     68     gencodecs(sys.argv[1])
     69