Add search param q

hel-repo · Nov 23, 2016 · 7bae74b · 7bae74b
1 parent a8bd373
commit 7bae74b
Show file tree

Hide file tree

Showing 3 changed files with 57 additions and 0 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -1,6 +1,7 @@
 3.1.0
 -----
 - Added ``owners`` search param.
+- Added ``q`` search param.
 - Added group ``~banned``.
 
 3.0.3

diff --git a/hel/tests/test_views.py b/hel/tests/test_views.py
@@ -260,6 +260,15 @@ def test_pkg_search_screen_url(self):
             (['http://img.example.com/img32'], [self.pkg3],)
         ]
 
+    @one_value_param('q')
+    def test_pkg_search_q(self):
+        return [
+            ('root', [self.pkg1, self.pkg2, self.pkg3],),
+            ('Change package "My first"', [self.pkg1],),
+            ('hello people xD', [],),
+            ('test img-2 2 21.', [self.pkg2],)
+        ]
+
     def test_bad_search_param(self):
         try:
             PackagesSearcher({'hi': ['test']})()

diff --git a/hel/utils/query.py b/hel/utils/query.py
@@ -247,6 +247,53 @@ def search(pkg):
 
         return search
 
+    @_only_one_param
+    def q(param):
+
+        def search(pkg):
+            phrases = parse_search_phrase(param)
+            for phrase in phrases:
+                found = False
+                for k in ["name", "description", "short_description"]:
+                    if phrase in pkg[k]:
+                        found = True
+                        break
+                if found:
+                    continue
+
+                for k in ["owners", "authors", "license", "tags"]:
+                    for v in pkg[k]:
+                        if phrase in v:
+                            found = True
+                            break
+                    if found:
+                        break
+                if found:
+                    continue
+
+                for k, screenshot in pkg["screenshots"].items():
+                    if phrase in screenshot:
+                        found = True
+                        break
+                if found:
+                    continue
+
+                for k, version in pkg["versions"].items():
+                    if phrase in version["changes"]:
+                        found = True
+                        break
+                if found:
+                    continue
+
+                # If we reached this line, `continue` lines weren't executed.
+                # It means that the phrase was not found.
+                # As all phrases must be found, we terminate the loop.
+                return False
+
+            return True
+
+        return search
+
 
 class PackagesSearcher: