第一个python小程序

　　刚刚开始学习python，花了点时间看了python语法，网上那个python简明教程感觉真不错又结合了一个园友写的系列文章，基本算是把python语法入门了。恰好一个朋友向我抱怨他的一个项目文档乱码问题，就想着用python写一个转换编码的程序来练练手。代码如下：

#!/usr/bin/python
#-*- coding: utf-8 -*-

# Filename: gb_to_utf8.py dirname
# GB2312 to UTF-8

import os
import os.path
import sys
import chardet

def convert_code(filename, in_enc = 'gbk', out_enc = 'utf-8'):
    """
    
    Arguments:
    - `filename`:
    - `in_enc`:
    - `out_enc`:
    """
    try:
        print 'convert:', filename
        f = open(filename, 'rb')
        data = f.read()
        f.close()
        in_enc = chardet.detect(data)['encoding']
        # print in_enc
        new_contert = data.decode(in_enc, 'ignore')
        new_data = new_contert.encode(out_enc, 'ignore')
        f = open(filename, 'w')
        f.write(new_data)
        f.close()
        print 'Success:', filename, 'converted from', in_enc, 'to', out_enc, '!'
        print
    except:
        print 'Error, fail convert:', filename
        print

def get_filename_list(dirname, suffix = ''):
    """
    
    Arguments:
    - `dirname`:
    - `suffix`:
    """
    flist = []
    for root, dirs, files in os.walk(dirname):
        for name in files:
            if name.endswith(suffix):
                flist.append(os.path.join(root, name))
    return flist

def explore(name):
    """
    """
    dirname = os.path.join(os.getcwd(), name)
    flist = get_filename_list(dirname)
    for fname in flist:
        convert_code(fname)
    print 'Finished!'

def main(argv):
    """
    """
    explore(argv)

if __name__ == '__main__':
    main(sys.argv[1])

程序用了chardet包，可能默认python是没有安装的，apt-get或者yum搞定。

需要转换编码文件的文件夹名字作为参数传递给程序： python convert.py filedir

程序调用os.walk()函数，就能得到文件夹下面所有文件名（包裹子文件夹下面的），放在一个list里面。接着遍历list，每次取出一个文件处理：读出文件内容，chardet.detect得到内容编码格式（存储在一个字典里面），然后用它的编码格式decode，然后用utf-8格式（可以换成任何你想要转换成为的格式）encode，重新写入文件，就转换成功了。