-
Notifications
You must be signed in to change notification settings - Fork 0
/
worker.py
68 lines (54 loc) · 1.65 KB
/
worker.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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
from libs.utils import getjson, save_json
from multiprocessing import Pool
from threading import Thread
import time
import threading
import socket
def get_update():
filepath = env_path.joinpath("task.json")
datafile = getjson(filepath)
data["waiting"] = datafile["waiting"]
data["assignments"] = datafile["assignments"]
def create_task(func, name):
class Thd(Thread):
def __init__(self):
Thread.__init__(self)
self.name = name
self.func = func
def run(self):
print(f"\n{time.ctime(time.time())} - starting {self.name}\n")
self.func()
print(f"\n{time.ctime(time.time())} - exiting {self.name}\n")
task = Thd()
task.start()
task.join()
print(f"\n{time.ctime(time.time())} - task {name}-thread terminated\n")
def run_task(i):
create_task(data["assignments"][i], i)
def main(*procs):
data["assignments"].extend(list(procs))
n = len(data["assignments"])
pn = range(n)
with Pool(n) as p:
p.map(run_task, pn)
def init():
get_update()
data["waiting"] = True
save_json(env_path.joinpath("task.json"), data)
while data["waiting"]:
try:
print(data)
time.sleep(3)
get_update()
except KeyboardInterrupt:
data["waiting"] = False
save_json(env_path.joinpath("task.json"), data)
print(data)
if __name__ == "__main__":
threads = [threading.Thread(target=scanport, kwargs=dict(port=i)) for i in ports]
#threads[0].start()
#threads[0].join()
for t in threads:
t.start()
[t.join() for t in threads]
print(ports)