summaryrefslogtreecommitdiffstats
path: root/israwhidebroken/controllers.py
blob: 8e666cae6cf61b6a7aea9d2d630894b37845d3d1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
import turbogears as tg
import fedora
from turbogears import controllers, expose, flash
from israwhidebroken.model import Test, Tree, TestResult, hub
from sqlobject import AND
from turbogears import identity, redirect
from cherrypy import request, response
# from israwhidebroken import json
# import logging
# log = logging.getLogger("israwhidebroken.controllers")

def to_int(fstr):
    '''Returns fstr converted to int, or None if fstr is None'''
    return fstr and int(float(fstr))

class Root(controllers.RootController):
    # Main index page
    @expose(template="israwhidebroken.templates.index")
    def index(self, c=0):
        if not c:
            newest_tree = Tree.select(orderBy='-compose_id')[0]
            c = newest_tree.compose_id
        # FIXME BLUH WIDGETIFY THIS
        return dict(in_qa='qa' in identity.current.groups,
                    admin='qa-admin' in identity.current.groups,
                    c=c,
                    tests=Test.select(),
                    trees=Tree.select(Tree.q.compose_id == c))

    # Helpers for JSON calls
    def _tree_select(self, *args, **kw):
        qlist = []
        for field in 'id', 'arch', 'compose_id', 'tree_time', 'repodata_time':
            if field in kw:
                val = kw[field]
                if field != 'arch':
                    val = to_int(val)
                qlist.append('%s == %s' % (field, Tree.sqlrepr(val)))
        print qlist
        return Tree.select(' AND '.join(qlist))

    # JSON RPC calls
    @expose(allow_json=True)
    def get_tests(self):
        tests = Test.select()
        return dict(tests=list(tests))

    @expose(allow_json=True)
    def get_trees(self, *args, **kw):
        tree_results = self._tree_select(*args, **kw)
        return dict(trees=list(tree_results))

    @identity.require(identity.in_group("qa-admin"))
    @expose(allow_json=True)
    def add_tree(self, arch, compose_id, tree_time=None, repodata_time=None):
        # check to see if tree already exists
        tree_result = self._tree_select(arch=arch, compose_id=compose_id,
                           tree_time=tree_time, repodata_time=repodata_time)
        if tree_result.count():
            return dict(exc='ValueError', tg_flash='Tree exists')
        t = Tree(arch=arch,
                 compose_id=to_int(compose_id),
                 tree_time=to_int(tree_time),
                 repodata_time=to_int(repodata_time))
        hub.commit()
        return dict(tree=t)

    @identity.require(identity.in_group("qa"))
    @expose(allow_json=True)
    def add_result(self, treeid, testid, result, overwrite=False):
        if not overwrite:
            # TODO return an error if there's already a result here
            pass
        tr = TestResult(tree=to_int(treeid),
                        test=to_int(testid),
                        result=to_int(result),
                        userid=identity.current.user_id,
                        bug_id=None)
        hub.commit() # XXX necessary?
        return dict(id=tr.id)

    @identity.require(identity.in_group("qa-admin"))
    @expose(allow_json=True)
    def delete_result(self, id):
        return dict(youwantedtodelete=id, thisisastub=True)

    # Identity stuff (login/logout)
    @expose(template="israwhidebroken.templates.login")
    @expose(allow_json=True)
    def login(self, forward_url=None, *args, **kw):

        if forward_url:
            if isinstance(forward_url, list):
                forward_url = forward_url.pop(0)
            else:
                del request.params['forward_url']

        # If the login was successful...
        if not identity.current.anonymous and identity.was_login_attempted() \
                and not identity.get_identity_errors():
            if 'json' == fedora.tg.util.request_format():
                return dict(user=identity.current.user)
            flash("Welcome, %s" % identity.current.user_name)
            redirect(tg.url(forward_url or '/', kw))

        if identity.was_login_attempted():
            msg = _("The credentials you supplied were not correct or "
                   "did not grant access to this resource.")
        elif identity.get_identity_errors():
            msg = _("You must provide your credentials before accessing "
                   "this resource.")
        else:
            msg = _("Please log in.")
            if not forward_url:
                forward_url = request.headers.get("Referer", "/")

        response.status = 401
        return dict(logging_in=True, message=msg,
            forward_url=forward_url, previous_url=request.path_info,
            original_parameters=request.params)

    @expose(allow_json=True)
    def logout(self):
        identity.current.logout()
        if 'json' in fedora.tg.util.request_format():
            return dict()
        redirect("/")