author | Patrick Mezard <patrick@mezard.eu> |
Sun, 19 Aug 2012 16:41:09 +0200 | |
branch | stable |
changeset 1126 | dbcb11553a3b |
parent 1125 | 20a680e7f09f |
child 1127 | 79d7aa832cf4 |
permissions | -rw-r--r-- |
1123
17bcbb020dda
tests: add killdaemons helper script
Matt Mackall <mpm@selenic.com>
parents:
diff
changeset
|
1 |
#!/usr/bin/env python |
17bcbb020dda
tests: add killdaemons helper script
Matt Mackall <mpm@selenic.com>
parents:
diff
changeset
|
2 |
|
1125
20a680e7f09f
pylint, pyflakes: remove unused or duplicate imports
Nicolas Dumazet <nicdumz.commits@gmail.com>
parents:
1124
diff
changeset
|
3 |
import os, time, errno, signal |
1123
17bcbb020dda
tests: add killdaemons helper script
Matt Mackall <mpm@selenic.com>
parents:
diff
changeset
|
4 |
|
1126
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
5 |
def killdaemons(pidfile, tryhard=True, remove=False, logfn=None): |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
6 |
if not logfn: |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
7 |
logfn = lambda s: s |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
8 |
# Kill off any leftover daemon processes |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
9 |
try: |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
10 |
fp = open(pidfile) |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
11 |
for line in fp: |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
12 |
try: |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
13 |
pid = int(line) |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
14 |
except ValueError: |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
15 |
continue |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
16 |
try: |
1123
17bcbb020dda
tests: add killdaemons helper script
Matt Mackall <mpm@selenic.com>
parents:
diff
changeset
|
17 |
os.kill(pid, 0) |
1126
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
18 |
logfn('# Killing daemon process %d' % pid) |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
19 |
os.kill(pid, signal.SIGTERM) |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
20 |
if tryhard: |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
21 |
for i in range(10): |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
22 |
time.sleep(0.05) |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
23 |
os.kill(pid, 0) |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
24 |
else: |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
25 |
time.sleep(0.1) |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
26 |
os.kill(pid, 0) |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
27 |
logfn('# Daemon process %d is stuck - really killing it' % pid) |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
28 |
os.kill(pid, signal.SIGKILL) |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
29 |
except OSError, err: |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
30 |
if err.errno != errno.ESRCH: |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
31 |
raise |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
32 |
fp.close() |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
33 |
if remove: |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
34 |
os.unlink(pidfile) |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
35 |
except IOError: |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
36 |
pass |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
37 |
|
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
38 |
if __name__ == '__main__': |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
39 |
killdaemons(os.environ['DAEMON_PIDS']) |
dbcb11553a3b
run-tests: do not duplicate killdaemons() code
Patrick Mezard <patrick@mezard.eu>
parents:
1125
diff
changeset
|
40 |