55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
from os import listdir
|
|
|
|
from .buffer import Buffer
|
|
|
|
class ResourceFile(object):
|
|
def __init__(self, fileName, buf=None):
|
|
# Ooooh, what does this button do?
|
|
# name(fileName)
|
|
|
|
if buf is not None:
|
|
buffer = buf
|
|
ownBuffer = False
|
|
else:
|
|
buffer = Buffer()
|
|
ownBuffer = true
|
|
|
|
try:
|
|
with open(fileName, 'rb') as f:
|
|
try:
|
|
sign = f.read(4)
|
|
if sign[0] != 'C' \
|
|
or sign[1] != 'R' \
|
|
or sign[2] != 'F' \
|
|
or sign[3]:
|
|
raise IOError("Failover")
|
|
|
|
major = f.unpack('i', 4)
|
|
minor = f.unpack('i', 4)
|
|
priority = f.unpack('i', 4)
|
|
|
|
if major != 2 or minor < 0:
|
|
raise Exception("Incompatible version of resource file '" + name + "'")
|
|
except IOError:
|
|
raise Exception("Invalid resource file '" + name + "'")
|
|
except IOError:
|
|
raise Exception("Error loading resource file '" + name + "'")
|
|
|
|
class ResourcesCollection:
|
|
def __init__(self, directories):
|
|
self.__files = []
|
|
self.__buffer = None
|
|
self.loadResourceFiles(directories)
|
|
processFiles()
|
|
|
|
def loadResourceFiles(self, directories):
|
|
for directory in directories:
|
|
try:
|
|
for name in listdir(directory):
|
|
if not name.startswith('.') and name.endswith('.res'):
|
|
self.__files.append(
|
|
ResourceFile(directory + '/' + name,
|
|
buf=self.__buffer))
|
|
except OSError:
|
|
pass
|