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@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])