Work on pseudo-REST dispatcher

1 files changed, 58 insertions, 0 deletions

diff --git a/beweb/beweb/prest.py b/beweb/beweb/prest.py
new file mode 100644
index 0000000..3a94ab5
--- /dev/null
+++ b/beweb/beweb/prest.py
@@ -0,0 +1,58 @@
+from unittest import TestCase
+import unittest
+"""A pseudo-REST dispatching method in which only the noun comes from the path.
+The action performed will depend on kwargs.
+"""
+class PrestHandler(object):
+    def __init__(self):
+        object.__init__(self)
+
+    def decode(self, path, data=None, has_id=False):
+        """Convert the """
+        if data is None:
+            data = {}
+        if len(path) < 2 or not hasattr(self, path[1]):
+            if len(path) == 1:
+                resource = self.instantiate(*data)
+            else:
+                resource = None
+            return self, resource, data, path[1:] 
+        if len(path) > 2:
+            data[path[1]] = path[2]
+        return getattr(self, path[1]).decode(path[2:], data)
+
+    def default(self, *args, **kwargs):
+        child, resource, data, extra = self.decode([None,] + list(args))
+        child.dispatch(*([data, resource]+extra), **kwargs)
+
+
+class PrestTester(TestCase):
+    def test_decode(self):
+        class ProjectHandler(PrestHandler):
+            def dispatch(self, project_data, project, *args, **kwargs):
+                self.project_id = project_data['project']
+                self.resource = project
+                self.args = args
+                self.kwargs = kwargs
+
+            def instantiate(self, project):
+                return [project]
+
+        foo = PrestHandler()
+        foo.project = ProjectHandler()
+        handler, resource, data, extra = foo.decode([None, 'project', '83', 
+                                                     'bloop', 'yeah'])
+        assert handler is foo.project
+        self.assertEqual({'project': '83'}, data)
+        self.assertEqual(['bloop', 'yeah'], extra)
+        foo.default(*['project', '27', 'extra'], **{'a':'b', 'b':'97'})
+        self.assertEqual(foo.project.args, ('extra',))
+                
+def test():
+    patchesTestSuite = unittest.makeSuite(PrestTester,'test')
+    runner = unittest.TextTestRunner(verbosity=0)
+    return runner.run(patchesTestSuite)
+    
+
+if __name__ == "__main__":
+    test()