From 01b35ab0b7c26638efa4b746fb9c4b6cd7cd4eef Mon Sep 17 00:00:00 2001
From: John Shanahan <shanahan.jrs@gmail.com>
Date: Mon, 19 Aug 2013 15:23:53 -0400
Subject: [PATCH] Added main function and wrapped everything in it.

---
 cheat | 58 ++++++++++++++++++++++++++++++++--------------------------
 1 file changed, 32 insertions(+), 26 deletions(-)

diff --git a/cheat b/cheat
index 8fc358d..a41c375 100755
--- a/cheat
+++ b/cheat
@@ -19,34 +19,40 @@ def cheat_files(cheat_directories):
                 for cheat in os.listdir(cheat_dir) if '.' not in cheat ]))
     return cheats
 
-# assemble a keyphrase out of all params passed to the script
-keyphrase  = ' '.join(sys.argv[1:])
-cheat_dirs = cheat_directories()
+def main():
+    """MAIN"""
 
-# verify that we have at least one cheat directory
-if not cheat_dirs:
-    print >> sys.stderr,\
-    'The ~/.cheat dir does not exist or the CHEATPATH var is not set.'
-    exit()
+    # assemble a keyphrase out of all params passed to the script
+    keyphrase  = ' '.join(sys.argv[1:])
+    cheat_dirs = cheat_directories()
 
-# list the files in the ~/.cheat directory
-cheatsheets = cheat_files(cheat_dirs)
+    # verify that we have at least one cheat directory
+    if not cheat_dirs:
+        print >> sys.stderr, \
+        'The ~/.cheat dir does not exist or the CHEATPATH var is not set.'
+        exit()
 
-# print help if requested
-if keyphrase.lower() in ['', 'cheat', 'help', '-h', '-help', '--help']:
-    print "Usage: cheat [keyphrase]\n"
-    print "Available keyphrases:"
-    max_command = max([ len(x) for x in cheatsheets.keys() ]) + 3
-    print '\n'.join(sorted([ '%s [%s]' % (key.ljust(max_command), value)\
-            for key, value in cheatsheets.items() ]))
-    exit()
+    # list the files in the ~/.cheat directory
+    cheatsheets = cheat_files(cheat_dirs)
 
-# print the cheatsheet if it exists
-if keyphrase in cheatsheets:
-    with open (os.path.join(cheatsheets[keyphrase], keyphrase), 'r')\
-            as cheatsheet:
-        print cheatsheet.read()
+    # print help if requested
+    if keyphrase.lower() in ['', 'cheat', 'help', '-h', '-help', '--help']:
+        print "Usage: cheat [keyphrase]\n"
+        print "Available keyphrases:"
+        max_command = max([ len(x) for x in cheatsheets.keys() ]) + 3
+        print '\n'.join(sorted([ '%s [%s]' % (key.ljust(max_command), value)\
+                for key, value in cheatsheets.items() ]))
+        exit()
 
-# if it does not, say so
-else:
-    print 'No cheatsheet found for %s.' % keyphrase
+    # print the cheatsheet if it exists
+    if keyphrase in cheatsheets:
+        with open (os.path.join(cheatsheets[keyphrase], keyphrase), 'r')\
+                as cheatsheet:
+            print cheatsheet.read()
+
+    # if it does not, say so
+    else:
+        print 'No cheatsheet found for %s.' % keyphrase
+
+if __name__ == '__main__':
+    main()