Ticket #2561: attachment-search.patch

trac/attachment.py

@@ -32 +32 @@
 from trac.mimeview import *
 from trac.perm import PermissionError, PermissionSystem, IPermissionPolicy
 from trac.resource import *
+from trac.search import ISearchSource, search_to_sql, shorten_result
 from trac.util import get_reporter_id, create_unique_file, content_disposition
 from trac.util.datefmt import to_timestamp, utc
 from trac.util.text import unicode_quote, unicode_unquote, pretty_size
@@ -297 +298 @@
 
     implements(IEnvironmentSetupParticipant, IRequestHandler,
                INavigationContributor, IWikiSyntaxProvider,
-               IResourceManager)
+               IResourceManager, ISearchSource)
 
     change_listeners = ExtensionPoint(IAttachmentChangeListener)
     manipulators = ExtensionPoint(IAttachmentManipulator)
@@ -515 +516 @@
             return _("Attachments of %(parent)s",
                      parent=get_resource_name(self.env, resource.parent))
 
+    # ISearchSource methods
+
+    def get_search_filters(self, req):
+        yield ('attachment', _('Attachments'))
+
+    def get_search_results(self, req, terms, filters):
+        if not 'attachment' in filters:
+            return
+        db = self.env.get_db_cnx()
+        sql_query, args = search_to_sql(db, ['filename', 'description', 
+                                        'author'], terms)
+        cursor = db.cursor()
+        cursor.execute("SELECT type,id,time,filename,description,author "
+                       "FROM attachment "
+                       "WHERE " + sql_query, args)
+        
+        for type, id, time, filename, desc, author in cursor:
+            attachment = Resource(type, id).child('attachment', filename)
+            if 'ATTACHMENT_VIEW' in req.perm(attachment):
+                yield (get_resource_url(self.env, attachment, req.href),
+                       "%s: %s" % (get_resource_shortname(self.env, 
+                                                          attachment.parent),
+                                   filename),
+                       datetime.fromtimestamp(time, utc), author,
+                       shorten_result(desc, terms))
+    
     # Internal methods
 
     def _do_save(self, req, attachment):