sitrep.py

#!/usr/bin/env python3

# Copyright 2022 Google LLC

# 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 os
import slack
import logging
from get_firestore_cases import get_firestore_cases

logger = logging.getLogger(__name__)


def sitrep(channel_id, user_id):
    """
    Lists the following details for all cases in the org:
    case id, priority, title, isEscalated, case creation time, last case update time,
    case status, and case creator.
    Additionally, provides a summary of the number of the number of cases open by
    priority, the total number of cases, and the total number of escalated cases.

    Parameters
    ----------
    channel_id : str
        unique string used to idenify a Slack channel. Used to send messages to the channel
    user_id : str
        the Slack user_id of the user who submitted the request. Used to send ephemeral
        messages to the user
    """
    client = slack.WebClient(token=os.environ.get('SLACK_TOKEN'))
    p1 = 0
    p2 = 0
    p3 = 0
    p4 = 0
    esc_count = 0
    report = ('This is the current state of Google Cloud Support cases:'
              '\n\ncase,priority,title,escalated,create_time,last_updated,state,'
              'case_creator')
    cases = get_firestore_cases()

    for case in cases:
        if case['priority'] == 'P1':
            p1 += 1
        elif case['priority'] == 'P2':
            p2 += 1
        elif case['priority'] == 'P3':
            p3 += 1
        else:
            p4 += 1

        if case['escalated']:
            esc_count += 1

        report = report + '\n{},{},{},{},{},{},{},{}'.format(
            case['case_number'],
            case['priority'],
            case['case_title'],
            case['escalated'],
            case['create_time'],
            case['update_time'],
            case['state'],
            case['case_creator'])

    report = (report + '\n\n'
              '\n{} P1 cases are open'
              '\n{} P2 cases are open'
              '\n{} P3 cases are open'
              '\n{} P4 cases are open'
              '\nTotal cases open: {}'
              '\nEscalated cases: {}').format(
                                              str(p1),
                                              str(p2),
                                              str(p3),
                                              str(p4),
                                              str(p1 + p2 + p3 + p4),
                                              str(esc_count))

    client.chat_postMessage(channel=channel_id, text=f"{report}")


if __name__ == "__main__":
    channel_id = os.environ.get('TEST_CHANNEL_ID')
    user_id = os.environ.get('TEST_USER_ID')
    sitrep(channel_id, user_id)