def main()

in cookbooks/aws-parallelcluster-environment/files/default/ec2_udev_rules/manageVolume.py [0:0]

• 28 lines of code
• 4 McCabe index (conditional complexity)

def main():
    try:
        parser = argparse.ArgumentParser(description="Attach or detach ebs volume")
        parser.add_argument(
            "--attach",
            action="store_true",
            help="Attach EBS volume",
            required=False,
            default=False,
        )
        parser.add_argument(
            "--detach",
            action="store_true",
            help="Detach EBS volume",
            required=False,
            default=False,
        )
        parser.add_argument(
            "--volume-id",
            required=True,
        )
        args = parser.parse_args()
        if not args.attach and not args.detach:
            raise Exception("Must specify attach or detach action.")
        handle_volume(args.volume_id, args.attach, args.detach)

    except Exception as e:
        print("ERROR: Failed to attach or detach volume, exception: %s", e)
        sys.exit(1)