Add zscore to Project document mapping (#5252)

Add zscore to `_project_docs` query to populate on reindex

tests/unit/packaging/test_search.py

@@ -35,6 +35,7 @@ def test_build_search():
         platform="any platform",
         created=datetime.datetime(1956, 1, 31),
         classifiers=["Alpha", "Beta"],
+        zscore=None,
     )
     obj = Project.from_db(release)
 
@@ -54,3 +55,4 @@ def test_build_search():
     assert obj["platform"] == "any platform"
     assert obj["created"] == datetime.datetime(1956, 1, 31)
     assert obj["classifiers"] == ["Alpha", "Beta"]
+    assert obj["zscore"] is None

warehouse/packaging/search.py

@@ -12,7 +12,7 @@
 
 import packaging.version
 
-from elasticsearch_dsl import Date, Document, Keyword, Text, analyzer
+from elasticsearch_dsl import Date, Document, Float, Keyword, Text, analyzer
 
 from warehouse.search.utils import doc_type
 
@@ -49,6 +49,7 @@ class Project(Document):
     platform = Keyword()
     created = Date()
     classifiers = Keyword(multi=True)
+    zscore = Float()
 
     @classmethod
     def from_db(cls, release):
@@ -71,5 +72,6 @@ def from_db(cls, release):
         obj["platform"] = release.platform
         obj["created"] = release.created
         obj["classifiers"] = release.classifiers
+        obj["zscore"] = release.zscore
 
         return obj

warehouse/search/tasks.py

@@ -85,6 +85,7 @@ def _project_docs(db, project_name=None):
             classifiers,
             Project.normalized_name,
             Project.name,
+            Project.zscore,
         )
         .select_from(releases_list)
         .join(Release, Release.id == releases_list.c.id)