YAO/YAO-agent
1
0
Fork 0
mirror of https://github.com/newnius/YAO-agent.git synced 2025-06-06 21:31:55 +00:00
Code Issues Releases Wiki Activity

update

Browse Source
This commit is contained in:
Newnius 2020-05-24 02:12:18 +08:00
parent 660fe82449
commit 0591ae0885
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
agent.py
View File

@ -33,6 +33,10 @@ id2token = {}
counter = {} counter = {}
event_counter = 0
client = docker.from_env()
def generate_token(stringLength=8): def generate_token(stringLength=8):
letters = string.ascii_lowercase letters = string.ascii_lowercase
@ -79,7 +83,8 @@ class MyHandler(BaseHTTPRequestHandler):
elif req.path == "/debug": elif req.path == "/debug":
msg = { msg = {
'pending_tasks': pending_tasks, 'pending_tasks': pending_tasks,
'id2token': id2token 'id2token': id2token,
'event_counter': event_counter
} }
self.send_response(200) self.send_response(200)
self.send_header('Content-type', 'application/json') self.send_header('Content-type', 'application/json')
@ -106,7 +111,6 @@ class MyHandler(BaseHTTPRequestHandler):
elif req.path == "/logs": elif req.path == "/logs":
try: try:
container_id = query.get('id')[0] container_id = query.get('id')[0]
client = docker.from_env()
container = client.containers.get(container_id) container = client.containers.get(container_id)
msg = {'code': 0, 'logs': str(container.logs().decode())} msg = {'code': 0, 'logs': str(container.logs().decode())}
except Exception as e: except Exception as e:
@ -119,7 +123,6 @@ class MyHandler(BaseHTTPRequestHandler):
elif req.path == "/status": elif req.path == "/status":
try: try:
container_id = query.get('id')[0] container_id = query.get('id')[0]
client = docker.from_env()
container = client.containers.list(all=True, filters={'id': container_id}) container = client.containers.list(all=True, filters={'id': container_id})
if len(container) > 0: if len(container) > 0:
container = container[0] container = container[0]
@ -201,7 +204,6 @@ class MyHandler(BaseHTTPRequestHandler):
docker_cmd docker_cmd
]) ])
client = docker.from_env()
container = client.containers.get('yao-agent-helper') container = client.containers.get('yao-agent-helper')
exit_code, output = container.exec_run(['sh', '-c', script]) exit_code, output = container.exec_run(['sh', '-c', script])
msg = {"code": 0, "id": output.decode('utf-8').rstrip('\n')} msg = {"code": 0, "id": output.decode('utf-8').rstrip('\n')}
@ -234,7 +236,6 @@ class MyHandler(BaseHTTPRequestHandler):
container_id = form.getvalue('id') container_id = form.getvalue('id')
try: try:
client = docker.from_env()
container = client.containers.get(container_id) container = client.containers.get(container_id)
container.stop(timeout=1) container.stop(timeout=1)
msg = {"code": 0, "error": "Success"} msg = {"code": 0, "error": "Success"}
@ -257,7 +258,6 @@ class MyHandler(BaseHTTPRequestHandler):
container_id = form.getvalue('id') container_id = form.getvalue('id')
try: try:
client = docker.from_env()
container = client.containers.get(container_id) container = client.containers.get(container_id)
container.remove(force=True) container.remove(force=True)
msg = {"code": 0, "error": "Success"} msg = {"code": 0, "error": "Success"}
@ -273,9 +273,10 @@ class MyHandler(BaseHTTPRequestHandler):
def event_trigger(): def event_trigger():
client = docker.from_env() global event_counter
for event in client.events(decode=True, filters={'event': 'die'}): for event in client.events(decode=True, filters={'event': 'die'}):
Thread(target=report).start() Thread(target=report).start()
event_counter += 1
print(event) print(event)