CheatSheets helpers added to the CheatSheets class, including pretty_print. display_sheet method also added which cleans up the argparse logic.

cheat

@@ -42,25 +42,6 @@ if os.name == 'posix' and 'CHEATCOLORS' in os.environ:
         pass
 
 
-def pretty_print(filename):
-    """Applies syntax highlighting to a cheatsheet and writes it to stdout"""
-    try:
-        if os.path.splitext(filename)[1]:
-            lexer = get_lexer_for_filename(filename)
-        else:
-            # shell is a sensible default when there is no extension
-            lexer = get_lexer_for_filename(filename + '.sh')
-
-    except ClassNotFound:
-        lexer = TextLexer()
-
-    with open(filename) as istream:
-        code = istream.read()
-
-    fmt = TerminalFormatter()
-    highlight(code, lexer, fmt, sys.stdout)
-
-
 class CheatSheets(object):
 
     dirs = None
@@ -77,7 +58,7 @@ class CheatSheets(object):
 
     def __cheat_directories(self):
         """Assembles a list of directories containing cheatsheets."""
-        default_directories = [DEFAULT_CHEAT_DIR]
+        default_directories = [DEFAULT_CHEAT_DIR, ]
         try:
             import cheatsheets
             default_directories.append(cheatsheets.cheat_dir)
@@ -104,7 +85,7 @@ class CheatSheets(object):
         return cheats
 
     def edit(self, cheat):
-        """Creates or edits a cheatsheet"""
+        """Creates or edits a cheatsheet."""
 
         # Assert that the EDITOR environment variable is set and that at least 3
         # arguments have been given
@@ -144,55 +125,85 @@ class CheatSheets(object):
             exit(1)
 
     def list(self):
-        """Lists the cheatsheets that are currently available"""
+        """Lists the cheatsheets that are currently available."""
         max_command = max([len(x) for x in self.sheets.keys()]) + 3
         return ('\n'.join(sorted(['%s [%s]' % (key.ljust(max_command), value)
                                   for key, value in self.sheets.items()])))
 
+    def display_sheet(self, sheet):
+        """Display a single sheet."""
+        filepath = os.path.join(self.sheets[sheet], sheet)
+        if self.is_vim_crypted(filepath):
+            self.vim_view(filepath)
+        elif USE_PYGMENTS:
+            self.pretty_print(filepath)
+        else:
+            with open(filepath) as istream:
+                for l in istream:
+                    sys.stdout.write(l)
+
+    def pretty_print(self, filepath):
+        """Applies syntax highlighting to a cheatsheet and writes it to
+        stdout."""
+        try:
+            if os.path.splitext(filepath)[1]:
+                lexer = get_lexer_for_filename(filepath)
+            else:
+                # shell is a sensible default when there is no extension
+                lexer = get_lexer_for_filename(filepath + '.sh')
+
+        except ClassNotFound:
+            lexer = TextLexer()
+
+        with open(filepath) as istream:
+            code = istream.read()
+
+        fmt = TerminalFormatter()
+        highlight(code, lexer, fmt, sys.stdout)
+
+    def is_vim_crypted(self, filepath):
+        """Helper method that returns True if the canonical file path is a Vim
+        Encrypted file."""
+        canonical_path = os.path.realpath(filepath)
+        return 'vim encrypted' in magic.from_file(canonical_path).lower()
+
+    def vim_view(self, filepath):
+        """Use Vim to read an Vim Encrypted file."""
+        canonical_path = os.path.realpath(filepath)
+        env_cmd = '/usr/bin/env'
+        if not os.path.isfile(env_cmd):
+            sys.stderr.write('No environment command {0}\n'.format(env_cmd))
+            exit(1)
+        try:
+            subprocess.call([env_cmd, 'vim', '-R', canonical_path, ])
+            return True
+        except Exception as e:
+            sys.stderr.write('Unknown exception: {0}\n'.format(unicode(e)))
+            exit(1)
+
 
 # Custom action for argparse
 class ListDirectories(argparse.Action):
-    """List cheat directories and exit"""
+    """List cheat directories and exit."""
     def __call__(self, parser, namespace, values, option_string=None):
         print('\n'.join(sheets.dirs))
         parser.exit()
 
 
 class ListCheatsheets(argparse.Action):
-    """List cheatsheets and exit"""
+    """List cheatsheets and exit."""
     def __call__(self, parser, namespace, values, option_string=None):
         print sheets.list()
         parser.exit()
 
 
 class EditSheet(argparse.Action):
-    """If the user wants to edit a cheatsheet"""
+    """If the user wants to edit a cheatsheet."""
     def __call__(self, parser, namespace, values, option_string=None):
         sheets.edit(values[0])
         parser.exit()
 
 
-def is_vim_crypted(filepath):
-    """Returns True if the canonical file path is a Vim Encrypted file."""
-    canonical_path = os.path.realpath(filepath)
-    return 'vim encrypted' in magic.from_file(canonical_path).lower()
-
-
-def vim_viewer(filepath):
-    """Use Vim to read an Vim Encrypted cheat file."""
-    canonical_path = os.path.realpath(filepath)
-    env_cmd = '/usr/bin/env'
-    if not os.path.isfile(env_cmd):
-        sys.stderr.write('No environment command {0}\n'.format(env_cmd))
-        exit(1)
-    try:
-        subprocess.call([env_cmd, 'vim', '-R', canonical_path, ])
-        return True
-    except Exception as e:
-        sys.stderr.write('Unknown exception: {0}\n'.format(unicode(e)))
-        exit(1)
-
-
 def main():
 
     global sheets
@@ -243,16 +254,7 @@ def main():
     if not sheet or sheet in ['help', 'cheat']:
         parser.print_help()
     elif sheet in sheets.sheets:
-        filename = os.path.join(sheets.sheets[sheet], sheet)
+        sheets.display_sheet(sheet)
-        if is_vim_crypted(filename):
-            vim_viewer(filename)
-        elif USE_PYGMENTS:
-            pretty_print(filename)
-        else:
-            with open(filename) as istream:
-                for l in istream:
-                    sys.stdout.write(l)
-
     # if it does not, say so
     else:
         print >> sys.stderr, ('No cheatsheet found for %s.' % sheet)