Source code for text_query

#!/usr/bin/python2.7
# Copyright 2010 Google Inc.
#
# 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.

__author__ = 'eyalf@google.com (Eyal Fink)'

import unicodedata
import logging
import re
import jautils


class TextQuery():
    """This class encapsulates the processing we are doing both for indexed
    strings like given_name and family_name and for a query string.  Currently
    the processing includes normalization (see doc below) and splitting to
    words.  Future stuff we might add: indexing of phone numbers, extracting
    of locations for geo-search, synonym support."""

    def __init__(self, query):
        self.query = query

        query = unicode(query or '')
        # Do we need a Japanese specific logic to normalize the query?
        if jautils.should_normalize(query):
            self.normalized = jautils.normalize(query)
        else:
            self.normalized = normalize(query)

        # Split out each CJK ideograph as its own word.
        # The main CJK ideograph range is from U+4E00 to U+9FFF.
        # CJK Extension A is from U+3400 to U+4DFF.
        cjk_separated = re.sub(ur'([\u3400-\u9fff])', r' \1 ', self.normalized)

        # Separate the query into words.
        self.words = cjk_separated.split()

        # query_words is redundant now but I'm leaving it since I don't want to
        # change the signature of TextQuery yet
        # TODO(ryok): get rid of this field?
        self.query_words = self.words


def normalize(string):
    """Normalize a string to all uppercase, remove accents, delete apostrophes,
    and replace non-letters with spaces."""
    string = unicode(string or '').strip().upper()
    letters = []
    """TODO(eyalf): we need to have a better list of types we are keeping
        one that will work for non latin languages"""
    for ch in unicodedata.normalize('NFD', string):
        category = unicodedata.category(ch)
        if category.startswith('L'):
            letters.append(ch)
        elif category != 'Mn' and ch != "'":  # Treat O'Hearn as OHEARN
            letters.append(' ')
    return ''.join(letters)