diff --git a/cg/freecad/Frames/pddl/freecad2pddl.py b/cg/freecad/Frames/pddl/freecad2pddl.py index 1c22950..66d6411 100644 --- a/cg/freecad/Frames/pddl/freecad2pddl.py +++ b/cg/freecad/Frames/pddl/freecad2pddl.py @@ -9,13 +9,14 @@ doc = App.ActiveDocument #функции для создания обьектов PDDL def add_types(): - types = doc.addObject('App::FeaturePython', 'Types') + types = doc.addObject('App::FeaturePython', 'types') types.addProperty("App::PropertyString", 'PDDL', 'PDDL').PDDL = 'PDDL' - types.addProperty("App::PropertyStringList", 'Type', 'PDDL').Type = 'Types' + types.addProperty("App::PropertyString", 'Type', 'PDDL').Type = 'Types' types.addProperty("App::PropertyStringList", 'Types', 'PDDL') sortEntity(types) print('Types of objects added successfully') return(types) +#слишком много types def add_parameters(): @@ -173,11 +174,18 @@ def export_to_file(file_path): def sortEntity(object): #если нету директории, то создадим ее - if doc.getObjectsByLabel(object.Type) == 0: - pddl_group = doc.addObject("App::DocumentObjectGroup", str(object.Type)) + # if len(doc.getObjectsByLabel(object.Type)) == 0 : + # pddl_group = doc.addObject("App::DocumentObjectGroup", object.Type) + # else: + pddl_group = doc.addObject("App::DocumentObjectGroup", object.Type) + + if len(doc.getObjectsByLabel(object.Type)) == 0: + pddl_group = doc.addObject("App::DocumentObjectGroup", object.Type) else: - pddl_group = doc.getObject(str(object.Type)) - pddl_group.addObject(object) + pddl_group = doc.getObjectsByLabel(object.Type)[0] + pddl_group.addObject(object) + + Gui.updateGui()