## 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