| 1 | import os, sys, resource, time, socket, threading |
| 2 | |
| 3 | class attrinfo(object): |
| 4 | def __init__(self, desc = None): |
| 5 | self.desc = desc |
| 6 | |
| 7 | class perfobj(object): |
| 8 | def __init__(self, *args, **kwargs): |
| 9 | super(perfobj, self).__init__() |
| 10 | |
| 11 | def pdm_protocols(self): |
| 12 | return [] |
| 13 | |
| 14 | class simpleattr(perfobj): |
| 15 | def __init__(self, func, info = None, *args, **kwargs): |
| 16 | super(simpleattr, self).__init__(*args, **kwargs) |
| 17 | self.func = func |
| 18 | if info is None: |
| 19 | info = attrinfo() |
| 20 | self.info = info |
| 21 | |
| 22 | def readattr(self): |
| 23 | return self.func() |
| 24 | |
| 25 | def attrinfo(self): |
| 26 | return self.info |
| 27 | |
| 28 | def pdm_protocols(self): |
| 29 | return super(simpleattr, self).pdm_protocols() + ["attr"] |
| 30 | |
| 31 | class valueattr(perfobj): |
| 32 | def __init__(self, init, info = None, *args, **kwargs): |
| 33 | super(valueattr, self).__init__(*args, **kwargs) |
| 34 | self.value = init |
| 35 | if info is None: |
| 36 | info = attrinfo() |
| 37 | self.info = info |
| 38 | |
| 39 | def readattr(self): |
| 40 | return self.value |
| 41 | |
| 42 | def attrinfo(self): |
| 43 | return self.info |
| 44 | |
| 45 | def pdm_protocols(self): |
| 46 | return super(valueattr, self).pdm_protocols() + ["attr"] |
| 47 | |
| 48 | |
| 49 | class eventobj(perfobj): |
| 50 | def __init__(self, *args, **kwargs): |
| 51 | super(eventobj, self).__init__(*args, **kwargs) |
| 52 | self.subscribers = set() |
| 53 | |
| 54 | def subscribe(self, cb): |
| 55 | if cb in self.subscribers: |
| 56 | raise ValueError("Already subscribed") |
| 57 | self.subscribers.add(cb) |
| 58 | |
| 59 | def unsubscribe(self, cb): |
| 60 | self.subscribers.remove(cb) |
| 61 | |
| 62 | def notify(self, event): |
| 63 | for cb in self.subscribers: |
| 64 | try: |
| 65 | cb(event) |
| 66 | except: pass |
| 67 | |
| 68 | def pdm_protocols(self): |
| 69 | return super(eventobj, self).pdm_protocols() + ["event"] |
| 70 | |
| 71 | class staticdir(perfobj): |
| 72 | def __init__(self, *args, **kwargs): |
| 73 | super(staticdir, self).__init__(*args, **kwargs) |
| 74 | self.map = {} |
| 75 | |
| 76 | def __setitem__(self, name, ob): |
| 77 | self.map[name] = ob |
| 78 | |
| 79 | def __delitem__(self, name): |
| 80 | del self.map[name] |
| 81 | |
| 82 | def __getitem__(self, name): |
| 83 | return self.map[name] |
| 84 | |
| 85 | def get(self, name, default = None): |
| 86 | return self.map.get(name, default) |
| 87 | |
| 88 | def listdir(self): |
| 89 | return self.map.keys() |
| 90 | |
| 91 | def lookup(self, name): |
| 92 | return self.map[name] |
| 93 | |
| 94 | def pdm_protocols(self): |
| 95 | return super(staticdir, self).pdm_protocols() + ["dir"] |
| 96 | |
| 97 | class event(object): |
| 98 | def __init__(self): |
| 99 | self.time = time.time() |
| 100 | |
| 101 | idlock = threading.Lock() |
| 102 | procevid = 0 |
| 103 | |
| 104 | def getprocid(): |
| 105 | global procevid |
| 106 | idlock.acquire() |
| 107 | try: |
| 108 | ret = procevid |
| 109 | procevid += 1 |
| 110 | return ret |
| 111 | finally: |
| 112 | idlock.release() |
| 113 | |
| 114 | class procevent(event): |
| 115 | def __init__(self, id): |
| 116 | super(procevent, self).__init__() |
| 117 | if isinstance(id, procevent): |
| 118 | self.id = id.id |
| 119 | else: |
| 120 | self.id = id |
| 121 | |
| 122 | class startevent(procevent): |
| 123 | def __init__(self): |
| 124 | super(startevent, self).__init__(getprocid()) |
| 125 | |
| 126 | class finishevent(procevent): |
| 127 | def __init__(self, start, aborted): |
| 128 | super(finishevent, self).__init__(start) |
| 129 | self.aborted = aborted |
| 130 | |
| 131 | sysres = staticdir() |
| 132 | itime = time.time() |
| 133 | ires = resource.getrusage(resource.RUSAGE_SELF) |
| 134 | def ct(): |
| 135 | ru = resource.getrusage(resource.RUSAGE_SELF) |
| 136 | return (ru.ru_utime - ires.ru_utime) + (ru.ru_stime - ires.ru_stime) |
| 137 | sysres["realtime"] = simpleattr(func = lambda: time.time() - itime) |
| 138 | sysres["cputime"] = simpleattr(func = ct) |
| 139 | sysres["utime"] = simpleattr(func = lambda: resource.getrusage(resource.RUSAGE_SELF).ru_utime - ires.ru_utime) |
| 140 | sysres["stime"] = simpleattr(func = lambda: resource.getrusage(resource.RUSAGE_SELF).ru_stime - ires.ru_stime) |
| 141 | sysres["maxrss"] = simpleattr(func = lambda: resource.getrusage(resource.RUSAGE_SELF).ru_maxrss) |
| 142 | sysres["rusage"] = simpleattr(func = lambda: resource.getrusage(resource.RUSAGE_SELF)) |
| 143 | |
| 144 | sysinfo = staticdir() |
| 145 | sysinfo["pid"] = simpleattr(func = os.getpid) |
| 146 | sysinfo["uname"] = simpleattr(func = os.uname) |
| 147 | sysinfo["hostname"] = simpleattr(func = socket.gethostname) |
| 148 | sysinfo["platform"] = valueattr(init = sys.platform) |