40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
from collections import UserDict, UserList
|
|
from . import ome_metadata_rs as rs # noqa
|
|
|
|
|
|
class Ome(UserDict):
|
|
@staticmethod
|
|
def from_xml(xml: str) -> Ome:
|
|
"""Create the OME structure from an XML string"""
|
|
new = Ome()
|
|
new.update(rs.ome(str(xml)))
|
|
return new
|
|
|
|
def __dir__(self) -> list[str]:
|
|
return list(self.keys()) + list(super().__dir__())
|
|
|
|
def __getattr__(self, key: str) -> Ome | OmeList | int | float | str:
|
|
try:
|
|
new = self.__getitem__(key)
|
|
except KeyError:
|
|
raise AttributeError(f"'Ome' object has no attribute '{key}'")
|
|
if isinstance(new, dict):
|
|
return Ome(**new)
|
|
elif isinstance(new, list):
|
|
return OmeList(new)
|
|
else:
|
|
return new
|
|
|
|
|
|
class OmeList(UserList):
|
|
def __getitem__(self, item: int) -> Ome | OmeList | int | float | str:
|
|
new = super().__getitem__(item)
|
|
if isinstance(new, dict):
|
|
return Ome(**new)
|
|
elif isinstance(new, list):
|
|
return OmeList(new)
|
|
else:
|
|
return new
|