mirror of
https://github.com/stashapp/stash.git
synced 2025-12-10 10:13:54 -06:00
5495d72849
* Restructure data layer part 2 (#2599) * Refactor and separate image model * Refactor image query builder * Handle relationships in image query builder * Remove relationship management methods * Refactor gallery model/query builder * Add scenes to gallery model * Convert scene model * Refactor scene models * Remove unused methods * Add unit tests for gallery * Add image tests * Add scene tests * Convert unnecessary scene value pointers to values * Convert unnecessary pointer values to values * Refactor scene partial * Add scene partial tests * Refactor ImagePartial * Add image partial tests * Refactor gallery partial update * Add partial gallery update tests * Use zero/null package for null values * Add files and scan system * Add sqlite implementation for files/folders * Add unit tests for files/folders * Image refactors * Update image data layer * Refactor gallery model and creation * Refactor scene model * Refactor scenes * Don't set title from filename * Allow galleries to freely add/remove images * Add multiple scene file support to graphql and UI * Add multiple file support for images in graphql/UI * Add multiple file for galleries in graphql/UI * Remove use of some deprecated fields * Remove scene path usage * Remove gallery path usage * Remove path from image * Move funscript to video file * Refactor caption detection * Migrate existing data * Add post commit/rollback hook system * Lint. Comment out import/export tests * Add WithDatabase read only wrapper * Prepend tasks to list * Add 32 pre-migration * Add warnings in release and migration notes
54 lines
1.7 KiB
Go
54 lines
1.7 KiB
Go
package autotag
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/stashapp/stash/pkg/match"
|
|
"github.com/stashapp/stash/pkg/models"
|
|
"github.com/stashapp/stash/pkg/scene"
|
|
)
|
|
|
|
func getSceneFileTagger(s *models.Scene, cache *match.Cache) tagger {
|
|
return tagger{
|
|
ID: s.ID,
|
|
Type: "scene",
|
|
Name: s.GetTitle(),
|
|
Path: s.Path(),
|
|
cache: cache,
|
|
}
|
|
}
|
|
|
|
// ScenePerformers tags the provided scene with performers whose name matches the scene's path.
|
|
func ScenePerformers(ctx context.Context, s *models.Scene, rw scene.PartialUpdater, performerReader match.PerformerAutoTagQueryer, cache *match.Cache) error {
|
|
t := getSceneFileTagger(s, cache)
|
|
|
|
return t.tagPerformers(ctx, performerReader, func(subjectID, otherID int) (bool, error) {
|
|
return scene.AddPerformer(ctx, rw, s, otherID)
|
|
})
|
|
}
|
|
|
|
// SceneStudios tags the provided scene with the first studio whose name matches the scene's path.
|
|
//
|
|
// Scenes will not be tagged if studio is already set.
|
|
func SceneStudios(ctx context.Context, s *models.Scene, rw SceneFinderUpdater, studioReader match.StudioAutoTagQueryer, cache *match.Cache) error {
|
|
if s.StudioID != nil {
|
|
// don't modify
|
|
return nil
|
|
}
|
|
|
|
t := getSceneFileTagger(s, cache)
|
|
|
|
return t.tagStudios(ctx, studioReader, func(subjectID, otherID int) (bool, error) {
|
|
return addSceneStudio(ctx, rw, s, otherID)
|
|
})
|
|
}
|
|
|
|
// SceneTags tags the provided scene with tags whose name matches the scene's path.
|
|
func SceneTags(ctx context.Context, s *models.Scene, rw scene.PartialUpdater, tagReader match.TagAutoTagQueryer, cache *match.Cache) error {
|
|
t := getSceneFileTagger(s, cache)
|
|
|
|
return t.tagTags(ctx, tagReader, func(subjectID, otherID int) (bool, error) {
|
|
return scene.AddTag(ctx, rw, s, otherID)
|
|
})
|
|
}
|