erik/mastodon
1
0
Fork 0
mirror of https://github.com/mastodon/mastodon.git synced 2024-08-20 21:08:15 -07:00
Code Issues Projects Releases Wiki Activity

Rename preloaded models constant in cli media

Browse source
This commit is contained in:
Matt Jankowski 2024-06-03 09:34:04 -04:00
parent aa4d8987a0
commit 33c62e3404
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
lib/mastodon/cli/media.rb
View file

@ -177,7 +177,7 @@ module Mastodon::CLI
attachment_name = path_segments[1].singularize attachment_name = path_segments[1].singularize
file_name = path_segments.last file_name = path_segments.last
next unless PRELOAD_MODEL_WHITELIST.include?(model_name) next unless PRELOADED_MODELS.include?(model_name)
record = model_name.constantize.find_by(id: record_id) record = model_name.constantize.find_by(id: record_id)
attachment = record&.public_send(attachment_name) attachment = record&.public_send(attachment_name)
@ -300,7 +300,7 @@ module Mastodon::CLI
model_name = path_segments.first.classify model_name = path_segments.first.classify
record_id = path_segments[2...-2].join.to_i record_id = path_segments[2...-2].join.to_i
fail_with_message "Cannot find corresponding model: #{model_name}" unless PRELOAD_MODEL_WHITELIST.include?(model_name) fail_with_message "Cannot find corresponding model: #{model_name}" unless PRELOADED_MODELS.include?(model_name)
record = model_name.constantize.find_by(id: record_id) record = model_name.constantize.find_by(id: record_id)
record = record.status if record.respond_to?(:status) record = record.status if record.respond_to?(:status)
@ -334,7 +334,7 @@ module Mastodon::CLI
) )
end end
PRELOAD_MODEL_WHITELIST = %w( PRELOADED_MODELS = %w(
Account Account
Backup Backup
CustomEmoji CustomEmoji
@ -356,7 +356,7 @@ module Mastodon::CLI
model_name = segments.first.classify model_name = segments.first.classify
record_id = segments[2...-2].join.to_i record_id = segments[2...-2].join.to_i
next unless PRELOAD_MODEL_WHITELIST.include?(model_name) next unless PRELOADED_MODELS.include?(model_name)
preload_map[model_name] << record_id preload_map[model_name] << record_id
end end