From b118f7f14fc93b60fc547d76bb5766ee1bd593ae Mon Sep 17 00:00:00 2001
From: David Halter <davidhalter88@gmail.com>
Date: Sat, 12 May 2012 01:24:19 +0200
Subject: [PATCH] builtin mixins support (functions custom defined)

---
 builtin.py           | 48 ++++++++++++++++++++++++++++++++++++++++----
 mixin/__builtin__.py |  7 +++++++
 2 files changed, 51 insertions(+), 4 deletions(-)
 create mode 100644 mixin/__builtin__.py

diff --git a/builtin.py b/builtin.py
index d0788399..b39d287e 100644
--- a/builtin.py
+++ b/builtin.py
@@ -101,6 +101,32 @@ class Parser(CachedModule):
     def _get_source(self):
         return self._generate_code(self.module)
 
+    def _load_mixins(self):
+        self.mixin_funcs = {}
+        if not self.path:
+            try:
+                f = open(os.path.sep.join(['mixin', self.name]) + '.py')
+            except IOError:
+                pass
+            else:
+                code = f.read()
+                # TODO implement classes, how? not used yet
+                regex = r'^(def|class)\s+([\w\d]+)'
+                matches = list(re.finditer(regex, code, re.MULTILINE))
+                positions = [m.start() for m in matches]
+                for i, pos in enumerate(positions):
+                    try:
+                        code_block = code[pos:positions[i+1]]
+                    except IndexError:
+                        code_block = code[pos:len(code)]
+                    structure_name = matches[i].group(1)
+                    name = matches[i].group(2)
+                    if structure_name == 'def':
+                        self.mixin_funcs[name] = code_block
+                    else:
+                        raise NotImplementedError
+                    print code_block
+
     def _generate_code(self, scope, depth=0):
         """
         Generate a string, which uses python syntax as an input to the
@@ -127,6 +153,9 @@ class Parser(CachedModule):
                     stmts[n] = exe
             return classes, funcs, stmts, members
 
+        if depth == 0:
+            self._load_mixins()
+
         code = ''
         try:
             try:
@@ -154,10 +183,21 @@ class Parser(CachedModule):
         # functions
         for name, func in funcs.iteritems():
             params, ret = parse_function_doc(func)
-            code += 'def %s(%s):\n' % (name, params)
-            block = '"""\n%s\n"""\n' % func.__doc__
-            block += '%s\n\n' % ret
-            code += parsing.indent_block(block)
+            doc_str = parsing.indent_block('"""\n%s\n"""\n' % func.__doc__)
+            try:
+                mixin = self.mixin_funcs[name]
+            except:
+                # normal code generation
+                code += 'def %s(%s):\n' % (name, params)
+                code += doc_str
+                code += parsing.indent_block('%s\n\n' % ret)
+            else:
+                # generation of code with mixins
+                # find doc_str place
+                pos = re.search(r'\):\s*\n', mixin).end()
+                if pos is None:
+                    raise Exception("Builtin function not parsed correctly")
+                code += mixin[:pos] + doc_str + mixin[pos:]
 
         # class members (functions)
         for name, func in members.iteritems():
diff --git a/mixin/__builtin__.py b/mixin/__builtin__.py
new file mode 100644
index 00000000..e39b76dc
--- /dev/null
+++ b/mixin/__builtin__.py
@@ -0,0 +1,7 @@
+
+def next(iterator, default=None):
+    if hasattr("next"):
+        return iterator.next()
+    else:
+        return iterator.__next__()
+    return default