Register all python widgets

Author: vidartf

js/scripts/generate-wrappers.js

@@ -653,6 +653,9 @@ function PythonWrapper(modulePath, className) {
     this.pyBaseRelativePath = path.relative(this.destDirAbsolutePath, pySrcDir);
     this.pyBaseRelativePath = relativePathToPythonImportPath(this.pyBaseRelativePath);
 
+    // check if manual file exists
+    this.hasOverride = fs.existsSync(this.pyDestPath);
+
     this.hasParameters = false;
 
     this.config = getClassConfig(this.className);
@@ -680,6 +683,7 @@ function PythonWrapper(modulePath, className) {
         superClass: this.superClass,
         properties: this.properties,
         dependencies: this.dependencies,
+        hasOverride: this.hasOverride,
     };
 
     // Render template

js/scripts/templates/py_wrapper.mustache

@@ -1,6 +1,6 @@
 import inspect
 
-from ipywidgets import Widget, DOMWidget, widget_serialization, Color
+from ipywidgets import Widget, DOMWidget, widget_serialization, Color, register
 from traitlets import (
     Unicode, Int, CInt, Instance, ForwardDeclaredInstance, This, Enum,
     Tuple, List, Dict, Float, CFloat, Bool, Union, Any,
@@ -17,6 +17,9 @@ from {{ superClass.pyRelativePath }} import {{ superClass.className }}
 from {{ dep.pyRelativePath }} import {{ dep.className }}
 {{/each}}
 
+{{#unless hasOverride}}
+@register
+{{/unless}}
 class {{ className }}({{ superClass.className }}):
     """{{ className }}
 

pythreejs/core/BufferAttribute.py

@@ -1,9 +1,11 @@
 import numpy as np
+from ipywidgets import register
 from traitlets import validate, TraitError
 
 from .BufferAttribute_autogen import BufferAttribute as BaseBufferAttribute
 
 
+@register
 class BufferAttribute(BaseBufferAttribute):
 
     @validate('array')

pythreejs/core/Object3D.py

@@ -1,8 +1,12 @@
+from ipywidgets import register
+
 from .._base.Three import ThreeWidget
 from .Object3D_autogen import Object3D as Object3DBase
 
+
+@register
 class Object3D(Object3DBase):
-    
+
     def add(self, child):
         children_list = list(self.children)
         children_list.append(child)
@@ -15,7 +19,7 @@ def remove(self, child):
 
     def lookAt(self, vector):
         self.exec_three_obj_method('lookAt', vector)
-    
+
     def rotateX(self, rad):
         self.exec_three_obj_method('rotateX', rad)
 

pythreejs/geometries/PlainBufferGeometry.py

@@ -1,7 +1,9 @@
 
+from ipywidgets import register
 from .PlainBufferGeometry_autogen import PlainBufferGeometry as PlainBufferGeometryBase
 
 
+@register
 class PlainBufferGeometry(PlainBufferGeometryBase):
 
     @classmethod

pythreejs/geometries/PlainGeometry.py

@@ -1,7 +1,8 @@
-
+from ipywidgets import register
 from .PlainGeometry_autogen import PlainGeometry as PlainGeometryBase
 
 
+@register
 class PlainGeometry(PlainGeometryBase):
 
     @classmethod

pythreejs/textures/DataTexture.py

@@ -1,9 +1,11 @@
 import numpy as np
+from ipywidgets import register
 from traitlets import validate, TraitError
 
 from .DataTexture_autogen import DataTexture as BaseDataTexture
 
 
+@register
 class DataTexture(BaseDataTexture):
 
     @validate('data')