Add possibility to add sections to cheatsheets

cheat/printer.py

@@ -52,12 +52,16 @@ def printcheats(self, template):
         :param template: Template to use with the format() function.
         """
 
-        for description in self.configparser['cheats']:
-            value = self.configparser['cheats'][description]
-            value = self.add_color(value) if self.print_colored else value
-            output = template.format(description.capitalize(), value)
+        sections = self.configparser.sections()
+        sections.remove('main')
 
-            print(output)
+        for section in sections:
+            print(section.upper())
+            for description in self.configparser[section]:
+                value = self.configparser[section][description]
+                value = self.add_color(value) if self.print_colored else value
+                output = template.format(description.capitalize(), value)
+                print(output)
 
 
 class InlinePrinter(Printer):
@@ -71,7 +75,13 @@ def width(self):
         Width of the longest ConfigParser entry.
         """
 
-        width = len(max(self.configparser['cheats'], key=len))
+        width = 1
+
+        # Calculate new width
+        for section in self.configparser.sections():
+            longest_width = len(max(self.configparser[section], key=len))
+            if longest_width > width:
+                width = longest_width
 
         return str(width)
 
@@ -104,8 +114,8 @@ class PrinterFactory:
     """
 
     printer_classes = {
-        "InlinePrinter": InlinePrinter,
-        "BreaklinePrinter": BreaklinePrinter
+        'InlinePrinter': InlinePrinter,
+        'BreaklinePrinter': BreaklinePrinter
     }
 
     @staticmethod

cheatsheets/example.ini

@@ -0,0 +1,8 @@
+[main]
+name=example
+desc=An example for start your own
+[common]
+Simple Example = example foobar
+Example with Parameter = example foo <PARAMETER>
+[workflow]
+Example in another section = example barfoo

tests/test_print.py

@@ -62,11 +62,11 @@ def test_InlinePrinter_colored(self):
         """
 
         # Done this way for better readablility.
-        lines = ["Test cheat a \x1b[94mlorem\x1b[1;m\n",
+        lines = ["CHEATS\n", "Test cheat a \x1b[94mlorem\x1b[1;m\n",
                  "Test cheat b \x1b[94mipsum\x1b[1;m\n",
                  "Test cheat c \x1b[94mdolor\x1b[1;m\n"]
 
-        expected_output = lines[0] + lines[1] + lines[2]
+        expected_output = ''.join(lines)
 
         printer = cp.InlinePrinter(self.cparser, u.Colors, print_colored=True)
 
@@ -81,11 +81,11 @@ def Test_InlinePrinter(self):
         """
 
         # Done this way for better readablility.
-        lines = ["Test cheat a lorem\n",
+        lines = ["CHEATS\n", "Test cheat a lorem\n",
                  "Test cheat b ipsum\n",
                  "Test cheat c dolor\n"]
 
-        expected_output = lines[0] + lines[1] + lines[2]
+        expected_output = ''.join(lines)
 
         printer = cp.InlinePrinter(self.cparser, u.Colors, print_colored=False)
 
@@ -109,11 +109,11 @@ def test_BreaklinePrinter(self):
         """
 
         # Done this way for better readablility.
-        lines = ["Test cheat a \n lorem\n",
+        lines = ["CHEATS\n", "Test cheat a \n lorem\n",
                  "Test cheat b \n ipsum\n",
                  "Test cheat c \n dolor\n"]
 
-        expected_output = lines[0] + lines[1] + lines[2]
+        expected_output = ''.join(lines)
 
         printer = cp.BreaklinePrinter(self.cparser, u.Colors, print_colored=False)
 
@@ -127,11 +127,11 @@ def test_Printer_printsheet(self):
         """
 
         # Done this way for better readablility.
-        lines = ["Test cheat a\n",
+        lines = ["CHEATS\n", "Test cheat a\n",
                  "Test cheat b\n",
                  "Test cheat c\n"]
 
-        expected_output = lines[0] + lines[1] + lines[2]
+        expected_output = ''.join(lines)
 
         printer = cp.Printer(self.cparser, u.Colors)
         template = "{0}"