Added rating to performers and studios (#1308)

pkg/sqlite/performer.go

@@ -279,6 +279,7 @@ func (qb *performerQueryBuilder) Query(performerFilter *models.PerformerFilterTy
 	query.handleStringCriterionInput(performerFilter.CareerLength, tableName+".career_length")
 	query.handleStringCriterionInput(performerFilter.Tattoos, tableName+".tattoos")
 	query.handleStringCriterionInput(performerFilter.Piercings, tableName+".piercings")
+	query.handleIntCriterionInput(performerFilter.Rating, tableName+".rating")
 	query.handleStringCriterionInput(performerFilter.HairColor, tableName+".hair_color")
 	query.handleStringCriterionInput(performerFilter.URL, tableName+".url")
 	query.handleIntCriterionInput(performerFilter.Weight, tableName+".weight")

pkg/sqlite/performer_test.go

@@ -617,6 +617,67 @@ func TestPerformerStashIDs(t *testing.T) {
 		t.Error(err.Error())
 	}
 }
+func TestPerformerQueryRating(t *testing.T) {
+	const rating = 3
+	ratingCriterion := models.IntCriterionInput{
+		Value:    rating,
+		Modifier: models.CriterionModifierEquals,
+	}
+
+	verifyPerformersRating(t, ratingCriterion)
+
+	ratingCriterion.Modifier = models.CriterionModifierNotEquals
+	verifyPerformersRating(t, ratingCriterion)
+
+	ratingCriterion.Modifier = models.CriterionModifierGreaterThan
+	verifyPerformersRating(t, ratingCriterion)
+
+	ratingCriterion.Modifier = models.CriterionModifierLessThan
+	verifyPerformersRating(t, ratingCriterion)
+
+	ratingCriterion.Modifier = models.CriterionModifierIsNull
+	verifyPerformersRating(t, ratingCriterion)
+
+	ratingCriterion.Modifier = models.CriterionModifierNotNull
+	verifyPerformersRating(t, ratingCriterion)
+}
+
+func verifyPerformersRating(t *testing.T, ratingCriterion models.IntCriterionInput) {
+	withTxn(func(r models.Repository) error {
+		sqb := r.Performer()
+		performerFilter := models.PerformerFilterType{
+			Rating: &ratingCriterion,
+		}
+
+		performers := queryPerformers(t, sqb, &performerFilter, nil)
+
+		for _, performer := range performers {
+			verifyInt64(t, performer.Rating, ratingCriterion)
+		}
+
+		return nil
+	})
+}
+
+func TestPerformerQueryIsMissingRating(t *testing.T) {
+	withTxn(func(r models.Repository) error {
+		sqb := r.Performer()
+		isMissing := "rating"
+		performerFilter := models.PerformerFilterType{
+			IsMissing: &isMissing,
+		}
+
+		performers := queryPerformers(t, sqb, &performerFilter, nil)
+
+		assert.True(t, len(performers) > 0)
+
+		for _, performer := range performers {
+			assert.True(t, !performer.Rating.Valid)
+		}
+
+		return nil
+	})
+}
 
 // TODO Update
 // TODO Destroy

pkg/sqlite/studio.go

@@ -183,10 +183,12 @@ func (qb *studioQueryBuilder) Query(studioFilter *models.StudioFilterType, findF
 		query.addArg(stashIDFilter)
 	}
 
+	if rating := studioFilter.Rating; rating != nil {
+		query.handleIntCriterionInput(studioFilter.Rating, "studios.rating")
+	}
 	query.handleCountCriterion(studioFilter.SceneCount, studioTable, sceneTable, studioIDColumn)
 	query.handleCountCriterion(studioFilter.ImageCount, studioTable, imageTable, studioIDColumn)
 	query.handleCountCriterion(studioFilter.GalleryCount, studioTable, galleryTable, studioIDColumn)
-
 	query.handleStringCriterionInput(studioFilter.URL, "studios.url")
 
 	if isMissingFilter := studioFilter.IsMissing; isMissingFilter != nil && *isMissingFilter != "" {

pkg/sqlite/studio_test.go

@@ -482,17 +482,42 @@ func TestStudioQueryURL(t *testing.T) {
 	verifyStudioQuery(t, filter, verifyFn)
 }
 
+func TestStudioQueryRating(t *testing.T) {
+	const rating = 3
+	ratingCriterion := models.IntCriterionInput{
+		Value:    rating,
+		Modifier: models.CriterionModifierEquals,
+	}
+
+	verifyStudiosRating(t, ratingCriterion)
+
+	ratingCriterion.Modifier = models.CriterionModifierNotEquals
+	verifyStudiosRating(t, ratingCriterion)
+
+	ratingCriterion.Modifier = models.CriterionModifierGreaterThan
+	verifyStudiosRating(t, ratingCriterion)
+
+	ratingCriterion.Modifier = models.CriterionModifierLessThan
+	verifyStudiosRating(t, ratingCriterion)
+
+	ratingCriterion.Modifier = models.CriterionModifierIsNull
+	verifyStudiosRating(t, ratingCriterion)
+
+	ratingCriterion.Modifier = models.CriterionModifierNotNull
+	verifyStudiosRating(t, ratingCriterion)
+}
+
 func verifyStudioQuery(t *testing.T, filter models.StudioFilterType, verifyFn func(s *models.Studio)) {
 	withTxn(func(r models.Repository) error {
 		t.Helper()
 		sqb := r.Studio()
 
-		galleries := queryStudio(t, sqb, &filter, nil)
+		studios := queryStudio(t, sqb, &filter, nil)
 
 		// assume it should find at least one
-		assert.Greater(t, len(galleries), 0)
+		assert.Greater(t, len(studios), 0)
 
-		for _, studio := range galleries {
+		for _, studio := range studios {
 			verifyFn(studio)
 		}
 
@@ -500,6 +525,51 @@ func verifyStudioQuery(t *testing.T, filter models.StudioFilterType, verifyFn fu
 	})
 }
 
+func verifyStudiosRating(t *testing.T, ratingCriterion models.IntCriterionInput) {
+	withTxn(func(r models.Repository) error {
+		sqb := r.Studio()
+		studioFilter := models.StudioFilterType{
+			Rating: &ratingCriterion,
+		}
+
+		studios, _, err := sqb.Query(&studioFilter, nil)
+
+		if err != nil {
+			t.Errorf("Error querying studio: %s", err.Error())
+		}
+
+		for _, studio := range studios {
+			verifyInt64(t, studio.Rating, ratingCriterion)
+		}
+
+		return nil
+	})
+}
+
+func TestStudioQueryIsMissingRating(t *testing.T) {
+	withTxn(func(r models.Repository) error {
+		sqb := r.Studio()
+		isMissing := "rating"
+		studioFilter := models.StudioFilterType{
+			IsMissing: &isMissing,
+		}
+
+		studios, _, err := sqb.Query(&studioFilter, nil)
+
+		if err != nil {
+			t.Errorf("Error querying studio: %s", err.Error())
+		}
+
+		assert.True(t, len(studios) > 0)
+
+		for _, studio := range studios {
+			assert.True(t, !studio.Rating.Valid)
+		}
+
+		return nil
+	})
+}
+
 func queryStudio(t *testing.T, sqb models.StudioReader, studioFilter *models.StudioFilterType, findFilter *models.FindFilterType) []*models.Studio {
 	studios, _, err := sqb.Query(studioFilter, findFilter)
 	if err != nil {