1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
import os
import sys
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 and not os.environ.get('CEPH_VOLUME_SKIP_NEEDS_ROOT', False):
raise exceptions.SuperUserError()
return func(*a, **kw)
return is_root
def catches(catch=None, handler=None, exit=True):
"""
Very simple decorator that tries any of the exception(s) passed in as
a single exception class or tuple (containing multiple ones) returning the
exception message and optionally handling the problem if it rises with the
handler if it is provided.
So instead of douing something like this::
def bar():
try:
some_call()
print("Success!")
except TypeError, exc:
print("Error while handling some call: %s" % exc)
sys.exit(1)
You would need to decorate it like this to have the same effect::
@catches(TypeError)
def bar():
some_call()
print("Success!")
If multiple exceptions need to be caught they need to be provided as a
tuple::
@catches((TypeError, AttributeError))
def bar():
some_call()
print("Success!")
"""
catch = catch or Exception
def decorate(f):
@wraps(f)
def newfunc(*a, **kw):
try:
return f(*a, **kw)
except catch as e:
import logging
logger = logging.getLogger('ceph_volume')
logger.exception('exception caught by decorator')
if os.environ.get('CEPH_VOLUME_DEBUG'):
raise
if handler:
return handler(e)
else:
sys.stderr.write(make_exception_message(e))
if exit:
sys.exit(1)
return newfunc
return decorate
#
# Decorator helpers
#
def make_exception_message(exc):
"""
An exception is passed in and this function
returns the proper string depending on the result
so it is readable enough.
"""
if str(exc):
return '%s %s: %s\n' % (terminal.red_arrow, exc.__class__.__name__, exc)
else:
return '%s %s\n' % (terminal.red_arrow, exc.__class__.__name__)
|
