redmine_dmsf/lib/tasks/dmsf_create_digests.rake

64 lines
1.8 KiB
Ruby

# encoding: utf-8
#
# Redmine plugin for Document Management System "Features"
#
# Copyright (C) 2011-16 Karel Pičman <karel.picman@kontron.com>
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
desc <<-END_DESC
DMSF maintenance task
* Create missing MD5 digest for all file revisions
Available options:
*dry_run - test, no changes to he database
Example:
bundle exec rake redmine:dmsf_create_digests RAILS_ENV="production"
bundle exec rake redmine:dmsf_create_digests dry_run=1 RAILS_ENV="production"
END_DESC
namespace :redmine do
task :dmsf_create_digests => :environment do
m = DmsfDigest.new
m.create_digests
end
end
class DmsfDigest
def initialize
@dry_run = ENV['dry_run']
end
def create_digests
revisions = DmsfFileRevision.where("digest IS NULL OR digest = ''").all
count = revisions.count
n = 0
revisions.each_with_index do |rev, i|
if rev.create_digest
rev.save unless @dry_run
n += 1
end
# Progress bar
print "\r#{i * 100 / count}%"
end
print "\r100%\n"
# Result
puts "#{n}/#{DmsfFileRevision.count} revisions updated."
end
end