Stop service if lock file was removed

1 files changed, 25 insertions, 0 deletions

diff --git a/lock.py b/lock.py
new file mode 100644
index 0000000..8be0651
--- /dev/null
+++ b/lock.py
@@ -0,0 +1,25 @@
+import os
+import tempfile
+
+lock_dir_path = os.path.join(tempfile.gettempdir(), "rwa-session-service")
+lock_file_path = os.path.join(lock_dir_path, "rwa-session-service.lock")
+
+
+def lock():
+    """Create lock file."""
+    os.makedirs(lock_dir_path, exist_ok=True)
+    with open(lock_file_path, "w") as f:
+        f.write("lock")
+
+
+def is_locked():
+    """Check if the lock file exists."""
+    return os.path.exists(lock_file_path)
+
+
+def unlock():
+    """Remove the lock file."""
+    try:
+        os.remove(lock_file_path)
+    except FileNotFoundError:
+        pass