## Amazon S3 manager
## Author: Michal Ludvig <michal@logix.cz>
##         http://www.logix.cz/michal
## License: GPL Version 2

import sys
import datetime
import Utils

class Progress(object):
	_stdout = sys.stdout

	def __init__(self, labels, total_size):
		self._stdout = sys.stdout
		self.new_file(labels, total_size)
	
	def new_file(self, labels, total_size):
		self.labels = labels
		self.total_size = total_size
		# Set initial_position to something in the
		# case we're not counting from 0. For instance
		# when appending to a partially downloaded file.
		# Setting initial_position will let the speed
		# be computed right.
		self.initial_position = 0
		self.current_position = self.initial_position
		self.time_start = datetime.datetime.now()
		self.time_last = self.time_start
		self.time_current = self.time_start

		self.display(new_file = True)
	
	def update(self, current_position = -1, delta_position = -1):
		self.time_last = self.time_current
		self.time_current = datetime.datetime.now()
		if current_position > -1:
			self.current_position = current_position
		elif delta_position > -1:
			self.current_position += delta_position
		#else:
		#	no update, just call display()
		self.display()

	def done(self, message):
		self.display(done_message = message)

	def output_labels(self):
		self._stdout.write(u"%(source)s -> %(destination)s  %(extra)s\n" % self.labels)
		self._stdout.flush()

	def display(self, new_file = False, done_message = None):
		"""
		display(new_file = False[/True], done = False[/True])

		Override this method to provide a nicer output.
		"""
		if new_file:
			self.output_labels()
			self.last_milestone = 0
			return

		if self.current_position == self.total_size:
			print_size = Utils.formatSize(self.current_position, True)
			if print_size[1] != "": print_size[1] += "B"
			timedelta = self.time_current - self.time_start
			sec_elapsed = timedelta.days * 86400 + timedelta.seconds + float(timedelta.microseconds)/1000000.0
			print_speed = Utils.formatSize((self.current_position - self.initial_position) / sec_elapsed, True, True)
			self._stdout.write("100%%  %s%s in %.2fs (%.2f %sB/s)\n" % 
				(print_size[0], print_size[1], sec_elapsed, print_speed[0], print_speed[1]))
			self._stdout.flush()
			return

		rel_position = selfself.current_position * 100 / self.total_size
		if rel_position >= self.last_milestone:
			self.last_milestone = (int(rel_position) / 5) * 5
			self._stdout.write("%d%% ", self.last_milestone)
			self._stdout.flush()
			return

class ProgressANSI(Progress):
    ## http://en.wikipedia.org/wiki/ANSI_escape_code
	SCI = '\x1b['
	ANSI_hide_cursor = SCI + "?25l"
	ANSI_show_cursor = SCI + "?25h"
	ANSI_save_cursor_pos = SCI + "s"
	ANSI_restore_cursor_pos = SCI + "u"
	ANSI_move_cursor_to_column = SCI + "%uG"
	ANSI_erase_to_eol = SCI + "0K"
	ANSI_erase_current_line = SCI + "2K"

	def display(self, new_file = False, done_message = None):
		"""
		display(new_file = False[/True], done_message = None)
		"""
		if new_file:
			self.output_labels()
			self._stdout.write(self.ANSI_save_cursor_pos)
			self._stdout.flush()
			return

		timedelta = self.time_current - self.time_start
		sec_elapsed = timedelta.days * 86400 + timedelta.seconds + float(timedelta.microseconds)/1000000.0
		if (sec_elapsed > 0):
			print_speed = Utils.formatSize((self.current_position - self.initial_position) / sec_elapsed, True, True)
		else:
			print_speed = (0, "")
		self._stdout.write(self.ANSI_restore_cursor_pos)
		self._stdout.write(self.ANSI_erase_to_eol)
		self._stdout.write("%(current)s of %(total)s   %(percent)3d%% in %(elapsed)ds  %(speed).2f %(speed_coeff)sB/s" % {
			"current" : str(self.current_position).rjust(len(str(self.total_size))),
			"total" : self.total_size,
			"percent" : self.total_size and (self.current_position * 100 / self.total_size) or 0,
			"elapsed" : sec_elapsed,
			"speed" : print_speed[0],
			"speed_coeff" : print_speed[1]
		})

		if done_message:
			self._stdout.write("  %s\n" % done_message)

		self._stdout.flush()

class ProgressCR(Progress):
    ## Uses CR char (Carriage Return) just like other progress bars do.
	CR_char = chr(13)

	def display(self, new_file = False, done_message = None):
		"""
		display(new_file = False[/True], done_message = None)
		"""
		if new_file:
			self.output_labels()
			return

		timedelta = self.time_current - self.time_start
		sec_elapsed = timedelta.days * 86400 + timedelta.seconds + float(timedelta.microseconds)/1000000.0
		if (sec_elapsed > 0):
			print_speed = Utils.formatSize((self.current_position - self.initial_position) / sec_elapsed, True, True)
		else:
			print_speed = (0, "")
		self._stdout.write(self.CR_char)
		output = " %(current)s of %(total)s   %(percent)3d%% in %(elapsed)4ds  %(speed)7.2f %(speed_coeff)sB/s" % {
			"current" : str(self.current_position).rjust(len(str(self.total_size))),
			"total" : self.total_size,
			"percent" : self.total_size and (self.current_position * 100 / self.total_size) or 0,
			"elapsed" : sec_elapsed,
			"speed" : print_speed[0],
			"speed_coeff" : print_speed[1]
		}
		self._stdout.write(output)
		if done_message:
			self._stdout.write("  %s\n" % done_message)

		self._stdout.flush()
