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

from Utils import getTreeFromXml, unicodise, deunicodise
from logging import debug, info, warning, error

try:
	import xml.etree.ElementTree as ET
except ImportError:
	import elementtree.ElementTree as ET

class S3Exception(Exception):
	def __init__(self, message = ""):
		self.message = unicodise(message)

	def __str__(self):
		## Call unicode(self) instead of self.message because
		## __unicode__() method could be overriden in subclasses!
		return deunicodise(unicode(self))

	def __unicode__(self):
		return self.message

	## (Base)Exception.message has been deprecated in Python 2.6
	def _get_message(self):
		return self._message
	def _set_message(self, message):
		self._message = message
	message = property(_get_message, _set_message)


class S3Error (S3Exception):
	def __init__(self, response):
		self.status = response["status"]
		self.reason = response["reason"]
		self.info = {
			"Code" : "",
			"Message" : "",
			"Resource" : ""
		}
		debug("S3Error: %s (%s)" % (self.status, self.reason))
		if response.has_key("headers"):
			for header in response["headers"]:
				debug("HttpHeader: %s: %s" % (header, response["headers"][header]))
		if response.has_key("data"):
			tree = getTreeFromXml(response["data"])
			error_node = tree
			if not error_node.tag == "Error":
				error_node = tree.find(".//Error")
			for child in error_node.getchildren():
				if child.text != "":
					debug("ErrorXML: " + child.tag + ": " + repr(child.text))
					self.info[child.tag] = child.text
		self.code = self.info["Code"]
		self.message = self.info["Message"]
		self.resource = self.info["Resource"]

	def __unicode__(self):
		retval = u"%d " % (self.status)
		retval += (u"(%s)" % (self.info.has_key("Code") and self.info["Code"] or self.reason))
		if self.info.has_key("Message"):
			retval += (u": %s" % self.info["Message"])
		return retval

class CloudFrontError(S3Error):
	pass
		
class S3UploadError(S3Exception):
	pass

class S3DownloadError(S3Exception):
	pass

class S3RequestError(S3Exception):
	pass

class InvalidFileError(S3Exception):
	pass

class ParameterError(S3Exception):
	pass