Merge pull request #9 from rwz/patch-1

[rust-101.git] / pycco-rs

diff --git a/pycco-rs b/pycco-rs
index bfb731c22d6d81fa87694a520315e8098ec567d2..b1d7f43d6274da6379f7e7afa2b4686b2f6f2bb6 100755 (executable)
--- a/pycco-rs
+++ b/pycco-rs
@@ -1,4 +1,4 @@
-#!/usr/bin/python
+#!/usr/bin/env python

 # A little wrapper around pycco, to add Rust support.
 import pycco, pycco_resources
 from pygments import lexers, formatters
 # A little wrapper around pycco, to add Rust support.
 import pycco, pycco_resources
 from pygments import lexers, formatters


@@ -19,9 +19,17 @@ for ext, l in pycco.main.languages.items():
     l["lexer"] = lexers.get_lexer_by_name(l["name"])
 # and monkey-patch for a custom CSS file
 html_src = pycco_resources.html
     l["lexer"] = lexers.get_lexer_by_name(l["name"])
 # and monkey-patch for a custom CSS file
 html_src = pycco_resources.html

-marker = '<link rel="stylesheet" href="{{ stylesheet }}">'
-custom_css = '<link rel="stylesheet" href="pycco_custom.css">'
-patched_html = html_src.replace(marker, marker+custom_css, 1)
-pycco.main.pycco_template = pycco.main.template(patched_html)
+
+css_marker = '<link rel="stylesheet" href="{{ stylesheet }}">'
+custom_css = '<link rel="stylesheet" href="pycco_custom.css"><meta name="viewport" content="width=device-width">'
+assert html_src.count(css_marker) == 1
+html_src = html_src.replace(css_marker, css_marker+custom_css, 1)
+
+title_marker = '<title>'
+assert html_src.count(title_marker) == 1
+html_src = html_src.replace(title_marker, title_marker + 'Rust-101: ', 1)
+
+pycco_resources.html = html_src
+pycco.main.pycco_template = pycco_resources.pycco_template = pycco_resources.template(pycco_resources.html)

 
 pycco.main.main()
 
 pycco.main.main()