Use `following` and `followers` scopes in CLI (#28154)

main
Matt Jankowski 2023-12-06 09:15:54 -05:00 committed by GitHub
parent 23b16aaab0
commit af66d3d836
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 8 deletions

View File

@ -472,15 +472,13 @@ module Mastodon::CLI
end end
total = 0 total = 0
total += Account.where(id: ::Follow.where(account: account).select(:target_account_id)).count if options[:follows] total += account.following.reorder(nil).count if options[:follows]
total += Account.where(id: ::Follow.where(target_account: account).select(:account_id)).count if options[:followers] total += account.followers.reorder(nil).count if options[:followers]
progress = create_progress_bar(total) progress = create_progress_bar(total)
processed = 0 processed = 0
if options[:follows] if options[:follows]
scope = Account.where(id: ::Follow.where(account: account).select(:target_account_id)) account.following.reorder(nil).find_each do |target_account|
scope.find_each do |target_account|
UnfollowService.new.call(account, target_account) UnfollowService.new.call(account, target_account)
rescue => e rescue => e
progress.log pastel.red("Error processing #{target_account.id}: #{e}") progress.log pastel.red("Error processing #{target_account.id}: #{e}")
@ -493,9 +491,7 @@ module Mastodon::CLI
end end
if options[:followers] if options[:followers]
scope = Account.where(id: ::Follow.where(target_account: account).select(:account_id)) account.followers.reorder(nil).find_each do |target_account|
scope.find_each do |target_account|
UnfollowService.new.call(target_account, account) UnfollowService.new.call(target_account, account)
rescue => e rescue => e
progress.log pastel.red("Error processing #{target_account.id}: #{e}") progress.log pastel.red("Error processing #{target_account.id}: #{e}")