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
|
# -*- coding: utf-8 -*-
#
# File: portato/helper.py
# This file is part of the Portato-Project, a graphical portage-frontend.
#
# Copyright (C) 2006 René 'Necoro' Neumann
# This is free software. You may redistribute copies of it under the terms of
# the GNU General Public License version 2.
# There is NO WARRANTY, to the extent permitted by law.
#
# Written by René 'Necoro' Neumann <necoro@necoro.net> et.al.
import traceback, os.path
DEBUG = True
def set_debug (d):
"""Sets the global DEBUG variable. Do not set it by your own - always use this function.
@param d: True to enable debugging; False otherwise
@type d: boolean"""
global DEBUG
DEBUG = d
def debug(*args, **kwargs):
"""Prints a debug message including filename and lineno.
A variable number of positional arguments are allowed.
If debug(obj0, obj1, obj2) is called, the text part of the output
looks like the output from print obj0, obj1, obj2.
If you pass the optional keyword-argument "name", it is used for the function-name instead of the original one."""
if not DEBUG : return
stack = traceback.extract_stack()
minus = -2
if "minus" in kwargs:
minus = minus - kwargs["minus"]
a, b, c, d = stack[minus]
a = os.path.basename(a)
out = []
for obj in args:
out.append(str(obj))
text = ' '.join(out)
if "name" in kwargs:
text = 'In %s (%s:%s): %s' % (kwargs["name"], a, b, text)
else:
text = 'In %s (%s:%s): %s' % (c, a, b, text)
text = "***DEBUG*** %s ***DEBUG***" % text
if "file" in kwargs:
f = open(kwargs["file"], "a+")
f.write(text+"\n")
f.close()
else:
print text
def am_i_root ():
"""Returns True if the current user is root, False otherwise.
@rtype: boolean"""
if os.getuid() == 0:
return True
else:
return False
|
