Fix error handling & add custom info printer
This commit is contained in:
askiiart
parent
1e05dd2cd6
commit
43af3125f1
1 changed files with 97 additions and 59 deletions
|
|
@ -1,69 +1,107 @@
|
|||
from docker_wrapper import Docker
|
||||
from pprint import pprint
|
||||
from docker_wrapper import Docker, NoContainersError
|
||||
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()
|
||||
print()
|
||||
def container_to_str(container):
|
||||
"""
|
||||
Returns info about a Docker container as a string
|
||||
:parameters:f
|
||||
container: The name of the container (str)
|
||||
|
||||
if container_i == 'q':
|
||||
exit(0)
|
||||
:returns:
|
||||
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:
|
||||
# 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')
|
||||
if not Docker.containers_exist():
|
||||
raise(NoContainersError('No containers exist! Please create a container and try again.'))
|
||||
|
||||
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()
|
||||
|
||||
if selection == 'menu':
|
||||
break
|
||||
if container_i == 'q':
|
||||
exit(0)
|
||||
|
||||
elif selection == 'view':
|
||||
pprint(Docker.container_info(container)) # TODO: Make better, custom printer
|
||||
while True:
|
||||
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':
|
||||
print('Starting...')
|
||||
status = Docker.start(container)
|
||||
selection = input()
|
||||
print()
|
||||
|
||||
elif selection == 'stop':
|
||||
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:
|
||||
if selection == 'menu':
|
||||
break
|
||||
|
||||
elif selection == 'menu':
|
||||
print('Returning to main menu...')
|
||||
break
|
||||
elif selection == 'view':
|
||||
print(container_to_str(container))
|
||||
|
||||
else:
|
||||
print('Selection invalid. Please try again.')
|
||||
print()
|
||||
elif selection == 'start':
|
||||
print('Starting...')
|
||||
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