Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-08-18 19:30:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

some typing

Browse Source
This commit is contained in:
Andrej730
2024-02-14 14:23:13 +05:00
parent 075b9e4775
commit f522e2beb7
3 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcpatch/ifcpatch/recipes/ExtractElements.py
+5 -4
View File
@@ -19,6 +19,7 @@
import ifcopenshell
import ifcopenshell.api
import ifcopenshell.util.selector
from typing import Union
class Patcher:
@@ -64,7 +65,7 @@ class Patcher:
self.create_spatial_tree()
self.file = self.new
def add_element(self, element):
def add_element(self, element) -> None:
new_element = self.append_asset(element)
if not new_element:
return
@@ -75,7 +76,7 @@ class Patcher:
self.add_decomposition_parents(spatial_element, new_spatial_element)
self.add_decomposition_parents(element, new_element)
def append_asset(self, element):
def append_asset(self, element) -> Union[ifcopenshell.entity_instance, None]:
try:
return self.new.by_guid(element.GlobalId)
except:
@@ -84,14 +85,14 @@ class Patcher:
return self.new.add(element)
return ifcopenshell.api.run("project.append_asset", self.new, library=self.file, element=element)
def add_decomposition_parents(self, element, new_element):
def add_decomposition_parents(self, element, new_element) -> None:
for rel in element.Decomposes:
parent = rel.RelatingObject
new_parent = self.append_asset(parent)
self.aggregates.setdefault(parent.GlobalId, set()).add(new_element)
self.add_decomposition_parents(parent, new_parent)
def create_spatial_tree(self):
def create_spatial_tree(self) -> None:
for relating_structure, related_elements in self.contained_ins.items():
self.new.createIfcRelContainedInSpatialStructure(
ifcopenshell.guid.new(),