Fix error handling & add custom info printer
This commit is contained in:
askiiart
parent
1e05dd2cd6
commit
43af3125f1
|
|
@ -1,69 +1,107 @@
|
||||||
from docker_wrapper import Docker
|
from docker_wrapper import Docker, NoContainersError
|
||||||
from pprint import pprint
|
from subprocess import getoutput
|
||||||
|
|
||||||
while True:
|
|
||||||
# Main Menu
|
|
||||||
print('Select the container to manage:')
|
|
||||||
containers = Docker.containers()
|
|
||||||
for i in range(len(containers)):
|
|
||||||
print(f' {i} - {containers[i]}')
|
|
||||||
print(' q - quit')
|
|
||||||
|
|
||||||
container_i = input()
|
def container_to_str(container):
|
||||||
print()
|
"""
|
||||||
|
Returns info about a Docker container as a string
|
||||||
|
:parameters:f
|
||||||
|
container: The name of the container (str)
|
||||||
|
|
||||||
if container_i == 'q':
|
:returns:
|
||||||
exit(0)
|
str: The container info as a string, formatted for printing
|
||||||
|
"""
|
||||||
|
# Header: "CONTAINER ID IMAGE COMMAND CREATED STATUS PORTS NAMES" (with way more spaces)
|
||||||
|
info = Docker.container_info(container)
|
||||||
|
info_str = f'{container}\n'
|
||||||
|
info_str += f' CONTAINER ID: {info["CONTAINER ID"]}\n'
|
||||||
|
info_str += f' IMAGE: {info["IMAGE"]}\n'
|
||||||
|
info_str += f' COMMAND: {info["COMMAND"]}\n'
|
||||||
|
info_str += f' CREATED: {info["CREATED"]}\n'
|
||||||
|
info_str += f' STATUS: {info["STATUS"]}\n'
|
||||||
|
info_str += f' PORTS: {info["PORTS"]}\n'
|
||||||
|
info_str += f' NAMES: {info["NAMES"]}'
|
||||||
|
|
||||||
|
return info_str
|
||||||
|
|
||||||
|
try:
|
||||||
while True:
|
while True:
|
||||||
# Container Menu
|
if not Docker.containers_exist():
|
||||||
container = containers[int(container_i)]
|
raise(NoContainersError('No containers exist! Please create a container and try again.'))
|
||||||
print()
|
|
||||||
print(f'You selected {container}. What would you like to do with it?')
|
|
||||||
print(f' view - View container info - (docker ps -a)')
|
|
||||||
print(f' start - Starts the container (docker start {container})')
|
|
||||||
print(f' stop - Stops the container (docker stop {container}')
|
|
||||||
print(f' rm - Deletes the container (docker rm {container})')
|
|
||||||
print( ' menu - Back to the main menu')
|
|
||||||
|
|
||||||
selection = input()
|
# Main Menu
|
||||||
|
print('Select the container to manage:')
|
||||||
|
|
||||||
|
containers = Docker.containers()
|
||||||
|
for i in range(len(containers)):
|
||||||
|
print(f' {i} - {containers[i]}')
|
||||||
|
print(' q - Quit')
|
||||||
|
|
||||||
|
container_i = input()
|
||||||
print()
|
print()
|
||||||
|
|
||||||
if selection == 'menu':
|
if container_i == 'q':
|
||||||
break
|
exit(0)
|
||||||
|
|
||||||
elif selection == 'view':
|
while True:
|
||||||
pprint(Docker.container_info(container)) # TODO: Make better, custom printer
|
if not Docker.containers_exist():
|
||||||
|
raise(NoContainersError('No containers exist! Please create a container and try again.'))
|
||||||
|
# Container Menu
|
||||||
|
container = containers[int(container_i)]
|
||||||
|
print()
|
||||||
|
print(f'You selected {container}. What would you like to do with it?')
|
||||||
|
print(f' view - View container info - (docker ps -a)')
|
||||||
|
print(f' start - Starts the container (docker start {container})')
|
||||||
|
print(f' stop - Stops the container (docker stop {container}')
|
||||||
|
print(f' rm - Deletes the container (docker rm {container})')
|
||||||
|
print( ' menu - Back to the main menu')
|
||||||
|
|
||||||
elif selection == 'start':
|
selection = input()
|
||||||
print('Starting...')
|
print()
|
||||||
status = Docker.start(container)
|
|
||||||
|
|
||||||
elif selection == 'stop':
|
if selection == 'menu':
|
||||||
print('Stopping...')
|
|
||||||
status = Docker.stop(container)
|
|
||||||
print('Done.')
|
|
||||||
|
|
||||||
elif selection == 'rm':
|
|
||||||
break_later = False
|
|
||||||
while selection != 'y' and selection != 'n' and selection != 'Y' and selection != 'N':
|
|
||||||
print(f'WARNING! This will DELETE {container}!')
|
|
||||||
print(f'Are you absolutely sure you want to delete {container}? (y/N)')
|
|
||||||
selection = input()
|
|
||||||
if selection == 'y' or selection == 'Y':
|
|
||||||
print('Deleting...')
|
|
||||||
status = Docker.rm(container)
|
|
||||||
print('Done')
|
|
||||||
break_later = True
|
|
||||||
elif selection == 'n' or selection == 'N':
|
|
||||||
print(f'Operation cancelled, returning to {container} menu...')
|
|
||||||
if break_later:
|
|
||||||
break
|
break
|
||||||
|
|
||||||
elif selection == 'menu':
|
elif selection == 'view':
|
||||||
print('Returning to main menu...')
|
print(container_to_str(container))
|
||||||
break
|
|
||||||
|
|
||||||
else:
|
elif selection == 'start':
|
||||||
print('Selection invalid. Please try again.')
|
print('Starting...')
|
||||||
print()
|
Docker.start(container)
|
||||||
|
|
||||||
|
elif selection == 'stop':
|
||||||
|
print('Stopping...')
|
||||||
|
Docker.stop(container)
|
||||||
|
print('Done')
|
||||||
|
|
||||||
|
elif selection == 'rm':
|
||||||
|
break_later = False
|
||||||
|
while selection != 'y' and selection != 'n' and selection != 'Y' and selection != 'N':
|
||||||
|
print(f'WARNING! This will DELETE {container}!')
|
||||||
|
print(
|
||||||
|
f'Are you absolutely sure you want to delete {container}? (y/N)')
|
||||||
|
selection = input()
|
||||||
|
if selection == 'y' or selection == 'Y':
|
||||||
|
print('Stopping...')
|
||||||
|
Docker.stop(container)
|
||||||
|
print('Deleting...')
|
||||||
|
Docker.rm(container)
|
||||||
|
print('Done')
|
||||||
|
break_later = True
|
||||||
|
elif selection == 'n' or selection == 'N':
|
||||||
|
print(f'Operation cancelled, returning to {container} menu...')
|
||||||
|
if break_later:
|
||||||
|
print()
|
||||||
|
break
|
||||||
|
|
||||||
|
elif selection == 'menu':
|
||||||
|
print('Returning to main menu...')
|
||||||
|
break
|
||||||
|
|
||||||
|
else:
|
||||||
|
print('Selection invalid. Please try again.')
|
||||||
|
print()
|
||||||
|
except NoContainersError as e:
|
||||||
|
print('Error:', e)
|
||||||
|
print('No containers found. Exiting...')
|
||||||
|
exit()
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue