一个文件分割脚本，请多多指教

Chowroc

2.       1 #!/usr/bin/python
3.       2 # -*- encoding: utf-8 -*-
4.       3
5.       4 # Author: 周鹏
6.       5 # Date: 2006-07-07
7.       6
8.       7 import os
9.       8 import sys
10.       9 import getopt
11.      10 import shutil
12.      11 import math
13.      12
14.      13 """此程序用来分割文件，由参数指定文件名及单位块大小，程序自动计算所需数量
15.      14 分割文件存放在 $filename.split 目录中
16.      15 然后可用如下方法重新组装：
17.      16 find $filename.split/* | sort | xargs -i cat {} >$new_filename
18.      17 $new_filename 最好不要在 $filename.split 目录中
19.      18 """
20.      19
21.      20 def split(fname, slice, unit, mode):
22.      21     if unit in ['m', 'M']:
23.      22         dd_unit = unit.upper()
24.      23     else:
25.      24         dd_unit = unit.lower()
26.      25     du_unit = unit.lower()
27.      26
28.      27     # 先检测文件是否存在及是否是普通文件
29.      28     if not os.path.isfile(fname):
30.      29         strerr = 'Not a regular file.'
31.      30         sys.stderr.write('%s\n' % strerr)
32.      31         sys.exit(1)
33.      32
34.      33     store = fname + '.split'
35.      34     if not os.path.exists(store):
36.      35         os.mkdir(store)
37.      36     elif mode == 1:
38.      37         shutil.rmtree(store)
39.      38         os.mkdir(store)
40.      39     else:
41.      40         strerr = 'File or directory exists.'
42.      41         sys.stderr.write('%s\n' % strerr)
43.      42         return 0
44.      43     bname = os.path.basename(fname)
45.      44
46.      45     temp = os.popen('du %s -%s' % (fname, du_unit)).read().split()
47.      46     if temp:
48.      47         fsize = int(temp[0])
49.      48         amount = fsize / slice
50.      49         if fsize % slice > 0: amount += 1
51.      50         ndigit = int(math.log10(amount)) + 1
52.      51         for i in range(amount):
53.      52             s = '%%0%sd' % ndigit
54.      53             exec "n = '%s' %% %s" % (s, i)
55.      54             target = '%s/%s.%s' % (store, bname, n)
56.      55             pid = os.fork()
57.      56             if pid == 0:
58.      57                 os.execlp(
59.      58                     'dd', 'dd', 'if=%s' % fname, 'of=%s' % target,
60.      59                     'bs=%s%s' % (slice, unit), 'skip=%d' % i, 'count=1'
61.      60                 )
62.      61                 assert 0, "Excuting dd failed."
63.      62             else:
64.      63                 os.wait()
65.      64         return 1
66.      65     else:
67.      66         return 0
68.      67
69.      68 def main(argv=sys.argv):
70.      69     try:
71.      70         opts, args = getopt.getopt(argv[1:], 'u:fh', ['force', 'help'])
72.      71
73.      72         unit = 'M'
74.      73         slice = 9
75.      74         mode = 0 # force or not
76.      75
77.      76         if not args:
78.      77             strerr = 'No filename given.'
79.      78             sys.stderr.write('%s\n' % strerr)
80.      79             print usage
81.      80             sys.exit(1)
82.      81
83.      82         fname = args[0]
84.      83
85.      84         for opt, value in opts:
86.      85             if opt in ['-u']:
87.      86                 unit = value[-1]
88.      87                 if unit not in ['b', 'B', 'k', 'K', 'm', 'M']:
89.      88                     strerr = 'Not a valid unit.'
90.      89                     sys.stderr.write('%s\n' % strerr)
91.      90                     sys.exit(1)
92.      91                 slice = value[:-1]
93.      92                 if slice.isdigit():
94.      93                     slice = int(slice)
95.      94                 else:
96.      95                     strerr = 'Not a valid value.'
97.      96                     sys.strerr.write('%s\n' % strerr)
98.      97                     sys.exit(1)
99.      98             elif opt in ['-f', '--force']:
100.      99                 mode = 1
101.     100             elif opt in ['-h', '--help']:
102.     101                 print usage
103.     102                 sys.exit(0)
104.     103
105.     104         if not split(fname, slice, unit, mode):
106.     105             strerr = 'File spliting failed.'
107.     106             sys.stderr.write('%s\n' % strerr)
108.     107             sys.exit(127)
109.     108
110.     109     except getopt.GetoptError, goEx:
111.     110         strerr = 'getopt error: %s, %s' % (goEx.opt, goEx.msg)
112.     111         sys.stderr.write('%s\n' % strerr)
113.     112         sys.exit(1)
114.     113
115.     114 if __name__ == '__main__':
116.     115     program = os.path.basename(sys.argv[0])
117.     116     usage = """program usage: %s [OPTIONS] FILENAME
118.     117     OPTIONS:
119.     118     -u $unit, 每个分割文件的大小：b, k, m...
120.     119     -f|--force, force，如果 $filename.split 存在就先删除
121.     120     -h|--help, print this message
122.     121     """ % program
123.     122     main()
124.     123 else:
125.     124     usage = "module usage %s " % __name__

复制代码