| 1 |
ninoborges |
3 |
##Splitter upper
|
| 2 |
|
|
|
| 3 |
|
|
import sys, os
|
| 4 |
|
|
kilobytes = 1024
|
| 5 |
|
|
megabytes = kilobytes * 1000
|
| 6 |
|
|
chunksize = int(1.4 * megabytes)
|
| 7 |
|
|
|
| 8 |
|
|
def split(fromfile, todir, chunksize=chunksize):
|
| 9 |
|
|
if not os.path.exists(todir):
|
| 10 |
|
|
os.mkdir(todir)
|
| 11 |
|
|
else:
|
| 12 |
|
|
for fname in os.listdir(todir):
|
| 13 |
|
|
os.remove(os.path.join(todir, fname))
|
| 14 |
|
|
partnum = 0
|
| 15 |
|
|
input = open(fromfile, 'rb')
|
| 16 |
|
|
while 1:
|
| 17 |
|
|
chunk = input.read(chunksize)
|
| 18 |
|
|
if not chunk: break
|
| 19 |
|
|
partnum = partnum+1
|
| 20 |
|
|
filename = os.path.join(todir, ('part%04d' %partnum))
|
| 21 |
|
|
fileobj = open(filename, 'wb')
|
| 22 |
|
|
fileobj.write(chunk)
|
| 23 |
|
|
fileobj.close()
|
| 24 |
|
|
input.close()
|
| 25 |
|
|
assert partnum <= 9999
|
| 26 |
|
|
return partnum
|
| 27 |
|
|
|
| 28 |
|
|
if __name__ == '__main__':
|
| 29 |
|
|
if len(sys.argv) == 2 and sys.argv[1] == 'help':
|
| 30 |
|
|
print 'Use: split.py [file-to-split target-dir [chunksize]]'
|
| 31 |
|
|
else:
|
| 32 |
|
|
if len(sys.argv) <3:
|
| 33 |
|
|
interactive = 1
|
| 34 |
|
|
fromfile = raw_input('File to be split? ')
|
| 35 |
|
|
todir = raw_input('Directory to store part files? ')
|
| 36 |
|
|
else:
|
| 37 |
|
|
interactive = 0
|
| 38 |
|
|
fromfile, todir = sys.argv[1:3]
|
| 39 |
|
|
if len(sys.argv) == 4: chunksize = int(sys.argv[3])
|
| 40 |
|
|
absfrom, absto = map(os.path.abspath, [fromfile, todir])
|
| 41 |
|
|
print 'Splitting', absfrom, 'to', absto, 'by', chunksize
|
| 42 |
|
|
try: parts = split(fromfile, todir, chunksize)
|
| 43 |
|
|
except:
|
| 44 |
|
|
print 'Error during split'
|
| 45 |
|
|
print sys.exc_type, sys.exc_value
|
| 46 |
|
|
else:
|
| 47 |
|
|
print 'split finished: ', parts, 'parts are in', absto
|
| 48 |
|
|
if interactive: raw_input('press enter key') |
