summaryrefslogtreecommitdiffstats
path: root/leafymiracle/views.py
blob: 65ee6260aae7d68dacc5cee5a4e6e401ba8bce4f (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
# Copyright (C) 2011 Luke Macken <lmacken@redhat.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from pyramid.httpexceptions import HTTPFound
from tw2.jqplugins.ui.base import set_ui_theme_name
from widgets import LeafyGraph
from widgets import LeafyDialog
from widgets import LeafySearchbar

import leafymiracle.models
import sqlalchemy
import simplejson
import webob

def view_root(context, request):
    return HTTPFound(location='/1')

def view_model(context, request):
    # TODO -- we need a fedora jquery-ui theme sitting around.
    set_ui_theme_name('hot-sneaks')
    return {'item':context, 'project':'leafymiracle',
            'jitwidget': LeafyGraph(rootObject=context),
            'dialogwidget': LeafyDialog,
            'searchbarwidget': LeafySearchbar,
           }


def search(context, request):
    """ NOTE :: this is *not* the `pyramid` way of doing things.

    DB stuff should not happen in the 'view' but should instead happen..
    elsewhere?   Quick hack to make this work before submission.

    --Ralph
    """

    term = request.params['term']
    cats = request.params.get('cats', 'Category,Group,Package')

    # Don't do any search if the term is too short
    if len(term) < 2:
        resp = webob.Response(request=request, content_type="application/json")
        resp.body = simplejson.dumps({'data':[]})
        return resp

    attrs = {
        'Category' : {
            'search_on' : ['name', 'description'],
        },
        'Group' : {
            'search_on' : ['name', 'description'],
        },
        'Package' : {
            'search_on' : ['name'],
        }
    }
    srch = '%%%s%%' % term
    cats = cats.split(',')
    results = []

    for cat in cats:
        if not cat in attrs.keys():
            raise ValueError, "'%s' is a disallowed category." % cat
        cls = getattr(leafymiracle.models, cat)
        entries = cls.query.filter(sqlalchemy.or_(
            *[getattr(cls, srch_attr).like(srch)
              for srch_attr in attrs[cat]['search_on']]
        ))
        results += [[unicode(e), e.id, cat] for e in entries]

    data = {
        'data' : [
            {
                'label' : label,
                'value' : value,
                'category' : category }
            for label, value, category in results
        ]
    }
    resp = webob.Response(request=request, content_type="application/json")
    resp.body = simplejson.dumps(data)
    return resp