From a08ca309cb5bd7971210a9247a38c9bbe376a6e6 Mon Sep 17 00:00:00 2001 From: Scott Gasch Date: Thu, 29 Jul 2021 09:11:05 -0700 Subject: Lockfile custom command, fixup minor things. --- lockfile.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'lockfile.py') diff --git a/lockfile.py b/lockfile.py index ee8c255..770beaa 100644 --- a/lockfile.py +++ b/lockfile.py @@ -37,9 +37,11 @@ class LockFile(object): *, do_signal_cleanup: bool = True, expiration_timestamp: Optional[float] = None, + override_command: Optional[str] = None, ) -> None: self.is_locked = False self.lockfile = lockfile_path + self.override_command = override_command if do_signal_cleanup: signal.signal(signal.SIGINT, self._signal) signal.signal(signal.SIGTERM, self._signal) @@ -117,10 +119,14 @@ class LockFile(object): self.release() def _get_lockfile_contents(self) -> str: + if self.override_command: + cmd = self.override_command + else: + cmd = ' '.join(sys.argv) contents = LockFileContents( pid = os.getpid(), - commandline = ' '.join(sys.argv), - expiration_timestamp = self.expiration_timestamp + cmd, + expiration_timestamp = self.expiration_timestamp, ) return json.dumps(contents.__dict__) -- cgit v1.3