#!/usr/bin/python3.6
import sys

try:
    from arc.lrms import pyCancel
except:
    sys.stderr.write('Failed to import pyCancel module\n')
    sys.exit(2)


if __name__ == '__main__':
    usage = 'Usage: %s [--config <arc.conf>] <jobid>' % (sys.argv[0])

    if len(sys.argv) != 4 and len(sys.argv) != 2:
        error(usage, 'cancel-SLURMPY-job')
        sys.exit(1)

    if len(sys.argv) == 4:
        if sys.argv[1] != "--config":
            error(usage, 'cancel-SLURMPY-job')
            sys.exit(1)
        sys.exit(pyCancel.main("slurm", sys.argv[3], sys.argv[2]))

    sys.exit(pyCancel.main("slurm", sys.argv[1]))
