datadogpy/datadog/dogshell/downtime.py at master · CodeBoarding/datadogpy · GitHub

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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
# Unless explicitly stated otherwise all files in this repository are licensed under the BSD-3-Clause License.
# This product includes software developed at Datadog (https://www.datadoghq.com/).
# Copyright 2015-Present Datadog, Inc
# stdlib
import json

# 3p
from datadog.util.format import pretty_json

# datadog
from datadog import api
from datadog.dogshell.common import report_errors, report_warnings


class DowntimeClient(object):
    @classmethod
    def setup_parser(cls, subparsers):
        parser = subparsers.add_parser("downtime", help="Create, edit, and delete downtimes")
        parser.add_argument(
            "--string_ids",
            action="store_true",
            dest="string_ids",
            help="Represent downtime IDs as strings instead of ints in JSON",
        )

        verb_parsers = parser.add_subparsers(title="Verbs", dest="verb")
        verb_parsers.required = True

        post_parser = verb_parsers.add_parser("post", help="Create a downtime")
        post_parser.add_argument("scope", help="scope to apply downtime to")
        post_parser.add_argument("start", help="POSIX timestamp to start the downtime", default=None)
        post_parser.add_argument("--end", help="POSIX timestamp to end the downtime", default=None)
        post_parser.add_argument(
            "--message", help="message to include with notifications" " for this downtime", default=None
        )
        post_parser.set_defaults(func=cls._schedule_downtime)

        update_parser = verb_parsers.add_parser("update", help="Update existing downtime")
        update_parser.add_argument("downtime_id", help="downtime to replace" " with the new definition")
        update_parser.add_argument("--scope", help="scope to apply downtime to")
        update_parser.add_argument("--start", help="POSIX timestamp to start" " the downtime", default=None)
        update_parser.add_argument("--end", help="POSIX timestamp to" " end the downtime", default=None)
        update_parser.add_argument(
            "--message", help="message to include with notifications" " for this downtime", default=None
        )
        update_parser.set_defaults(func=cls._update_downtime)

        show_parser = verb_parsers.add_parser("show", help="Show a downtime definition")
        show_parser.add_argument("downtime_id", help="downtime to show")
        show_parser.set_defaults(func=cls._show_downtime)

        show_all_parser = verb_parsers.add_parser("show_all", help="Show a list of all downtimes")
        show_all_parser.add_argument(
            "--current_only", help="only return downtimes that" " are active when the request is made", default=None
        )
        show_all_parser.set_defaults(func=cls._show_all_downtime)

        delete_parser = verb_parsers.add_parser("delete", help="Delete a downtime")
        delete_parser.add_argument("downtime_id", help="downtime to delete")
        delete_parser.set_defaults(func=cls._cancel_downtime)

        cancel_parser = verb_parsers.add_parser("cancel_by_scope", help="Cancel all downtimes with a given scope")
        cancel_parser.add_argument("scope", help="The scope of the downtimes to cancel")
        cancel_parser.set_defaults(func=cls._cancel_downtime_by_scope)

    @classmethod
    def _schedule_downtime(cls, args):
        api._timeout = args.timeout
        format = args.format
        res = api.Downtime.create(scope=args.scope, start=args.start, end=args.end, message=args.message)
        report_warnings(res)
        report_errors(res)
        if format == "pretty":
            print(pretty_json(res))
        else:
            print(json.dumps(res))

    @classmethod
    def _update_downtime(cls, args):
        api._timeout = args.timeout
        format = args.format
        res = api.Downtime.update(
            args.downtime_id, scope=args.scope, start=args.start, end=args.end, message=args.message
        )
        report_warnings(res)
        report_errors(res)
        if format == "pretty":
            print(pretty_json(res))
        else:
            print(json.dumps(res))

    @classmethod
    def _cancel_downtime(cls, args):
        api._timeout = args.timeout
        res = api.Downtime.delete(args.downtime_id)
        if res is not None:
            report_warnings(res)
            report_errors(res)

    @classmethod
    def _show_downtime(cls, args):
        api._timeout = args.timeout
        format = args.format
        res = api.Downtime.get(args.downtime_id)
        report_warnings(res)
        report_errors(res)
        if format == "pretty":
            print(pretty_json(res))
        else:
            print(json.dumps(res))

    @classmethod
    def _show_all_downtime(cls, args):
        api._timeout = args.timeout
        format = args.format
        res = api.Downtime.get_all(current_only=args.current_only)
        report_warnings(res)
        report_errors(res)
        if format == "pretty":
            print(pretty_json(res))
        else:
            print(json.dumps(res))

    @classmethod
    def _cancel_downtime_by_scope(cls, args):
        api._timeout = args.timeout
        format = args.format
        res = api.Downtime.cancel_downtime_by_scope(scope=args.scope)
        if format == "pretty":
            print(pretty_json(res))
        else:
            print(json.dumps(res))