From 945a960f1c2664b51d8d414e03a917b1e07459e8 Mon Sep 17 00:00:00 2001
From: "J.A. de Jong - Redu-Sone B.V., ASCEE V.O.F" <j.a.dejong@ascee.nl>
Date: Sun, 1 Mar 2020 15:06:01 +0100
Subject: [PATCH] Added a machine-local LASP shelve, next to the overall
 (multiple machine-wide) one

---
 lasp/lasp_common.py | 40 ++++++++++++++++++++++++++++++++++------
 1 file changed, 34 insertions(+), 6 deletions(-)

diff --git a/lasp/lasp_common.py b/lasp/lasp_common.py
index aec29da..39b065d 100644
--- a/lasp/lasp_common.py
+++ b/lasp/lasp_common.py
@@ -1,19 +1,25 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
+import os
+import platform
+import shelve
+
+import appdirs
 import numpy as np
+
 from .wrappers import Window as wWindow
-import appdirs, os, shelve
+
 """
 Common definitions used throughout the code.
 """
 
-__all__ = ['P_REF', 'FreqWeighting', 'TimeWeighting', 'getTime', 
-           'getFreq', 'lasp_shelve',
-           'W_REF', 'U_REF', 'I_REF']
+__all__ = [
+    'P_REF', 'FreqWeighting', 'TimeWeighting', 'getTime', 'getFreq',
+    'lasp_shelve', 'this_lasp_shelve', 'W_REF', 'U_REF', 'I_REF'
+]
 
 lasp_appdir = appdirs.user_data_dir('Lasp', 'ASCEE')
 
-
 if not os.path.exists(lasp_appdir):
     try:
         os.mkdir(lasp_appdir)
@@ -29,7 +35,8 @@ class lasp_shelve:
     def __enter__(self):
         if lasp_shelve.shelve is None:
             assert lasp_shelve.refcount == 0
-            lasp_shelve.shelve = shelve.open(os.path.join(lasp_appdir, 'config.shelve'))
+            lasp_shelve.shelve = shelve.open(
+                os.path.join(lasp_appdir, 'config.shelve'))
         lasp_shelve.refcount += 1
         return lasp_shelve.shelve
 
@@ -39,6 +46,27 @@ class lasp_shelve:
             lasp_shelve.shelve.close()
             lasp_shelve.shelve = None
 
+
+class this_lasp_shelve:
+    refcount = 0
+    shelve = None
+
+    def __enter__(self):
+        if this_lasp_shelve.shelve is None:
+            assert lasp_shelve.refcount == 0
+            node = platform.node()
+            this_lasp_shelve.shelve = shelve.open(
+                os.path.join(lasp_appdir, f'{node}_config.shelve'))
+        this_lasp_shelve.refcount += 1
+        return this_lasp_shelve.shelve
+
+    def __exit__(self, type, value, traceback):
+        this_lasp_shelve.refcount -= 1
+        if this_lasp_shelve.refcount == 0:
+            this_lasp_shelve.shelve.close()
+            this_lasp_shelve.shelve = None
+
+
 # Reference sound pressure level
 P_REF = 2e-5