diff options
Diffstat (limited to 'lockfile.py')
| -rw-r--r-- | lockfile.py | 10 |
1 files changed, 8 insertions, 2 deletions
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__) |