94 lines
3.7 KiB
Python
Executable File
94 lines
3.7 KiB
Python
Executable File
#!/usr/bin/env python2
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2015 Mirantis, Inc.
|
|
#
|
|
# 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 argparse
|
|
import nodes
|
|
import logging
|
|
import sys
|
|
import loadconf
|
|
import flock
|
|
|
|
def main(argv=None):
|
|
if argv is None:
|
|
argv = sys.argv
|
|
|
|
parser = argparse.ArgumentParser(description='need to add description')
|
|
parser.add_argument('--config', default='config.yaml',
|
|
help='Config file')
|
|
parser.add_argument('-o', '--dest-file', default='/tmp/',
|
|
help='output archive file')
|
|
# The following parameter has not been implemented yet.
|
|
parser.add_argument('-f', '--nodes',
|
|
help='nodes file', default='nodes.json')
|
|
parser.add_argument('-e', '--extended', default="0",
|
|
help='exec once by role cmdfiles')
|
|
parser.add_argument('-c', '--cluster', help='cluster id')
|
|
parser.add_argument('-l', '--logs',
|
|
help='Collect logs from fuel-node',
|
|
action="store_const", dest='getlogs',
|
|
const=True, default=False)
|
|
parser.add_argument('--only-logs',
|
|
action="store_const", dest="only_logs",
|
|
const=True, default=False,
|
|
help='Collect only logs from fuel-node')
|
|
parser.add_argument('-d', '--debug',
|
|
help="Print lots of debugging statements",
|
|
action="store_const", dest="loglevel",
|
|
const=logging.DEBUG,
|
|
default=logging.WARNING,)
|
|
parser.add_argument('-v', '--verbose',
|
|
help="Be verbose",
|
|
action="store_const", dest="loglevel",
|
|
const=logging.INFO,)
|
|
|
|
args = parser.parse_args(argv[1:])
|
|
logging.basicConfig(level=args.loglevel,
|
|
format='%(asctime)s %(levelname)s %(message)s')
|
|
args.extended = args.extended == "1"
|
|
conf = loadconf.load_conf(args.config)
|
|
n = nodes.Nodes(conf=conf,
|
|
extended=args.extended,
|
|
cluster=args.cluster,
|
|
destdir=args.dest_file)
|
|
# nodes.print_nodes()
|
|
if not args.only_logs:
|
|
n.get_node_file_list()
|
|
n.launch_ssh(conf['out-dir'])
|
|
n.get_conf_files(conf['out-dir'])
|
|
n.create_archive_general(conf['out-dir'], '/tmp/timmy-gen.tar.bz2', 60)
|
|
if args.only_logs or args.getlogs:
|
|
lock = flock.FLock('/tmp/timmy-logs.lock')
|
|
if not lock.lock():
|
|
logging.warning('Unable to obtain lock, skipping "logs"-part')
|
|
return 1
|
|
n.get_node_file_list()
|
|
n.calculate_log_size(conf['find']['template'])
|
|
if n.is_enough_space():
|
|
n.get_log_files(conf['out-dir'])
|
|
n.create_archive_logs(conf['find']['template'],
|
|
conf['logs-archive'],
|
|
conf['compress-timeout'])
|
|
n.add_logs_archive(conf['out-dir'], nodes.lkey,
|
|
conf['logs-archive'], 120)
|
|
n.compress_archive(conf['logs-archive'], conf['compress-timeout'])
|
|
|
|
n.print_nodes()
|
|
return 0
|
|
|
|
if __name__ == '__main__':
|
|
exit(main(sys.argv))
|