מסע אל העבר

קוד: בחירת הכל

#!/usr/bin/env python3

import argparse
import errno
import os
import struct
import sys

def create_directory(name):
    try:
        os.makedirs(name)
    except OSError as e:
        if e.errno != errno.EEXIST:
            raise

def get_sound(num, offset, data, inputFile):
    name = 'tempfile{}-{}.voc'.format(num, hex(offset))
    size = 0
    inputFile.seek(offset, 0)
    with open('voc/' + name.format(num), 'wb') as tempFile:
        tempFile.write(inputFile.read(27))
        size = inputFile.read(3)
        tempFile.write(size)
        size = int.from_bytes(size, byteorder='little', signed=False)
        tempFile.write(inputFile.read(size + 1))
        size = tempFile.tell()
    return size - 1

if __name__=='__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('filename')
    args = parser.parse_args()

    create_directory('voc')
    create_directory('dat')

    with open(args.filename, 'rb') as f:
        s = f.read()
        count = 0
        offset = s.find(b'Creative')
        with open('dat/' + 'file{}-start-{}.dat'.format(count, hex(offset)), 'wb') as datFile:
            datFile.write(s[:offset])
        add = offset
        while add != -1:
            skip = get_sound(count, offset, s, f)
            newS = s[1 + offset:]
            add = newS.find(b'Creative')
            if skip != add and add != -1:
                with open('dat/' + 'file{}-{}-{}.dat'.format(count + 1, hex(offset + skip), hex(offset + add)), 'wb') as datFile:
                    datFile.write(s[(offset + skip):(offset+add)])
            offset = offset + 1 + add
            count += 1
        with open('dat/' + 'file{}-{}-end.dat'.format(count, hex(offset + skip)), 'wb') as datFile:
            datFile.write(s[(offset + skip):])