Fix scraping stash-box performers with null birthdates

stashapp · Oct 29, 2024 · b7fdefd · b7fdefd
1 parent 89f539e
commit b7fdefd
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 81 deletions.
diff --git a/graphql/stash-box/query.graphql b/graphql/stash-box/query.graphql
@@ -30,11 +30,6 @@ fragment TagFragment on Tag {
   id
 }
 
-fragment FuzzyDateFragment on FuzzyDate {
-  date
-  accuracy
-}
-
 fragment MeasurementsFragment on Measurements {
   band_size
   cup_size
@@ -60,9 +55,7 @@ fragment PerformerFragment on Performer {
   images {
     ...ImageFragment
   }
-  birthdate {
-    ...FuzzyDateFragment
-  }
+  birth_date
   ethnicity
   country
   eye_color

diff --git a/pkg/scraper/stashbox/graphql/generated_client.go b/pkg/scraper/stashbox/graphql/generated_client.go
diff --git a/pkg/scraper/stashbox/stash_box.go b/pkg/scraper/stashbox/stash_box.go
@@ -648,9 +648,8 @@ func performerFragmentToScrapedPerformer(p graphql.PerformerFragment) *models.Sc
 		sp.Height = &hs
 	}
 
-	if p.Birthdate != nil {
-		b := p.Birthdate.Date
-		sp.Birthdate = &b
+	if p.BirthDate != nil {
+		sp.Birthdate = padFuzzyDate(p.BirthDate)
 	}
 
 	if p.Gender != nil {
@@ -1356,3 +1355,15 @@ func (c *Client) submitDraft(ctx context.Context, query string, input interface{
 
 	return err
 }
+
+func padFuzzyDate(date *string) *string {
+	var paddedDate string
+	if len(*date) == 10 {
+		return date
+	} else if len(*date) == 7 {
+		paddedDate = fmt.Sprintf("%s-01", *date)
+	} else if len(*date) == 4 {
+		paddedDate = fmt.Sprintf("%s-01-01", *date)
+	}
+	return &paddedDate
+}