Move stuff to a separate file, & fix big bugs

This commit is contained in:
askiiart 2022-12-14 22:23:07 -06:00
parent 836a5ff07e
commit 49f1ca76d2
2 changed files with 60 additions and 9 deletions

View file

@ -10,7 +10,7 @@ def docker_info():
""" """
raw_info = getoutput('docker ps') raw_info = getoutput('docker ps')
# 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')] header = raw_info[:raw_info.find('\n')+1]
header_temp = header header_temp = header
header_indices = {} header_indices = {}
while ' ' in header_temp: # Split header while ' ' in header_temp: # Split header
@ -28,21 +28,24 @@ def docker_info():
containers.append(line.strip()[line.strip().rfind(' ')+1:]) containers.append(line.strip()[line.strip().rfind(' ')+1:])
# Fill in info # Fill in info
for container in containers: for i in range(len(containers)):
info[container] = {} info[containers[i]] = {}
for column in header_indices: for column in header_indices:
end_i = header_indices[column] + len(column) end_i = header_indices[column] + len(column)
info[container][column] = header[header_indices[column]:end_i] info[containers[i]][column] = raw_info[i][header_indices[column]:end_i]
debug = True debug = True
if debug: if debug:
print('Raw info:\n', raw_info) print('Raw info:\n', raw_info, '\n')
print('Header:\n', header) print('Header:\n', header, '\n')
print('Header indices:\n', header_indices) print('Header indices:\n', header_indices, '\n')
print('Containers:\n', containers) print('Containers:\n', containers, '\n')
print('Info:\n', info) print('Info:\n', info, '\n')
return info return info
if '\n' not in getoutput('docker ps'):
print('No containers running')
exit()
docker_info() docker_info()

48
docker_parser.py Normal file
View file

@ -0,0 +1,48 @@
from subprocess import getoutput
import pprint
class NoContainersError(Exception):
pass
class DockerParser:
def containers_info(raw_info=getoutput('docker container list')):
"""
Gets info about all the Docker containers
:return: Nested dict of containers info
"""
if '\n' not in raw_info:
raise(NoContainersError('A Docker container is required to run this program. Please create a docker container and try again.'))
raw_info = getoutput('docker ps')
# 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
raw_info = raw_info[raw_info.find('\n')+1:] # Remove header (example above)
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]
return info
if __name__ == '__main__':
try:
pprint.pprint(DockerParser.containers_info())
except NoContainersError as e:
print(e)