X-Git-Url: https://git.ralfj.de/rust-101.git/blobdiff_plain/cfc6a4901346f45284dcf591375f759e7193d561..706bf6cb37885ca97a49f772de00b535cf3dbf9f:/pycco-rs

diff --git a/pycco-rs b/pycco-rs
index 072dd2e..b1d7f43 100755
--- a/pycco-rs
+++ b/pycco-rs
@@ -21,12 +21,15 @@ for ext, l in pycco.main.languages.items():
 html_src = pycco_resources.html
 
 css_marker = '<link rel="stylesheet" href="{{ stylesheet }}">'
-custom_css = '<link rel="stylesheet" href="pycco_custom.css">'
-patched_html = html_src.replace(css_marker, css_marker+custom_css, 1)
+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>'
-patched_html = html_src.replace(title_marker, title_marker + 'Rust-101: ', 1)
+assert html_src.count(title_marker) == 1
+html_src = html_src.replace(title_marker, title_marker + 'Rust-101: ', 1)
 
-pycco.main.pycco_template = pycco.main.template(patched_html)
+pycco_resources.html = html_src
+pycco.main.pycco_template = pycco_resources.pycco_template = pycco_resources.template(pycco_resources.html)
 
 pycco.main.main()