Make all variables class instance variables

Fixes #29

Author: alan412

src/blocks/mrc_class_method_def.ts

@@ -24,6 +24,8 @@ import { MRC_STYLE_CLASS_BLOCKS } from '../themes/styles';
 import { createFieldNonEditableText } from '../fields/FieldNonEditableText'
 import * as ChangeFramework from './utils/change_framework'
 import { getLegalName } from './utils/python';
+import { Order } from 'blockly/python';
+import { ExtendedPythonGenerator } from '../editor/extended_python_generator';
 
 export const BLOCK_NAME = 'mrc_class_method_def';
 
@@ -351,10 +353,6 @@ export const setup = function() {
   Blockly.Blocks[PARAM_CONTAINER_BLOCK_NAME] = METHOD_PARAM_CONTAINER;
 };
 
-import { Order } from 'blockly/python';
-import { ExtendedPythonGenerator } from '../editor/extended_python_generator';
-
-
 export const pythonFromBlock = function (
     block: ClassMethodDefBlock,
     generator: ExtendedPythonGenerator,
@@ -394,7 +392,7 @@ export const pythonFromBlock = function (
         xfix2 = xfix1;
     }
     if(block.mrcPythonMethodName == '__init__'){
-        branch = generator.INDENT + "self.mechanisms = []\n" + branch;
+        branch = generator.defineClassVariables(block.workspace) + branch;
     }    
     if (returnValue) {
         returnValue = generator.INDENT + 'return ' + returnValue + '\n';

src/blocks/setup_custom_blocks.ts

@@ -1,4 +1,3 @@
-import * as Blockly from 'blockly';
 
 import * as CallPythonFunction from './mrc_call_python_function';
 import * as GetPythonEnumValue from './mrc_get_python_enum_value';

src/editor/extended_python_generator.ts

@@ -39,6 +39,38 @@ export class ExtendedPythonGenerator extends PythonGenerator {
     super('Python');
   }
 
+  init(workspace: Blockly.Workspace){
+    super.init(workspace);
+    // This will have all variables in the defintion 'variables' so we will need to destroy it and make our own
+    delete this.definitions_['variables'];
+
+    const defvars = [];
+    // Add developer variables (not created or named by the user).
+    const devVarList = Blockly.Variables.allDeveloperVariables(workspace);
+    for (let i = 0; i < devVarList.length; i++) {
+      defvars.push(
+        this.nameDB_!.getName(devVarList[i], Blockly.Names.DEVELOPER_VARIABLE_TYPE) + ' = None',
+      );
+    }
+    this.definitions_['variables'] = defvars.join('\n');     
+    // user variables are dealt with in init code generation
+  }
+
+  defineClassVariables(workspace: Blockly.Workspace) : string{
+      const classVars = Blockly.Variables.allUsedVarModels(workspace);
+
+      let variableDefinitions = '';
+      
+      for (let i = 0; i < classVars.length; i++) {
+          variableDefinitions += this.INDENT + this.getVariableName(classVars[i].getId()) + ' = None\n';
+      }
+      return variableDefinitions;
+  }
+  getVariableName(name : string) : string{
+    const varName = super.getVariableName(name);
+    return "self." + varName;
+  }
+
   workspaceToCode(workspace: Blockly.Workspace, context: GeneratorContext): string {
     this.workspace = workspace;
     this.context = context;