jgrewe/nixview-python
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

[file handler] returns ItemDescriptors instead of dicts

Browse Source
This commit is contained in:
Jan Grewe 2020-12-30 17:03:01 +01:00
parent 2016b6dad1
commit 84f4fd96e6
1 changed files with 22 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
file_handler.py
View File

@ -1,17 +1,18 @@
import os import os
import nixio as nix import nixio as nix
from nixio.pycore.util.util import check_entity_type
class ItemDescriptor(): class ItemDescriptor():
def __init__(self) -> None: def __init__(self, name=None, id=None, type=None, value=None, definition=None, entity_type=None) -> None:
super().__init__() super().__init__()
self.name = None self.name = name
self.type = None self.type = type
self.id = None self.id = id
self.definition = None self.definition = definition
self.value = None self.value = value
self.entity_type = None self.entity_type = entity_type
class Singleton(type): class Singleton(type):
@ -78,13 +79,14 @@ class FileHandler(metaclass=Singleton):
def get_subsections(section): def get_subsections(section):
sub_sections = [] sub_sections = []
for s in section.sections: for s in section.sections:
sub_sections.append({"name": s.name, "id": s.id, "item_type":s.type, "description": s.definition}) sub_sections.append(ItemDescriptor(s.name, s.id, s.type, definition=s.definition, entity_type="Section"))
return sub_sections return sub_sections
def get_properties(section): def get_properties(section):
props = [] props = []
for p in section.props: for p in section.props:
props.append({"name": p.name, "id":p.id, "unit": p.unit}) value = "unset"
props.append(ItemDescriptor(p.name, p.id, value=value, entity_type="Property"))
return props return props
sections = [] sections = []
@ -98,21 +100,22 @@ class FileHandler(metaclass=Singleton):
properties.extend(get_properties(fs)) properties.extend(get_properties(fs))
return sections, properties return sections, properties
def _entity_info(self, entities): def _entity_info(self, entities, entity_type):
infos = [] infos = []
for e in entities: for e in entities:
infos.append({"name": e.name, "id": e.id, "item_type": e.type, "description": e.definition}) itd = ItemDescriptor(e.name, e.id, e.type, definition=e.definition, entity_type=entity_type)
infos.append(itd)
return infos return infos
def request_blocks(self): def request_blocks(self):
return self._entity_info(self._nix_file.blocks) return self._entity_info(self._nix_file.blocks, "Block")
def request_data_arrays(self, block_id): def request_data_arrays(self, block_id):
return self._entity_info(self._nix_file.blocks[block_id].data_arrays) return self._entity_info(self._nix_file.blocks[block_id].data_arrays, "DataArray")
def request_tags(self, block_id): def request_tags(self, block_id):
tags = self._entity_info(self._nix_file.blocks[block_id].tags) tags = self._entity_info(self._nix_file.blocks[block_id].tags, "Tag")
tags.extend(self._entity_info(self._nix_file.blocks[block_id].multi_tags)) tags.extend(self._entity_info(self._nix_file.blocks[block_id].multi_tags), "MultiTag")
return tags return tags
def request_references(self, block_id, tag_id, is_mtag): def request_references(self, block_id, tag_id, is_mtag):
@ -122,7 +125,7 @@ class FileHandler(metaclass=Singleton):
t = b.multi_tags[tag_id] t = b.multi_tags[tag_id]
else: else:
t = b.tags[tag_id] t = b.tags[tag_id]
return self._entity_info(t.references) return self._entity_info(t.references, "DataArray")
def request_features(self, block_id, tag_id, is_mtag): def request_features(self, block_id, tag_id, is_mtag):
b = self._nix_file.blocks[block_id] b = self._nix_file.blocks[block_id]
@ -133,12 +136,7 @@ class FileHandler(metaclass=Singleton):
t = b.tags[tag_id] t = b.tags[tag_id]
feats = [] feats = []
for f in t.features: for f in t.features:
itd = ItemDescriptor() itd = ItemDescriptor(f.data.name, f.id, f.type, definition=f.data.definition, entity_type="Feature")
itd.name = f.data.name
itd.id = f.id
itd.type = f.type
itd.definition = f.data.definition
itd.entity_type = "nix.Feature"
feats.append(itd) feats.append(itd)
return feats return feats
@ -146,5 +144,6 @@ class FileHandler(metaclass=Singleton):
dimensions = [] dimensions = []
for i, d in enumerate(self._nix_file.blocks[block_id].data_arrays[array_id].dimensions): for i, d in enumerate(self._nix_file.blocks[block_id].data_arrays[array_id].dimensions):
dim_name = "%i. dim: %s" % (i+1, d.label if hasattr(d, "label") else "") dim_name = "%i. dim: %s" % (i+1, d.label if hasattr(d, "label") else "")
dimensions.append({"name": dim_name, "item_type": d.dimension_type, "id":None, "description": None}) dim_type= "%s %s" % (d.dimension_type, "dimension")
dimensions.append(ItemDescriptor(dim_name, type=dim_type, entity_type="Dimension"))
return dimensions return dimensions