Make common methods threading safe. pytest enhancements
[osm/common.git] / osm_common / fsbase.py
index 6f82cd3..b941c21 100644 (file)
 # limitations under the License.
 
 
+import logging
 from http import HTTPStatus
+from osm_common.common_utils import FakeLock
+from threading import Lock
 
 __author__ = "Alfonso Tierno <alfonso.tiernosepulveda@telefonica.com>"
 
@@ -28,8 +31,24 @@ class FsException(Exception):
 
 
 class FsBase(object):
-    def __init__(self):
-        pass
+    def __init__(self, logger_name='fs', lock=False):
+        """
+        Constructor of FsBase
+        :param logger_name: logging name
+        :param lock: Used to protect simultaneous access to the same instance class by several threads:
+            False, None: Do not protect, this object will only be accessed by one thread
+            True: This object needs to be protected by several threads accessing.
+            Lock object. Use thi Lock for the threads access protection
+        """
+        self.logger = logging.getLogger(logger_name)
+        if not lock:
+            self.lock = FakeLock()
+        elif lock is True:
+            self.lock = Lock()
+        elif isinstance(lock, Lock):
+            self.lock = lock
+        else:
+            raise ValueError("lock parameter must be a Lock class or boolean")
 
     def get_params(self):
         return {}