summaryrefslogtreecommitdiffstats
path: root/func/utils.py
blob: c2fbb9fdb5cdb3950e8d7ebc2a65d6d47429c0c6 (plain)
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
"""
Copyright 2007, Red Hat, Inc
see AUTHORS

This software may be freely redistributed under the terms of the GNU
general public license.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
"""

import os
import string
import sys
import traceback
import xmlrpclib

REMOTE_CANARY = "***REMOTE_ERROR***"

# this is kind of handy, so keep it around for now
# but we really need to fix out server side logging and error
# reporting so we don't need it
def trace_me():
    x = traceback.extract_stack()
    bar = string.join(traceback.format_list(x))
    return bar


def daemonize(pidfile=None):
    """
    Daemonize this process with the UNIX double-fork trick.
    Writes the new PID to the provided file name if not None.
    """

    print pidfile
    pid = os.fork()
    if pid > 0:
        sys.exit(0)
    os.setsid()
    os.umask(0)
    pid = os.fork()


    if pid > 0:
        if pidfile is not None:
            open(pidfile, "w").write(str(pid))
        sys.exit(0)

def remove_exceptions(results):
    """
    Used by forkbomb/jobthing to avoid storing exceptions in database
    because you know those don't serialize so well :)
    # FIXME: this needs cleanup
    """

    if results is None:
        return REMOTE_CANARY

    if str(results).startswith("<Fault"):
        return REMOTE_CANARY

    if type(results) == xmlrpclib.Fault:
        return REMOTE_CANARY
    
    if type(results) == dict:
        new_results = {}
        for x in results.keys():
            value = results[x]
            if str(value).find("<Fault") == -1:
                # there are interesting issues with the way it is imported and type()
                # so that is why this hack is here.  type(x) != xmlrpclib.Fault appears to miss some things
                new_results[x] = value
            else:
                new_results[x] = REMOTE_CANARY
        return new_results

    return results