forked from iamlemec/wikidiff
-
Notifications
You must be signed in to change notification settings - Fork 0
/
ktool.py
90 lines (74 loc) · 1.9 KB
/
ktool.py
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
from subprocess import run, PIPE
import click
project = 'thesecretaryofwar'
zone = 'us-east1-b'
def getoutput(cmd, shell=True, check=True):
return run(
cmd,
shell=shell,
check=True,
universal_newlines=True,
stdout=PIPE
).stdout
def getpods(status=None, all=True):
all = '-a' if all else ''
if status is None:
ret = getoutput(
"kubectl get pods {all} -o json | jq -r '.items[].metadata.name'".format(all=all)
)
else:
ret = getoutput(
"kubectl get pods {all} -o json | jq -r '.items[] | select(.status.phase == \"{status}\") | .metadata.name'".format(all=all, status=status)
)
return ret.strip().split('\n')
def getnodes():
return getoutput("kubectl get nodes -o json | jq -r '.items[].metadata.name'").strip().split('\n')
def runall(cmd):
for node in getnodes():
click.echo(node)
runnode(cmd, node)
click.echo()
def runnode(cmd, node):
run(
'gcloud compute --project "{project}" ssh --zone "{zone}" "{node}" --command="{cmd}"'.format(
cmd=cmd,
node=node,
project=project,
zone=zone
),
shell=True
)
@click.group()
def cli():
pass
@cli.group()
def nodes():
pass
@nodes.command()
def ls():
for node in getnodes():
click.echo(node)
@nodes.command()
def df():
runall("df -h | grep '/var$'")
@nodes.command()
def gc():
runall('docker rm \$(docker ps -q -f status=exited)')
@cli.group()
def pods():
pass
@pods.command()
def ls():
for pod in getpods(status='Succeeded'):
click.echo(pod)
@pods.command()
def gc():
for pod in getpods(status='Succeeded'):
click.echo(pod)
run('kubectl delete pod {pod}'.format(pod=pod), shell=True)
@cli.command()
@click.argument('command')
def cmd(command):
runall(command)
if __name__ == '__main__':
cli()