updating make-html.py

Akuli · Akuli · commit 1bacc91b4564 · 2017-03-11T09:25:12.000+02:00
diff --git a/basics/getting-started.md b/basics/getting-started.md
@@ -63,7 +63,7 @@ worked just fine. Later we'll learn what `(3, 14)` is.
 ## Using Python as a calculator
 
 ```diff
--WARNING: This part contains boring math. Be careful!
+-WARNING: This part contains boring math. Be careful!-
 ```
 
 Let's type some math stuff into Python and see what it does.
diff --git a/make-html.py b/make-html.py
@@ -135,8 +135,29 @@ def block_code(self, code, lang=None):
             formatter = pygments.formatters.HtmlFormatter(
                 style=self.pygments_style, noclasses=True)
             return pygments.highlight(code, lexer, formatter)
-        # we can't highlight it
-        return super().block_code(code, lang)
+
+        elif lang == 'diff':
+            # http://stackoverflow.com/a/39413824
+            result = []
+            for line in code.split('\n'):
+                line = line.strip()
+                if not line:
+                    continue
+
+                if line.startswith('+'):
+                    result.append('<p><font color="green">%s</font></p>'
+                                  % line.strip('+'))
+                elif line.startswith('-'):
+                    result.append('<p><font color="red">%s</font></p>'
+                                  % line.strip('-'))
+                else:
+                    result.append('<p>%s</p>' % line)
+
+            return '\n'.join(result)
+
+        else:
+            # we can't highlight it
+            return super().block_code(code, lang)
 
     def image(self, src, title, text):
         """Return an image inside a link."""
