Make search work for multiple media types

couchpotato/core/providers/info/_modifier/main.py

@@ -32,9 +32,32 @@ class MovieResultModifier(Plugin):
     }
 
     def __init__(self):
+        addEvent('result.modify.info.search', self.returnByType)
         addEvent('result.modify.movie.search', self.combineOnIMDB)
         addEvent('result.modify.movie.info', self.checkLibrary)
 
+    def returnByType(self, results):
+
+        new_results = {'unknown':[]}
+        for r in results:
+            if r.get('type'):
+                type_name = r.get('type') + 's'
+                if not new_results.has_key(type_name):
+                    new_results[type_name] = []
+
+                new_results[type_name].append(r)
+            else:
+                new_results['unknown'].append(r)
+
+        if len(new_results['unknown']) == 0:
+            del new_results['unknown']
+
+        # Combine movies, needs a cleaner way..
+        if new_results.has_key('movies'):
+            new_results['movies'] = self.combineOnIMDB(new_results['movies'])
+
+        return new_results
+
     def combineOnIMDB(self, results):
 
         temp = {}