week06

env/lib/python3.12/site-packages/pyshark/packet/common.py

@@ -0,0 +1,41 @@
+import sys
+import functools
+
+import termcolor
+
+
+class Pickleable(object):
+    """
+    Base class that implements getstate/setstate, since most of the classes are overriding getattr.
+    """
+
+    def __getstate__(self):
+        return self.__dict__
+
+    def __setstate__(self, data):
+        self.__dict__.update(data)
+
+
+class SlotsPickleable(object):
+    __slots__ = []
+
+    def __getstate__(self):
+        ret = {}
+        for slot in self.__slots__:
+            ret[slot] = getattr(self, slot)
+        return ret
+
+    def __setstate__(self, data):
+        for key, val in data.items():
+            setattr(self, key, val)
+
+
+@functools.wraps(termcolor.colored)
+def colored(text, *args, **kwargs):
+    try:
+        enable_color = sys.stdout.isatty()
+    except (AttributeError, NotImplementedError, FileNotFoundError):
+        enable_color = False
+    if enable_color:
+        return termcolor.colored(text, *args, **kwargs)
+    return text