mirror of
https://github.com/facebook/sapling.git
synced 2024-10-07 07:17:55 +03:00
584656dff3
Summary: Turned on the auto formatter. Ran `arc lint --apply-patches --take BLACK **/*.py`. Then run `arc lint` again so some other autofixers like spellchecker etc. looked at the code base. Manually accept the changes whenever they make sense, or use a workaround (ex. changing "dict()" to "dict constructor") where autofix is false positive. Disabled linters on files that are hard (i18n/polib.py) to fix, or less interesting to fix (hgsubversion tests), or cannot be fixed without breaking OSS build (FBPYTHON4). Conflicted linters (test-check-module-imports.t, part of test-check-code.t, test-check-pyflakes.t) are removed or disabled. Duplicated linters (test-check-pyflakes.t, test-check-pylint.t) are removed. An issue of the auto-formatter is lines are no longer guarnateed to be <= 80 chars. But that seems less important comparing with the benefit auto-formatter provides. As we're here, also remove test-check-py3-compat.t, as it is currently broken if `PYTHON3=/bin/python3` is set. Reviewed By: wez, phillco, simpkins, pkaush, singhsrb Differential Revision: D8173629 fbshipit-source-id: 90e248ae0c5e6eaadbe25520a6ee42d32005621b
135 lines
4.2 KiB
Python
Executable File
135 lines
4.2 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import errno
|
|
import os
|
|
import signal
|
|
import sys
|
|
import time
|
|
|
|
|
|
if os.name == "nt":
|
|
import ctypes
|
|
|
|
_BOOL = ctypes.c_long
|
|
_DWORD = ctypes.c_ulong
|
|
_UINT = ctypes.c_uint
|
|
_HANDLE = ctypes.c_void_p
|
|
|
|
ctypes.windll.kernel32.CloseHandle.argtypes = [_HANDLE]
|
|
ctypes.windll.kernel32.CloseHandle.restype = _BOOL
|
|
|
|
ctypes.windll.kernel32.GetLastError.argtypes = []
|
|
ctypes.windll.kernel32.GetLastError.restype = _DWORD
|
|
|
|
ctypes.windll.kernel32.OpenProcess.argtypes = [_DWORD, _BOOL, _DWORD]
|
|
ctypes.windll.kernel32.OpenProcess.restype = _HANDLE
|
|
|
|
ctypes.windll.kernel32.TerminateProcess.argtypes = [_HANDLE, _UINT]
|
|
ctypes.windll.kernel32.TerminateProcess.restype = _BOOL
|
|
|
|
ctypes.windll.kernel32.WaitForSingleObject.argtypes = [_HANDLE, _DWORD]
|
|
ctypes.windll.kernel32.WaitForSingleObject.restype = _DWORD
|
|
|
|
def _check(ret, expectederr=None):
|
|
if ret == 0:
|
|
winerrno = ctypes.GetLastError()
|
|
if winerrno == expectederr:
|
|
return True
|
|
raise ctypes.WinError(winerrno)
|
|
|
|
def kill(pid, logfn, tryhard=True):
|
|
logfn("# Killing daemon process %d" % pid)
|
|
PROCESS_TERMINATE = 1
|
|
PROCESS_QUERY_INFORMATION = 0x400
|
|
SYNCHRONIZE = 0x00100000
|
|
WAIT_OBJECT_0 = 0
|
|
WAIT_TIMEOUT = 258
|
|
WAIT_FAILED = _DWORD(0xFFFFFFFF).value
|
|
handle = ctypes.windll.kernel32.OpenProcess(
|
|
PROCESS_TERMINATE | SYNCHRONIZE | PROCESS_QUERY_INFORMATION, False, pid
|
|
)
|
|
if handle is None:
|
|
_check(0, 87) # err 87 when process not found
|
|
return # process not found, already finished
|
|
try:
|
|
r = ctypes.windll.kernel32.WaitForSingleObject(handle, 100)
|
|
if r == WAIT_OBJECT_0:
|
|
pass # terminated, but process handle still available
|
|
elif r == WAIT_TIMEOUT:
|
|
_check(ctypes.windll.kernel32.TerminateProcess(handle, -1))
|
|
elif r == WAIT_FAILED:
|
|
_check(0) # err stored in GetLastError()
|
|
|
|
# TODO?: forcefully kill when timeout
|
|
# and ?shorter waiting time? when tryhard==True
|
|
r = ctypes.windll.kernel32.WaitForSingleObject(handle, 100)
|
|
# timeout = 100 ms
|
|
if r == WAIT_OBJECT_0:
|
|
pass # process is terminated
|
|
elif r == WAIT_TIMEOUT:
|
|
logfn("# Daemon process %d is stuck")
|
|
elif r == WAIT_FAILED:
|
|
_check(0) # err stored in GetLastError()
|
|
except: # re-raises
|
|
ctypes.windll.kernel32.CloseHandle(handle) # no _check, keep error
|
|
raise
|
|
_check(ctypes.windll.kernel32.CloseHandle(handle))
|
|
|
|
|
|
else:
|
|
|
|
def kill(pid, logfn, tryhard=True):
|
|
try:
|
|
os.kill(pid, 0)
|
|
logfn("# Killing daemon process %d" % pid)
|
|
os.kill(pid, signal.SIGTERM)
|
|
if tryhard:
|
|
for i in range(10):
|
|
time.sleep(0.05)
|
|
os.kill(pid, 0)
|
|
else:
|
|
time.sleep(0.1)
|
|
os.kill(pid, 0)
|
|
logfn("# Daemon process %d is stuck - really killing it" % pid)
|
|
os.kill(pid, signal.SIGKILL)
|
|
except OSError as err:
|
|
if err.errno != errno.ESRCH:
|
|
raise
|
|
|
|
|
|
def killdaemons(pidfile, tryhard=True, remove=False, logfn=None):
|
|
if not logfn:
|
|
logfn = lambda s: s
|
|
# Kill off any leftover daemon processes
|
|
try:
|
|
pids = []
|
|
with open(pidfile) as fp:
|
|
for line in fp:
|
|
try:
|
|
pid = int(line)
|
|
if pid <= 0:
|
|
raise ValueError
|
|
except ValueError:
|
|
logfn(
|
|
"# Not killing daemon process %s - invalid pid" % line.rstrip()
|
|
)
|
|
continue
|
|
pids.append(pid)
|
|
for pid in pids:
|
|
kill(pid, logfn, tryhard)
|
|
if remove:
|
|
os.unlink(pidfile)
|
|
except IOError:
|
|
pass
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) > 1:
|
|
path, = sys.argv[1:]
|
|
else:
|
|
path = os.environ["DAEMON_PIDS"]
|
|
|
|
killdaemons(path)
|