git » pytipc » master » tree

[master] / examples / cs.py 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
import os
import sys
import time
import threading


def withthreads(srv, cli):
	t1 = threading.Thread(target = srv)
	t1.start()
	cli()


def withfork(srv, cli):
	ppid = os.getpid()
	cpid = os.fork()
	if cpid < 0:
		raise
	elif cpid == 0:
		# child
		try:
			srv()
		finally:
			os.kill(ppid, 15)
	else:
		# parent
		try:
			cli()
		finally:
			os.kill(cpid, 15)


def run(srv, cli):
	if int(os.environ.get('USE_THREADS', '0')):
		withthreads(srv, cli)
	else:
		withfork(srv, cli)