mktest-html.py

Go to the documentation of this file.

# -*- coding: utf-8 -*-
# @COPYRIGHT_begin
#
# Copyright [2010-2014] Institute of Nuclear Physics PAN, Krakow, Poland
#
# Licensed under the Apache License, Version 2.0 (the "License");
#    you may not use this file except in compliance with the License.
#    You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS,
#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#    See the License for the specific language governing permissions and
#    limitations under the License.
#
# @COPYRIGHT_end

import sys
from mktest import Test
import os
import settings
import traceback
# Django templates
from django.template import loader, Context
from django.conf import settings as django_settings


def main():
    if len(sys.argv) <= 2:
        print "Usage: %s [html log file] [test_number1|test_name1] [test_number2|test_name2] ..." % sys.argv[0]
        sys.exit(1)

    results = []
    for script in sys.argv[2:]:
        try:
            s = None
            for t in os.listdir(os.path.dirname(sys.argv[0]) + '/scripts/'):
                if t[-3:] == '.py' and t[0:2] != '__' and (script == t[1:4] or script == t[:-3]):
                    s = t[:-3]
                    break

            exec("from scripts import %s as scenario" % s)
        except Exception, e:
            traceback.print_exc(limit=sys.getrecursionlimit())
            print "Cannot find test! %s" % str(e)
            sys.exit(0)

        try:
            t = Test(settings.CLM_ADDRESS, scenario)
            t.results = []
            t.start()
        except Exception, e:
            print "Error: %s" % str(e)

        results.extend(t.results)

    # Append actin numbers
    for i in range(len(results)):
        results[i]['id'] = i

    # Create django template
    loader.settings.configure(TEMPLATE_DIRS=['%s/django-templates' % os.path.dirname(sys.argv[0])], DEBUG=True)
    template = loader.get_template_from_string(open("%s/django-templates/template.html" % os.path.dirname(sys.argv[0])).read())

    c = Context({'results': results})
    t = template.render(c)
    log = open(sys.argv[1], "w")
    log.write(t)
    log.close()


if __name__ == '__main__':
    main()