Useful decorators

decorators.py

import os
from functools import wraps


def sudo_required(func):
    """Check sudo permission, if it hasn't, it'll exit non-zero."""

    @wraps(func)
    def wrapper(*args, **kwargs):
        if os.getuid() != 0:
            print("Sudo access is required!", file=sys.stderr)
            raise SystemExit(1)
        return func(*args, **kwargs)

    return wrapper