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 os
class NoContainersError(Exception):
pass
class DockerParser:
class Docker:
def ps(raw_info=getoutput('docker ps')):
"""
Gets info about all running Docker containers from docker ps
@ -43,7 +44,7 @@ class DockerParser:
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
:return: Nested dict of containers info
@ -53,34 +54,85 @@ class DockerParser:
raw_info = getoutput('docker container list')
# Header: "CONTAINER ID IMAGE COMMAND CREATED STATUS PORTS NAMES" (with way more spaces)
header = raw_info[:raw_info.find('\n')+1]
header_temp = header
header_indices = {}
while ' ' in header_temp: # Split header
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
header_indices = {'CONTAINER ID': header.find('CONTAINER ID'), 'IMAGE': header.find('IMAGE'),
'COMMAND': header.find('COMMAND'), 'CREATED': header.find('CREATED'), 'STATUS': header.find('STATUS'),
'PORTS': header.find('PORTS'), 'NAMES': header.find('NAMES')}
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 = {}
# Find container names
containers = []
for line in raw_info.split('\n'):
containers.append(line.strip()[line.strip().rfind(' ')+1:])
# Fill in info
for i in range(len(containers)):
info[containers[i]] = {}
for column in header_indices:
end_i = header_indices[column] + len(column)
info[containers[i]][column] = raw_info[i][header_indices[column]:end_i]
header_indices_keys = list(header_indices)
for j in range(len(header_indices_keys)): # TODO: Fix
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
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__':
try:
pprint.pprint(DockerParser.containers_info())
pprint.pprint(Docker.containers_info())
except NoContainersError as e:
print(e)