demarches-normaliennes/lib/tasks/task_helper.rb

71 lines
1.5 KiB
Ruby
Raw Normal View History

# Write the given objects to the standard output except if Rake is configured
# to be quiet.
#
# This is useful when running tests (when Rake is configured to be quiet),
# to avoid spamming the output with extra informations.
def rake_puts(*args)
if Rake.verbose
puts(*args)
end
end
def rake_print(*args)
if Rake.verbose
print(*args)
end
end
class ProgressReport
def initialize(total)
@start = Time.zone.now
rake_puts
set_progress(total: total, count: 0)
end
def inc
set_progress(count: @count + 1)
if @per_10_000 % 10 == 0
print_progress
end
end
def finish
if @count > 0 && @per_10_000 != 10_000
set_progress(total: @count)
print_progress
end
rake_puts
end
2019-05-29 11:48:41 +02:00
private
def set_progress(total: nil, count: nil)
if total.present?
@total = total
end
if count.present?
@count = count
@total = [@count, @total].max
end
if @total&.nonzero?
@per_10_000 = 10_000 * @count / @total
end
end
def print_progress
elapsed = Time.zone.now - @start
2018-12-18 21:55:45 +01:00
percent = format('%5.1f%%', @per_10_000 / 100.0)
total = @total.to_s
count = @count.to_s.rjust(total.length)
rake_print("\r#{percent} (#{count}/#{total}) [#{format_duration(elapsed)}/#{format_duration(elapsed * 10_000.0 / @per_10_000)}]")
end
def format_duration(seconds)
if seconds.finite?
Time.zone.at(seconds).utc.strftime('%H:%M:%S')
else
'--:--:--'
end
end
end