Add super to derived methods

Author: alan412

src/blocks/mrc_class_method_def.ts

@@ -365,9 +365,9 @@ export const pythonFromBlock = function (
         branch = generator.INDENT + 'super.__init__(' + class_specific + ')\n' +
             generator.defineClassVariables() + branch;
     }
-    else if (funcName == 'update'){
+    else if (generator.inBaseClassMethod(blocklyName)){
         // Special case for update, to also call the update method of the base class
-        branch = generator.INDENT + 'super.update()\n' + branch;
+        branch = generator.INDENT + 'super.' + blocklyName + '()\n' + branch;
     }
     if (returnValue) {
         returnValue = generator.INDENT + 'return ' + returnValue + '\n';

src/editor/extended_python_generator.ts

@@ -212,6 +212,34 @@ export class ExtendedPythonGenerator extends PythonGenerator {
     }
     return ''
   }
+
+  /**
+   * This returns the list of methods that are derived from so that mrc_class_method_def 
+   * knows whether to call super() or not.
+   * @returns list of method names
+   */
+  getBaseClassMethods() : string[] {
+    let classParent = this.context?.getClassParent();
+    if (classParent == 'OpMode'){
+      return ['start', 'loop', 'stop'];
+    }
+    else if (classParent == 'Mechanism') {
+      return ['start', 'update', 'stop'];
+    }
+    else if (classParent == 'RobotBase'){
+      return ['start', 'update', 'stop'];
+    }
+    return [];
+  }
+
+  /**
+   * Returns true if the method is in the base class.
+   * @param methodName the name of the method to check
+   */
+  inBaseClassMethod(methodName: string): boolean {
+    const baseMethods = this.getBaseClassMethods();
+    return baseMethods.includes(methodName);
+  }
 }
 
 export const extendedPythonGenerator = new ExtendedPythonGenerator();