04文件分割器脚本

'''
将文件分割成很多组分,join.py将它们合并回
'''

import sys,os

kilobytes=1024
megabytes=kilobytes*1000
chunksize=int(1.4*megabytes)    #默认设置，大约一个软盘的容量

def split(fromfile,todir,chunksize=chunksize):
    if not os.path.exists(todir):
        os.mkdir(todir)
    else:
        for fname in os.listdir(todir):
            os.remove(os.path.join(todir,fname))
    partnum=0
    file_open=open(fromfile, 'rb')   #二进制模式：不解码，没有行尾
    while True:
        chunk=file_open.read(chunksize) #获取下一个组分文件
        if not chunk:break
        partnum+=1
        filename=os.path.join(todir,('part%04d'%partnum))
        fileobj=open(filename,'wb')
        fileobj.write(chunk)
        fileobj.close()
    file_open.close()
    assert partnum<=9999
    return partnum

if __name__ == '__main__':
    if len(sys.argv)==2 and sys.argv[1]=='-help':
        print('Use:split.py [file-to-split target-dir [chunksize]]')
    else:
        if len(sys.argv)<3:
            interactive=True
            fromfile=input('File to be split?')
            todir=input('Directory to store part files?')
        else:
            interactive=False
            fromfile,todir=sys.argv[1:3]
            if len(sys.argv)==4:chunksize=int(sys.argv[3])#如果传入第3个参数，则令此参数为分割的大小
        absfrom,absto=map(os.path.abspath,[fromfile,todir])
        print('Splitting',absfrom,'to',absto,'by','chunksize')

        try:
            parts=split(fromfile,todir,chunksize)
        except:
            print('Error during split:')
            print(sys.exc_info()[0],sys.exc_info()[1])
        else:
            print('Split finished:',parts,'parts are in ',absto)
        if interactive:input('Press Enter key')

 

　　通过检查命令行参数的数目来判断其使用模式。在命令行模式下，你在命令行中列出待分割的文件和输出目录，还可以选择性的利用第三个命令行参数重写默认的组分文件大小。

　　在交互模式下，脚本通过input在控制台窗口询问文件名和输出目录，并在结束时暂停，等待键盘输入后退出。

　　这段代码谨慎地在二进制（rb，wb）下打开输入及输出文件，因为它需要可移植地处理可执行文件和音频文件等，而不仅是处理文本。

　　这个脚本还特意手动关闭文件。以前我们使用一行代码就足够了：open(partname,'wb').write(chunk)  这种简化的形式依赖于一个事实，即python在对象被回收时（即文件对象由于不再有引用而被垃圾回收时）自动关闭文件。在这行单行代码中，文件对象可能马上就被回收了，因为open的结果临时存在于一个表达式中，并从未被长时存在的名称引用过。