tmuxp freeze

Author: tony

tmuxp/cli.py

@@ -18,12 +18,13 @@
 import argparse
 import argcomplete
 import logging
+import pkg_resources
 import kaptan
-from . import config
-from distutils.util import strtobool
-from . import log, util, exc, WorkspaceBuilder, Server
+from . import log, util, exc, WorkspaceBuilder, Server, config
 from .util import ascii_lowercase, input
-import pkg_resources
+from .workspacebuilder import freeze
+from distutils.util import strtobool
+
 
 __version__ = pkg_resources.require("tmuxp")[0].version
 
@@ -281,6 +282,70 @@ def load_workspace(config_file, args):
             sys.exit()
 
 
+def command_freeze(args):
+    """ Import teamocil config to tmuxp format. """
+
+    t = Server(
+        socket_name=args.socket_name,
+        socket_path=args.socket_path
+    )
+
+    logger.error(args)
+    session = t.findWhere({
+        'session_name': args.session_name
+    })
+
+    sconf = freeze(session)
+    configparser = kaptan.Kaptan()
+    newconfig = config.inline(sconf)
+    configparser.import_config(newconfig)
+    config_format = prompt_choices('Convert to', choices=[
+                                    'yaml', 'json'], default='yaml')
+
+    if config_format == 'yaml':
+        newconfig = configparser.export(
+            'yaml', indent=2, default_flow_style=False, safe=True
+        )
+    elif config_format == 'json':
+        newconfig = configparser.export('json', indent=2)
+    else:
+        sys.exit('Unknown config format.')
+
+    print(newconfig)
+    print(
+        '---------------------------------------------------------------')
+    print(
+        'Configuration import does its best to convert teamocil files.\n')
+    if prompt_yes_no(
+        'The new config *WILL* require adjusting afterwards. Save config?'
+    ):
+        dest = None
+        while not dest:
+            dest_prompt = prompt('Save to: ', os.path.abspath(
+                os.path.join(config_dir, 'myimport.%s' % config_format)))
+            if os.path.exists(dest_prompt):
+                print('%s exists. Pick a new filename.' % dest_prompt)
+                continue
+
+            dest = dest_prompt
+
+        dest = os.path.abspath(os.path.relpath(dest))
+        if prompt_yes_no('Write to %s?' % dest):
+            buf = open(dest, 'w')
+            buf.write(newconfig)
+            buf.close()
+
+            print('Saved to %s.' % dest)
+    else:
+        print(
+            'tmuxp has examples in JSON and YAML format at <http://tmuxp.readthedocs.org/en/latest/examples.html>\n'
+            'View tmuxp docs at <http://tmuxp.readthedocs.org/>'
+        )
+        sys.exit()
+
+
+
+
 def command_load(args):
     """ Load a session from a tmuxp session file. """
     if args.list:
@@ -542,6 +607,9 @@ def command_convert(args):
                 print('written new config to <%s>.' % (newfile))
 
 
+
+
+
 def command_attach_session(args):
     """ Command to attach / switch client to a tmux session."""
     commands = []
@@ -624,6 +692,14 @@ def get_parser():
         type=str,
     ).completer = SessionCompleter
 
+    freeze = subparsers.add_parser('freeze')
+    freeze.set_defaults(callback=command_freeze)
+
+    freeze.add_argument(
+        dest='session_name',
+        type=str,
+    ).completer = SessionCompleter
+
     load = subparsers.add_parser('load')
 
     loadgroup = load.add_mutually_exclusive_group(required=True)
@@ -758,6 +834,8 @@ def main():
         command_import_teamocil(args)
     elif args.callback is command_import_tmuxinator:
         command_import_tmuxinator(args)
+    elif args.callback is command_freeze:
+        command_freeze(args)
     elif args.callback is command_attach_session:
         command_attach_session(args)
     elif args.callback is command_kill_session:

tmuxp/testsuite/test_workspacefreezer.py

@@ -2,12 +2,13 @@
 from __future__ import absolute_import, division, print_function, with_statement
 
 import os
+import sys
 import unittest
 import logging
 import time
 import kaptan
 from .. import Window, config, exc
-from ..workspacebuilder import WorkspaceBuilder
+from ..workspacebuilder import WorkspaceBuilder, freeze
 from .helpers import TmuxTestCase
 
 logger = logging.getLogger(__name__)
@@ -16,41 +17,6 @@
 example_dir = os.path.abspath(os.path.join(current_dir, '..', '..'))
 
 
-def freeza(session):
-    sconf = {}
-
-    sconf['session_name'] = session['session_name']
-
-    sconf['windows'] = []
-    for w in session.windows:
-        wconf = {}
-        wconf['options'] = w.show_window_options()
-        wconf['window_name'] = w.get('window_name')
-        wconf['panes'] = []
-        logger.error(w)
-        logger.error(dict(w))
-
-        for p in w.panes:
-            pconf = {}
-            pconf['shell_command'] = []
-            pconf['shell_command'].append('cd ' + p.get('pane_current_path'))
-            pconf['shell_command'].append(p.get('pane_current_command'))
-            wconf['panes'].append(pconf)
-            logger.error(p)
-            logger.error(dict(p))
-
-
-        sconf['windows'].append(wconf)
-
-    logger.error(sconf)
-
-    return sconf
-
-
-
-
-
-
 class FreezeTest(TmuxTestCase):
 
     yaml_config = '''
@@ -77,7 +43,11 @@ class FreezeTest(TmuxTestCase):
         - htop
     '''
 
-    def test_split_windows(self):
+    def test_focus(self):
+        # assure the built yaml config has focus
+        pass
+
+    def test_freeze_config(self):
         sconfig = kaptan.Kaptan(handler='yaml')
         sconfig = sconfig.import_config(self.yaml_config).get()
 
@@ -89,7 +59,7 @@ def test_split_windows(self):
         time.sleep(1)
 
         session = self.session
-        sconf = freeza(session)
+        sconf = freeze(session)
 
         config.check_consistency(sconf)
 
@@ -102,7 +72,6 @@ def test_split_windows(self):
         yaml = kaptanconf.export(
             'yaml', indent=2, default_flow_style=False, safe=True)
 
-
         logger.error(json)
         logger.error(yaml)
 

tmuxp/workspacebuilder.py

@@ -215,3 +215,37 @@ def iter_create_panes(self, w, wconf):
             w.server._update_panes()
 
             yield p
+
+
+def freeze(session):
+    sconf = {}
+
+    sconf['session_name'] = session['session_name']
+
+    sconf['windows'] = []
+    for w in session.windows:
+        wconf = {}
+        wconf['options'] = w.show_window_options()
+        wconf['window_name'] = w.get('window_name')
+        wconf['panes'] = []
+        logger.error(w)
+        logger.error(dict(w))
+
+        if all(w.panes[0].get('pane_current_path') == p.get('pane_current_path') for p in w.panes):
+            wconf['shell_command_before'] = w.panes[0].get('pane_current_path')
+
+        for p in w.panes:
+            pconf = {}
+            pconf['shell_command'] = []
+            if 'shell_command_before' not in wconf:
+                pconf['shell_command'].append('cd ' + p.get('pane_current_path'))
+            pconf['shell_command'].append(p.get('pane_current_command'))
+            wconf['panes'].append(pconf)
+            logger.error(p)
+            logger.error(dict(p))
+
+
+        sconf['windows'].append(wconf)
+
+    return sconf
+