mirror of
https://github.com/avatao-content/baseimage-tutorial-framework
synced 2024-11-17 07:01:33 +00:00
41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
|
from os import remove
|
||
|
|
from fcntl import flock, LOCK_EX, LOCK_UN
|
||
|
|
|
||
|
|
|
||
|
|
class RefCounter:
|
||
|
|
def __init__(self, lockpath):
|
||
|
|
self.lockpath = lockpath
|
||
|
|
self._lockfile = open(self.lockpath, 'a+')
|
||
|
|
flock(self._lockfile, LOCK_EX)
|
||
|
|
counter = self._read_counter()
|
||
|
|
self._write_counter(counter+1)
|
||
|
|
flock(self._lockfile, LOCK_UN)
|
||
|
|
|
||
|
|
def _read_counter(self):
|
||
|
|
self._lockfile.seek(0)
|
||
|
|
try:
|
||
|
|
counter = int(self._lockfile.read())
|
||
|
|
except ValueError:
|
||
|
|
counter = 0
|
||
|
|
return counter
|
||
|
|
|
||
|
|
def _write_counter(self, counter):
|
||
|
|
self._lockfile.seek(0)
|
||
|
|
self._lockfile.truncate()
|
||
|
|
self._lockfile.write(str(counter))
|
||
|
|
self._lockfile.flush()
|
||
|
|
|
||
|
|
def teardown_instance(self):
|
||
|
|
flock(self._lockfile, LOCK_EX)
|
||
|
|
counter = self._read_counter()
|
||
|
|
if counter <= 1:
|
||
|
|
remove(self.lockpath)
|
||
|
|
self.deallocate()
|
||
|
|
else:
|
||
|
|
self._write_counter(counter-1)
|
||
|
|
flock(self._lockfile, LOCK_UN)
|
||
|
|
self._lockfile.close()
|
||
|
|
|
||
|
|
def deallocate(self):
|
||
|
|
pass
|