From 939ad5451c07ef4f3cd1eb72f733e45dda0db05a Mon Sep 17 00:00:00 2001
From: Jan Petykiewicz <anewusername@gmail.com>
Date: Fri, 8 Jan 2021 21:21:17 -0800
Subject: [PATCH] enable pickling of LockableImpl

---
 masque/traits/lockable.py | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/masque/traits/lockable.py b/masque/traits/lockable.py
index 5a0f06f..06b817c 100644
--- a/masque/traits/lockable.py
+++ b/masque/traits/lockable.py
@@ -1,4 +1,4 @@
-from typing import TypeVar
+from typing import TypeVar, Dict, Tuple, Any
 from abc import ABCMeta, abstractmethod
 
 from ..error import PatternLockedError
@@ -12,7 +12,7 @@ class Lockable(metaclass=ABCMeta):
     """
     Abstract class for all lockable entities
     """
-    __slots__ = ()
+    __slots__ = ()      # type: Tuple[str, ...]
 
     '''
     ---- Methods
@@ -68,7 +68,7 @@ class LockableImpl(Lockable, metaclass=ABCMeta):
     """
     Simple implementation of Lockable
     """
-    __slots__ = ()
+    __slots__ = ()      # type: Tuple[str, ...]
 
     locked: bool
     """ If `True`, disallows changes to the object """
@@ -81,6 +81,16 @@ class LockableImpl(Lockable, metaclass=ABCMeta):
             raise PatternLockedError()
         object.__setattr__(self, name, value)
 
+    def __getstate__(self) -> Dict[str, Any]:
+        if hasattr(self, '__slots__'):
+            return {key: getattr(self, key) for key in self.__slots__}
+        else:
+            return self.__dict__
+
+    def __setstate__(self, state: Dict[str, Any]) -> None:
+        for k, v in state.items():
+            object.__setattr__(self, k, v)
+
     def lock(self: I) -> I:
         object.__setattr__(self, 'locked', True)
         return self