1ce8ecd8be
Since 7.0 the diagnostic snapshow aren't shared by Nginx anymore, and any request to download goes through Nailgun and therefore requires authentication token. Closes-Bug: #1492207 Change-Id: I29e1a5142dd7c7050ba58b68f1161eba339cf53f Signed-off-by: Igor Kalnitsky <igor@kalnitsky.org>
81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
# Copyright 2014 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 sys
|
|
|
|
import yaml
|
|
|
|
from fuelclient.cli.actions.base import Action
|
|
import fuelclient.cli.arguments as Args
|
|
from fuelclient.cli.formatting import download_snapshot_with_progress_bar
|
|
from fuelclient.client import APIClient
|
|
from fuelclient.objects.task import SnapshotTask
|
|
|
|
|
|
class SnapshotAction(Action):
|
|
"""Generate and download snapshot.
|
|
"""
|
|
action_name = "snapshot"
|
|
|
|
def __init__(self):
|
|
super(SnapshotAction, self).__init__()
|
|
self.args = (
|
|
Args.get_dir_arg("Directory to which download snapshot."),
|
|
Args.get_boolean_arg("conf",
|
|
help_="Provide this flag to generate conf")
|
|
)
|
|
self.flag_func_map = (
|
|
('conf', self.get_snapshot_config),
|
|
(None, self.get_snapshot),
|
|
)
|
|
|
|
def get_snapshot(self, params):
|
|
"""To download diagnostic snapshot:
|
|
fuel snapshot
|
|
|
|
To download diagnostic snapshot to specific directory:
|
|
fuel snapshot --dir path/to/directory
|
|
|
|
To specify config for snapshoting
|
|
fuel snapshot < conf.yaml
|
|
|
|
"""
|
|
if sys.stdin.isatty():
|
|
conf = {}
|
|
else:
|
|
conf = yaml.load(sys.stdin.read())
|
|
|
|
snapshot_task = SnapshotTask.start_snapshot_task(conf)
|
|
self.serializer.print_to_output(
|
|
snapshot_task.data,
|
|
"Generating dump..."
|
|
)
|
|
snapshot_task.wait()
|
|
download_snapshot_with_progress_bar(
|
|
snapshot_task.connection.root + snapshot_task.data["message"],
|
|
auth_token=APIClient.auth_token,
|
|
directory=params.dir
|
|
)
|
|
|
|
def get_snapshot_config(self, params):
|
|
"""Download default config for snapshot
|
|
|
|
fuel snapshot --conf > dump_conf.yaml
|
|
|
|
To use json formatter
|
|
fuel snapshot --conf --json
|
|
"""
|
|
conf = SnapshotTask.get_default_config()
|
|
self.serializer.write_to_file(sys.stdout, conf)
|