web/ldt/ldt_utils/projectindexer.py
author ymh <ymh.work@gmail.com>
Sat, 12 Jun 2010 04:25:05 +0200
changeset 0 cc4a51750724
child 62 39b2dab4f939
permissions -rw-r--r--
first commit
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
0
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     1
import tempfile
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     2
import os
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     3
import os.path
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     4
import shutil
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     5
import ldt.utils.xml
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     6
from ldt import settings
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     7
import xml
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     8
import xml.dom
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     9
import xml.dom.minidom
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    10
import xml.dom.ext
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    11
import xml.xpath
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    12
import lucene
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    13
from ldt.ldt_utils import STORE
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    14
from ldt.ldt_utils import ANALYZER
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    15
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    16
def Property(func):
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    17
    return property(**func()) 
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    18
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    19
class ProjectIndexer(object):
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    20
    def __init__(self, projectList, writer, decoupage_blackList = settings.DECOUPAGE_BLACKLIST):
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    21
        self.__projectList = projectList
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    22
        self.__decoupage_blacklist = decoupage_blackList
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    23
        self.__writer = writer
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    24
            
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    25
    @Property
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    26
    def decoupage_blacklist(): #@NoSelf
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    27
        doc = """get blacklist""" #@UnusedVariable
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    28
       
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    29
        def fget(self):
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    30
            if self.__decoupage_blacklist is None:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    31
                self.__decoupage_blacklist = ()
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    32
            return self.__decoupage_blacklist
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    33
           
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    34
        def fset(self, value):
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    35
            self.__decoupage_blacklist = value
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    36
           
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    37
        def fdel(self):
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    38
            del self.__decoupage_blacklist
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    39
           
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    40
        return locals()
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    41
               
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    42
    def index_all(self):
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    43
        for project in self.__projectList:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    44
            self.index_project(project)
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    45
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    46
    def index_project(self, project):
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    47
        # ldt.utils.log.debug("Indexing project : "+str(project.ldt_id))
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    48
        ldt=project.ldt
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    49
        doc = xml.dom.minidom.parseString(ldt.encode( "utf-8" ))
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    50
        
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    51
        self.__writer.deleteDocuments(lucene.Term("ldt_id", project.ldt_id))
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    52
            
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    53
        con = xml.xpath.Context.Context(doc, 1, 1, None)
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    54
        res = xml.xpath.Evaluate("/iri/annotations/content", context=con)
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    55
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    56
        for content in res:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    57
            contentId = content.getAttribute("id")
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    58
            
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    59
            res =xml.xpath.Evaluate("ensemble", content)
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    60
            for ensemble in res:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    61
                ensembleId = ensemble.getAttribute("id")
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    62
                
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    63
                for decoupageNode in ensemble.childNodes:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    64
                    # ldt.utils.log.debug("Indexing project decoupage : "+ repr(decoupageNode.nodeType) + " in " + repr(self.decoupage_blacklist))
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    65
                    if decoupageNode.nodeType != xml.dom.Node.ELEMENT_NODE or decoupageNode.tagName != "decoupage"  or decoupageNode.getAttribute("id") in self.decoupage_blacklist:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    66
                        continue
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    67
                
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    68
                    decoupId = decoupageNode.getAttribute("id")
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    69
                    res = xml.xpath.Evaluate("elements/element", decoupageNode)
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    70
                    for elementNode in res:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    71
                        doc = lucene.Document()
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    72
                        elementId = elementNode.getAttribute("id")
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    73
                        tags = elementNode.getAttribute("tags")
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    74
                        
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    75
                        if tags is not None:                            
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    76
                            tags.replace(",", ";")
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    77
                        
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    78
                        if tags is None or len(tags) == 0:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    79
                            tags = ""
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    80
                            restagnode = xml.xpath.Evaluate("tag/text()", elementNode)
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    81
                            for tagnode in restagnode:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    82
                                tags = tags + " ; " + tagnode.data
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    83
                                
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    84
                        if tags is None or len(tags) == 0:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    85
                            tags = ""
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    86
                            restagnode = xml.xpath.Evaluate("tags/tag/text()", elementNode)
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    87
                            for tagnode in restagnode:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    88
                                tags = tags + " ; " + tagnode.data                            
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    89
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    90
                        title = ""
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    91
                        for txtRes in xml.xpath.Evaluate("title/text()", elementNode): 
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    92
                            title = title + txtRes.data 
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    93
                
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    94
                        abstract = ""
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    95
                        for txtRes in xml.xpath.Evaluate("abstract/text()", elementNode): 
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    96
                            abstract = abstract + txtRes.data 
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    97
                            
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    98
                        doc.add(lucene.Field("ldt_id", project.ldt_id, lucene.Field.Store.YES, lucene.Field.Index.UN_TOKENIZED))              
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    99
                        doc.add(lucene.Field("iri_id", contentId, lucene.Field.Store.YES, lucene.Field.Index.UN_TOKENIZED))
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
   100
                        doc.add(lucene.Field("ensemble_id", ensembleId, lucene.Field.Store.YES, lucene.Field.Index.NO))
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
   101
                        doc.add(lucene.Field("decoupage_id", decoupId, lucene.Field.Store.YES, lucene.Field.Index.NO))
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
   102
                        doc.add(lucene.Field("element_id", elementId, lucene.Field.Store.YES, lucene.Field.Index.NO))
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
   103
                        doc.add(lucene.Field("tags", tags, lucene.Field.Store.NO, lucene.Field.Index.TOKENIZED))
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
   104
                        doc.add(lucene.Field("title", title, lucene.Field.Store.NO, lucene.Field.Index.TOKENIZED))
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
   105
                        doc.add(lucene.Field("abstract", abstract, lucene.Field.Store.NO, lucene.Field.Index.TOKENIZED))
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
   106
                        doc.add(lucene.Field("all", " ".join([tags, title, abstract]), lucene.Field.Store.NO, lucene.Field.Index.TOKENIZED))
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
   107
            
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
   108
                        self.__writer.addDocument(doc)
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
   109
    
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
   110
        self.__writer.flush()