96 lines
3.3 KiB
Python
Executable File
96 lines
3.3 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright 2010 United States Government as represented by the
|
|
# Administrator of the National Aeronautics and Space Administration.
|
|
# All Rights Reserved.
|
|
#
|
|
# 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.
|
|
|
|
|
|
import base64
|
|
import json
|
|
import logging
|
|
import os
|
|
import shutil
|
|
import sys
|
|
import urllib
|
|
import urllib2
|
|
try:
|
|
import cPickle as pickle
|
|
except Exception:
|
|
import pickle
|
|
|
|
|
|
class SpoolSentry(object):
|
|
def __init__(self, spool_dir, sentry_url, key=None):
|
|
self.spool_dir = spool_dir
|
|
self.sentry_url = sentry_url
|
|
self.key = key
|
|
|
|
def process(self):
|
|
for fname in os.listdir(self.spool_dir):
|
|
if fname == "processed":
|
|
continue
|
|
try:
|
|
sourcefile = "%s/%s" % (self.spool_dir, fname)
|
|
with open(sourcefile) as f:
|
|
fdata = f.read()
|
|
data_from_json = json.loads(fdata)
|
|
data = self.build_data(data_from_json)
|
|
self.send_data(data)
|
|
destfile = "%s/processed/%s" % (self.spool_dir, fname)
|
|
shutil.move(sourcefile, destfile)
|
|
except Exception:
|
|
logging.exception("Unable to upload record %s", fname)
|
|
raise
|
|
|
|
def build_data(self, filejson):
|
|
env = {'SERVER_NAME': 'unknown', 'SERVER_PORT': '0000',
|
|
'SCRIPT_NAME': '/unknown/', 'PATH_INFO': 'unknown'}
|
|
if filejson['env']:
|
|
env = json.loads(filejson['env'])
|
|
url = "http://%s:%s%s%s" % (env['SERVER_NAME'], env['SERVER_PORT'],
|
|
env['SCRIPT_NAME'], env['PATH_INFO'])
|
|
rv = {'logger': filejson['logger'], 'level': logging.ERROR,
|
|
'server_name': filejson['host'], 'url': url,
|
|
'message': filejson['message'],
|
|
'traceback': filejson['traceback']}
|
|
rv['data'] = {}
|
|
if filejson['env']:
|
|
rv['data']['META'] = env
|
|
if filejson['request_id']:
|
|
rv['data']['request_id'] = filejson['request_id']
|
|
return rv
|
|
|
|
def send_data(self, data):
|
|
data = {'data': base64.b64encode(pickle.dumps(data).encode('zlib')),
|
|
'key': self.key}
|
|
req = urllib2.Request(self.sentry_url)
|
|
res = urllib2.urlopen(req, urllib.urlencode(data))
|
|
if res.getcode() != 200:
|
|
raise Exception("Bad HTTP code: %s" % res.getcode())
|
|
txt = res.read()
|
|
|
|
if __name__ == '__main__':
|
|
sentryurl = 'http://127.0.0.1/sentry/store/'
|
|
key = ''
|
|
spooldir = '/var/spool/nova'
|
|
if len(sys.argv) > 1:
|
|
sentryurl = sys.argv[1]
|
|
if len(sys.argv) > 2:
|
|
key = sys.argv[2]
|
|
if len(sys.argv) > 3:
|
|
spooldir = sys.argv[3]
|
|
SpoolSentry(spooldir, sentryurl, key).process()
|