python-zulip-api/zulip/integrations/openshift/post_deploy

#!/usr/bin/env python3
#
# Zulip notification post-receive hook.

import os
import subprocess
import sys
from typing import Dict

sys.path.insert(0, os.path.dirname(__file__))
import zulip_openshift_config as config

VERSION = "0.1"

if config.ZULIP_API_PATH is not None:
    sys.path.append(config.ZULIP_API_PATH)

import zulip

client = zulip.Client(
    email=config.ZULIP_USER,
    site=config.ZULIP_SITE,
    api_key=config.ZULIP_API_KEY,
    client="ZulipOpenShift/" + VERSION,
)


def get_deployment_details() -> Dict[str, str]:
    # "gear deployments" output example:
    #   Activation time - Deployment ID - Git Ref - Git SHA1
    #   2017-01-07 15:40:30 -0500 - 9e2b7143 - master - b9ce57c - ACTIVE
    dep = subprocess.check_output(["gear", "deployments"], universal_newlines=True).splitlines()[1]
    splits = dep.split(" - ")

    return dict(
        app_name=os.environ["OPENSHIFT_APP_NAME"],
        url=os.environ["OPENSHIFT_APP_DNS"],
        branch=splits[2],
        commit_id=splits[3],
    )


def send_bot_message(deployment: Dict[str, str]) -> None:
    destination = config.deployment_notice_destination(deployment["branch"])
    if destination is None:
        # No message should be sent
        return
    message = config.format_deployment_message(**deployment)

    client.send_message(
        {
            "type": "stream",
            "to": destination["stream"],
            "subject": destination["subject"],
            "content": message,
        }
    )

    return


deployment = get_deployment_details()
send_bot_message(deployment)