waybar custom script to display open gerrit reviews

gerrit_waybar.py

#!/usr/bin/env python3
##!/usr/bin/python3
# -*- coding: utf-8 -*-

import os
import sys
import json
import subprocess


import logging

#    "custom/gerrit": {
#        "format": "G{}",
#        "exec": "/home/<USERNAME>/.config/gerrit/gerrit_incoming_crs.waybar.py",
#        "interval": 60,
#        "return-type": "json",
#        "on-click": "exec google-chrome https://gerrit.<YOUR_DOMAIN>/#/dashboard/self"
#    },
 
# Create the file
# and output every level since 'DEBUG' is used
# and remove all headers in the output
# using empty format=''

#logging.basicConfig(filename='/tmp/log.txt', level=logging.DEBUG, format='')

gerrit_user = '<USERNAME>'
gerrit_host = 'gerrit.<YOUR_DOMAIN>'
gerrit_port = '29418'

#sys.setdefaultencoding('utf8')

def query_gerrit():
    ssh_auth_sock_query="find /tmp/ -type s -path '/tmp/ssh-*/agent.*' -user $(whoami) 2>/dev/null | tail -1"
    ssh_auth_sock_results = subprocess.check_output(
            ssh_auth_sock_query, stderr=subprocess.STDOUT, universal_newlines=True, shell=True)

    #logging.debug("ssh_auth_sock_results: %s" % ssh_auth_sock_results)
    my_env = os.environ.copy()
    my_env['SSH_AUTH_SOCK'] = ssh_auth_sock_results.strip()
    gerrit_query = ['ssh', '%s@%s' % (gerrit_user, gerrit_host), '-p', gerrit_port,
                    'gerrit', 'query', '--format=JSON', '--all-approvals',
                    'status:open', 'reviewer:"%s"' % gerrit_user]
    try:
        results = subprocess.check_output(
            gerrit_query, stderr=subprocess.STDOUT, universal_newlines=True, env=my_env)
        return strip_unused_results(results)
    except Exception:
        logging.error('Unable to query gerrit')
        logging.error('---')
        logging.error("Ensure you've uploaded your SSH key to gerrit")
        sys.exit(1)
def strip_unused_results(results):
    # last line is blank
    # second to last line is an aggregate
    results = results.split('\n')
    results = results[:-2] if len(results) > 2 else []
    return [json.loads(x) for x in results]


def is_self(result):
    return result.get('owner', {}).get('username', '') == gerrit_user

def filter_self(results):
    return [x for x in results if not is_self(x)]

def is_approved(result):
    approval_total = 0
    patch_sets = result.get("patchSets", {})
    for patch in patch_sets:
        approvals = patch.get("approvals",[])
        for approval in approvals:
            value = int(approval.get("value",0))
            approval_total += value
    if approval_total >= 2:
        return True
    else:
        return False

def filter_approved(results):
    return [x for x in results if not is_approved(x)]

def print_results(results):
    num_results = len(results)
    if num_results == 0:
        print(json.dumps({
            "text": "0",
            "tooltip": "href=https://%s/#/dashboard/self" % (gerrit_host)
                    }))
    else:
        print(json.dumps({
            "text": "⚡%s" % (num_results),
            "tooltip": "href=https://%s/#/dashboard/self" % (gerrit_host)
                    }))

def main():
    all_unapproved = []
    all_incoming = filter_self(query_gerrit())
    all_unapproved = filter_approved(all_incoming)
    print_results(all_unapproved)

if __name__ == "__main__":
    main()