diff --git a/main.py b/main.py index 5eb9ca5..08f9740 100644 --- a/main.py +++ b/main.py @@ -35,7 +35,14 @@ def md(html, **options): def code_parser(el): - return el['data-language'] if el.has_attr('data-language') else 'text' + if not el.has_attr('data-language'): + return 'text' + + lang = el['data-language'] + if lang == 'jscript': + return 'js' + + return lang mkdirp('output')