27 |
|
|
28 |
|
cdef extern from "statgrab.h": |
29 |
|
cdef extern int sg_init() |
30 |
+ |
cdef extern int sg_shutdown() |
31 |
+ |
cdef extern int sg_snapshot() |
32 |
|
cdef extern int sg_drop_privileges() |
33 |
|
|
34 |
|
ctypedef enum sg_error: |
274 |
|
py_SG_PROCESS_STATE_UNKNOWN = SG_PROCESS_STATE_UNKNOWN |
275 |
|
|
276 |
|
|
277 |
< |
class Result: |
277 |
> |
class Result(dict): |
278 |
|
def __init__(self, attrs): |
279 |
< |
self.attrs = attrs |
280 |
< |
for attr in attrs: |
279 |
< |
setattr(self, attr, attrs[attr]) |
280 |
< |
def __getitem__(self, item): |
281 |
< |
return getattr(self, item) |
282 |
< |
def __repr__(self): |
283 |
< |
return str(self.attrs) |
279 |
> |
self.attrs = attrs # to maintain compatibility |
280 |
> |
super(Result, self).__init__(attrs) |
281 |
|
|
282 |
+ |
def __getattr__(self, item): |
283 |
+ |
try: |
284 |
+ |
return self.__getitem__(item) |
285 |
+ |
except KeyError: |
286 |
+ |
raise AttributeError(item) |
287 |
+ |
|
288 |
|
class StatgrabException(Exception): |
289 |
|
def __init__(self, value): |
290 |
|
self.value = value |
298 |
|
else: |
299 |
|
return False |
300 |
|
|
301 |
+ |
def py_sg_shutdown(): |
302 |
+ |
if sg_shutdown() == 0: |
303 |
+ |
return True |
304 |
+ |
else: |
305 |
+ |
return False |
306 |
+ |
|
307 |
+ |
def py_sg_snapshot(): |
308 |
+ |
if sg_snapshot() == 0: |
309 |
+ |
return True |
310 |
+ |
else: |
311 |
+ |
return False |
312 |
+ |
|
313 |
|
def py_sg_drop_privileges(): |
314 |
|
if sg_drop_privileges() == 0: |
315 |
|
return True |
612 |
|
raise StatgrabException, 'sg_get_process_stats() returned NULL' |
613 |
|
list = [] |
614 |
|
for i from 0 <= i < entries: |
615 |
< |
if s.process_name == NULL: |
616 |
< |
s.process_name = '' |
617 |
< |
if s.proctitle == NULL: |
618 |
< |
s.proctitle = '' |
615 |
> |
if s.process_name is NULL: |
616 |
> |
process_name = '' |
617 |
> |
else: |
618 |
> |
process_name = s.process_name |
619 |
> |
if s.proctitle is NULL: |
620 |
> |
proctitle = '' |
621 |
> |
else: |
622 |
> |
proctitle = s.proctitle |
623 |
|
list.append(Result( |
624 |
< |
{'process_name': s.process_name, |
625 |
< |
'proctitle' : s.proctitle, |
624 |
> |
{'process_name': process_name, |
625 |
> |
'proctitle' : proctitle, |
626 |
|
'pid' : s.pid, |
627 |
|
'parent' : s.parent, |
628 |
|
'pgid' : s.pgid, |