Skip to content
Snippets Groups Projects
Select Git revision
  • master default protected
1 result
Blame Permalink
SearchResult.py 815 B
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 






class SearchResult:
    def __init__(self, result, query):
        self.id = 'paperid' + str(result['paper'][0][:8].encode('utf-8')).replace(" ", "")
        self.title = result['title']
        self.abstract = self.highlightQuery(result['abstract'], query)
        self.url = self.getUrlFormat(result['paper'][0])
    
    def getUrlFormat(self, file):
        return "http://www.aclweb.org/anthology/" + file[0:8] + ".pdf"

    def highlightQuery(self, text, query):
        keywords = [w.lower() for w in query.split()]
        words = text.split()

        for i, word in enumerate(words):
            for keyword in keywords:
                if keyword in word.lower():
                    words[i] = '<b>' + word + '</b>'
                    break
        return ' '.join(words).rstrip("\"").rstrip(".") + "."