fba/locking.py

   1 # Fedi API Block - An aggregator for fetching blocking data from fediverse nodes
   2 # Copyright (C) 2023 Free Software Foundation
   3 #
   4 # This program is free software: you can redistribute it and/or modify
   5 # it under the terms of the GNU Affero General Public License as published
   6 # by the Free Software Foundation, either version 3 of the License, or
   7 # (at your option) any later version.
   8 #
   9 # This program is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 # GNU Affero General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Affero General Public License
  15 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
  16
  17 import os
  18 import sys
  19 import tempfile
  20 import zc.lockfile
  21
  22 # Lock file
  23 lockfile = tempfile.gettempdir() + '/fba.lock'
  24 LOCK = None
  25
  26 def acquire():
  27     global LOCK
  28     # DEBUG: print("DEBUG: CALLED!")
  29
  30     try:
  31         # DEBUG: print(f"DEBUG: Acquiring lock: '{lockfile}'")
  32         LOCK = zc.lockfile.LockFile(lockfile)
  33         # DEBUG: print("DEBUG: Lock obtained.")
  34
  35     except zc.lockfile.LockError:
  36         print(f"ERROR: Cannot aquire lock: '{lockfile}'")
  37         sys.exit(100)
  38
  39     # DEBUG: print("DEBUG: EXIT!")
  40
  41 def release():
  42     # DEBUG: print("DEBUG: CALLED!")
  43     if LOCK is not None:
  44         # DEBUG: print("DEBUG: Releasing lock ...")
  45         LOCK.close()
  46         # DEBUG: print(f"DEBUG: Deleting lockfile='{lockfile}' ...")
  47         os.remove(lockfile)
  48
  49     # DEBUG: print("DEBUG: EXIT!")