kubecontext.1s.py Update to work

kubecontext.1s.py

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

# <bitbar.title>Kubeconfig Context Changer</bitbar.title>
# <bitbar.version>v1.0</bitbar.version>
# <bitbar.author>Chris Opland</bitbar.author>
# <bitbar.author.github>copland</bitbar.author.github>
# <bitbar.desc>Displays active kubeconfig context and allows you to easily change contexts.</bitbar.desc>
# <bitbar.dependencies>python,kubectl</bitbar.dependencies>

from collections import namedtuple
from distutils import spawn
import os
import subprocess

Context = namedtuple('Context', ['name', 'active'])

os.environ['PATH'] = '/usr/local/bin:/usr/bin:%s' % os.getenv('PATH')
KUBECTL_PATH = spawn.find_executable('kubectl')


def get_active(contexts):
    return next((x for x in contexts if x.active), 'CONTEXT_NOT_SET')


def load_contexts():
    cmd = [
        KUBECTL_PATH,
        'config',
        'get-contexts',
        '--no-headers'
    ]
    out = subprocess.check_output(cmd)
    
    # Encoding '\n' to bytes like object
    lines = out.split('\n'.encode('utf-8'))
    contexts = []
    for line in lines:
        columns = line.split()
        if columns == []:
            continue
        elif columns[0] == "*":
            contexts.append(Context(columns[1], True))
        else:
            contexts.append(Context(columns[0], False))
    return contexts


def display(contexts):
    active = get_active(contexts)
    
    # Handle case of `CONTEXT_NOT_SET` string returned
    if type(active) is str:
        print(active)
    else:
        print(active.name)
    print('---')
    for context in sorted(contexts, key=lambda x: x.name):
        vardict = {
            'context': context.name,
            'kubectl': KUBECTL_PATH
        }
        print("{context} | bash={kubectl} param1=config param2=use-context param3={context} terminal=false".format(**vardict))


if __name__ == '__main__':
    CONTEXTS = load_contexts()
    display(CONTEXTS)