jaculabilis/danbooru
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

uploads: allow searching uploads and media assets by metatag.

Browse Source 
Allow searching the /uploads and /media_assets pages by the following metatags:

* id:
* md5:
* width:
* height:
* duration:
* mpixels:
* ratio:
* filesize:
* filetype:
* date:
* age:
* status:<processing|active|deleted|expunged|failed> (for /media_assets)
* status:<pending|processing|active|failed> (for /uploads)
* is:<filetype>, is:<status>
* exif:

Examples:

* https://betabooru.donmai.us/media_assets?search[ai_tags_match]=filetype:png
* https://betabooru.donmai.us/uploads?search[ai_tags_match]=filetype:png

Note that in /uploads search, the id:, date:, and age: metatags refer to the upload media asset, not
the upload itself.

Note also that uploads may contain multiple assets, so for example searching uploads by
`filetype:png` will return all uploads containing at least one PNG file, even if they contain other
non-PNG files.
This commit is contained in:
evazion
2022-12-07 00:53:04 -06:00
parent 062a67086e
commit 2c33539be7
8 changed files with 232 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
app/logical/ai_tag_query.rb → app/logical/media_asset_query.rb
View File

@@ -1,8 +1,7 @@
# frozen_string_literal: true
# An AITagQuery is a tag search performed on media assets using AI tags. Only
# basic tags are allowed, no metatags.
class AITagQuery
# A MediaAssetQuery is a tag search performed on media assets (or upload media assets) using AI tags.
class MediaAssetQuery
extend Memoist
attr_reader :search_string
@@ -36,7 +35,7 @@ class AITagQuery
ai_tag = AITag.named(node.name).where(score: score_range)
relation.where(ai_tag.where(AITag.arel_table[:media_asset_id].eq(relation.arel_table[foreign_key])).arel.exists)
in :metatag
relation.none
metatag_matches(node.name, node.value, relation)
in :wildcard
relation.none
in :not
@@ -54,4 +53,37 @@ class AITagQuery
end
end
end
def metatag_matches(name, value, relation)
case name
when "id"
relation.attribute_matches(value, :id)
when "md5"
relation.attribute_matches(value, "media_assets.md5", :md5)
when "width"
relation.attribute_matches(value, "media_assets.image_width")
when "height"
relation.attribute_matches(value, "media_assets.image_height")
when "duration"
relation.attribute_matches(value, "media_assets.duration", :float)
when "mpixels"
relation.attribute_matches(value, "(media_assets.image_width * media_assets.image_height) / 1000000.0", :float)
when "ratio"
relation.attribute_matches(value, "ROUND(media_assets.image_width::numeric / media_assets.image_height::numeric, 2)", :ratio)
when "filesize"
relation.attribute_matches(value, "media_assets.file_size", :filesize)
when "filetype"
relation.attribute_matches(value, "media_assets.file_ext", :enum)
when "date"
relation.attribute_matches(value, :created_at, :date)
when "age"
relation.attribute_matches(value, :created_at, :age)
when "status"
relation.attribute_matches(value, :status, :enum)
when "is"
relation.is_matches(value)
when "exif"
relation.exif_matches(value)
end
end
end