Hot-keys on this page

r m x p   toggle line displays

j k   next/prev highlighted chunk

0   (zero) top of page

1   (one) first highlighted chunk

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

# Copyright ? 2012 New Dream Network, LLC (DreamHost) 

# 

# Licensed under the Apache License, Version 2.0 (the "License"); you may 

# not use this file except in compliance with the License. You may obtain 

# a copy of the License at 

# 

#      http://www.apache.org/licenses/LICENSE-2.0 

# 

# Unless required by applicable law or agreed to in writing, software 

# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT 

# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the 

# License for the specific language governing permissions and limitations 

# under the License. 

""" 

Middleware to replace the plain text message body of an error 

response with one formatted so the client can parse it. 

 

Based on pecan.middleware.errordocument 

""" 

 

import json 

from xml import etree as et 

 

import webob 

 

from magnum.openstack.common._i18n import _ 

from magnum.openstack.common._i18n import _LE 

from magnum.openstack.common import log 

 

LOG = log.getLogger(__name__) 

 

 

class ParsableErrorMiddleware(object): 

    """Replace error body with something the client can parse.""" 

    def __init__(self, app): 

        self.app = app 

 

    def __call__(self, environ, start_response): 

        # Request for this state, modified by replace_start_response() 

        # and used when an error is being reported. 

        state = {} 

 

        def replacement_start_response(status, headers, exc_info=None): 

            """Overrides the default response to make errors parsable.""" 

            try: 

                status_code = int(status.split(' ')[0]) 

                state['status_code'] = status_code 

            except (ValueError, TypeError):  # pragma: nocover 

                raise Exception(_( 

                    'ErrorDocumentMiddleware received an invalid ' 

                    'status %s') % status) 

            else: 

                if (state['status_code'] // 100) not in (2, 3): 

                    # Remove some headers so we can replace them later 

                    # when we have the full error message and can 

                    # compute the length. 

                    headers = [(h, v) 

                               for (h, v) in headers 

                               if h not in ('Content-Length', 'Content-Type') 

                               ] 

                # Save the headers in case we need to modify them. 

                state['headers'] = headers 

                return start_response(status, headers, exc_info) 

 

        app_iter = self.app(environ, replacement_start_response) 

        if (state['status_code'] // 100) not in (2, 3): 

            req = webob.Request(environ) 

70            if (req.accept.best_match(['application/json', 'application/xml']) 

                == 'application/xml'): 

                try: 

                    # simple check xml is valid 

                    body = [et.ElementTree.tostring( 

                            et.ElementTree.fromstring('<error_message>' 

                                                      + '\n'.join(app_iter) 

                                                      + '</error_message>'))] 

                except et.ElementTree.ParseError as err: 

                    LOG.error(_LE('Error parsing HTTP response: %s'), err) 

                    body = ['<error_message>%s' % state['status_code'] 

                            + '</error_message>'] 

                state['headers'].append(('Content-Type', 'application/xml')) 

            else: 

                body = [json.dumps({'error_message': '\n'.join(app_iter)})] 

                state['headers'].append(('Content-Type', 'application/json')) 

            state['headers'].append(('Content-Length', len(body[0]))) 

        else: 

            body = app_iter 

        return body