Initial commit of the new MPTT based document indexing app

apps/document_indexing/__init__.py

@@ -0,0 +1 @@
+

apps/document_indexing/admin.py

@@ -1,23 +1,25 @@
 from django.contrib import admin
 
-from document_indexing.models import DocumentIndex
+from mptt.admin import MPTTModelAdmin
 
-#from filesystem_serving.admin import DocumentMetadataIndexInline
+from document_indexing.models import Index, IndexInstance
 
 
-#class MetadataIndexInline(admin.StackedInline):
-#    model = MetadataIndex
-#    extra = 1
-#    classes = ('collapse-open',)
-#    allow_add = True
+class IndexInstanceInline(admin.StackedInline):
+    model = IndexInstance
+    extra = 1
+    classes = ('collapse-open',)
+    allow_add = True
 
 
-class DocumentIndexAdmin(admin.ModelAdmin):
-    pass
-    #inlines = [
-    #    DocumentMetadataIndexInline,
-    #
+class IndexAdmin(MPTTModelAdmin):
+    list_display = ('expression', 'enabled', 'link_document')
+    
+    
+class IndexInstanceAdmin(MPTTModelAdmin):
+    model = IndexInstance
+    list_display = ('value', 'index', 'get_document_list_display')
 
 
-
-admin.site.register(DocumentIndex, DocumentIndexAdmin)
+admin.site.register(Index, IndexAdmin)
+admin.site.register(IndexInstance, IndexInstanceAdmin)

apps/document_indexing/api.py

@@ -0,0 +1,53 @@
+from django.utils.translation import ugettext_lazy as _
+
+from metadata.classes import MetadataObject
+
+#from filesystem_serving.conf.settings import MAX_RENAME_COUNT
+
+from document_indexing.models import Index, IndexInstance
+from document_indexing.conf.settings import AVAILABLE_INDEXING_FUNCTIONS
+
+
+def evaluate_index(eval_dict, document, node, parent_index_instance=None):
+    warnings = []
+    if node.enabled:
+        try:
+            result = eval(node.expression, eval_dict, AVAILABLE_INDEXING_FUNCTIONS)
+            index_instance, created = IndexInstance.objects.get_or_create(index=node, value=result, parent=parent_index_instance)
+            if node.link_document:
+                index_instance.documents.add(document)
+                
+            for children in node.get_children():
+                children_warnings = evaluate_index(eval_dict, document, children, index_instance)
+                warnings.extend(children_warnings)
+            
+        except NameError, exc:
+            warnings.append(_(u'Error in metadata indexing expression: %s') % exc)
+            #raise NameError()
+            #This should be a warning not an error
+            #pass
+        except Exception, exc:
+            warnings.append(_(u'Unable to create metadata indexing directory: %s') % exc)
+            
+    return warnings
+
+
+def update_indexes(document):
+    print 'update_indexes'
+    warnings = []
+
+    eval_dict = {}
+    eval_dict['document'] = document
+    metadata_dict = dict([(metadata.metadata_type.name, metadata.value) for metadata in document.documentmetadata_set.all() if metadata.value])
+    eval_dict['metadata'] = MetadataObject(metadata_dict)
+
+    for root in Index.objects.filter(parent=None):
+        #for node in root.get_children():
+        index_warnings = evaluate_index(eval_dict, document, root)
+        warnings.extend(index_warnings)
+
+    return warnings
+    
+    
+def delete_indexes(document):
+    print 'delete_indexes'

apps/document_indexing/models.py

@@ -1,20 +1,48 @@
 from django.db import models
 from django.utils.translation import ugettext_lazy as _
 
+from mptt.models import MPTTModel, TreeForeignKey
+
+from documents.models import Document
+
 from document_indexing.conf.settings import AVAILABLE_INDEXING_FUNCTIONS
 
-available_indexing_functions_string = (_(u' Available functions: %s') % u','.join([u'%s()' % name for name, function in AVAILABLE_INDEXING_FUNCTIONS.items()])) if AVAILABLE_INDEXING_FUNCTIONS else u''
+available_indexing_functions_string = (_(u'Available functions: %s') % u','.join([u'%s()' % name for name, function in AVAILABLE_INDEXING_FUNCTIONS.items()])) if AVAILABLE_INDEXING_FUNCTIONS else u''
 
 
-class DocumentIndex(models.Model):
-    expression = models.CharField(max_length=128,
-        verbose_name=_(u'indexing expression'),
-        help_text=_(u'Enter a python string expression to be evaluated.  The slash caracter "/" acts as a directory delimiter.%s') % available_indexing_functions_string)
+class Index(MPTTModel):
+    parent = TreeForeignKey('self', null=True, blank=True, related_name='index_meta_class')
+    expression = models.CharField(max_length=128, verbose_name=_(u'indexing expression'), help_text=_(u'Enter a python string expression to be evaluated.'))
+        # % available_indexing_functions_string)
     enabled = models.BooleanField(default=True, verbose_name=_(u'enabled'))
-
+    link_document = models.BooleanField(default=False, verbose_name=_(u'link document'))
+    
     def __unicode__(self):
-        return unicode(self.expression)
+        return self.expression if not self.link_document else u'%s/[document]' % self.expression
 
     class Meta:
-        verbose_name = _(u'metadata index')
-        verbose_name_plural = _(u'metadata indexes')
+        verbose_name = _(u'index')
+        verbose_name_plural = _(u'indexes')
+
+
+class IndexInstance(MPTTModel):
+    parent = TreeForeignKey('self', null=True, blank=True, related_name='index_meta_instance')
+    index = models.ForeignKey(Index, verbose_name=_(u'index'))
+    value = models.CharField(max_length=128, blank=True, verbose_name=_(u'value'))
+    documents = models.ManyToManyField(Document, verbose_name=_(u'documents'))
+    
+    def __unicode__(self):
+        return self.value
+
+    def get_document_list_display(self):
+        return u', '.join([d.file_filename for d in self.documents.all()])
+        
+
+    class Meta:
+        verbose_name = _(u'index instance')
+        verbose_name_plural = _(u'indexes instances')
+
+
+# TODO
+# class DocumentRenameCount
+# FK=IndexInstance