57da6ac140
Fixes #17952. This PR is a continuation of #20719. It mainly involves adding `self.doc`. This new attribute is then also used in the rest of the modified classes. In some cases code was slightly reordered for consistency. The `_finish_command_on_doc_close` function has been modifief to also handle `Gui.Snapper.ui.escape` and the code in gui_arcs.py has been updated accordingly. Additionally: Removed the `done` and `spacedone` variables in BimWall.py by tweaking the code.
71 lines
3.3 KiB
Python
71 lines
3.3 KiB
Python
# SPDX-License-Identifier: LGPL-2.1-or-later
|
|
|
|
# ***************************************************************************
|
|
# * *
|
|
# * Copyright (c) 2025 FreeCAD Project Association *
|
|
# * *
|
|
# * This file is part of FreeCAD. *
|
|
# * *
|
|
# * FreeCAD is free software: you can redistribute it and/or modify it *
|
|
# * under the terms of the GNU Lesser General Public License as *
|
|
# * published by the Free Software Foundation, either version 2.1 of the *
|
|
# * License, or (at your option) any later version. *
|
|
# * *
|
|
# * FreeCAD is distributed in the hope that it will be useful, but *
|
|
# * WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
# * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU *
|
|
# * Lesser General Public License for more details. *
|
|
# * *
|
|
# * You should have received a copy of the GNU Lesser General Public *
|
|
# * License along with FreeCAD. If not, see *
|
|
# * <https://www.gnu.org/licenses/>. *
|
|
# * *
|
|
# ***************************************************************************
|
|
|
|
import FreeCAD as App
|
|
|
|
if App.GuiUp:
|
|
import FreeCADGui as Gui
|
|
from draftutils.todo import ToDo
|
|
|
|
class _Draft_DocObserver():
|
|
|
|
# See: /src/Gui/DocumentObserverPython.h
|
|
|
|
def slotDeletedDocument(self, gui_doc):
|
|
_finish_command_on_doc_close(gui_doc)
|
|
|
|
_doc_observer = None
|
|
|
|
def _doc_observer_start():
|
|
global _doc_observer
|
|
if _doc_observer is None:
|
|
_doc_observer = _Draft_DocObserver()
|
|
Gui.addDocumentObserver(_doc_observer)
|
|
|
|
def _doc_observer_stop():
|
|
global _doc_observer
|
|
try:
|
|
if _doc_observer is not None:
|
|
Gui.removeDocumentObserver(_doc_observer)
|
|
except:
|
|
pass
|
|
_doc_observer = None
|
|
|
|
def _finish_command_on_doc_close(gui_doc):
|
|
"""Finish the active Draft or BIM command if the related document has been
|
|
closed. Only works for commands that have set `App.activeDraftCommand.doc`
|
|
and use a task panel.
|
|
"""
|
|
if getattr(App, "activeDraftCommand", None) \
|
|
and getattr(App.activeDraftCommand, "doc", None) == gui_doc.Document:
|
|
if hasattr(App.activeDraftCommand, "ui"):
|
|
if hasattr(App.activeDraftCommand.ui, "reject"):
|
|
ToDo.delay(App.activeDraftCommand.ui.reject, None)
|
|
elif hasattr(App.activeDraftCommand.ui, "escape"):
|
|
ToDo.delay(App.activeDraftCommand.ui.escape, None)
|
|
elif hasattr(Gui, "Snapper") \
|
|
and hasattr(Gui.Snapper, "ui") \
|
|
and hasattr(Gui.Snapper.ui, "escape"):
|
|
ToDo.delay(Gui.Snapper.ui.escape, None)
|