FEM: use super class FemCommands for ShellThickness and BeamSection C…

…ommands
FreeCAD · Nov 5, 2015 · 39d6d68 · 39d6d68
1 parent 59bafd5
commit 39d6d68
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 24 deletions.
diff --git a/src/Mod/Fem/_CommandFemBeamSection.py b/src/Mod/Fem/_CommandFemBeamSection.py
@@ -26,32 +26,29 @@
 
 
 import FreeCAD
+from FemCommands import FemCommands
 
 if FreeCAD.GuiUp:
     import FreeCADGui
     import FemGui
     from PySide import QtCore
 
 
-class _CommandFemBeamSection:
+class _CommandFemBeamSection(FemCommands):
     "The Fem_BeamSection command definition"
-    def GetResources(self):
-        return {'Pixmap': 'fem-beam-section',
-                'MenuText': QtCore.QT_TRANSLATE_NOOP("Fem_BeamSection", "FEM Beam Cross Section Definition ..."),
-                'Accel': "C, B",
-                'ToolTip': QtCore.QT_TRANSLATE_NOOP("Fem_BeamSection", "Creates a FEM Beam Cross Section")}
+    def __init__(self):
+        super(_CommandFemBeamSection, self).__init__()
+        self.resources = {'Pixmap': 'fem-beam-section',
+                          'MenuText': QtCore.QT_TRANSLATE_NOOP("Fem_BeamSection", "FEM Beam Cross Section Definition ..."),
+                          'Accel': "C, B",
+                          'ToolTip': QtCore.QT_TRANSLATE_NOOP("Fem_BeamSection", "Creates a FEM Beam Cross Section")}
+        self.is_active = 'with_analysis'
 
     def Activated(self):
         FreeCAD.ActiveDocument.openTransaction("Create FemBeamSection")
         FreeCADGui.addModule("FemBeamSection")
         FreeCADGui.doCommand("FemGui.getActiveAnalysis().Member = FemGui.getActiveAnalysis().Member + [FemBeamSection.makeFemBeamSection()]")
 
-    def IsActive(self):
-        if FemGui.getActiveAnalysis():
-            return True
-        else:
-            return False
-
 
 if FreeCAD.GuiUp:
     FreeCADGui.addCommand('Fem_BeamSection', _CommandFemBeamSection())
diff --git a/src/Mod/Fem/_CommandFemShellThickness.py b/src/Mod/Fem/_CommandFemShellThickness.py
@@ -26,32 +26,29 @@
 
 
 import FreeCAD
+from FemCommands import FemCommands
 
 if FreeCAD.GuiUp:
     import FreeCADGui
     import FemGui
     from PySide import QtCore
 
 
-class _CommandFemShellThickness:
+class _CommandFemShellThickness(FemCommands):
     "The Fem_ShellThickness command definition"
-    def GetResources(self):
-        return {'Pixmap': 'fem-shell-thickness',
-                'MenuText': QtCore.QT_TRANSLATE_NOOP("Fem_ShellThickness", "FEM Shell Plate Thickness Definition ..."),
-                'Accel': "C, S",
-                'ToolTip': QtCore.QT_TRANSLATE_NOOP("Fem_ShellThickness", "Creates a FEM Shell Thickness")}
+    def __init__(self):
+        super(_CommandFemShellThickness, self).__init__()
+        self.resources = {'Pixmap': 'fem-shell-thickness',
+                          'MenuText': QtCore.QT_TRANSLATE_NOOP("Fem_ShellThickness", "FEM Shell Plate Thickness Definition ..."),
+                          'Accel': "C, S",
+                          'ToolTip': QtCore.QT_TRANSLATE_NOOP("Fem_ShellThickness", "Creates a FEM Shell Thickness")}
+        self.is_active = 'with_analysis'
 
     def Activated(self):
         FreeCAD.ActiveDocument.openTransaction("Create FemShellThickness")
         FreeCADGui.addModule("FemShellThickness")
         FreeCADGui.doCommand("FemGui.getActiveAnalysis().Member = FemGui.getActiveAnalysis().Member + [FemShellThickness.makeFemShellThickness()]")
 
-    def IsActive(self):
-        if FemGui.getActiveAnalysis():
-            return True
-        else:
-            return False
-
 
 if FreeCAD.GuiUp:
     FreeCADGui.addCommand('Fem_ShellThickness', _CommandFemShellThickness())