1
diff --git a/bootstack-ops/sosreport.py b/bootstack-ops/sosreport.py
2
0
new file mode 100755
0
new file mode 100755
3
index 0000000..9ff82f2
4
--- /dev/null
5
+++ b/bootstack-ops/sosreport.py
6
@@ -0,0 +1,131 @@
7
1
#!/usr/bin/env python3
8
2
import argparse
9
3
import json
10
4
import os
11
5
import shutil
12
6
import subprocess
13
7
14
8
TIMEOUT = 1800
15
9
16
10
17
11
parser = argparse.ArgumentParser(
18
12
    formatter_class=argparse.ArgumentDefaultsHelpFormatter,
19
13
    description='genererate sosreport')
20
14
21
15
# make target exclusive, to simplify logic
22
16
group = parser.add_mutually_exclusive_group(required=True)
23
17
group.add_argument('-a', '--app', metavar='APP', dest='apps', help='comma separated application names')
24
18
group.add_argument('-u', '--unit', metavar='UNIT', dest='units', help='comma separated unit names')
25
19
group.add_argument('--machine', metavar='MACHINE', dest='machines', help='comma separated machine ids')
26
20
parser.add_argument('case', metavar='CASE', help='salesforce case number, e.g.: 00272654')
27
21
28
22
cli = parser.parse_args()
29
23
CASE = cli.case
30
24
31
25
32
26
if cli.apps:
33
27
    JUJU_RUN_TARGETS = ['--app', cli.apps]
34
28
    # expand apps to units for juju scp
35
29
    juju_status_json = subprocess.check_output(['juju', 'status', '--format', 'json']).decode('utf8')
36
30
    juju_status_dict = json.loads(juju_status_json)
37
31
    juju_apps = juju_status_dict['applications']
38
32
    JUJU_SCP_TARGETS = set([])
39
33
    for app in cli.apps.split(','):
40
34
        for unit in juju_apps[app]['units']:
41
35
            JUJU_SCP_TARGETS.add(unit)
42
36
elif cli.units:
43
37
    JUJU_RUN_TARGETS = ['--unit', cli.units]
44
38
    JUJU_SCP_TARGETS = set(cli.units.split(','))
45
39
elif cli.machines:
46
40
    JUJU_RUN_TARGETS = ['--machine', cli.machines]
47
41
    JUJU_SCP_TARGETS = set(cli.machines.split(','))
48
42
49
43
50
44
def juju_run(cmd):
51
45
    # default itmeout is 5m, which is not enough for some case
52
46
    fullcmd = [
53
47
        'juju', 'run', '--show-log', '--timeout', '{}s'.format(TIMEOUT)
54
48
    ] + JUJU_RUN_TARGETS + ['--', cmd]
55
49
    print(' '.join(fullcmd))
56
50
    subprocess.check_call(fullcmd)
57
51
58
52
59
53
def juju_scp(target):
60
54
    fullcmd = [
61
55
        'juju', 'scp', '--no-host-key-checks', '--', '-r',
62
56
        '{target}:{CASE_DIR}/*'.format(target=target, CASE_DIR=CASE_DIR),
63
57
        CASE_DIR,
64
58
    ]
65
59
    print(' '.join(fullcmd))
66
60
    subprocess.check_call(fullcmd)
67
61
68
62
69
63
# use same dir path for both infra node and units
70
64
CASE_DIR = '/tmp/sosreport'
71
65
# rm and re-create the dir
72
66
print('deleting local dir {} '.format(CASE_DIR))
73
67
shutil.rmtree(CASE_DIR, ignore_errors=True)
74
68
os.makedirs(CASE_DIR, exist_ok=False)
75
69
76
70
cmds = [
77
71
    'sudo rm -rf {CASE_DIR}'.format(CASE_DIR=CASE_DIR),
78
72
    'mkdir -p {CASE_DIR}'.format(CASE_DIR=CASE_DIR),
79
73
    'sudo apt-get install --yes sosreport',
80
74
    'sudo ionice -c 3 sosreport --alloptions --all-logs --batch --case-id={CASE} --tmp-dir {CASE_DIR}'.format(CASE=CASE, CASE_DIR=CASE_DIR),
81
75
    'sudo chown -R ubuntu {CASE_DIR}'.format(CASE_DIR=CASE_DIR),
82
76
]
83
77
84
78
# Do not use f string, which requires python 3.6+, but we still have 3.4 for many clouds
85
79
for cmd in cmds:
86
80
    juju_run(cmd)
87
81
88
82
89
83
for target in JUJU_SCP_TARGETS:
90
84
    juju_scp(target)
91
85
92
86
93
87
print('removing reports ...')
94
88
juju_run('rm --verbose -r {}'.format(CASE_DIR))
95
89
96
90
# make sosreports readable to all users
97
91
# in case you need to scp it back to your local machine with your own user
98
92
subprocess.check_call(['chmod', '-R', 'a+r', CASE_DIR])
99
93
print('sosreports are generated successfully and collected into {}:'.format(CASE_DIR))
100
94
subprocess.check_call(['ls', '-alh', CASE_DIR])
101
95
102
96
103
97
UPLOAD_TARGET = 'files.support.canonical.com'
104
98
print('checking {UPLOAD_TARGET} ssh'.format(UPLOAD_TARGET=UPLOAD_TARGET))
105
99
returncode = subprocess.call(['nc', '-w3', '-zv', UPLOAD_TARGET, '22'])
106
100
can_upload = (returncode == 0)
107
101
108
102
UPLOAD_CMD = ('scp '
109
103
              '-o "StrictHostKeyChecking=no" '
110
104
              '-o "PasswordAuthentication=yes" '
111
105
              '-o "ConnectTimeout={TIMEOUT}" '
112
106
              '{CASE_DIR}/* '
113
107
              'ubuntu@{UPLOAD_TARGET}:').format(
114
108
                  UPLOAD_TARGET=UPLOAD_TARGET,
115
109
                  CASE_DIR=CASE_DIR,
116
110
                  TIMEOUT=TIMEOUT)
117
111
118
112
HELP_YES = """
119
113
Please check files in {CASE_DIR}. If they are ok, upload them with:
120
114
121
115
    {UPLOAD_CMD}
122
116
123
117
""".format(CASE_DIR=CASE_DIR, UPLOAD_CMD=UPLOAD_CMD)
124
118
125
119
HELP_NO = """
126
120
It seems this node has no access to {UPLOAD_TARGET}, run following cmds on your *local machine* to download files and upload from there instead:
127
121
128
122
    rm -rf {CASE_DIR} && mkdir -p {CASE_DIR}
129
123
    scp -r maas.XXXX.bootstack:{CASE_DIR}/* {CASE_DIR}
130
124
    {UPLOAD_CMD}
131
125
132
126
""".format(CASE_DIR=CASE_DIR, UPLOAD_TARGET=UPLOAD_TARGET, UPLOAD_CMD=UPLOAD_CMD)
133
127
134
128
if can_upload:
135
129
    print(HELP_YES)
136
130
else:
137
131
    print(HELP_NO)
Status:	Rejected
Rejected by:	Haw Loeung on 2022-10-28
Proposed branch:	bootstack-ops:sosreport-python
Merge into:	bootstack-ops:master
Diff against target:	137 lines (+131/-0) 1 file modified bootstack-ops/sosreport.py (+131/-0)
Related bugs:	Link a bug report
Reviewer	Date Requested	Status
Chris Sanders	2020-05-05	Pending
Joe Guo		Pending
Review via email: mp+382629@code.launchpad.net