Mini Shell

#!/opt/cloudlinux/venv/bin/python3 -bb

import sys

from clsudo import (Clsudo, NoSuchUser, UnableToReadFile, UnableToWriteFile)


def exit_with_error(message, status=1):
    sys.stderr.write("%s\n" % (str(message,)))
    sys.exit(status)


def main():
    try:
        sudo = Clsudo()
        sudo.remove_user(sys.argv[1])
    except IndexError:
        exit_with_error("A username to remove is expected")
    except (NoSuchUser, UnableToReadFile, UnableToWriteFile) as e:
        exit_with_error(e)
    except Exception as e:
        exit_with_error(e)


if __name__ == '__main__':
    main()