import struct
from demparse import Block

class Demo(object):
    def __init__(self, fileorfilename):
        if isinstance(fileorfilename, str):
            fileorfilename = open(fileorfilename, 'rb')
        self.file = fileorfilename
        self.cdtrack = self.file.readline()

    def __iter__(self):
        while 1:
            b = self.file.read(4)
            if not b:
                return
            blocksize = struct.unpack('<i', b)[0]
            self.curblockdata = b + self.file.read(12 + blocksize)
            yield Block(self.curblockdata[4:])


def test():
    for block in Demo('original.dem'):
        blocktime = None
        for message in block:
            #print message
            if message.__class__.__name__ == "Time":
                blocktime = message.time
                if blocktime > 10:
                    return 

            if message.__class__.__name__ == "ServerInfo":
                print message
                print block.next()
            #if message.__class__.__name__ == "UpdateEntity":
                #print blocktime, message.entity, message.origin
                #import code
                #code.interact(local=locals())
                #return

if __name__ == "__main__":
    if 0:
        import profile
        profile.run("test()")
    else:
        test()