def get_type(self, record: shapefile._Record):
record_dict = record.as_dict()
for key, value in record_dict.items():
if 'type' == key.lower():
for esdl_type in esdl.eClassifiers:
t: EClass = esdl.getEClassifier(esdl_type).eClass
if esdl.EnergyAsset.eClass in t.eAllSuperTypes():
# print('checking', esdl_type)
if value.lower() in esdl_type.lower():
print('Using {} as type for this shape'.format(
esdl_type))
return esdl.getEClassifier(esdl_type)
return esdl.GenericProducer
def categorize_object_attributes_and_references(self, object, attributes,
references):
attr_dict = {attr['name']: attr for attr in attributes}
ref_dict = {ref['name']: ref for ref in references}
view_mode = get_session('mapeditor_view_mode')
this_view_mode_config = view_modes_config[view_mode]
categorized_list = dict()
for key in this_view_mode_config:
categorized_list[key] = list()
for objtype in this_view_mode_config[key]:
if isinstance(object, esdl.getEClassifier(objtype)):
for feature in this_view_mode_config[key][objtype]:
if feature in attr_dict:
attr_info = deepcopy(attr_dict[feature])
del attr_dict[feature]
categorized_list[key].append(attr_info)
elif feature in ref_dict:
ref_info = deepcopy(ref_dict[feature])
del ref_dict[feature]
categorized_list[key].append(ref_info)
categorized_list['Advanced'] = list()
for feature in attr_dict:
attr_info = deepcopy(attr_dict[feature])
categorized_list['Advanced'].append(attr_info)
for feature in ref_dict:
attr_info = deepcopy(ref_dict[feature])
categorized_list['Advanced'].append(attr_info)
return categorized_list
def get_esdl_objects_of_type(self, asset_type):
asset_list = list()
active_es_id = get_session('active_es_id')
esh = get_handler()
es = esh.get_energy_system(active_es_id)
# module = importlib.import_module('esdl.esdl')
# esdl_asset_class = getattr(module, asset_type)
for asset in es.instance[0].area.asset:
if isinstance(asset, esdl.getEClassifier(asset_type)):
asset_list.append(asset)
return asset_list