Changeset 705

Timestamp:

06/30/08 13:25:47 (5 months ago)

Author:

sylvain

Message:

added LinkIndex? to support atom:link elements

Files:

• oss/amplee/amplee/indexer/__init__.py (modified) (3 diffs)

oss/amplee/amplee/indexer/__init__.py

@@ -63 +63 @@
-import re
-import shelve
+from urlparse import urlparse
-#rom sets import Set
-
@@ -69 +70 @@
-__all__ = ['Indexer', 'BaseIndex', 'PublishedIndex', 'AuthorIndex',
-           'CategoryIndex', 'UpdatedIndex', 'DateIndex', 'MemoryContainer',
-
+ 'LinkIndex',
-           'KeywordIndex', 'AtomIDIndexer', 'EditedIndex']
-
@@ -855 +856 @@
-        """
-        return self.load(entry_id)
+
+class LinkIndex(BaseIndex):
+    def __init__(self, name, container=None, index_href=True, index_rel=False):
+        """
+        Simple atom category indexing.
+
+        @type name: string
+        @param name: an internal identifier used to reference
+         the indexer.
+
+        @type container: object
+        @param container: instance of a class
+          implementing the dictionnary interface.
+
+        @type index_href: bool
+        @param: when C{True} this leads to the href attribute to be used as
+         an indexing key for a member.
+
+        
+        @type index_rel: bool
+        @param: when C{True} this leads to the rel attribute to be used as
+         an indexing key for a member.
+        """
+        BaseIndex.__init__(self, name, container)
+        self.index_href = index_href
+        self.index_rel = index_rel
+
+    def update(self, member):
+        """
+        Updates the index based on the link elements of the provided member.
+
+        @type member: L{MemberResource}
+        @param member: Member object to index.
+        """
+        entry = member.atom.entry
+        links = entry.xml_xpath('atom:link')
+        if links:
+            for link in links:
+                if self.index_href:
+                    if link.is_attribute('href'):
+                        hashed_key = sha.new('%s' % str(link.href)).hexdigest()
+                        self.store(hashed_key, (member.collection.name_or_id, member.member_id))
+                        scheme, host = urlparse(str(link.href))[:2]
+                        base = '://'.join((scheme, host))
+                        hashed_key = sha.new('%s' % base).hexdigest()
+                        self.store(hashed_key, (member.collection.name_or_id, member.member_id))
+                if self.index_rel:
+                    if link.is_attribute('rel'):
+                        hashed_key = sha.new('%s' % str(link.rel)).hexdigest()
+                        self.store(hashed_key, (member.collection.name_or_id, member.member_id))
+                      
+    def lookup(self, href=None, rel=None):
+        """
+        Returns a set of member identifiers matching the href and rel provided.
+
+        @type href: string
+        @param href: value to be matched against indexed href attributes
+
+        @type rel: string
+        @param rel: value to be matched against indexed rel attributes
+
+        @rtype: L{set}
+        @return: joint result set of the query
+        """
+        result = set()
+        
+        if href:
+            hashed_key = sha.new('%s' % href).hexdigest()
+            result |= self.load(hashed_key)
+            
+        if rel:
+            hashed_key = sha.new('%s' % rel).hexdigest()
+            result |= self.load(hashed_key)
+
+        return result