70 lines
2.2 KiB
Python
70 lines
2.2 KiB
Python
from typing import TYPE_CHECKING
|
|
from logging import Logger
|
|
|
|
if TYPE_CHECKING:
|
|
from .connection import Connection
|
|
|
|
|
|
LOGGER = Logger(__name__)
|
|
|
|
|
|
class SimpleNode:
|
|
_normal_attrs = ['path', '_data', '_connection']
|
|
|
|
def __init__(self, path: str, data: dict, connection: 'Connection'):
|
|
self.path = path
|
|
self._data = data
|
|
self._connection = connection
|
|
|
|
def download(self, key: str = 'jcr:data') -> bytes:
|
|
if ':' + key not in self._data:
|
|
LOGGER.warning(f"Key :{key} is not present, binary probably not available")
|
|
size = self._data.get(':' + key)
|
|
if not isinstance(size, int):
|
|
LOGGER.warning("Size is not present")
|
|
# TODO value denotes file size, warn/deny large files?
|
|
return self._connection.download_binary(self.path + '/' + key)
|
|
|
|
def __getitem__(self, item):
|
|
return getattr(self, item)
|
|
|
|
def __setitem__(self, key, value):
|
|
return self.__setattr__(key, value)
|
|
|
|
def __setattr__(self, key, value):
|
|
if key in self._normal_attrs:
|
|
return super(SimpleNode, self).__setattr__(key, value)
|
|
if (':' + key) in self._data:
|
|
value_type = self._data[':' + key]
|
|
if isinstance(value_type, int):
|
|
value_type = 'Binary'
|
|
else:
|
|
if isinstance(value, str):
|
|
value_type = 'String'
|
|
elif isinstance(value, int):
|
|
value_type = 'Long'
|
|
elif isinstance(value, bool):
|
|
value_type = 'Boolean'
|
|
else:
|
|
raise ValueError(f"Unknown value type {type(value)!r}")
|
|
|
|
self._connection._patch_builder.set_value(self.path + '/' + key, value, value_type)
|
|
|
|
def __getattr__(self, item: str):
|
|
try:
|
|
return super(SimpleNode, self).__getattr__(item)
|
|
except AttributeError:
|
|
pass
|
|
|
|
try:
|
|
value = self._data.get(item)
|
|
except KeyError:
|
|
raise AttributeError()
|
|
|
|
if isinstance(value, dict):
|
|
return self._connection.get_simple_node(self.path + '/' + item)
|
|
return value
|
|
|
|
def __dir__(self):
|
|
return super(SimpleNode, self).__dir__() + list(self._data.keys())
|