stoneridge/srinfogatherer.py

74 строки
2.7 KiB
Python

#!/usr/bin/env python
# This Source Code Form is subject to the terms of the Mozilla Public License,
# v. 2.0. If a copy of the MPL was not distributed with this file, You can
# obtain one at http://mozilla.org/MPL/2.0/.
import ConfigParser
import json
import logging
import os
import platform
import stoneridge
class StoneRidgeInfoGatherer(object):
"""Gathers information about the platform stone ridge is running on as well
as the build that stone ridge is being run against
"""
def run(self):
logging.debug('info gatherer running')
bindir = stoneridge.get_config('run', 'bin')
info_file = os.path.join(bindir, 'application.ini')
logging.debug('parsing ini file at %s' % (info_file,))
cp = ConfigParser.SafeConfigParser()
cp.read([info_file])
build_info = {}
build_info['name'] = cp.get('App', 'Name')
build_info['version'] = cp.get('App', 'Version')
build_info['revision'] = cp.get('App', 'SourceStamp')
build_info['branch'] = stoneridge.get_config('run', 'netconfig')
# Due to the way the graph server works, we need to create a unique
# build id for each build/os/netconfig combination. We also want to
# keep the unmodified build ID around for posterity.
build_info['original_buildid'] = cp.get('App', 'BuildID')
# Build ID is limited to 16 characters in the receiving database, and
# our suffix is 2 characters, so we truncate the original to 14
# characters before adding our suffix. It should already be only 14
# characters, but we do this Just In Case.
buildid_base = build_info['original_buildid'][:14]
build_info['id'] = buildid_base + stoneridge.get_buildid_suffix()
machine_info = {}
machine_info['name'] = platform.node()
machine_info['os'] = stoneridge.get_config('machine', 'os')
machine_info['osversion'] = stoneridge.get_os_version()
machine_info['platform'] = platform.machine()
info = {'test_machine': machine_info,
'test_build': build_info,
'testrun': {},
'date': stoneridge.get_config_int('run', 'tstamp')}
logging.debug('gathered info: %s' % (info,))
outdir = stoneridge.get_config('run', 'out')
if not os.path.exists(outdir):
logging.debug('making outdir %s' % (outdir,))
os.mkdir(outdir)
with file(os.path.join(outdir, 'info.json'), 'wb') as f:
logging.debug('dumping json to file')
json.dump(info, f)
@stoneridge.main
def main():
parser = stoneridge.TestRunArgumentParser()
parser.parse_args()
info_gatherer = StoneRidgeInfoGatherer()
info_gatherer.run()