mirror of
https://github.com/scinote-eln/scinote-web.git
synced 2024-11-14 05:05:55 +08:00
109 lines
3.4 KiB
Ruby
109 lines
3.4 KiB
Ruby
namespace :data do
|
|
Rails.logger = Logger.new(STDOUT)
|
|
|
|
desc "Remove expired temporary files"
|
|
task clean_temp_files: :environment do
|
|
Rails.logger.info "Cleaning temporary files older than 3 days"
|
|
TempFile.where("created_at < ?", 3.days.ago).each do |tmp_file|
|
|
|
|
TempFile.transaction do
|
|
begin
|
|
tmp_file.destroy!
|
|
rescue Exception => e
|
|
Rails.logger.error "Failed to destroy temporary file #{tmp_file.id}: #{e}"
|
|
raise ActiveRecord::Rollback
|
|
else
|
|
Rails.logger.info "Temporary file ##{tmp_file.id} removed"
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
def destroy_users(users)
|
|
users.each do |user|
|
|
|
|
User.transaction do
|
|
begin
|
|
# Destroy user_team, and possibly team
|
|
if user.teams.count > 0
|
|
oids = user.teams.pluck(:id)
|
|
oids.each do |oid|
|
|
team = Team.find(oid)
|
|
user_team = user.user_teams.where(team: team).first
|
|
destroy_team = (team.users.count == 1 && team.created_by == user)
|
|
if !user_team.destroy(nil) then
|
|
raise Exception
|
|
end
|
|
team.destroy! if destroy_team
|
|
end
|
|
end
|
|
|
|
user.destroy!
|
|
rescue Exception => e
|
|
Rails.logger.error "Failed to destroy unconfirmed user #{user.id}: #{e}"
|
|
raise ActiveRecord::Rollback
|
|
else
|
|
Rails.logger.info "Unconfirmed user ##{user.id} removed"
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
desc "Remove unconfirmed user accounts"
|
|
task clean_unconfirmed_users: :environment do
|
|
Rails.logger.info "Cleaning unconfirmed users"
|
|
|
|
# First, remove the users who signed up by themselves
|
|
users = User
|
|
.where(confirmed_at: nil)
|
|
.where.not(confirmation_token: nil)
|
|
.where(invitation_token: nil)
|
|
.where("created_at < ?", Devise.confirm_within.ago)
|
|
destroy_users(users)
|
|
|
|
# Now, remove users who were invited
|
|
users = User
|
|
.where(confirmed_at: nil)
|
|
.where(invitation_accepted_at: nil)
|
|
.where(confirmation_token: nil)
|
|
.where.not(invitation_token: nil)
|
|
.where("created_at < ?", Devise.invite_for.ago)
|
|
destroy_users(users)
|
|
|
|
# Remove users who didn't finish signup with LinkedIn
|
|
users = User.joins(:user_identities)
|
|
.where(confirmed_at: nil)
|
|
.where('users.created_at < ?', Devise.confirm_within.ago)
|
|
destroy_users(users)
|
|
end
|
|
|
|
desc "Remove temporary and obsolete data"
|
|
task clean: [:environment, :clean_temp_files, :clean_unconfirmed_users]
|
|
|
|
desc 'Export team to directory'
|
|
task :team_export, [:team_id] => [:environment] do |_, args|
|
|
Rails.logger.info(
|
|
"Exporting team with ID:#{args[:team_id]} to directory in tmp"
|
|
)
|
|
te = TeamExporter.new(args[:team_id])
|
|
te.export_to_dir if te
|
|
end
|
|
|
|
desc 'Import team from directory'
|
|
task :team_import, [:dir_path] => [:environment] do |_, args|
|
|
Rails.logger.info(
|
|
"Importing team from directory #{args[:dir_path]}"
|
|
)
|
|
TeamImporter.new.import_from_dir(args[:dir_path])
|
|
end
|
|
|
|
desc 'Delete team and all data inside the team'
|
|
task :team_delete, [:team_id] => [:environment] do |_, args|
|
|
Rails.logger.info(
|
|
"Deleting team with ID:#{args[:team_id]} and all data inside the team"
|
|
)
|
|
team = Team.find_by_id(args[:team_id])
|
|
raise StandardError, 'Can not load team' unless team
|
|
UserDataDeletion.delete_team_data(team) if team
|
|
end
|
|
end
|