create_sysdig_capture.py

#!/usr/bin/env python
#
# Creates a sysdig capture, waits for termination and prints the download URL.
#

import sys
import time

from sdcclient import SdcClient

#
# Parse arguments
#
if len(sys.argv) not in (5, 6):
    print(('usage: %s <sysdig-token> hostname capture_name duration [filter]' % sys.argv[0]))
    print('You can find your token at https://app.sysdigcloud.com/#/settings/user')
    sys.exit(1)

sdc_token = sys.argv[1]
hostname = sys.argv[2]
capture_name = sys.argv[3]
duration = sys.argv[4]
capture_filter = ''

if len(sys.argv) == 6:
    capture_filter = sys.argv[5]

#
# Instantiate the SDC client
#
sdclient = SdcClient(sdc_token)

ok, res = sdclient.create_sysdig_capture(hostname, capture_name, int(duration), capture_filter)

#
# Show the list of metrics
#
if ok:
    capture = res['dump']
else:
    print(res)
    sys.exit(1)

while True:
    ok, res = sdclient.poll_sysdig_capture(capture)
    if ok:
        capture = res['dump']
    else:
        print(res)
        sys.exit(1)

    print(('Capture is in state ' + capture['status']))
    if capture['status'] in ('requested', 'capturing', 'uploading'):
        pass
    elif capture['status'] in ('error', 'uploadingError'):
        sys.exit(1)
    elif capture['status'] in ('done', 'uploaded'):
        print(('Download at: ' + sdclient.url + capture['downloadURL']))
        sys.exit(0)

    time.sleep(1)