from django.contrib.postgres.search import SearchVector, SearchQuery, SearchRank

from ...product.models import Product
from ...order.models import Order
from ...userprofile.models import User


def search_products(phrase):
    '''Dashboard full text product search'''
    sv = (SearchVector('name', weight='A') +
          SearchVector('description', weight='B'))
    rank = SearchRank(sv, SearchQuery(phrase))
    return Product.objects.annotate(rank=rank).filter(
        rank__gte=0.2).order_by('-rank')


def search_orders(phrase):
    '''Dashboard full text order search

    When phrase is convertable to int, no full text search is performed,
    just order with according id is looked up.

    '''
    try:
        order_id = int(phrase.strip())
        return Order.objects.filter(id=order_id)
    except ValueError:
        pass

    sv = (
        SearchVector('user__default_shipping_address__first_name',
                     weight='B') +
        SearchVector('user__default_shipping_address__last_name', weight='B') +
        SearchVector('user__email', weight='A'))
    rank = SearchRank(sv, SearchQuery(phrase))
    return Order.objects.annotate(rank=rank).filter(
        rank__gte=0.2).order_by('-rank')


def search_users(phrase):
    '''Dashboard full text user search'''
    sv = (SearchVector('email', weight='A') +
          SearchVector('default_billing_address__first_name', weight='B') +
          SearchVector('default_billing_address__last_name', weight='B'))
    rank = SearchRank(sv, SearchQuery(phrase))
    return User.objects.annotate(rank=rank).filter(
        rank__gte=0.2).order_by('-rank')


def search(phrase):
    '''Dashboard full text postgres products, orders and users search

    Composes independent search querysets into dictionary result.

    Args:
        phrase (str): searched phrase

    '''
    return {
        'products': search_products(phrase),
        'orders': search_orders(phrase),
        'users': search_users(phrase)
    }

@@ -36,16 +36,17 @@
  import 
+(
 SearchVe
@@ -63,16 +63,60 @@
 chQuery,
+%0A                                           
  SearchR
@@ -118,16 +118,17 @@
 archRank
+)
 %0A%0Afrom .

Align with 79, not 80 line limit for codeclimate sake