YAO/YAO-agent
1
0
Fork 0
mirror of https://github.com/newnius/YAO-agent.git synced 2025-12-15 13:56:44 +00:00
Code Issues Releases Wiki Activity

bugfix, add query stat

Browse Source
This commit is contained in:
Newnius
2019-03-18 15:59:54 +08:00
parent 71bd27f530
commit c4e390d310
4 changed files with 133 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
executor.py
View File

@@ -3,8 +3,11 @@ import docker
def run():
client = docker.from_env()
#print(client.containers.run(image="alpine", command="echo 'Hello World'", environment={"KEY": "value"}))
print(client.containers.run(image="nvidia/cuda:9.0-base", command="nvidia-smi", environment={"KEY": "value"}, runtime="nvidia"))
try:
print(client.containers.run(image="alpine", command="nvid", environment={"KEY": "value"}))
# print(client.containers.run(image="nvidia/cuda:9.0-base", command="nvidia-smi", environment={"KEY": "value"}, runtime="nvidia"))
except Exception as e:
print(e.__class__.__name__, e)
def run_in_background():
@@ -19,4 +22,31 @@ def list_containers():
print(container.id)
run()
def get_logs(id):
try:
client = docker.from_env()
container = client.containers.get(id)
print(container.logs().decode())
except Exception as e:
print(e)
def get_status(id):
client = docker.from_env()
container = client.containers.list(all=True, filters={'id': id})
status = {}
if len(container) > 0:
container= container[0]
status['id'] = container.short_id
status['image'] = container.attrs['Config']['Image']
status['image_digest'] = container.attrs['Image']
status['command'] = container.attrs['Config']['Cmd']
status['createdAt'] = container.attrs['Created']
status['finishedAt'] = container.attrs['State']['FinishedAt']
status['status'] = container.status
if status['command'] is not None:
status['command'] = ' '.join(container.attrs['Config']['Cmd'])
print(status)
get_status('')