dosage/scripts/mktestpage.py

#!/usr/bin/env python
# Copyright (C) 2012 Bastian Kleineidam
from __future__ import print_function
import sys
import os
import time
sys.path.append(os.path.join(os.path.dirname(__file__), ".."))
from dosagelib.scraper import get_scrapers

htmltemplate = """
---
extends: base.j2
title: Dosage by Bastian Kleineidam
description: a commandline webcomic downloader and archiver
---
{% block js %}
<script src="media/js/masonry.min.js"></script>
{% endblock js %}

{% block content %}
<div class="inner clearfix">
<section id="main-content">

<h2>Dosage test results from %(date)s</h2>
<p>Note that it is almost impossible to get a 100%% OK test run
due to temporary site failures.</p>
<div id="testresults">
%(content)s
</div>
<script>
window.onload = function() {
  var wall = new Masonry( document.getElementById('container'), {
    columnWidth: 240
  });
};
</script>
{% endblock content %}
"""


def get_mtime (filename):
    """Return modification time of filename."""
    return os.path.getmtime(filename)


def strdate(t):
    return time.strftime("%d.%m.%Y", time.localtime(t))


def get_test_name(line):
    classname = line.split('::')[1][4:]
    for scraper in get_scrapers():
        if scraper.__name__ == classname:
            try:
                url = scraper.starter()
            except Exception:
                url = None
            return scraper.get_name(), url
    raise ValueError("Scraper %r not found" % classname)


def get_test(line):
    name, url = get_test_name(line)
    result = "OK" if line.startswith(". ") else "FAILED"
    return [name, url, result, ""]


def get_content(filename):
    tests = []
    with open(filename, "r") as f:
        print("Tests parsed: 0", end=" ", file=sys.stderr)
        num_tests = 0
        add_reason = False
        for line in f:
            if line.startswith((". ", "F ")) and "test_comics" in line:
                add_reason = line.startswith("F ")
                num_tests += 1
                try:
                    tests.append(get_test(line))
                except Exception as msg:
                    print("WARNING:", msg, file=sys.stderr)
                    continue
            elif add_reason and line.startswith(" E "):
                reason = line[3:].strip()
                tests[-1][-1] = reason
            if num_tests % 5 == 0:
                print(num_tests, end=" ", file=sys.stderr)
    tests.sort()
    res = []
    for name, url, result, reason in tests:
        css = result.lower()
        if len(name) > 25 and '/' in name:
            name = name.replace('/', '/ ')
        if url:
            inner = '<a href="%s" title="%s" class="%s">%s %s</a>' % (url, reason, css, name, result)
        else:
            inner = '<span title="%s" class="%s">%s %s</span>' % (reason, css, name, result)
        res.append('    <div class="item">%s</div>' % inner)
    return os.linesep.join(res)


def main(args):
    filename = args[0]
    modified = get_mtime(filename)
    content = get_content(filename)
    attrs = {"date": strdate(modified), "content": content}
    print(htmltemplate % attrs)
    return 0


if __name__ == '__main__':
    sys.exit(main(sys.argv[1:]))