From: Joe Presbrey <presbrey@mit.edu>
Date: Fri, 13 Mar 2009 08:20:50 +0000 (+0000)
Subject: begin restructuring
X-Git-Url: http://andersk.mit.edu/gitweb/sql.git/commitdiff_plain/5cb62ba942e735545295a8cce19be93688557719

begin restructuring

git-svn-id: svn://presbrey.mit.edu/sql@170 a142d4bd-2cfb-0310-9673-cb33a7e74f58
---

diff --git a/lib/python/sql/__init__.py b/lib/python/mitsql/__init__.py
similarity index 100%
rename from lib/python/sql/__init__.py
rename to lib/python/mitsql/__init__.py
diff --git a/lib/python/sql/db.py b/lib/python/mitsql/db.py
similarity index 100%
rename from lib/python/sql/db.py
rename to lib/python/mitsql/db.py
diff --git a/lib/python/mitsql/interface/__init__.py b/lib/python/mitsql/interface/__init__.py
new file mode 100644
index 0000000..41c4f7a
--- /dev/null
+++ b/lib/python/mitsql/interface/__init__.py
@@ -0,0 +1,15 @@
+import cherrypy
+
+from main import Root as root
+root = root()
+
+def map_uri():
+    print root.default()
+    #from cherrypy.lib.httpauth import parseAuthorization
+    #if not cherrypy.request.login:
+    #    authorization = cherrypy.request.headers.get('Authorization', None)
+    #    if authorization:
+    #        cherrypy.request.login = parseAuthorization(authorization)
+
+cherrypy.tools.map_uri = cherrypy.Tool('on_start_resource', map_uri)
+#cherrypy.tools.map_uri = cherrypy.Tool('before_request_body', map_uri)
diff --git a/lib/python/mitsql/interface/main.py b/lib/python/mitsql/interface/main.py
new file mode 100644
index 0000000..39f8591
--- /dev/null
+++ b/lib/python/mitsql/interface/main.py
@@ -0,0 +1,13 @@
+import cherrypy
+from pprint import pformat
+
+class Root(object):
+    def default(self, *args, **kw):
+        cherrypy.response.headers['Content-type'] = 'text/plain'
+        return pformat(dict(filter(lambda x: not x[0][:2] == '__', cherrypy.request.__dict__.items())))
+    default.exposed = True
+
+    def test(self):
+        return 'test'
+    test.exposed = True
+
diff --git a/lib/python/mitsql/templates.py b/lib/python/mitsql/templates.py
new file mode 100644
index 0000000..f728005
--- /dev/null
+++ b/lib/python/mitsql/templates.py
@@ -0,0 +1,155 @@
+# Joe Presbrey  <presbrey@mit.edu>
+
+import os, sys, types, time
+import threading 
+from pyinotify import WatchManager, Notifier, EventsCodes
+from Cheetah.Template import Template as _Template
+
+__all__ = ['Templates']
+
+class Template(object):
+    def __init__(self, name, ns={}, path=['templates']):
+        self._name, self._ns, self._path = name, ns, path+[name]
+        assert self._template()
+    
+    def _template(self):
+        return sys.modules['.'.join(self._path)]
+    
+    def __getattr__(self, name):
+        return Template(name, self._ns, self._path)
+    
+    def __call__(self, **kw):
+        namespaces = self._ns.copy()
+        namespaces.update(kw)
+        return getattr(self._template(),
+                       self._name)(namespaces=namespaces).respond()
+
+class Templates(object):
+    _base = ''
+    _mask = EventsCodes.ALL_FLAGS['IN_CREATE'] \
+          | EventsCodes.ALL_FLAGS['IN_MODIFY'] \
+          | EventsCodes.ALL_FLAGS['IN_DELETE'] \
+          | EventsCodes.ALL_FLAGS['IN_ATTRIB']
+    _watch_manager = None
+    _notifier = None
+    _thread = None
+    
+    _loaded = False
+    _files = {}
+    _modules = {}
+    
+    def __init__(self, template_path, global_namespace={}):
+        self._base = os.path.abspath(template_path)
+        self._event_init()
+        self._assert_module('templates', [template_path])
+        self._globals = global_namespace
+        self.load()
+        self._thread = threading.Thread(target=self.loop)
+        self._thread.setDaemon(True)
+        self._thread.start()
+
+    def _tree_walk(self, tree, path, func):
+        path = list(path)
+        if len(path) > 1:
+            x = path.pop(0)
+            if not x in tree:
+                tree[x] = {}
+            return self._tree_walk(tree[x], path, func)
+        elif callable(func):
+            return func(tree, path[0])
+
+    def _tree_add(self, tree, path, value):
+        def _add(tree, key):
+            tree[key] = value
+            return value
+        print self._tree_walk(tree, path, _add)
+    
+    def _assert_module(self, name, path=None):
+        if not name in self._modules:
+            self._modules[name] = types.ModuleType(name)
+            if path:
+                self._modules[name].__path__ = path
+            sys.modules[name] = self._modules[name]
+    
+    def _get_template_path(self, template_file):
+        r = template_file.startswith(self._base) \
+          and template_file[len(self._base)+1:].split(os.path.sep) or []
+        if r[-1].endswith('.tmpl'):
+            r[-1] = r[-1][:-5]
+        return tuple(r)
+    
+    def _compile(self, template_path):
+        template = _Template.compile(file=self._files[template_path],
+                                     returnAClass=False,
+                                     moduleName='.'.join(['templates']+list(template_path)),
+                                     className=template_path[-1],
+                                     useCache=False,
+                                     compilerSettings={'useStackFrames':False})
+        return template
+    
+    def add(self, template_file):
+        template_path = self._get_template_path(template_file)
+        template_name = '.'.join(['templates']+list(template_path))
+        template_pkg = '.'.join(['templates']+list(template_path[:-1]))
+        template_py = os.path.join(os.path.dirname(template_file), '%s.py' % template_path[-1])
+        self._assert_module(template_pkg, [os.path.dirname(template_file)])
+        self._files[template_path] = template_file
+        code = self._compile(template_path)
+        f_code = file(template_py, 'w')
+        f_code.write(code)
+        f_code.close()
+        if self._loaded and \
+           template_name in sys.modules:
+            reload(sys.modules[template_name])
+    
+    def remove(self, template_file):
+        #print 'removing', template_file
+        pass
+    
+    def _walk_load(self, args, dname, fnames):
+        for fname in fnames:
+            if fname.endswith('.tmpl'):
+                template_file = os.path.join(dname, fname)
+                self.add(template_file)
+    
+    def load(self):
+        if not self._loaded:
+            os.path.walk(self._base, self._walk_load, None)
+            for path, filename in self._files.items():
+                __import__('.'.join(['templates']+list(path)))
+            self._loaded = True
+    
+    def _event_handler(self, event):
+        if event.name.endswith('.tmpl') and not event.name.startswith('.'):
+            if event.maskname == 'IN_DELETE':
+                self.remove(event.pathname)
+            else:
+                time.sleep(0.5)
+                self.add(event.pathname)
+                
+    
+    def _event_init(self):
+        self._watch_manager = WatchManager()
+        self._watch_manager.add_watch(self._base,
+                                      self._mask,
+                                      self._event_handler,
+                                      rec=True,
+                                      auto_add=True)
+        self._notifier = Notifier(self._watch_manager)
+    
+    def loop(self):
+        self.load()
+        while True:
+            try:
+                self._notifier.process_events()
+                if self._notifier.check_events():
+                    self._notifier.read_events()
+            except KeyboardInterrupt:
+                self._notifier.stop()
+                break
+    
+    def __getattr__(self, key):
+        return Template(key, self._globals)
+
+if __name__ == '__main__':
+    t = Templates('/home/joe/templates')
diff --git a/lib/python/sql/util.py b/lib/python/mitsql/util.py
similarity index 100%
rename from lib/python/sql/util.py
rename to lib/python/mitsql/util.py