Better handle nested erros from docker client

l1_precision_fix
Tyler Goodlet 2022-02-18 12:13:38 -05:00
parent 445b82283d
commit 4bcc301c01
1 changed files with 24 additions and 5 deletions

View File

@ -32,7 +32,7 @@ import tractor
import docker import docker
import json import json
from docker.models.containers import Container from docker.models.containers import Container
from docker.errors import DockerException from docker.errors import DockerException, APIError
from requests.exceptions import ConnectionError from requests.exceptions import ConnectionError
from ..log import get_logger # , get_console_log from ..log import get_logger # , get_console_log
@ -102,9 +102,22 @@ async def open_docker(
# prolly no daemon started # prolly no daemon started
raise DockerNotStarted('!?!?') raise DockerNotStarted('!?!?')
except DockerException as err: except (
DockerException,
APIError,
) as err:
def unpack_msg(err: Exception) -> str:
args = getattr(err, 'args', None)
if args:
return args
# could be more specific so let's check if it's just perms. # could be more specific so let's check if it's just perms.
if 'PermissionError' in err.args[0]: if err.args:
errs = err.args
for err in errs:
msg = unpack_msg(err)
if msg and 'PermissionError' in msg:
raise DockerException('You dint run as root yo!') raise DockerException('You dint run as root yo!')
# not perms? # not perms?
@ -233,7 +246,13 @@ async def start_ahab(
loglevel='runtime', loglevel='runtime',
) as tn: ) as tn:
portal = await tn.start_actor('ahab', enable_modules=[__name__]) portal = await tn.start_actor(
'marketstored',
enable_modules=[__name__]
)
# de-escalate root perms to the original user
# after the docker supervisor actor is spawned.
if config._parent_user: if config._parent_user:
import pwd import pwd
os.setuid( os.setuid(