Trunk: Merged the Devel-branch (branches/griatch) into /trunk. This constitutes a major refactoring of Evennia. Development will now continue in trunk. See the wiki and the past posts to the mailing list for info. /Griatch

src/help/manager.py

@@ -0,0 +1,86 @@
+"""
+Custom manager for HelpEntry objects.
+"""
+from django.db import models
+from src.utils import logger
+
+class HelpEntryManager(models.Manager):
+    """
+    This implements different ways to search for help entries.
+    """
+    def find_topicmatch(self, topicstr, exact=False):
+        """
+        Searches for matching topics based on player's input.
+        """        
+        if topicstr.isdigit():
+            return self.filter(id=topicstr)
+        topics = self.filter(db_key__iexact=topicstr)
+        if not topics and not exact: 
+            topics = self.filter(db_key__istartswith=topicstr)            
+            if not topics:
+                topics = self.filter(db_key__icontains=topicstr)            
+        return topics
+
+    def find_apropos(self, topicstr):
+        """
+        Do a very loose search, returning all help entries containing
+        the search criterion in their titles. 
+        """
+        return self.filter(db_key__icontains=topicstr)
+        
+    def find_topicsuggestions(self, topicstr):
+        """
+        Do a fuzzy match, preferably within the category of the
+        current topic.
+        """                
+        topics = self.find_apropos(topicstr)       
+        # we need to clean away the given help entry.
+        return [topic for topic in topics
+                if topic.key.lower() != topicstr.lower()]
+                    
+    def find_topics_with_category(self, help_category):
+        """
+        Search topics having a particular category
+        """
+        topics = self.filter(db_help_category__iexact=help_category)
+        return topics
+
+    def get_all_topics(self):
+        """
+        Return all topics.
+        """
+        return self.all()
+
+    def get_all_categories(self, pobject):
+        """
+        Return all defined category names with at least one
+        topic in them.
+        """
+        return list(set(topic.help_category for topic in self.all()))
+
+    def all_to_category(self, default_category):
+        """
+        Shifts all help entries in database to default_category.
+        This action cannot be reverted. It is used primarily by
+        the engine when importing a default help database, making
+        sure this ends up in one easily separated category. 
+        """
+        topics = self.all()
+        for topic in topics:
+            topic.help_category = default_category
+            topic.save()
+        string = "Help database moved to category %s" % default_category
+        logger.log_infomsg(string)
+
+    def search_help(self, ostring, help_category=None):
+        """
+        Retrieve a search entry object.
+
+        ostring - the help topic to look for
+        category - limit the search to a particular help topic
+        """
+        ostring = ostring.strip().lower() 
+        help_entries = self.filter(db_topicstr=ostring)
+        if help_category:
+            help_entries.filter(db_help_category=help_category)
+        return help_entries