예제 #1
0
파일: bills.py 프로젝트: annerajb/billy
    def get_queryset(self):

        FilterBillsForm = get_filter_bills_form(None)

        # Setup the paginator.
        get = self.request.GET.get
        show_per_page = getattr(self, 'show_per_page', 10)
        show_per_page = int(get('show_per_page', show_per_page))
        page = int(get('page', 1))
        if 100 < show_per_page:
            show_per_page = 100

        if not self.request.GET:
            spec = {}
            cursor = db.bills.find(spec)
            cursor.sort([('updated_at', pymongo.DESCENDING)])
            return self.paginator(cursor, page=page,
                      show_per_page=show_per_page)

        form = FilterBillsForm(self.request.GET)
        params = [
            'chamber',
            'subjects',
            'sponsor__leg_id',
            'actions__type',
            'type']
        search_text = form.data.get('search_text')

        if settings.ENABLE_ELASTICSEARCH:
            kwargs = {}

            chamber = form.data.get('chamber')
            if chamber:
                kwargs['chamber'] = chamber

            subjects = form.data.getlist('subjects')
            if subjects:
                kwargs['subjects'] = {'$all': filter(None, subjects)}

            cursor = Bill.search(search_text, **kwargs)
            cursor.sort([('updated_at', pymongo.DESCENDING)])

        else:
            # Elastic search not enabled--query mongo normally.
            # Mainly here for local work on search views.
            for key in params:
                val = form.data.get(key)
                if val:
                    key = key.replace('__', '.')
                    spec[key] = val

            if search_text:
                spec['title'] = {'$regex': search_text, '$options': 'i'}

            cursor = db.bills.find(spec)
            cursor.sort([('updated_at', pymongo.DESCENDING)])

        return self.paginator(cursor, page=page,
                              show_per_page=show_per_page)
예제 #2
0
파일: handlers.py 프로젝트: annerajb/billy
    def read(self, request):

        bill_fields = {'title': 1, 'created_at': 1, 'updated_at': 1,
                       'bill_id': 1, 'type': 1, 'state': 1, 'level': 1,
                       'country': 1, 'session': 1, 'chamber': 1, 'subjects': 1,
                       '_type': 1, 'id': 1}
        # replace with request's fields if they exist
        bill_fields = _build_field_list(request, bill_fields)

        # normal mongo search logic
        base_fields = _build_mongo_filter(request, ('state', 'chamber',
                                                    'subjects', 'bill_id',
                                                    'bill_id__in'))

        # process extra attributes
        query = request.GET.get('q')
        search_window = request.GET.get('search_window', 'all')
        since = request.GET.get('updated_since', None)
        sponsor_id = request.GET.get('sponsor_id')

        try:
            query = Bill.search(query,
                                search_window=search_window,
                                updated_since=since, sponsor_id=sponsor_id,
                                bill_fields=bill_fields,
                                **base_fields)
        except ValueError as e:
            resp = rc.BAD_REQUEST
            resp.write(': %s' % e)
            return resp

        # add pagination
        page = request.GET.get('page')
        per_page = request.GET.get('per_page')
        if page and not per_page:
            per_page = 50
        if per_page and not page:
            page = 1

        if page:
            page = int(page)
            per_page = int(per_page)
            query = query.limit(per_page).skip(per_page*(page-1))
        else:
            # limit response size
            if query.count() > 5000:
                resp = rc.BAD_REQUEST
                resp.write(': request too large, try narrowing your search by '
                           'adding more filters.')
                return resp

        # sorting
        sort = request.GET.get('sort')
        if sort == 'updated_at':
            query = query.sort([('updated_at', -1)])
        elif sort == 'created_at':
            query = query.sort([('created_at', -1)])

        return list(query)
예제 #3
0
def search(request, abbr):
    '''
    Context:
        - search_text
        - abbr
        - metadata
        - found_by_id
        - bill_results
        - more_bills_available
        - legislators_list
        - nav_active

    Tempaltes:
        - billy/web/public/search_results_no_query.html
        - billy/web/public/search_results_bills_legislators.html
        - billy/web/public/bills_list_row_with_abbr_and_session.html
    '''
    if not request.GET:
        return render(request, templatename('search_results_no_query'),
                      {'abbr': abbr})

    search_text = unicode(request.GET['search_text']).encode('utf8')

    # First try to get by bill_id.
    if re.search(r'\d', search_text):
        url = '/%s/bills?' % abbr
        url += urllib.urlencode([('search_text', search_text)])
        return redirect(url)

    else:
        found_by_id = False
        kwargs = {}
        if abbr != 'all':
            kwargs['abbr'] = abbr
        bill_results = Bill.search(search_text, sort='last', **kwargs)

        # Limit the bills if it's a search.
        bill_result_count = len(bill_results)
        more_bills_available = (bill_result_count > 5)
        bill_results = bill_results[:5]

        # See if any legislator names match. First split up name to avoid
        # the Richard S. Madaleno problem. See Jira issue OS-32.
        textbits = search_text.split()
        textbits = filter(lambda s: 2 < len(s), textbits)
        textbits = filter(lambda s: '.' not in s, textbits)
        andspec = []
        for text in textbits:
            andspec.append({'full_name': {'$regex': text, '$options': 'i'}})
        if andspec:
            spec = {'$and': andspec}
        else:
            spec = {'full_name': {'$regex': search_text, '$options': 'i'}}

        # Run the query.
        if abbr != 'all':
            spec[settings.LEVEL_FIELD] = abbr
        legislator_results = list(
            db.legislators.find(spec).sort([('active', -1)]))

    if abbr != 'all':
        metadata = Metadata.get_object(abbr)
    else:
        metadata = None

    return render(
        request,
        templatename('search_results_bills_legislators'),
        dict(
            search_text=search_text,
            abbr=abbr,
            metadata=metadata,
            found_by_id=found_by_id,
            bill_results=bill_results,
            bill_result_count=bill_result_count,
            more_bills_available=more_bills_available,
            legislators_list=legislator_results,
            column_headers_tmplname=None,  # not used
            rowtemplate_name=templatename('bills_list_row_with'
                                          '_abbr_and_session'),
            show_chamber_column=True,
            nav_active=None))
예제 #4
0
def search(request, abbr):
    '''
    Context:
        - search_text
        - abbr
        - metadata
        - found_by_id
        - bill_results
        - more_bills_available
        - legislators_list
        - more_legislators_available
        - bill_column_headers_tmplname
        - nav_active

    Tempaltes:
        - billy/web/public/search_results_no_query.html
        - billy/web/public/search_results_bills_legislators.html
        - billy/web/public/bills_list_row_with_abbr_and_session.html
    '''
    if not request.GET:
        return render(request, templatename('search_results_no_query'),
                      {'abbr': abbr})

    search_text = request.GET['search_text']

    # First try to get by bill_id.
    if re.search(r'\d', search_text):
        url = '/%s/bills?' % abbr
        url += urllib.urlencode([('search_text', search_text)])
        return redirect(url)

    else:
        found_by_id = False
        kwargs = {}
        if abbr != 'all':
            kwargs['abbr'] = abbr
        bill_results = Bill.search(search_text, **kwargs)

        # add sorting
        bill_results = bill_results.sort([('action_dates.last', -1)])

        # Limit the bills if it's a search.
        more_bills_available = (5 < bill_results.count())
        bill_results = bill_results.limit(5)

        # See if any legislator names match.
        spec = {'full_name': {'$regex': search_text, '$options': 'i'}}
        if abbr != 'all':
            spec[settings.LEVEL_FIELD] = abbr
        legislator_results = db.legislators.find(spec)
        more_legislators_available = (5 < legislator_results.count())
        legislator_results = legislator_results.limit(5)

    if abbr != 'all':
        metadata = Metadata.get_object(abbr)
    else:
        metadata = None

    return render(request, templatename('search_results_bills_legislators'),
        dict(search_text=search_text,
             abbr=abbr,
             metadata=metadata,
             found_by_id=found_by_id,
             bill_results=bill_results,
             more_bills_available=more_bills_available,
             legislators_list=legislator_results,
             more_legislators_available=more_legislators_available,
             column_headers_tmplname=templatename('search_column_headers'),
             rowtemplate_name=templatename('bills_list_row_with'
                                           '_abbr_and_session'),
             show_chamber_column=True,
             nav_active=None))
예제 #5
0
파일: bills.py 프로젝트: jmdupont/billy
    def get_queryset(self):
        abbr = self.kwargs['abbr']
        if abbr != 'all':
            metadata = Metadata.get_object(abbr)
        else:
            metadata = None
        FilterBillsForm = get_filter_bills_form(metadata)

        # start with the spec
        spec = {}
        if abbr != 'all':
            spec['abbr'] = abbr

        # Setup the paginator.
        get = self.request.GET.get
        show_per_page = getattr(self, 'show_per_page', 10)
        show_per_page = int(get('show_per_page', show_per_page))
        try:
            page = int(get('page', 1))
        except ValueError:
            raise Http404('no such page')
        if show_per_page > 100:
            show_per_page = 100

        # If search params are given:
        form = FilterBillsForm(self.request.GET)

        search_text = form.data.get('search_text')

        if form.data:
            form_abbr = form.data.get('abbr')
            if form_abbr:
                spec['abbr'] = form_abbr

            chamber = form.data.get('chamber')
            if chamber:
                spec['chamber'] = chamber

            subjects = form.data.getlist('subjects')
            if subjects:
                spec['subjects'] = subjects

            sponsor_id = form.data.get('sponsor__leg_id')
            if sponsor_id:
                spec['sponsor_id'] = sponsor_id

            if 'status' in form.data:
                status_choices = form.data.getlist('status')
                spec['status'] = status_choices

            type_ = form.data.get('type')
            if type_:
                spec['type_'] = type_

            session = form.data.get('session')
            if session:
                spec['session'] = session

        sort = self.request.GET.get('sort', 'last')

        result = Bill.search(search_text, sort=sort, **spec)

        return self.paginator(result, page=page, show_per_page=show_per_page)
예제 #6
0
    def read(self, request):
        bill_fields = {'title': 1, 'created_at': 1, 'updated_at': 1,
                       'bill_id': 1, 'type': 1, settings.LEVEL_FIELD: 1,
                       'session': 1, 'chamber': 1, 'subjects': 1, '_type': 1,
                       'id': 1}
        # replace with request's fields if they exist
        bill_fields = _build_field_list(request, bill_fields)

        # normal mongo search logic
        base_fields = _build_mongo_filter(request, ('chamber', 'bill_id',
                                                    'bill_id__in'))

        # process extra attributes
        query = request.GET.get('q')
        abbr = request.GET.get(settings.LEVEL_FIELD)
        if abbr:
            abbr = abbr.lower()
        search_window = request.GET.get('search_window', 'all')
        since = request.GET.get('updated_since', None)
        last_action_since = request.GET.get('last_action_since', None)
        sponsor_id = request.GET.get('sponsor_id')
        subjects = request.GET.getlist('subject')
        type_ = request.GET.get('type')
        status = request.GET.getlist('status')

        # sorting
        sort = request.GET.get('sort', 'last')
        if sort == 'last_action':
            sort = 'last'

        try:
            query = Bill.search(query,
                                abbr=abbr,
                                search_window=search_window,
                                updated_since=since,
                                last_action_since=last_action_since,
                                sponsor_id=sponsor_id,
                                subjects=subjects, type_=type_, status=status,
                                sort=sort, bill_fields=bill_fields,
                                **base_fields)
        except ValueError as e:
            resp = rc.BAD_REQUEST
            resp.write('%s' % e)
            return resp

        # add pagination
        page = request.GET.get('page')
        per_page = request.GET.get('per_page')
        if page and not per_page:
            per_page = 50
        if per_page and not page:
            page = 1

        if page:
            page = int(page)
            per_page = int(per_page)
            start = per_page * (page - 1)
            end = start + per_page
            bills = query[start:end]
        else:
            # limit response size
            if len(query) > 10000:
                resp = rc.BAD_REQUEST
                resp.write('request too large, try narrowing your search by '
                           'adding more filters.')
                return resp
            bills = query[:]

        bills = list(bills)
        # attach votes if necessary
        bill_ids = [bill['_id'] for bill in bills]
        vote_fields = _get_vote_fields(bill_fields) or []
        if 'votes' in bill_fields or vote_fields:
            # add bill_id to vote_fields for relating back
            votes = list(db.votes.find({'bill_id': {'$in': bill_ids}},
                                       fields=vote_fields + ['bill_id']))
            votes_by_bill = defaultdict(list)
            for vote in votes:
                votes_by_bill[vote['bill_id']].append(vote)
                # remove bill_id unless they really requested it
                if 'bill_id' not in vote_fields:
                    vote.pop('bill_id')
            for bill in bills:
                bill['votes'] = votes_by_bill[bill['_id']]

        return bills
예제 #7
0
    def get_queryset(self):
        abbr = self.kwargs['abbr']
        if abbr != 'all':
            metadata = Metadata.get_object(abbr)
        else:
            metadata = None
        FilterBillsForm = get_filter_bills_form(metadata)

        # start with the spec
        spec = {}
        if abbr != 'all':
            spec['abbr'] = abbr

        # Setup the paginator.
        get = self.request.GET.get
        show_per_page = getattr(self, 'show_per_page', 10)
        show_per_page = int(get('show_per_page', show_per_page))
        try:
            page = int(get('page', 1))
        except ValueError:
            raise Http404('no such page')
        if show_per_page > 100:
            show_per_page = 100

        # If search params are given:
        form = FilterBillsForm(self.request.GET)

        search_text = form.data.get('search_text')

        if form.data:
            form_abbr = form.data.get('abbr')
            if form_abbr:
                spec['abbr'] = form_abbr

            chamber = form.data.get('chamber')
            if chamber:
                spec['chamber'] = chamber

            subjects = form.data.getlist('subjects')
            if subjects:
                spec['subjects'] = subjects

            sponsor_id = form.data.get('sponsor__leg_id')
            if sponsor_id:
                spec['sponsor_id'] = sponsor_id

            if 'status' in form.data:
                status_choices = form.data.getlist('status')
                spec['status'] = status_choices

            type_ = form.data.get('type')
            if type_:
                spec['type_'] = type_

            session = form.data.get('session')
            if session:
                spec['session'] = session

        sort = self.request.GET.get('sort', 'last')

        result = Bill.search(search_text, sort=sort, **spec)

        return self.paginator(result, page=page, show_per_page=show_per_page)
예제 #8
0
파일: region.py 프로젝트: JT5D/billy
def search(request, abbr):

    if not request.GET:
        return render(request, templatename('search_results_no_query'),
                      {'abbr': abbr})

    search_text = request.GET['search_text']

    # First try to get by bill_id.
    if re.search(r'\d', search_text):
        url = '/%s/bills?' % abbr
        url += urllib.urlencode([('search_text', search_text)])
        return redirect(url)

    else:
        found_by_id = False
        if settings.ENABLE_ELASTICSEARCH:
            kwargs = {}
            if abbr != 'all':
                kwargs['state'] = abbr
            bill_results = Bill.search(search_text, **kwargs)
        else:
            spec = {'title': {'$regex': search_text, '$options': 'i'}}
            if abbr != 'all':
                spec.update(state=abbr)
            bill_results = db.bills.find(spec)

        # add sorting
        bill_results = bill_results.sort([('action_dates.last', -1)])

        # Limit the bills if it's a search.
        more_bills_available = (5 < bill_results.count())
        bill_results = bill_results.limit(5)

        # See if any legislator names match.
        spec = {'full_name': {'$regex': search_text, '$options': 'i'}}
        if abbr != 'all':
            spec.update(state=abbr)
        legislator_results = db.legislators.find(spec)
        more_legislators_available = (5 < legislator_results.count())
        legislator_results = legislator_results.limit(5)

    if abbr != 'all':
        metadata = Metadata.get_object(abbr)
    else:
        metadata = None

    return render(request, templatename('search_results_bills_legislators'),
        dict(search_text=search_text,
             abbr=abbr,
             metadata=metadata,
             found_by_id=found_by_id,
             bill_results=bill_results,
             more_bills_available=more_bills_available,
             legislators_list=legislator_results,
             more_legislators_available=more_legislators_available,
             bill_column_headers=('State', 'Title', 'Session', 'Introduced',
                                  'Recent Action',),
             rowtemplate_name=templatename('bills_list_row_with'
                                           '_state_and_session'),
             show_chamber_column=True,
             statenav_active=None))
예제 #9
0
파일: handlers.py 프로젝트: tjasmith/billy
    def read(self, request):
        bill_fields = {
            'title': 1,
            'created_at': 1,
            'updated_at': 1,
            'bill_id': 1,
            'type': 1,
            settings.LEVEL_FIELD: 1,
            'session': 1,
            'chamber': 1,
            'subjects': 1,
            '_type': 1,
            'id': 1
        }
        # replace with request's fields if they exist
        bill_fields = _build_field_list(request, bill_fields)

        # normal mongo search logic
        base_fields = _build_mongo_filter(
            request, ('chamber', 'bill_id', 'bill_id__in'))

        # process extra attributes
        query = request.GET.get('q')
        abbr = request.GET.get(settings.LEVEL_FIELD)
        if abbr:
            abbr = abbr.lower()
        search_window = request.GET.get('search_window', 'all')
        since = request.GET.get('updated_since', None)
        last_action_since = request.GET.get('last_action_since', None)
        sponsor_id = request.GET.get('sponsor_id')
        subjects = request.GET.getlist('subject')
        type_ = request.GET.get('type')
        status = request.GET.getlist('status')

        # sorting
        sort = request.GET.get('sort', 'last')
        if sort == 'last_action':
            sort = 'last'

        try:
            query = Bill.search(query,
                                abbr=abbr,
                                search_window=search_window,
                                updated_since=since,
                                last_action_since=last_action_since,
                                sponsor_id=sponsor_id,
                                subjects=subjects,
                                type_=type_,
                                status=status,
                                sort=sort,
                                bill_fields=bill_fields,
                                **base_fields)
        except ValueError as e:
            resp = rc.BAD_REQUEST
            resp.write('%s' % e)
            return resp

        # add pagination
        page = request.GET.get('page')
        per_page = request.GET.get('per_page')
        if page and not per_page:
            per_page = 50
        if per_page and not page:
            page = 1

        if page:
            page = int(page)
            per_page = int(per_page)
            start = per_page * (page - 1)
            end = start + per_page
            bills = query[start:end]
        else:
            # limit response size
            if len(query) > 10000:
                resp = rc.BAD_REQUEST
                resp.write('request too large, try narrowing your search by '
                           'adding more filters.')
                return resp
            bills = query[:]

        bills = list(bills)
        # attach votes if necessary
        bill_ids = [bill['_id'] for bill in bills]
        vote_fields = _get_vote_fields(bill_fields) or []
        if 'votes' in bill_fields or vote_fields:
            # add bill_id to vote_fields for relating back
            votes = list(
                db.votes.find({'bill_id': {
                    '$in': bill_ids
                }},
                              fields=vote_fields + ['bill_id']))
            votes_by_bill = defaultdict(list)
            for vote in votes:
                votes_by_bill[vote['bill_id']].append(vote)
                # remove bill_id unless they really requested it
                if 'bill_id' not in vote_fields:
                    vote.pop('bill_id')
            for bill in bills:
                bill['votes'] = votes_by_bill[bill['_id']]

        return bills
예제 #10
0
파일: bills.py 프로젝트: kvimal13/billy
    def get_queryset(self):
        abbr = self.kwargs["abbr"]
        if abbr != "all":
            metadata = Metadata.get_object(abbr)
        else:
            metadata = None
        FilterBillsForm = get_filter_bills_form(metadata)

        # start with the spec
        spec = {}
        if abbr != "all":
            spec["abbr"] = abbr

        # Setup the paginator.
        get = self.request.GET.get
        show_per_page = getattr(self, "show_per_page", 10)
        show_per_page = int(get("show_per_page", show_per_page))
        try:
            page = int(get("page", 1))
        except ValueError:
            raise Http404("no such page")
        if show_per_page > 100:
            show_per_page = 100

        # If search params are given:
        form = FilterBillsForm(self.request.GET)

        search_text = form.data.get("search_text")

        if form.data:
            form_abbr = form.data.get("abbr")
            if form_abbr:
                spec["abbr"] = form_abbr

            chamber = form.data.get("chamber")
            if chamber:
                spec["chamber"] = chamber

            subjects = form.data.getlist("subjects")
            if subjects:
                spec["subjects"] = subjects

            sponsor_id = form.data.get("sponsor__leg_id")
            if sponsor_id:
                spec["sponsor_id"] = sponsor_id

            if "status" in form.data:
                status_choices = form.data.getlist("status")
                spec["status"] = status_choices

            type_ = form.data.get("type")
            if type_:
                spec["type_"] = type_

            session = form.data.get("session")
            if session:
                spec["session"] = session

        sort = self.request.GET.get("sort", "last")

        result = Bill.search(search_text, sort=sort, **spec)

        return self.paginator(result, page=page, show_per_page=show_per_page)
예제 #11
0
파일: region.py 프로젝트: h4ck3rm1k3/billy
def search(request, abbr):
    '''
    Context:
        - search_text
        - abbr
        - metadata
        - found_by_id
        - bill_results
        - more_bills_available
        - legislators_list
        - nav_active

    Tempaltes:
        - billy/web/public/search_results_no_query.html
        - billy/web/public/search_results_bills_legislators.html
        - billy/web/public/bills_list_row_with_abbr_and_session.html
    '''
    if not request.GET:
        return render(request, templatename('search_results_no_query'),
                      {'abbr': abbr})

    search_text = unicode(request.GET['search_text']).encode('utf8')

    # First try to get by bill_id.
    if re.search(r'\d', search_text):
        url = '/%s/bills?' % abbr
        url += urllib.urlencode([('search_text', search_text)])
        return redirect(url)

    else:
        found_by_id = False
        kwargs = {}
        if abbr != 'all':
            kwargs['abbr'] = abbr
        bill_results = Bill.search(search_text, sort='last', **kwargs)

        # Limit the bills if it's a search.
        more_bills_available = (len(bill_results) > 5)
        bill_result_count = len(bill_results)
        bill_results = bill_results[:5]

        # See if any legislator names match. First split up name to avoid
        # the Richard S. Madaleno problem. See Jira issue OS-32.
        textbits = search_text.split()
        textbits = filter(lambda s: 2 < len(s), textbits)
        textbits = filter(lambda s: '.' not in s, textbits)
        andspec = []
        for text in textbits:
            andspec.append({'full_name': {'$regex': text, '$options': 'i'}})
        if andspec:
            spec = {'$and': andspec}
        else:
            spec = {'full_name': {'$regex': search_text, '$options': 'i'}}

        # Run the query.
        if abbr != 'all':
            spec[settings.LEVEL_FIELD] = abbr
        legislator_results = list(db.legislators.find(spec).sort(
            [('active', -1)]))

    if abbr != 'all':
        metadata = Metadata.get_object(abbr)
    else:
        metadata = None

    return render(
        request, templatename('search_results_bills_legislators'),
        dict(search_text=search_text,
             abbr=abbr,
             metadata=metadata,
             found_by_id=found_by_id,
             bill_results=bill_results,
             bill_result_count=bill_result_count,
             more_bills_available=more_bills_available,
             legislators_list=legislator_results,
             column_headers_tmplname=None,  # not used
             rowtemplate_name=templatename('bills_list_row_with'
                                           '_abbr_and_session'),
             show_chamber_column=True,
             nav_active=None))
예제 #12
0
파일: bills.py 프로젝트: ecitizens/billy
    def get_queryset(self):
        abbr = self.kwargs['abbr']
        if abbr != 'all':
            metadata = Metadata.get_object(abbr)
        else:
            metadata = None
        FilterBillsForm = get_filter_bills_form(metadata)

        # start with the spec
        spec = {}
        if abbr != 'all':
            spec['abbr'] = abbr

        # Setup the paginator.
        get = self.request.GET.get
        show_per_page = getattr(self, 'show_per_page', 10)
        show_per_page = int(get('show_per_page', show_per_page))
        page = int(get('page', 1))
        if show_per_page > 100:
            show_per_page = 100

        # If search params are given:
        form = FilterBillsForm(self.request.GET)

        search_text = form.data.get('search_text')

        chamber = form.data.get('chamber')
        if chamber:
            spec['chamber'] = chamber

        subjects = form.data.get('subjects')
        if subjects:
            spec['subjects'] = {'$all': [filter(None, subjects)]}

        sponsor_id = form.data.get('sponsor__leg_id')
        if sponsor_id:
            spec['sponsor_id'] = sponsor_id

        status = form.data.get('status')
        if status:
            spec['status'] = {'action_dates.%s' % status: {'$ne': None}}

        type_ = form.data.get('type')
        if type_:
            spec['type_'] = type_

        session = form.data.get('session')
        if session:
            spec['session'] = session

        cursor = Bill.search(search_text, **spec)

        sort = self.request.GET.get('sort', 'last')
        if sort not in ('first', 'last', 'signed', 'passed_upper',
                        'passed_lower'):
            sort = 'last'
        sort_key = 'action_dates.{0}'.format(sort)

        # do sorting on the cursor
        cursor.sort([(sort_key, pymongo.DESCENDING)])

        return self.paginator(cursor, page=page,
                              show_per_page=show_per_page)
예제 #13
0
파일: region.py 프로젝트: annerajb/billy
def search(request, scope):

    abbr = None
    search_text = request.GET['q']
    scope_name = None
    spec = {}

    # If the input looks like a bill id, try to fetch the bill.
    if re.search(r'\d', search_text):
        bill_id = fix_bill_id(search_text).upper()
        collection = db.bills
        spec.update(bill_id=bill_id)

        if scope != 'all':
            abbr = scope

        docs = collection.find(spec, limit=10)

        # If there were actual results, return a bill_id result view.
        if 0 < docs.count():

            def sortkey(doc):
                session = doc['session']
                years = re.findall(r'\d{4}', session)
                try:
                    return int(years[-1])
                except IndexError:
                    return session

            docs = sorted(docs, key=operator.itemgetter('session'),
                          reverse=True)

            return render(request, templatename('search_results_bill_id'),
              dict(bill_id=bill_id,
               abbr=abbr,
               rowtemplate_name=templatename('bills_list_row_with_state_and_session'),
               object_list=IteratorPaginator(docs),
               use_table=True,
               column_headers=('Title', 'Session', 'Introduced',
                               'Recent Action', 'Votes'),
               statenav_active=None))

    # The input didn't contain \d{4}, so assuming it's not a bill,
    # search bill title and legislator names.
    if settings.ENABLE_ELASTICSEARCH:
        kwargs = {}
        if scope != 'all':
            kwargs['state'] = scope
        bill_results = Bill.search(search_text, **kwargs)
    else:
        spec = {'title': {'$regex': search_text, '$options': 'i'}}
        if scope != 'all':
            abbr = scope
            scope_name = Metadata.get_object(abbr)['name']
            spec.update(state=abbr)
        bill_results = db.bills.find(spec)

    # See if any legislator names match.
    spec = {'full_name': {'$regex': search_text, '$options': 'i'}}
    if scope != 'all':
        abbr = scope
        scope_name = Metadata.get_object(abbr)['name']
        spec.update(state=abbr)
    legislator_results = db.legislators.find(spec)

    return render(request, templatename('search_results_bills_legislators'),
        dict(search_text=search_text,
             abbr=abbr,
             scope_name=scope_name,
             bills_list=bill_results.limit(5),
             more_bills_available=(5 < bill_results.count()),
             legislators_list=legislator_results.limit(5),
             more_legislators_available=(5 < legislator_results.count()),
             bill_column_headers=('State', 'Title', 'Session', 'Introduced',
                                  'Recent Action',),
             rowtemplate_name=templatename('bills_list_row_with_state_and_session'),
             show_chamber_column=True,
             statenav_active=None))
예제 #14
0
파일: bills.py 프로젝트: JT5D/billy
    def get_queryset(self):
        abbr = self.kwargs['abbr']
        if abbr != 'all':
            metadata = Metadata.get_object(abbr)
        else:
            metadata = None
        FilterBillsForm = get_filter_bills_form(metadata)

        # start with the spec
        spec = {}
        if abbr != 'all':
            spec['state'] = abbr

        # Setup the paginator.
        get = self.request.GET.get
        show_per_page = getattr(self, 'show_per_page', 10)
        show_per_page = int(get('show_per_page', show_per_page))
        page = int(get('page', 1))
        if show_per_page > 100:
            show_per_page = 100

        # If search params are given:
        form = FilterBillsForm(self.request.GET)

        # First try to get by bill_id.
        search_text = form.data.get('search_text')
        if search_text:
            found_by_bill_id = search_by_bill_id(self.kwargs['abbr'],
                                                 search_text)
            if found_by_bill_id:
                return IteratorPaginator(found_by_bill_id)

        if settings.ENABLE_ELASTICSEARCH:
            chamber = form.data.get('chamber')
            if chamber:
                spec['chamber'] = chamber

            subjects = form.data.get('subjects')
            if subjects:
                spec['subjects'] = {'$all': [filter(None, subjects)]}

            sponsor_id = form.data.get('sponsor__leg_id')
            if sponsor_id:
                spec['sponsor_id'] = sponsor_id

            status = form.data.get('status')
            if status:
                spec['status'] = {'action_dates.%s' % status: {'$ne': None}}

            type_ = form.data.get('type')
            if type_:
                spec['type_'] = type_

            session = form.data.get('session')
            if session:
                spec['session'] = session

            cursor = Bill.search(search_text, **spec)
        else:
            # Elastic search not enabled--query mongo normally.
            # Mainly here for local work on search views.
            params = ['chamber', 'subjects', 'sponsor__leg_id',
                      'actions__type', 'type', 'status', 'session']
            for key in params:
                val = form.data.get(key)
                if val:
                    key = key.replace('__', '.')
                    spec[key] = val

            if search_text:
                spec['title'] = {'$regex': search_text, '$options': 'i'}

            cursor = db.bills.find(spec)

        sort = self.request.GET.get('sort', 'last')
        if sort not in ('first', 'last', 'signed', 'passed_upper',
                        'passed_lower'):
            sort = 'last'
        sort_key = 'action_dates.{0}'.format(sort)

        # do sorting on the cursor
        cursor.sort([(sort_key, pymongo.DESCENDING)])

        return self.paginator(cursor, page=page,
                              show_per_page=show_per_page)
예제 #15
0
파일: region.py 프로젝트: VersaHQ/billy
def search(request, abbr):
    '''
    Context:
        - search_text
        - abbr
        - metadata
        - found_by_id
        - bill_results
        - more_bills_available
        - legislators_list
        - more_legislators_available
        - nav_active

    Tempaltes:
        - billy/web/public/search_results_no_query.html
        - billy/web/public/search_results_bills_legislators.html
        - billy/web/public/bills_list_row_with_abbr_and_session.html
    '''
    if not request.GET:
        return render(request, templatename('search_results_no_query'),
                      {'abbr': abbr})

    search_text = request.GET['search_text']

    # First try to get by bill_id.
    if re.search(r'\d', search_text):
        url = '/%s/bills?' % abbr
        url += urllib.urlencode([('search_text', search_text)])
        return redirect(url)

    else:
        found_by_id = False
        kwargs = {}
        if abbr != 'all':
            kwargs['abbr'] = abbr
        bill_results = Bill.search(search_text, **kwargs)

        # add sorting
        bill_results = bill_results.sort([('action_dates.last', -1)])

        # Limit the bills if it's a search.
        more_bills_available = (5 < bill_results.count())
        bill_results = bill_results.limit(5)

        # See if any legislator names match.
        spec = {'full_name': {'$regex': search_text, '$options': 'i'}}
        if abbr != 'all':
            spec[settings.LEVEL_FIELD] = abbr
        legislator_results = db.legislators.find(spec)
        more_legislators_available = (5 < legislator_results.count())
        legislator_results = legislator_results.limit(5)

    if abbr != 'all':
        metadata = Metadata.get_object(abbr)
    else:
        metadata = None

    return render(
        request,
        templatename('search_results_bills_legislators'),
        dict(
            search_text=search_text,
            abbr=abbr,
            metadata=metadata,
            found_by_id=found_by_id,
            bill_results=bill_results,
            more_bills_available=more_bills_available,
            legislators_list=legislator_results,
            more_legislators_available=more_legislators_available,
            column_headers_tmplname=None,  # not used
            rowtemplate_name=templatename('bills_list_row_with'
                                          '_abbr_and_session'),
            show_chamber_column=True,
            nav_active=None))
예제 #16
0
파일: bills.py 프로젝트: VersaHQ/billy
    def get_queryset(self):
        abbr = self.kwargs['abbr']
        if abbr != 'all':
            metadata = Metadata.get_object(abbr)
        else:
            metadata = None
        FilterBillsForm = get_filter_bills_form(metadata)

        # start with the spec
        spec = {}
        if abbr != 'all':
            spec['abbr'] = abbr

        # Setup the paginator.
        get = self.request.GET.get
        show_per_page = getattr(self, 'show_per_page', 10)
        show_per_page = int(get('show_per_page', show_per_page))
        page = int(get('page', 1))
        if show_per_page > 100:
            show_per_page = 100

        # If search params are given:
        form = FilterBillsForm(self.request.GET)

        # First try to get by bill_id.
        search_text = form.data.get('search_text')
        if search_text:
            found_by_bill_id = search_by_bill_id(self.kwargs['abbr'],
                                                 search_text)
            if found_by_bill_id:
                return IteratorPaginator(found_by_bill_id)

        chamber = form.data.get('chamber')
        if chamber:
            spec['chamber'] = chamber

        subjects = form.data.get('subjects')
        if subjects:
            spec['subjects'] = {'$all': [filter(None, subjects)]}

        sponsor_id = form.data.get('sponsor__leg_id')
        if sponsor_id:
            spec['sponsor_id'] = sponsor_id

        status = form.data.get('status')
        if status:
            spec['status'] = {'action_dates.%s' % status: {'$ne': None}}

        type_ = form.data.get('type')
        if type_:
            spec['type_'] = type_

        session = form.data.get('session')
        if session:
            spec['session'] = session

        cursor = Bill.search(search_text, **spec)

        sort = self.request.GET.get('sort', 'last')
        if sort not in ('first', 'last', 'signed', 'passed_upper',
                        'passed_lower'):
            sort = 'last'
        sort_key = 'action_dates.{0}'.format(sort)

        # do sorting on the cursor
        cursor.sort([(sort_key, pymongo.DESCENDING)])

        return self.paginator(cursor, page=page, show_per_page=show_per_page)