Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-08-09 17:31:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

ifctester - some typing

Browse Source
This commit is contained in:
Andrej730
2024-02-13 11:56:27 +05:00
parent 23c65742e1
commit d710a01df5
1 changed files with 19 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifctester/ifctester/facet.py
+19 -7
View File
@@ -23,7 +23,7 @@ import ifcopenshell.util.element
import ifcopenshell.util.classification
from functools import lru_cache
from xmlschema.validators import identities
from typing import List
from typing import List, Union
def cast_to_value(from_value, to_value):
@@ -86,7 +86,9 @@ class Facet:
setattr(self, name, value)
return self
def filter(self, ifc_file, elements):
def filter(
self, ifc_file: ifcopenshell.file, elements: List[ifcopenshell.entity_instance]
) -> List[ifcopenshell.entity_instance]:
return [e for e in elements if self(e)]
def to_string(self, clause_type, specification=None, requirement=None):
@@ -211,7 +213,9 @@ class Attribute(Facet):
]
super().__init__(name, value, minOccurs, maxOccurs, instructions)
def filter(self, ifc_file, elements):
def filter(
self, ifc_file: ifcopenshell.file, elements: Union[ifcopenshell.entity_instance, None]
) -> List[ifcopenshell.entity_instance]:
if isinstance(elements, list):
return super().filter(ifc_file, elements)
@@ -337,7 +341,9 @@ class Classification(Facet):
super().__init__(value, system, uri, minOccurs, maxOccurs, instructions)
def filter(self, ifc_file, elements):
def filter(
self, ifc_file: ifcopenshell.file, elements: Union[ifcopenshell.entity_instance, None]
) -> List[ifcopenshell.entity_instance]:
if isinstance(elements, list):
return super().filter(ifc_file, elements)
return ifc_file.by_type("IfcObjectDefinition")
@@ -400,7 +406,9 @@ class PartOf(Facet):
]
super().__init__(name, predefinedType, relation, minOccurs, maxOccurs, instructions)
def filter(self, ifc_file, elements):
def filter(
self, ifc_file: ifcopenshell.file, elements: Union[ifcopenshell.entity_instance, None]
) -> List[ifcopenshell.entity_instance]:
if isinstance(elements, list):
return super().filter(ifc_file, elements)
return list(ifc_file) # Lazy
@@ -618,7 +626,9 @@ class Property(Facet):
]
super().__init__(propertySet, name, value, datatype, uri, minOccurs, maxOccurs, instructions)
def filter(self, ifc_file, elements):
def filter(
self, ifc_file: ifcopenshell.file, elements: Union[ifcopenshell.entity_instance, None]
) -> List[ifcopenshell.entity_instance]:
if isinstance(elements, list):
return super().filter(ifc_file, elements)
if ifc_file.schema == "IFC2X3":
@@ -883,7 +893,9 @@ class Material(Facet):
]
super().__init__(value, uri, minOccurs, maxOccurs, instructions)
def filter(self, ifc_file, elements):
def filter(
self, ifc_file: ifcopenshell.file, elements: Union[ifcopenshell.entity_instance, None]
) -> List[ifcopenshell.entity_instance]:
if isinstance(elements, list):
return super().filter(ifc_file, elements)
return ifc_file.by_type("IfcObjectDefinition")