clxud
109 lines
3.6 KiBLFS
Python
109 lines
3.6 KiBLFS
Python
from ..FileSystem.File import _BaseFile
|
|
from ..Exceptions import FileSystemException
|
|
|
|
class Path:
|
|
def __init__(self, path: list[str] | str, dir_ptr = 0):
|
|
if isinstance(path, list):
|
|
self.path = path
|
|
else:
|
|
self.path = path.split("/")
|
|
self.dir_ptr = dir_ptr
|
|
|
|
def advance(self):
|
|
if len(self.path) <= self.dir_ptr:
|
|
pass
|
|
if self.path[self.dir_ptr].endswith("\x00"):
|
|
return
|
|
self.dir_ptr += 1
|
|
|
|
def fetch(self):
|
|
if len(self.path) <= self.dir_ptr:
|
|
pass
|
|
if self.path[self.dir_ptr].endswith("\x00"):
|
|
return self.path[self.dir_ptr][:-1]
|
|
else:
|
|
self.advance()
|
|
return self.path[self.dir_ptr-1]
|
|
|
|
|
|
def get(self):
|
|
if len(self.path) <= self.dir_ptr:
|
|
pass
|
|
return self.path[self.dir_ptr]
|
|
|
|
|
|
class FileSystemQuery:
|
|
"""
|
|
an object used to query a FileSystem / Directory for a specified file
|
|
if supplied a file_obj (an object that inherits _BaseFile), it can also be used to add a file to a directory
|
|
|
|
"""
|
|
def __init__(self, file, dir_ptr = 0, file_obj = None):
|
|
if isinstance(file, list):
|
|
*self.path, self.name = file
|
|
else:
|
|
*self.path, self.name = file.split('/')
|
|
self.dir_ptr = dir_ptr # value used in file lookup to tell the Directory what path to use for lookup
|
|
self.file_obj: _BaseFile = file_obj
|
|
|
|
|
|
def get_next(self):
|
|
"""
|
|
gets the current path of the query and advances dir_ptr by one
|
|
returns 1, dir_name when supplied a directory
|
|
returns 2, name when at file name
|
|
"""
|
|
if self.dir_ptr > len(self.path):
|
|
raise FileSystemException(f'Tried to access a higher level directory than applicable in current FileSystemQuery. Path: {self.path}; Name: {self.name}')
|
|
if self.dir_ptr == len(self.path):
|
|
self.dir_ptr += 1
|
|
return 2, self.name
|
|
else:
|
|
self.dir_ptr += 1
|
|
return 1, self.path[self.dir_ptr-1]
|
|
|
|
|
|
def get_current(self):
|
|
"""
|
|
same as get_next, but doesnt advance dir_ptr
|
|
"""
|
|
if self.dir_ptr > len(self.path):
|
|
raise FileSystemException(f'Tried to access a higher level directory than applicable in current FileSystemQuery. Path: {self.path}; Name: {self.name}')
|
|
if self.dir_ptr == len(self.path):
|
|
return 2, self.name
|
|
else:
|
|
return 1, self.path[self.dir_ptr-1]
|
|
|
|
class MassFileSystemQuery:
|
|
"""
|
|
an object used to query a FileSystem / Directory for files
|
|
|
|
any string inside a directory lookup that ends with a \x00 (aka null terminated)
|
|
|
|
exclude operations will be checked first, then include on what was left
|
|
:param dir_include: any directory that matches to the input will be included for file searching
|
|
:param dir_exclude:
|
|
:param file_name_include:
|
|
:param file_exclude:
|
|
"""
|
|
def __init__(self, dir_include, dir_exclude, file_include, file_exclude):
|
|
if not isinstance(dir_include, list):
|
|
self.dir_include = [dir_include]
|
|
else:
|
|
self.dir_include = dir_include
|
|
|
|
if not isinstance(dir_exclude, list):
|
|
self.dir_exclude = [dir_exclude]
|
|
else:
|
|
self.dir_exclude = dir_exclude
|
|
|
|
if not isinstance(file_include, list):
|
|
self.file_include = [file_include]
|
|
else:
|
|
self.file_include = file_include
|
|
|
|
if not isinstance(file_exclude, list):
|
|
self.file_exclude = [file_exclude]
|
|
else:
|
|
self.file_exclude = file_exclude
|