Lionel TrainTown Deluxe Sprites

Code:
#extractor script for Train Town .tbv files

# by DarkWolf



import struct

import sys

import os



def get_base(path):

    tmp = os.path.basename(path)

    pieces = tmp.split('.')

    return pieces[0]

    

def check_dir(path):

  if not os.path.exists(path):

    os.makedirs(path)



def main():

    if len(sys.argv) != 2:

        print("Please pass in a file path");

        sys.exit(1)

        

    header_struct = struct.Struct('<2I')

    name_struct = struct.Struct('<24sI')

    out_base = f'extracted{os.path.sep}'

    check_dir(out_base)

    input_path = sys.argv[1]

    input_base = f'{out_base}{get_base(input_path)}'

    check_dir(input_base)

    

    with open(input_path, 'rb') as fp:

        fp.seek(0x0B)

        header_items = struct.unpack('<H', fp.read(2))[0]

        

        fp.seek(0x29)

        offsets = []

        for ndx in range(0, header_items):

            data = header_struct.unpack(fp.read(header_struct.size))

            offsets.append(data[1])

        

        for addr in offsets:

            fp.seek(addr)

            data = name_struct.unpack(fp.read(name_struct.size))

            name = data[0].decode('ascii').rstrip('\x00')

            length = data[1]

            

            out_path = f'{input_base}{os.path.sep}{name}'

            with open(out_path, 'wb') as outfp:

                outfp.write(fp.read(length))

            



if __name__ == "__main__":

    main()