Request helper functions

[twisted]

Request helper functions

[notoriousno]

I figured out how to do this with a bit of Twisted basics. I started by subclassing the Site and Request classes (twisted.web.server). In the Request subclass, I added the helper functions. Then I overloaded the buildProto() function in the Site subclass so the Site.requestFactory is set to my subclassed Request. Then I slightly modified what klein.app.Klein.run does and start a TCP server using my subclasses. Works like a charm!

from klein import Klein
from twisted.web import http, server
from twisted.internet import reactor


#----- Subclasses -----#
class KleinHTTPRequest(server.Request):
    """ Request subclass """

    def getter(self, key, default=None):
        # ...

    def getOne(self, key, default=None):
        # ...

    def getOneCast(self, key, cast, default=None, decoding=None):
       # ...

class KleinSite(server.Site):
    """ Site subclass """

    def buildProtocol(self, addr):
        channel = http.HTTPFactory.buildProtocol(self, addr)
        channel.requestFactory = KleinHTTPRequest            # produce subclassed Request
        channel.site = self
        return channel


#----- Routes -----#
app = Klein()

@app.route('/hello', methods=['POST'])
def home(request):
    return b'hello ' + request.getOne('name', 'world')


#----- Start server -----#
reactor.listenTCP(8000, KleinSite(app.resource()), interface='localhost')
reactor.run()