# -*- coding:utf-8 -*-
import os
import chardet
from ConfigParser import RawConfigParser as rcp
def cdWalker(cdrom, cdcfile):
'''
光盘扫描主程序
@param cdrom: 光盘访问路径
@param cdcfile:输出的光盘信息记录文件(包含路径,绝对,相对都可以)
@return: 无, 直接输出成*.cdc文件
@attention: 从V0.7开始不使用此扫描函数,使用IniCDinfo()
'''
export = ""
for root, dirs, files in os.walk(cdrom):
#print formatCDinfo(root, dirs, files)
export += formatCDinfo(root, dirs, files)
open(cdcfile, 'w').write(export)
#格式化文件信息输出
def formatCDinfo(root, dirs, files):
'''
光盘信息记录格式化函数
@note: 直接使用os.walk()函数的输出信息进行重组
@param root: 当前根
@param dirs:当前根中的所有目录
@param files: 当前根中的所有文件
@return: 字串,组织好的当前目录信息
'''
export = "\n" + root + "\n"
for d in dirs:
export += "-d" + root + _smartcode(d) + "\n"
for f in files:
export += "-f %s %s \n" % (root, _smartcode(f))
export += "="*70
return export
def iniCDinifo(cdrom, cdcfile):
'''
光盘信息.ini格式化函数
@note: 直接利用os.walk()函数的输出信息由ConfigParser.RawConfigParser
进行重组处理或.ini格式文件输出并记录
@param cdrom: 光盘访问路径
@param cdcfile; 输出的光盘信息记录文件(包含路径, 绝对,相对都可以)
@return: 无, 直接输出成组织好的类.ini的*.cdc文件
'''
walker = {}
for root, dirs, files in os.walk(cdrom):
walker[root] = (dirs, files)
cfg = rcp()
cfg.add_section("Info")
cfg.add_section("Comment")
cfg.set("Info", "ImagePath", cdrom)
cfg.set("Info", "Volume", cdcfile)
dirs = walker.keys()
i = 0
for d in dirs:
i += 1
cfg.set("Comment", str(i), d)
for p in walker:
cfg.add_section(p)
for f in walker[p][1]:
cfg.set(p, f, os.stat("%s/%s" % (p, f)).st_size)
f = open(cdcfile, 'w')
cfg.write(f)
f.close()
def _smartcode(stream):
'''
智能能匹配编码
@param stream: 需要匹配的字符串
@return: 相应编码后的字符串
'''
ustring = stream
codedetect = chardet.detect(ustring)["encoding"]
#print codedetect
try:
#print ustring
ustring = unicode(ustring, codedetect)
#print ustring
return "%s %s " % ("", ustring.encode('gb2312'))
#return "%s %s " % ("", ustring.encode('utf8'))
except:
return u"bad unicode encode try!"
if __name__ =='__main__':
cdrom = "F:/Whu"
cdcfile = "F:/Whu/axiaoqiang.cdc"
cdWalker(cdrom, cdcfile)
#iniCDinifo(cdrom, cdcfile)
#//python/1186