Ticket #467: server.2.py

# Copyright (c) 2007-2009 The PyAMF Project.
# See LICENSE for details.

"""
ByteArray test server.
"""

from pyamf.amf3 import ByteArray
from pyamf import register_class
import os, os.path


class Example(object):
  def  __init__(self):
      self.id = 1
      self.data = ByteArray()


def getExample():
    return Example()

def echoExample(example):
    return example

register_class(Example, 'Example')


def app(environ, start_response):
            if environ['PATH_INFO'] == '/crossdomain.xml':
                fn = os.path.join(os.getcwd(), os.path.dirname(__file__),
                    'crossdomain.xml')
    
                fp = open(fn, 'rt')
                buffer = fp.readlines()
                fp.close()
    
                start_response('200 OK', [
                    ('Content-Type', 'application/xml'),
                    ('Content-Length', str(len(''.join(buffer))))
                ])
    
                return buffer
    
            return gw(environ, start_response)

services = {
    'get': getExample,
    'echo': echoExample
}

if __name__ == '__main__':
    from pyamf.remoting.gateway.wsgi import WSGIGateway
    from wsgiref import simple_server
    
    gw = WSGIGateway(services)
    
    def app(environ, start_response):
            if environ['PATH_INFO'] == '/crossdomain.xml':
                fn = os.path.join(os.getcwd(), os.path.dirname(__file__),
                    'crossdomain.xml')
    
                fp = open(fn, 'rt')
                buffer = fp.readlines()
                fp.close()
    
                start_response('200 OK', [
                    ('Content-Type', 'application/xml'),
                    ('Content-Length', str(len(''.join(buffer))))
                ])
    
                return buffer
    
            return gw(environ, start_response)

    

    httpd = simple_server.WSGIServer(
        ('localhost', 8000),
        simple_server.WSGIRequestHandler,
    )

    httpd.set_app(app)

    print "Running ByteArray Test AMF gateway on http://localhost:8000"

    try:
        httpd.serve_forever()
    except KeyboardInterrupt:
        pass