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

typing

Browse Source
This commit is contained in:
Andrej730
2024-04-03 17:35:02 +05:00
parent 19dcbf5ffb
commit af3e2e0cf3
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/blenderbim/blenderbim/tool/pset.py
+7 -2
View File
@@ -18,14 +18,18 @@
import bpy import bpy
import ifcopenshell import ifcopenshell
import ifcopenshell.util.element
import blenderbim.core.tool import blenderbim.core.tool
import blenderbim.tool as tool import blenderbim.tool as tool
import blenderbim.bim.schema import blenderbim.bim.schema
from typing import Union
class Pset(blenderbim.core.tool.Pset): class Pset(blenderbim.core.tool.Pset):
@classmethod @classmethod
def get_element_pset(cls, element, pset_name): def get_element_pset(
cls, element: ifcopenshell.entity_instance, pset_name: str
) -> Union[ifcopenshell.entity_instance, None]:
pset = ifcopenshell.util.element.get_pset(element, pset_name) pset = ifcopenshell.util.element.get_pset(element, pset_name)
if pset: if pset:
return tool.Ifc.get().by_id(pset["id"]) return tool.Ifc.get().by_id(pset["id"])
@@ -50,13 +54,14 @@ class Pset(blenderbim.core.tool.Pset):
return bpy.context.scene.WorkSchedulePsetProperties return bpy.context.scene.WorkSchedulePsetProperties
elif obj_type == "Group": elif obj_type == "Group":
return bpy.context.scene.GroupPsetProperties return bpy.context.scene.GroupPsetProperties
@classmethod @classmethod
def get_pset_name(cls, obj, obj_type): def get_pset_name(cls, obj, obj_type):
pset = cls.get_pset_props(obj, obj_type) pset = cls.get_pset_props(obj, obj_type)
return pset.pset_name return pset.pset_name
@classmethod @classmethod
def is_pset_applicable(cls, element, pset_name): def is_pset_applicable(cls, element: ifcopenshell.entity_instance, pset_name: str) -> bool:
return bool( return bool(
pset_name pset_name
in blenderbim.bim.schema.ifc.psetqto.get_applicable_names( in blenderbim.bim.schema.ifc.psetqto.get_applicable_names(