summaryrefslogtreecommitdiff
path: root/cloudinit/cmd/cloud_id.py
blob: 0cdc96754c59662f056495b95ea9d9c09dde6f78 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
# This file is part of cloud-init. See LICENSE file for license information.

"""Commandline utility to list the canonical cloud-id for an instance."""

import argparse
import json
import sys

from cloudinit.util import error
from cloudinit.sources import (
    INSTANCE_JSON_FILE, METADATA_UNKNOWN, canonical_cloud_id)

DEFAULT_INSTANCE_JSON = '/run/cloud-init/%s' % INSTANCE_JSON_FILE

NAME = 'cloud-id'


def get_parser(parser=None):
    """Build or extend an arg parser for the cloud-id utility.

    @param parser: Optional existing ArgumentParser instance representing the
        query subcommand which will be extended to support the args of
        this utility.

    @returns: ArgumentParser with proper argument configuration.
    """
    if not parser:
        parser = argparse.ArgumentParser(
            prog=NAME,
            description='Report the canonical cloud-id for this instance')
    parser.add_argument(
        '-j', '--json', action='store_true', default=False,
        help='Report all standardized cloud-id information as json.')
    parser.add_argument(
        '-l', '--long', action='store_true', default=False,
        help='Report extended cloud-id information as tab-delimited string.')
    parser.add_argument(
        '-i', '--instance-data', type=str, default=DEFAULT_INSTANCE_JSON,
        help=('Path to instance-data.json file. Default is %s' %
              DEFAULT_INSTANCE_JSON))
    return parser


def handle_args(name, args):
    """Handle calls to 'cloud-id' cli.

    Print the canonical cloud-id on which the instance is running.

    @return: 0 on success, 1 otherwise.
    """
    try:
        instance_data = json.load(open(args.instance_data))
    except IOError:
        return error(
            "File not found '%s'. Provide a path to instance data json file"
            ' using --instance-data' % args.instance_data)
    except ValueError as e:
        return error(
            "File '%s' is not valid json. %s" % (args.instance_data, e))
    v1 = instance_data.get('v1', {})
    cloud_id = canonical_cloud_id(
        v1.get('cloud_name', METADATA_UNKNOWN),
        v1.get('region', METADATA_UNKNOWN),
        v1.get('platform', METADATA_UNKNOWN))
    if args.json:
        v1['cloud_id'] = cloud_id
        response = json.dumps(   # Pretty, sorted json
            v1, indent=1, sort_keys=True, separators=(',', ': '))
    elif args.long:
        response = '%s\t%s' % (cloud_id, v1.get('region', METADATA_UNKNOWN))
    else:
        response = cloud_id
    sys.stdout.write('%s\n' % response)
    return 0


def main():
    """Tool to query specific instance-data values."""
    parser = get_parser()
    sys.exit(handle_args(NAME, parser.parse_args()))


if __name__ == '__main__':
    main()

# vi: ts=4 expandtab