← Back to branch summary

~ziad-sawalha/keystone/trunk

« back to all changes in this revision

Viewing changes to echo/echo/server.py

  • Committer: Brian Lamar
  • Date: 2011-05-24 15:36:52 UTC
  • mto: This revision was merged to the branch mainline in revision 103.
  • Revision ID: git-v1:6fdc83234f821a3b893c92ab496e6170818fefd2
-Removed .project file from project and added it to .gitignore
-Moved pylintrc -> .pylintrc, personal preference that this file should be available, but not seen
-Moved echo to examples directory, seemed a bit odd to be in the top level
-Moved management directory to tools, seemed a bit odd to be in the top level
-Moved pip-requires to tools/, and updated the reference to it in README.md

Show diffs side-by-side

added added

removed removed

Lines of Context:
echo/echo/server.py
1
 
# vim: tabstop=4 shiftwidth=4 softtabstop=4
2
 
# Copyright (c) 2010-2011 OpenStack, LLC.
3
 
#
4
 
# Licensed under the Apache License, Version 2.0 (the "License");
5
 
# you may not use this file except in compliance with the License.
6
 
# You may obtain a copy of the License at
7
 
#
8
 
#    http://www.apache.org/licenses/LICENSE-2.0
9
 
#
10
 
# Unless required by applicable law or agreed to in writing, software
11
 
# distributed under the License is distributed on an "AS IS" BASIS,
12
 
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
13
 
# implied.
14
 
# See the License for the specific language governing permissions and
15
 
# limitations under the License.
16
 
 
17
 
import eventlet
18
 
from eventlet import wsgi
19
 
from lxml import etree
20
 
import os
21
 
from paste.deploy import loadapp
22
 
import sys
23
 
from webob.exc import HTTPUnauthorized
24
 
 
25
 
 
26
 
# If ../echo/__init__.py exists, add ../ to Python search path, so that
27
 
# it will override what happens to be installed in /usr/(local/)lib/python...
28
 
POSSIBLE_TOPDIR = os.path.normpath(os.path.join(os.path.abspath(sys.argv[0]),
29
 
                                   os.pardir,
30
 
                                   os.pardir))
31
 
if os.path.exists(os.path.join(POSSIBLE_TOPDIR, 'echo', '__init__.py')):
32
 
    # also use the local keystone
33
 
    KEYSTONE_TOPDIR = os.path.normpath(os.path.join(POSSIBLE_TOPDIR,
34
 
                                                    os.pardir))
35
 
    if os.path.exists(os.path.join(KEYSTONE_TOPDIR,
36
 
                                   'keystone',
37
 
                                   '__init__.py')):
38
 
        sys.path.insert(0, KEYSTONE_TOPDIR)
39
 
    sys.path.insert(0, POSSIBLE_TOPDIR)
40
 
 
41
 
 
42
 
"""
43
 
Echo: a dummy service for OpenStack auth testing. It returns request info.
44
 
"""
45
 
 
46
 
 
47
 
class EchoApp(object):
48
 
    def __init__(self, environ, start_response):
49
 
        self.envr = environ
50
 
        self.start = start_response
51
 
        self.dom = self.toDOM(environ)
52
 
        echo_xsl = os.path.join(os.path.abspath(\
53
 
            os.path.dirname(__file__)), "xsl/echo.xsl")
54
 
        self.transform = etree.XSLT(etree.parse(echo_xsl))
55
 
 
56
 
    def __iter__(self):
57
 
        # We expect an X_AUTHORIZATION header to be passed in
58
 
        # We assume the request is coming from a trusted source. Middleware
59
 
        # is used to perform that validation.
60
 
        if 'HTTP_X_AUTHORIZATION' not in self.envr:
61
 
            self.start('401 Unauthorized', [('Content-Type',
62
 
                                             'application/json')])
63
 
            return iter(["401 Unauthorized"])
64
 
 
65
 
        if 'HTTP_X_IDENTITY_STATUS' not in self.envr:
66
 
            identity_status = "Unknown"
67
 
        else:
68
 
            identity_status = self.envr["HTTP_X_IDENTITY_STATUS"]
69
 
 
70
 
        print '  Received:'
71
 
        print '  Auth Status:', identity_status
72
 
        if 'HTTP_X_AUTHORIZATION' in self.envr:
73
 
            print '  Identity   :', self.envr['HTTP_X_AUTHORIZATION']
74
 
        if 'HTTP_X_TENANT' in self.envr:
75
 
            print '  Tenant     :', self.envr['HTTP_X_TENANT']
76
 
        if 'HTTP_X_GROUP' in self.envr:
77
 
            print '  Group      :', self.envr['HTTP_X_GROUP']
78
 
 
79
 
        accept = self.envr.get("HTTP_ACCEPT", "application/json")
80
 
        if accept == "application/xml":
81
 
            return self.toXML()
82
 
        else:
83
 
            return self.toJSON()
84
 
 
85
 
    def toJSON(self):
86
 
        self.start('200 OK', [('Content-Type', 'application/json')])
87
 
        yield str(self.transform(self.dom))
88
 
 
89
 
    def toXML(self):
90
 
        self.start('200 OK', [('Content-Type', 'application/xml')])
91
 
        yield etree.tostring(self.dom)
92
 
 
93
 
    def toDOM(self, environ):
94
 
        echo = etree.Element("{http://docs.openstack.org/echo/api/v1.0}echo",
95
 
                             method=environ["REQUEST_METHOD"],
96
 
                             pathInfo=environ["PATH_INFO"],
97
 
                             queryString=environ.get('QUERY_STRING', ""))
98
 
        content = etree.Element(
99
 
            "{http://docs.openstack.org/echo/api/v1.0}content")
100
 
        content.set("type", environ["CONTENT_TYPE"])
101
 
        content.text = ""
102
 
        inReq = environ["wsgi.input"]
103
 
        for line in inReq:
104
 
            content.text = content.text + line
105
 
        echo.append(content)
106
 
        return echo
107
 
 
108
 
 
109
 
def app_factory(global_conf, **local_conf):
110
 
    return EchoApp
111
 
 
112
 
if __name__ == "__main__":
113
 
    def usage():
114
 
        print "Runs Echo, the canonical OpenStack service, " \
115
 
                "with auth middleware"
116
 
        print "Options:"
117
 
        print "-h, --help  : show this usage information"
118
 
        print "-b, --basic : run with basic auth (uses echo_basic.ini)"
119
 
        print "-r, --remote: run with remote auth on port 8100" \
120
 
                "(uses echo_remote.ini)"
121
 
        print "-i, --ini filename: run with specified ini file"
122
 
        print "-p, --port: specifies port to listen on (default is 8090)"
123
 
        print "by default will run with local, token auth (uses echo.ini)"
124
 
 
125
 
    import getopt
126
 
    try:
127
 
        opts, args = getopt.getopt(sys.argv[1:],
128
 
                                   "hbrp:i:",
129
 
                                   ["help", "basic", "remote", "port", "ini"])
130
 
    except getopt.GetoptError:
131
 
        usage()
132
 
        sys.exit()
133
 
 
134
 
    port = 0
135
 
    ini = "echo.ini"
136
 
    auth_name = "local Token Auth"
137
 
 
138
 
    for opt, arg in opts:
139
 
        if opt in ["-h", "--help"]:
140
 
            usage()
141
 
            sys.exit()
142
 
        elif opt in ["-p", "--port"]:
143
 
            port = int(arg)
144
 
        elif opt in ["-i", "--ini"]:
145
 
            auth_name = "with custom ini: %s" % arg
146
 
            ini = arg
147
 
        elif opt in ["-b", "--basic"]:
148
 
            auth_name = "Basic Auth"
149
 
            ini = "echo_basic.ini"
150
 
        elif opt in ["-r", "--remote"]:
151
 
            auth_name = "remote Token Auth"
152
 
            ini = "echo_remote.ini"
153
 
            if not port:
154
 
                port = 8100
155
 
 
156
 
    if not port:
157
 
        port = 8090
158
 
    print "Running with", auth_name
159
 
    app = loadapp("config:" + \
160
 
        os.path.join(os.path.abspath(os.path.dirname(__file__)),
161
 
        ini), global_conf={"log_name": "echo.log"})
162
 
    listener = eventlet.listen(('', port))
163
 
    pool = eventlet.GreenPool(1000)
164
 
    wsgi.server(listener, app, custom_pool=pool)