kiwi-scp/src/kiwi/subcommands/logs.py

# system
import logging

# local
from ._subcommand import ServiceCommand
from .utils.dockercommand import DockerCommand


class LogsCommand(ServiceCommand):
    def __init__(self):
        super().__init__(
            'logs', nargs='*',
            description="Show logs of a project or service(s) of a project"
        )

        # -f switch: Follow logs
        self._sub_parser.add_argument(
            '-f', '--follow', action='store_true',
            help="output appended data as log grows"
        )

    def run(self, config, args):
        # include timestamps
        compose_cmd = ['logs', '-t']

        # handle following the log output
        if args.follow:
            compose_cmd = [*compose_cmd, '-f', '--tail=10']

        # append if one or more services are given
        if args.services:
            compose_cmd = [*compose_cmd, *args.services]

        # use 'less' viewer if output will be static
        if args.follow:
            DockerCommand('docker-compose').run(config, args, compose_cmd)
        else:
            DockerCommand('docker-compose').run_less(config, args, compose_cmd)