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