ceph-volume: decorators: add a check for super user privileges

Signed-off-by: Alfredo Deza <adeza@redhat.com>

diff --git a/src/ceph-volume/ceph_volume/decorators.py b/src/ceph-volume/ceph_volume/decorators.py
index f41f89fb4d0f820e05a8b96585c536cc3298592c..d8bea58fe92b536d50959f697468784ba426f0ae 100644 (file)
--- a/src/ceph-volume/ceph_volume/decorators.py
+++ b/src/ceph-volume/ceph_volume/decorators.py
@@ -1,9 +1,22 @@
 import os
 import sys
-from ceph_volume import terminal
+from ceph_volume import terminal, exceptions
 from functools import wraps
 
 
+def needs_root(func):
+    """
+    Check for super user privileges on functions/methods. Raise
+    ``SuperUserError`` with a nice message.
+    """
+    @wraps(func)
+    def is_root(*a, **kw):
+        if not os.getuid() == 0:
+            raise exceptions.SuperUserError()
+        return func
+    return is_root
+
+
 def catches(catch=None, handler=None, exit=True):
     """
     Very simple decorator that tries any of the exception(s) passed in as