Rename preloaded models constant in cli media (#30531)

This commit is contained in:
Matt Jankowski 2024-10-04 04:30:08 -04:00 committed by GitHub
parent 82e7d53d54
commit cf28104317
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

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)
@ -296,7 +296,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)
@ -333,7 +333,7 @@ module Mastodon::CLI
SQL SQL
end end
PRELOAD_MODEL_WHITELIST = %w( PRELOADED_MODELS = %w(
Account Account
Backup Backup
CustomEmoji CustomEmoji
@ -355,7 +355,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