bin.py

import wsgiref.handlers
from django.utils import simplejson
from google.appengine.ext import webapp
from google.appengine.ext.webapp import template
from google.appengine.api import urlfetch
from models import Bin, Post
import urllib
import re
import hashlib
from cgi import FieldStorage
import logging
from google.appengine.api.labs import taskqueue

class NotFound(Exception): pass


class BinHandler(webapp.RequestHandler):
    def get(self):
        path = self.request.path
        if path[-1] == '/':
            self.redirect(path[:-1])
        path, feed = re.subn(r'^(.+)\/feed$', r'\1', path)
        bin = self._get_bin(path)
        if self.request.query_string:
            self._record_post(bin, True)
            self.redirect('/%s' % bin.name)
        else:
            posts = bin.post_set.order('-created').fetch(50)
            self.response.out.write(template.render('templates/bin.%s' % ('atom' if feed else 'html'), 
                {'bin':bin, 'posts':posts, 'request':self.request}))

    def post(self):
        bin = self._get_bin(self.request.path)
        post = self._record_post(bin)
        # TODO: This should maybe be a header thing
        if 'http://' in self.request.query_string:
            params = dict(self.request.POST.items())
            params['_url'] = self.request.query_string
            urlfetch.fetch(url='http://hookah.progrium.com/dispatch',
                            payload=urllib.urlencode(params), method='POST')
        taskqueue.add(url='/tasks/newpost', params={'ip': post.remote_addr, 'size': post.size, 'bin': bin.name})
        self.response.set_status(201)
        self.response.headers['Location'] = str("/%s" % bin.name)
        self.response.out.write('<html><head><meta http-equiv="refresh" content="0;url=/%s" /></head><body>201 Created. Redirecting...</body></html>' % bin.name)
    
    def head(self):
        bin = self._get_bin(self.request.path)
        if self.request.query_string:
            self._record_post(bin, True)
        else:
            self._record_post(bin)

    def handle_exception(self, exception, debug_mode):
        if isinstance(exception, NotFound):
            self.error(404)
        else:
            super(BinHandler, self).handle_exception(exception, debug_mode)

    def _record_post(self, bin, use_get=False):
        post = Post(bin=bin, remote_addr=self.request.remote_addr)
        post.headers        = dict(self.request.headers)
        try:
            post.body           = self.request.body
        except UnicodeDecodeError:
            #post.body_binary    = self.request.body
            pass
        post.query_string   = self.request.query_string
        post.form_data = []
        data_source = self.request.GET if use_get else self.request.POST
        post.size = len(post.body) if post.body else 0
        for k,v in data_source.items():
            if isinstance(v, FieldStorage):
                file_body = v.file.read()
                post.form_data.append([k, {
                    'file_name': v.filename,
                    'file_extension': v.filename.split('.')[-1],
                    'file_digest': hashlib.md5(file_body).hexdigest(),
                    'file_size': round(len(file_body) / 1024.0, 1),
                }])
                post.size += len(file_body)
            else:
                post.form_data.append([k,v])
        post.put()
        return post

    def _get_bin(self, path):
        name = path[1:].split('/')[0]
        bin = Bin.all().filter('name =', name).get()
        if bin:
            return bin
        else:
            raise NotFound()


if __name__ == '__main__':
    wsgiref.handlers.CGIHandler().run(webapp.WSGIApplication([
        ('/.*', BinHandler),
        ], debug=True))