1
0
mirror of https://github.com/Escartem/AnimeWwise.git synced 2026-06-07 17:00:23 +08:00

fix file loading

This commit is contained in:
Escartem
2025-01-07 14:55:10 +01:00
parent f28afbb87c
commit d7f897ffdc
3 changed files with 7 additions and 7 deletions

View File

@@ -6,23 +6,22 @@ class Allocator:
def __init__(self):
self.files = {}
def load_file(self, path):
filename = os.path.basename(path)
def load_file(self, path, name):
with open(path, "r+b") as f:
mmap_object = mmap.mmap(f.fileno(), 0, access=mmap.ACCESS_READ)
self.files[filename] = mmap_object
self.files[name] = mmap_object
def unload_file(self, name):
self.files[os.path.basename(name)].close()
self.files[name].close()
def read_at(self, file, offset, size):
mmap_object = self.files[os.path.basename(file)]
mmap_object = self.files[file]
mmap_object.seek(offset)
data = mmap_object.read(size)
return data
def free_mem(self):
for file in list(self.files.keys()):
self.files[os.path.basename(file)].close()
self.files[file].close()
self.files.clear()