Probably finished

This commit is contained in:
--global 2022-12-15 09:56:06 -06:00
parent b4dcdbc905
commit 3ac23af420

View file

@ -1,10 +1,11 @@
from subprocess import getoutput from subprocess import getoutput, getstatusoutput
import pprint import pprint
import os
class NoContainersError(Exception): class NoContainersError(Exception):
pass pass
class DockerParser: class Docker:
def ps(raw_info=getoutput('docker ps')): def ps(raw_info=getoutput('docker ps')):
""" """
Gets info about all running Docker containers from docker ps Gets info about all running Docker containers from docker ps
@ -43,7 +44,7 @@ class DockerParser:
return info return info
def containers_info(raw_info=getoutput('docker container list')): def container_list(raw_info=getoutput('docker container list')):
""" """
Gets info about all the Docker containers Gets info about all the Docker containers
:return: Nested dict of containers info :return: Nested dict of containers info
@ -53,15 +54,12 @@ class DockerParser:
raw_info = getoutput('docker container list') raw_info = getoutput('docker container list')
# Header: "CONTAINER ID IMAGE COMMAND CREATED STATUS PORTS NAMES" (with way more spaces) # Header: "CONTAINER ID IMAGE COMMAND CREATED STATUS PORTS NAMES" (with way more spaces)
header = raw_info[:raw_info.find('\n')+1] header = raw_info[:raw_info.find('\n')+1]
header_temp = header header_indices = {'CONTAINER ID': header.find('CONTAINER ID'), 'IMAGE': header.find('IMAGE'),
header_indices = {} 'COMMAND': header.find('COMMAND'), 'CREATED': header.find('CREATED'), 'STATUS': header.find('STATUS'),
while ' ' in header_temp: # Split header 'PORTS': header.find('PORTS'), 'NAMES': header.find('NAMES')}
temp = header_temp[header_temp.rfind(' ')+2:].strip()
header_indices[temp] = header.find(temp)
header_temp = header_temp[:header_temp.rfind(' ')].strip()
header_indices[header[:header.find(' ')]] = 0
raw_info = raw_info[raw_info.find('\n')+1:] # Remove header (example above) # Remove header (example above)
raw_info = raw_info[raw_info.find('\n')+1:]
info = {} info = {}
# Find container names # Find container names
@ -72,15 +70,69 @@ class DockerParser:
# Fill in info # Fill in info
for i in range(len(containers)): for i in range(len(containers)):
info[containers[i]] = {} info[containers[i]] = {}
for column in header_indices: header_indices_keys = list(header_indices)
end_i = header_indices[column] + len(column) for j in range(len(header_indices_keys)): # TODO: Fix
info[containers[i]][column] = raw_info[i][header_indices[column]:end_i] start_i = header_indices[header_indices_keys[j]]
if j+1 != len(header_indices):
end_i = header_indices[header_indices_keys[j+1]]
else:
end_i = len(header)
info[containers[i]][header_indices_keys[j]] = \
raw_info.split('\n')[i][start_i:end_i].strip()
return info return info
def compose(dir):
"""
Composes whatever is in dir
:parameters:
dir: The directory of a container; must contain docker-compose.yml file
:returns:
tuple: (exit_code, compose_output)
"""
cwd = os.getcwd()
os.chdir(dir)
status = getstatusoutput('docker ps')
os.chdir(cwd)
return status
def start(container):
"""
Starts a container
:parameters:
container: The name of the container to start
:returns:
int: The exit code of docker start
"""
return getstatusoutput(f'docker start {container}')[0]
def stop(container):
"""
Stops a container
:parameters:
container: The name of the container to stop
:returns:
int: The exit code of docker stop
"""
return getstatusoutput(f'docker stop {container}')[0]
def rm(container):
"""
Deletes a container
:parameters:
container: The name of the container to remove
:returns:
int: The exit code of docker rm
"""
return getstatusoutput(f'docker rm {container}')[0]
if __name__ == '__main__': if __name__ == '__main__':
try: try:
pprint.pprint(DockerParser.containers_info()) pprint.pprint(Docker.containers_info())
except NoContainersError as e: except NoContainersError as e:
print(e) print(e)