← Back to branch summary

~0x44/nova/bug838466

« back to all changes in this revision

Viewing changes to bin/nova-spoolsentry

  • Committer: Eric Day
  • Date: 2010-10-21 18:49:51 UTC
  • mto: This revision was merged to the branch mainline in revision 377.
  • Revision ID: eday@oddments.org-20101021184951-x0vs3s8y7mc0aeyy
PEP8 and pylint cleanup. There should be no functional changes here, just style changes to get violations down.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bin/nova-spoolsentry
1
 
#!/usr/bin/env python
2
 
# vim: tabstop=4 shiftwidth=4 softtabstop=4
3
 
 
4
 
# Copyright 2010 United States Government as represented by the
5
 
# Administrator of the National Aeronautics and Space Administration.
6
 
# All Rights Reserved.
7
 
#
8
 
#    Licensed under the Apache License, Version 2.0 (the "License"); you may
9
 
#    not use this file except in compliance with the License. You may obtain
10
 
#    a copy of the License at
11
 
#
12
 
#         http://www.apache.org/licenses/LICENSE-2.0
13
 
#
14
 
#    Unless required by applicable law or agreed to in writing, software
15
 
#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
16
 
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
17
 
#    License for the specific language governing permissions and limitations
18
 
#    under the License.
19
 
 
20
 
 
21
 
import base64
22
 
import json
23
 
import logging
24
 
import os
25
 
import shutil
26
 
import sys
27
 
import urllib
28
 
import urllib2
29
 
try:
30
 
    import cPickle as pickle
31
 
except:
32
 
    import pickle
33
 
 
34
 
 
35
 
class SpoolSentry(object):
36
 
    def __init__(self, spool_dir, sentry_url, key=None):
37
 
        self.spool_dir = spool_dir
38
 
        self.sentry_url = sentry_url
39
 
        self.key = key
40
 
 
41
 
    def process(self):
42
 
        for fname in os.listdir(self.spool_dir):
43
 
            if fname == "processed":
44
 
                continue
45
 
            try:
46
 
                sourcefile = "%s/%s" % (self.spool_dir, fname)
47
 
                with open(sourcefile) as f:
48
 
                    fdata = f.read()
49
 
                data_from_json = json.loads(fdata)
50
 
                data = self.build_data(data_from_json)
51
 
                self.send_data(data)
52
 
                destfile = "%s/processed/%s" % (self.spool_dir, fname)
53
 
                shutil.move(sourcefile, destfile)
54
 
            except:
55
 
                logging.exception("Unable to upload record %s", fname)
56
 
                raise
57
 
 
58
 
    def build_data(self, filejson):
59
 
        env = {'SERVER_NAME': 'unknown', 'SERVER_PORT': '0000',
60
 
               'SCRIPT_NAME': '/unknown/', 'PATH_INFO': 'unknown'}
61
 
        if filejson['env']:
62
 
            env = json.loads(filejson['env'])
63
 
        url = "http://%s:%s%s%s" % (env['SERVER_NAME'], env['SERVER_PORT'],
64
 
                                    env['SCRIPT_NAME'], env['PATH_INFO'])
65
 
        rv = {'logger': filejson['logger'], 'level': logging.ERROR,
66
 
              'server_name': filejson['host'], 'url': url,
67
 
              'message': filejson['message'],
68
 
              'traceback': filejson['traceback']}
69
 
        rv['data'] = {}
70
 
        if filejson['env']:
71
 
            rv['data']['META'] = env
72
 
        if filejson['request_id']:
73
 
            rv['data']['request_id'] = filejson['request_id']
74
 
        return rv
75
 
 
76
 
    def send_data(self, data):
77
 
        data = {'data': base64.b64encode(pickle.dumps(data).encode('zlib')),
78
 
                'key': self.key}
79
 
        req = urllib2.Request(self.sentry_url)
80
 
        res = urllib2.urlopen(req, urllib.urlencode(data))
81
 
        if res.getcode() != 200:
82
 
            raise Exception("Bad HTTP code: %s" % res.getcode())
83
 
        txt = res.read()
84
 
 
85
 
if __name__ == '__main__':
86
 
    sentryurl = 'http://127.0.0.1/sentry/store/'
87
 
    key = ''
88
 
    spooldir = '/var/spool/nova'
89
 
    if len(sys.argv) > 1:
90
 
        sentryurl = sys.argv[1]
91
 
    if len(sys.argv) > 2:
92
 
        key = sys.argv[2]
93
 
    if len(sys.argv) > 3:
94
 
        spooldir = sys.argv[3]
95
 
    SpoolSentry(spooldir, sentryurl, key).process()