diff --git a/app/models/ai_tag.rb b/app/models/ai_tag.rb index ccf487704..b2db59c6c 100644 --- a/app/models/ai_tag.rb +++ b/app/models/ai_tag.rb @@ -50,4 +50,8 @@ class AITag < ApplicationRecord post.has_tag?(tag.name) end end + + def self.available_includes + %i[media_asset post tag] + end end diff --git a/app/models/media_asset.rb b/app/models/media_asset.rb index a54ab154d..d27ea81e9 100644 --- a/app/models/media_asset.rb +++ b/app/models/media_asset.rb @@ -388,4 +388,8 @@ class MediaAsset < ApplicationRecord def initialize_file_key self.file_key = MediaAsset.generate_file_key end + + def self.available_includes + %i[post media_metadata pixiv_ugoira_frame_data ai_tags] + end end diff --git a/app/models/post.rb b/app/models/post.rb index 0a1c64d3d..333609ea5 100644 --- a/app/models/post.rb +++ b/app/models/post.rb @@ -69,6 +69,7 @@ class Post < ApplicationRecord has_many :disapprovals, :class_name => "PostDisapproval", :dependent => :destroy has_many :favorites, dependent: :destroy has_many :replacements, class_name: "PostReplacement", :dependent => :destroy + has_many :ai_tags, through: :media_asset attr_accessor :old_tag_string, :old_parent_id, :old_source, :old_rating, :has_constraints, :disable_versioning, :post_edit @@ -1639,7 +1640,7 @@ class Post < ApplicationRecord %i[ uploader approver flags appeals parent children notes comments approvals disapprovals replacements pixiv_ugoira_frame_data - artist_commentary + artist_commentary media_asset ai_tags ] end end