New experimental wsgi module

0da00e19 · willmcgugan · c3560e7f · 0da00e19 · 0da00e19 · 0da00e19
Commit 0da00e19 authored Aug 15, 2010 by willmcgugan
Show whitespace changes
Inline Side-by-side

Showing with 239 additions and 0 deletions

fs/expose/wsgi/__init__.py
+1 -0

fs/expose/wsgi/dirtemplate.py
+85 -0

fs/expose/wsgi/serve_home.py
+5 -0

fs/expose/wsgi/wsgi.py
+148 -0

No files found.
--- a/fs/expose/wsgi/__init__.py
+++ b/fs/expose/wsgi/__init__.py
+from wsgi import serve_fs
--- a/fs/expose/wsgi/dirtemplate.py
+++ b/fs/expose/wsgi/dirtemplate.py
+template = """
+<html>
+
+<head>
+<title>${path}</title>
+<style type="text/css">
+
+body {
+    font-family:Arial, Verdana;   
+    margin:0px;
+    padding:0px; 
+}
+
+table.dirlist {
+    margin:0 auto;    
+    font-size:13px;
+    color:#666;
+}
+
+table.dirlist tr.r1 {
+    background-color:#f4f4f4;
+}
+
+table.dirlist td {
+    padding:6px 12px;
+    margin:0px;
+}
+
+table.dirlist td a.link-dir {
+    font-weight:bold;
+}
+
+table.dirlist td a {
+    text-decoration:none;    
+}
+
+table.dirlist td a:hover {
+    text-decoration:underline;
+}
+
+table.dirlist tr {
+    padding:4px;
+    margin:0px;
+}
+
+table.dirlist tr:hover {
+    background-color:#e9e9e9;
+}
+
+
+</style>
+
+</head>
+
+<body>
+
+<div class="dirlist-container">
+
+<table class="dirlist">
+        
+    <tbody>
+    
+        % for i, entry in enumerate(dirlist):
+        
+        <tr class="${entry['type']} r${i%2}">
+                    
+            <td><a class="link-${entry['type']}" href="${ entry['path'] }">${entry['name']}</a></td>
+            <td>${entry['size']}</td>
+            <td>${entry['created_time']}</td>
+        
+        </tr>
+        % endfor
+    
+    </tbody>
+
+</table>
+
+</div>
+
+</body>
+
+</html>
+
+"""
\ No newline at end of file
--- a/fs/expose/wsgi/serve_home.py
+++ b/fs/expose/wsgi/serve_home.py
+from fs.osfs import OSFS
+from wsgi import serve_fs
+osfs = OSFS('~/')
+application = serve_fs(osfs)
\ No newline at end of file
--- a/fs/expose/wsgi/wsgi.py
+++ b/fs/expose/wsgi/wsgi.py
+import json
+import urlparse
+import mimetypes
+
+from fs.errors import FSError
+from fs.path import basename, pathsplit
+
+from datetime import datetime
+
+try:
+    from mako.template import Template
+except ImportError:
+    print "Requires mako templates http://www.makotemplates.org/"
+    raise
+
+
+class Request(object):
+    """Very simple request object"""
+    def __init__(self, environ, start_response):
+        self.environ = environ
+        self.start_response = start_response        
+        self.path = environ.get('PATH_INFO')
+        
+
+class WSGIServer(object):
+    """Light-weight WSGI server that exposes an FS"""
+    
+    def __init__(self, serve_fs, indexes=True, dir_template=None, chunk_size=16*1024*1024):
+        
+        if dir_template is None:
+            from dirtemplate import template as dir_template            
+        
+        self.serve_fs = serve_fs
+        self.indexes = indexes        
+        self.chunk_size = chunk_size
+        
+        self.dir_template = Template(dir_template)     
+     
+    def __call__(self, environ, start_response):
+        
+        request = Request(environ, start_response)
+        
+        if not self.serve_fs.exists(request.path):
+            return self.serve_404(request)     
+        
+        if self.serve_fs.isdir(request.path):
+            if not self.indexes:
+                return self.serve_404(request)
+            return self.serve_dir(request)
+        else:
+            return self.serve_file(request)
+    
+    
+    def serve_file(self, request):    
+        """Serve a file, guessing a mime-type"""
+        path = request.path                                            
+        serving_file = None
+        try:            
+            serving_file = self.serve_fs.open(path, 'rb')
+        except Exception, e:
+            if serving_file is not None:
+                serving_file.close()
+            return self.serve_500(request, str(e))
+        
+        mime_type = mimetypes.guess_type(basename(path))        
+        file_size = self.serve_fs.getsize(path)
+        headers = [('Content-Type', mime_type),
+                   ('Content-Length', str(file_size))]
+        
+        def gen_file():
+            try:
+                while True:
+                    data = serving_file.read(self.chunk_size)
+                    if not data:
+                        break
+                    yield data
+            finally:
+                serving_file.close()
+                
+        request.start_response('200 OK',
+                               headers)
+        return gen_file()                
+    
+    def serve_dir(self, request):
+        """Serve an index page"""
+        fs = self.serve_fs
+        isdir = fs.isdir        
+        path = request.path                     
+        dirinfo = fs.listdirinfo(path, full=True, absolute=True)        
+        entries = []
+        
+        for p, info in dirinfo:
+            entry = {}
+            entry['path'] = p
+            entry['name'] = basename(p)
+            entry['size'] = info.get('size', 'unknown')
+            entry['created_time'] = info.get('created_time')                                                        
+            if isdir(p):
+                entry['type'] = 'dir'
+            else:
+                entry['type'] = 'file'                
+                
+            entries.append(entry)
+            
+        # Put dirs first, and sort by reverse created time order
+        no_time = datetime(1970, 1, 1, 1, 0)
+        entries.sort(key=lambda k:(k['type'] == 'dir', k.get('created_time') or no_time), reverse=True)
+        
+        # Turn datetime to text and tweak names
+        for entry in entries:
+            t = entry.get('created_time')
+            if t and hasattr(t, 'ctime'):
+                entry['created_time'] = t.ctime()
+            if entry['type'] == 'dir':
+                entry['name'] += '/'
+    
+        # Add an up dir link for non-root
+        if path not in ('', '/'):
+            entries.insert(0, dict(name='../', path='../', type="dir", size='', created_time='..'))
+            
+        # Render the mako template
+        html = self.dir_template.render(**dict(fs=self.serve_fs,
+                                               path=path,
+                                               dirlist=entries))
+        
+        request.start_response('200 OK', [('Content-Type', 'text/html'),
+                                          ('Content-Length', '%i' % len(html))])
+        
+        return [html]
+        
+            
+    def serve_404(self, request, msg='Not found'):
+        """Serves a Not found page"""
+        request.start_response('404 NOT FOUND', [('Content-Type', 'text/html')])
+        return [msg]
+    
+    def serve_500(self, request, msg='Unable to complete request'):
+        """Serves an internal server error page"""        
+        request.start_response('500 INTERNAL SERVER ERROR', [('Content-Type', 'text/html')])
+        return [msg]
+        
+        
+def serve_fs(fs, indexes=True):
+    """Serves an FS object via WSGI"""
+    application = WSGIServer(fs, indexes)
+    return application     
+            
\ No newline at end of file