From e00aff3f7d3ab665f66217c237774a47a62bbec1 Mon Sep 17 00:00:00 2001
From: Tony Narlock <tony@git-pull.com>
Date: Thu, 2 Jun 2016 18:02:38 -0500
Subject: [PATCH] some scratch on decoupling child objects into queryset

---
 libtmux/server.py | 79 +++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 77 insertions(+), 2 deletions(-)

diff --git a/libtmux/server.py b/libtmux/server.py
index 0f6c2887a..400c0d86d 100644
--- a/libtmux/server.py
+++ b/libtmux/server.py
@@ -13,11 +13,85 @@
 
 from . import exc, formats
 from .common import EnvironmentMixin, TmuxRelationalObject, tmux_cmd
-from .session import Session
+from .session import Session, Window
 
 logger = logging.getLogger(__name__)
 
 
+class BaseQuerySet(object):
+    """Object to help with iterating through results."""
+
+    tmux_class = None
+    id = None
+    format_prefix = None
+    format_fields = None
+
+    def __init__(self, server):
+        self.server = server
+
+        assert self.tmux_class
+        assert self.id
+        assert self.format_prefix
+        assert self.format_fields
+
+    def list_cmd(self, command, format_fields):
+        tmux_formats = ['#{%s}' % f for f in format_fields]
+
+        tmux_args = (
+            '-F%s' % '\t'.join(tmux_formats),   # output
+        )
+
+        proc = self.server.cmd(
+            command,
+            *tmux_args
+        )
+
+        if proc.stderr:
+            raise exc.LibTmuxException(proc.stderr)
+
+        return proc
+
+    def list_raw(self, format_fields=format_fields):
+        proc = self.list_raw(format_fields)
+        sessions = proc.stdout
+
+        # combine format keys with values returned from ``tmux list-sessions``
+        sessions = [dict(zip(
+            format_fields, session.split('\t'))) for session in sessions]
+
+        # clear up empty dict
+        sessions = [
+            dict((k, v) for k, v in session.items() if v)
+            for session in sessions
+        ]
+
+        return sessions
+
+
+class SessionQuerySet(object):
+    """Object linked to server to filter and iterate through sessions."""
+    tmux_class = Session
+    id = 'session_id'
+    format_prefix = 'session_'
+    format_fields = formats.SESSION_FORMATS
+
+    def list_cmd(self, format_fields=format_fields):
+        return super(self, SessionQuerySet).list_cmd(
+            'list-sessions', format_fields
+        )
+
+
+class Window(object):
+    """Object linked to server to filter and iterate through sessions."""
+    tmux_class = Window
+    id = 'session_id'
+    format_prefix = 'session_'
+    format_fields = formats.SESSION_FORMATS
+
+    def __len__(self):
+        return int(self.session.get('session_windows'))
+
+
 class Server(TmuxRelationalObject, EnvironmentMixin):
 
     """The :term:`tmux(1)` server.
@@ -130,7 +204,7 @@ def _list_sessions(self):
         tmux_formats = ['#{%s}' % format for format in sformats]
         sessions = proc.stdout
 
-        # combine format keys with values returned from ``tmux list-windows``
+        # combine format keys with values returned from ``tmux list-sessions``
         sessions = [dict(zip(
             sformats, session.split('\t'))) for session in sessions]
 
@@ -162,6 +236,7 @@ def list_sessions(self):
     def sessions(self):
         """Property / alias to return :meth:`~.list_sessions`."""
         return self.list_sessions()
+
     #: Alias of :attr:`sessions`, , used by :class:`TmuxRelationalObject`
     children = sessions