Draft: Handle wantedLayer==None

This commit is contained in:
Chris Hennes
2023-03-23 22:18:23 -05:00
parent 59fa7c9205
commit 2d7a77169e

View File

@@ -333,17 +333,18 @@ def locateLayer(wantedLayer, color=None, drawstyle=None):
"""
# layers is a global variable.
# It should probably be passed as an argument.
wantedLayerName = decodeName(wantedLayer)
for l in layers:
if wantedLayerName == l.Label:
return l
if wantedLayer is None:
wantedLayer = '0'
for layer in layers:
if layer.Label == wantedLayer:
return layer
if dxfUseDraftVisGroups:
newLayer = Draft.make_layer(name=wantedLayer,
line_color=color,
draw_style=drawstyle)
line_color=(0.0,0.0,0.0) if not color else color,
draw_style="Solid" if not drawstyle else drawstyle)
else:
newLayer = doc.addObject("App::DocumentObjectGroup", wantedLayer)
newLayer.Label = wantedLayerName
newLayer.Label = wantedLayer
layers.append(newLayer)
return newLayer