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

type.py - some typing

Browse Source
This commit is contained in:
Andrej730
2024-02-28 16:24:22 +05:00
parent 931fd9cc9f
commit e46f18f686
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcopenshell-python/ifcopenshell/util/type.py
+3 -2
View File
@@ -19,6 +19,7 @@
import os
import json
import ifcopenshell.util.schema
from typing import List
cwd = os.path.dirname(os.path.realpath(__file__))
@@ -51,11 +52,11 @@ for schema in mapped_schemas:
type_to_entity_map[schema][element_type] = [e for e in elements if guessed_element in e]
def get_applicable_types(ifc_class, schema="IFC4"):
def get_applicable_types(ifc_class: str, schema="IFC4") -> List[str]:
schema = ifcopenshell.util.schema.get_fallback_schema(schema.upper())
return entity_to_type_map[schema].get(ifc_class, [])
def get_applicable_entities(ifc_type_class, schema="IFC4"):
def get_applicable_entities(ifc_type_class: str, schema="IFC4") -> List[str]:
schema = ifcopenshell.util.schema.get_fallback_schema(schema.upper())
return type_to_entity_map[schema].get(ifc_type_class, [])