1 files changed, 75 insertions, 0 deletions
diff --git a/contrib/python/pypodman/pypodman/lib/actions/logs_action.py b/contrib/python/pypodman/pypodman/lib/actions/logs_action.py
new file mode 100644
index 000000000..764a4b9c7
--- /dev/null
+++ b/contrib/python/pypodman/pypodman/lib/actions/logs_action.py
@@ -0,0 +1,75 @@
+"""Remote client command for retrieving container logs."""
+import argparse
+import logging
+import sys
+from collections import deque
+
+import podman
+from pypodman.lib import AbstractActionBase
+
+
+class PositiveIntAction(argparse.Action):
+    """Validate number given is positive integer."""
+
+    def __call__(self, parser, namespace, values, option_string=None):
+        """Validate input."""
+        if values > 0:
+            setattr(namespace, self.dest, values)
+            return
+
+        msg = 'Must be a positive integer.'
+        raise argparse.ArgumentError(self, msg)
+
+
+class Logs(AbstractActionBase):
+    """Class for retrieving logs from container."""
+
+    @classmethod
+    def subparser(cls, parent):
+        """Add Logs command to parent parser."""
+        parser = parent.add_parser('logs', help='retrieve logs from container')
+        parser.add_argument(
+            '--tail',
+            metavar='LINES',
+            action=PositiveIntAction,
+            type=int,
+            help='Output the specified number of LINES at the end of the logs')
+        parser.add_argument(
+            'container',
+            nargs=1,
+            help='retrieve container logs',
+        )
+        parser.set_defaults(class_=cls, method='logs')
+
+    def __init__(self, args):
+        """Construct Logs class."""
+        super().__init__(args)
+
+    def logs(self):
+        """Retrieve logs from containers."""
+        try:
+            ident = self._args.container[0]
+            try:
+                logging.debug('Get container "%s" logs', ident)
+                ctnr = self.client.containers.get(ident)
+            except podman.ContainerNotFound as e:
+                sys.stdout.flush()
+                print(
+                    'Container "{}" not found'.format(e.name),
+                    file=sys.stderr,
+                    flush=True)
+            else:
+                if self._args.tail:
+                    logs = iter(deque(ctnr.logs(), maxlen=self._args.tail))
+                else:
+                    logs = ctnr.logs()
+
+                for line in logs:
+                    sys.stdout.write(line)
+        except podman.ErrorOccurred as e:
+            sys.stdout.flush()
+            print(
+                '{}'.format(e.reason).capitalize(),
+                file=sys.stderr,
+                flush=True)
+            return 1