add wrappers for build-in modules

Author: Leandro Inocencio

example_auto_nodes/module_nodes.py

@@ -2,27 +2,18 @@
                                     get_functions_from_module,
                                     get_functions_from_type)
 
-import math
 import os
 import sys
 import random
 import numpy
 import numpydoc.docscrape
 import inspect
 
-# add basic math functions to math library
-math.add = lambda x, y: x + y
-math.sub = lambda x, y: x - y
-math.mul = lambda x, y: x * y
-math.div = lambda x, y: x / y
-
-
-def get_value(self,index):
-    return self[index]
-
-
-def get_item(self,index):
-    return self.items()[index]
+import example_auto_nodes.wrappers.math as _math
+import example_auto_nodes.wrappers.list as _list
+import example_auto_nodes.wrappers.dict as _dict
+import example_auto_nodes.wrappers.str as _str
+import example_auto_nodes.wrappers.tuple as _tuple
 
 
 class MathModuleNode(ModuleNode):
@@ -36,7 +27,7 @@ class MathModuleNode(ModuleNode):
     # set the initial default node name.
     NODE_NAME = 'math module'
 
-    module_functions = get_functions_from_type(math)
+    module_functions = get_functions_from_type(_math)
 
     def __init__(self):
         super(MathModuleNode, self).__init__(float, float)
@@ -162,8 +153,8 @@ class StringFunctionsNode(ModuleNode):
     # set the initial default node name.
     NODE_NAME = 'String Functions'
 
-    module_functions = get_functions_from_type(str)
-    module_functions["get value"] = get_value
+    module_functions = get_functions_from_type(_str)
+
 
     def __init__(self):
         super(StringFunctionsNode, self).__init__()
@@ -180,8 +171,7 @@ class ListFunctionsNode(ModuleNode):
     # set the initial default node name.
     NODE_NAME = 'List Functions'
 
-    module_functions = get_functions_from_type(list)
-    module_functions["get value"] = get_value
+    module_functions = get_functions_from_type(_list)
 
     def __init__(self):
         super(ListFunctionsNode, self).__init__()
@@ -198,8 +188,7 @@ class DictFunctionsNode(ModuleNode):
     # set the initial default node name.
     NODE_NAME = 'Dict Functions'
 
-    module_functions = get_functions_from_type(list)
-    module_functions["get item"] = get_item
+    module_functions = get_functions_from_type(_dict)
 
     def __init__(self):
         super(DictFunctionsNode, self).__init__()
@@ -216,8 +205,7 @@ class TupleFunctionsNode(ModuleNode):
     # set the initial default node name.
     NODE_NAME = 'Tuple Functions'
 
-    module_functions = get_functions_from_type(tuple)
-    module_functions["get value"] = get_value
+    module_functions = get_functions_from_type(_tuple)
 
     def __init__(self):
         super(TupleFunctionsNode, self).__init__()

example_auto_nodes/wrappers/__init__.py

@@ -0,0 +1,45 @@
+def clear(self):
+    self.clear()
+    return self
+
+
+def copy(self):
+    return self.copy()
+
+
+def fromkeys(self, iterable, value=None):
+    return self.fromkeys(iterable, value)
+
+
+def get(self, key, default=None):
+    return self.get(key, default)
+
+
+def items(self):
+    return list(self.items())
+
+
+def keys(self):
+    return list(self.keys())
+
+
+def pop(self, index):
+    return self.pop(index)
+
+
+def popitem(self):
+    return self.popitem()
+
+
+def setdefault(self, key, default=None):
+    self.setdefault(key, default)
+    return self
+
+
+def update(self, e):
+    self.update(e)
+    return self
+
+
+def values(self):
+    return list(self.values())

example_auto_nodes/wrappers/dict.py

@@ -0,0 +1,52 @@
+def append(self, object):
+    self.append(object)
+    return self
+
+
+def clear(self):
+    self.clear()
+    return self
+
+
+def copy(self):
+    return self.copy()
+
+
+def count(self, value):
+    return self.count(value)
+
+
+def extend(self, iterable):
+    self.extend(iterable)
+    return self
+
+def index(self, value):
+    return self.index(value)
+
+
+def insert(self, index, obj):
+    self.insert(index, obj)
+    return self
+
+
+def pop(self, index=-1):
+    return self.pop(index)
+
+
+def remove(self, value):
+    return self.remove(value)
+
+
+def reverse(self):
+    self.reverse()
+    return self
+
+
+def sort(self):
+    self.sort()
+    return self
+
+
+# custom functions
+def get(self, index):
+    return self.__getitem__(index)

example_auto_nodes/wrappers/list.py

@@ -0,0 +1,52 @@
+from math import *
+
+_pi = pi
+_e = e
+_tau = tau
+_inf = inf
+_nan = nan
+
+
+# add contants as functions
+
+def pi():
+    print('pi', _pi)
+    return _pi
+
+
+def e():
+    print('e', _e)
+    return _e
+
+
+def tau():
+    print('tau', _tau)
+    return _tau
+
+
+def inf():
+    print('inf', _inf)
+    return _inf
+
+
+def nan():
+    print('nan', _nan)
+    return _nan
+
+
+# add basic math functions to math library
+
+def add(x, y):
+    return x + y
+
+
+def sub(x, y):
+    return x - y
+
+
+def mul(x, y):
+    return x * y
+
+
+def div(x, y):
+    return x / y