einstein/einsteingame/messages.py
Gergely Polonkai 8806876046 workwork
2015-11-04 17:30:48 +01:00

69 lines
2.1 KiB
Python

import struct
from .buffer import Buffer
from .formatter import Formatter
class Messages(object):
def __init__(self):
self.__messages = {}
def load(self, resources):
buf = Buffer()
resources.forEachInGroup('messages', lambda res: self.loadFromResource(res, buf))
def loadFromResource(self, resource, buf):
if not resource:
return
cnt = resource.getVariantsCount()
for variant in resource.variants:
if variant:
try:
score = variant.i18nScore
data = variant.getData()
self.loadBundle(score, data)
except Exception as e:
print("Error loading text bundle {}: {}".format(resource.getName(), e))
raise
def loadBundle(self, score, data):
if data[0:3] != b"CMF":
raise Exception("Invalid format of message file!")
version = struct.unpack('i', data[3:7])[0]
if version != 1:
raise Exception("Unknown version of message file!")
offset = struct.unpack('i', data[len(data) - 4:])[0]
cnt = struct.unpack('i', data[offset:offset + 4])[0]
offset += 4
for i in range(0, cnt):
sz = struct.unpack('i', data[offset:offset + 4])[0]
offset += 4;
if sz > 0:
name = data[offset:offset + sz].decode('utf-8')
msgOffset = struct.unpack('i', data[offset + sz:offset + sz + 4])[0]
if name in self.__messages:
ss = self.__messages[name]
if ss['score'] <= score:
ss['score'] = score
ss['message'] = Formatter(data, msgOffset)
else:
self.__messages[name] = {
"score": score,
"message": Formatter(data, msgOffset),
}
print("{}: {}".format(name, self.__messages[name]['message'].getMessage()))
offset += sz + 4
msg = Messages()