#!/usr/bin/env python
# ampserver.py -- A Twisted AMP server.
# From: Twisted-8.0.1/doc/core/examples/ampserver.py

from twisted.internet import reactor
from twisted.internet.protocol import Factory
from twisted.protocols import amp

class Sum(amp.Command):
    arguments = [('a', amp.Integer()),
                 ('b', amp.Integer())]
    response = [('total', amp.Integer())]

class Divide(amp.Command):
    arguments = [('numerator', amp.Integer()),
                 ('denominator', amp.Integer())]
    response = [('result', amp.Float())]
    errors = {ZeroDivisionError: 'ZERO_DIVISION'}

class Math(amp.AMP):
    def sum(self, a, b):
        total = a + b
        print 'Did a sum: %d + %d = %d' % (a, b, total)
        return {'total': total}
    Sum.responder(sum)

    def divide(self, numerator, denominator):
        result = numerator / denominator
        print 'Divided: %d / %d = %d' % (numerator, denominator, total)
        return {'result': result}
    Divide.responder(divide)

if __name__ == '__main__':
    pf = Factory()
    pf.protocol = Math
    
    reactor.listenTCP(1234, pf)
    reactor.run()
