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
|
from __future__ import annotations
import inspect
import os
import sys
from contextlib import contextmanager
from io import StringIO
IS_WIN32 = sys.platform == "win32"
class ProcInfo:
def __init__(self, pid, uid, stat, args):
self.pid = pid
self.uid = uid
self.stat = stat
self.args = args
def __repr__(self):
return f"ProcInfo({self.pid!r}, {self.uid!r}, {self.stat!r}, {self.args!r})"
@contextmanager
def captured_stdout(stdin=""):
"""
Captures stdout (similar to the redirect_stdout in Python 3.4+, but with slightly different arguments)
"""
prevstdin = sys.stdin
prevstdout = sys.stdout
sys.stdin = StringIO(stdin)
sys.stdout = StringIO()
try:
yield sys.stdout
finally:
sys.stdin = prevstdin
sys.stdout = prevstdout
class StaticProperty:
"""This acts like a static property, allowing access via class or object.
This is a non-data descriptor."""
def __init__(self, function):
self._function = function
self.__doc__ = function.__doc__
def __get__(self, obj, klass=None):
return self._function()
def getdoc(obj):
"""
This gets a docstring if available, and cleans it, but does not look up docs in
inheritance tree (Pre Python 3.5 behavior of ``inspect.getdoc``).
"""
try:
doc = obj.__doc__
except AttributeError:
return None
if not isinstance(doc, str):
return None
return inspect.cleandoc(doc)
def read_fd_decode_safely(fd, size=4096):
"""
This reads a utf-8 file descriptor and returns a chunk, growing up to
three bytes if needed to decode the character at the end.
Returns the data and the decoded text.
"""
data = os.read(fd.fileno(), size)
for _ in range(3):
try:
return data, data.decode("utf-8")
except UnicodeDecodeError as e:
if e.reason != "unexpected end of data":
raise
data += os.read(fd.fileno(), 1)
return data, data.decode("utf-8")
|
