s2gs file -- a fun little mystery for us - Page 2

Traceback (most recent call last):
  File "extract.py", line 89, in <module>
    main()
  File "extract.py", line 80, in main
    buffer.read_data_struct()
  File "build\bdist.win-amd64\egg\sc2reader\utils.py", line 270, in read_data_struct
    datatype = self.read_byte()
  File "build\bdist.win-amd64\egg\sc2reader\utils.py", line 150, in read_byte
    return ord(self.read_basic(1))
TypeError: ord() expected a character, but string of length 0 found

import zlib, sys, pprint
from sc2reader.utils import ReplayBuffer

races = {'Prot':'Protoss','Zerg':'Zerg','Terr':'Terran','RAND':'Random'}
data_names = [
    'R',
    'U',
    'S',
    'O',
    'AUR',
    'RCR',
    'WC',
    'UT',
    'KUC',
    'SB',
    'SRC',
    
]
data_names_pretty = [
    'Resources',
    'Units',
    'Structures',
    'Overview',
    'Average Unspent Resources',
    'Resource Collection Rate',
    'Workers Created',
    'Units Trained',
    'Killed Unit Count',
    'Structures Built',
    'Structures Razed Count'
]

def getRealm(str):
    if str == '\x00\x00S2':
        return "EU"
    
    return "?"
def getPlayers(data):
    players = []
    parr = data[0][3]
    for i in range(16):
        if not (parr[i][0][1] == 0):
            players.append(getPlayer(data, i))

    return players
def getPlayer(data, index):
    pinfo = data[0][3][index]
    pdata = data[3][0]

    player = {
        'id': "{}/{}/{}".format(getRealm(pinfo[0][1][0][1]), pinfo[0][1][0][2], pinfo[0][1][0][3]),
        'race' : races[pinfo[2]]
        }

    stats = {}

    for i in range(len(pdata)):
        stats[data_names[i]] = pdata[i][1][index][0][0]
    stats[data_names[len(pdata)]] = data[4][0][0][1][index][0][0]
    player['stats'] = stats

    return player
    

def main():
    p = sys.argv[2] == "players"
    with open(sys.argv[1],"rb") as s2gs:
        raw_data = zlib.decompress(s2gs.read()[16:])
        buffer = ReplayBuffer(raw_data)
        
        data = [
            buffer.read_data_struct(),
            buffer.read_data_struct(),
            buffer.read_data_struct(),
            buffer.read_data_struct(),
            buffer.read_data_struct(),
            buffer.read_data_struct(),
            buffer.read_data_struct(),
            buffer.read_data_struct(),
            buffer.read_data_struct()
            ]
        
        if p:
            pprint.PrettyPrinter(indent=2).pprint(getPlayers(data))
        else:
            pprint.PrettyPrinter(indent=2).pprint(data)
            
main()

[ { 'id': 'EU/1/535265',
    'race': 'Protoss',
    'stats': { 'AUR': 958,
               'KUC': 82,
               'O': 62650,
               'R': 17350,
               'RCR': 1384,
               'S': 4875,
               'SB': 34,
               'SRC': 5,
               'U': 38525,
               'UT': 96,
               'WC': 61}},
  { 'id': 'EU/1/756106',
    'race': 'Zerg',
    'stats': { 'AUR': 794,
               'KUC': 105,
               'O': 62131,
               'R': 18331,
               'RCR': 1360,
               'S': 4100,
               'SB': 17,
               'SRC': 7,
               'U': 38300,
               'UT': 183,
               'WC': 70}},
  { 'id': 'EU/1/838056',
    'race': 'Zerg',
    'stats': { 'AUR': 1134,
               'KUC': 20,
               'O': 42450,
               'R': 18550,
               'RCR': 1354,
               'S': 2050,
               'SB': 17,
               'SRC': 0,
               'U': 20150,
               'UT': 169,
               'WC': 72}},
  { 'id': 'EU/1/2158213',
    'race': 'Protoss',
    'stats': { 'AUR': 587,
               'KUC': 49,
               'O': 49150,
               'R': 16950,
               'RCR': 1130,
               'S': 2950,
               'SB': 28,
               'SRC': 0,
               'U': 28550,
               'UT': 90,
               'WC': 45}}]