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