data-transport/transport/couch.py

"""
Data-Transport
Steve L. Nyemba, The Phi Technology

This file is a wrapper around couchdb using IBM Cloudant SDK that has an interface to couchdb

"""
import cloudant
import json
		else:
			self.filename = None

	# def isready(self):
	# 	#
	# 	# Is the basic information about the database valid
	# 	#
	# 	p = Couchdb.isready(self)
		
	# 	if p == False:
	# 		return False
	# 	#
	# 	# The database name is set and correct at this point
	# 	# We insure the document of the given user has the requested attachment.
	# 	# 
		
	# 	doc = self.dbase.get(self.uid)
		
	# 	if '_attachments' in doc:
	# 		r = self.filename in doc['_attachments'].keys()
			
	# 	else:
	# 		r = False
		
	# 	return r	
	def stream(self):
		#
		# @TODO Need to get this working ...
		#
		document = cloudant.document.Document(self.dbase,self.uid)
		# content = self.dbase.fetch_attachment(self.uid,self.filename).split('\n') ;
		content = self.get_attachment(self.filename)
		for row in content:
			yield row
		
	def read(self,size=-1):
		if self.filename is not None:
			self.stream()
		else:
			return self.basic_read()
	def basic_read(self):
		document = cloudant.document.Document(self.dbase,self.uid)
		
		# document = self.dbase.get(self.uid)
		if document.exists() :			
			document.fetch()
			document = dict(document)
			del document['_rev']
		else:
			document = {}
		return document

class CouchWriter(Couch,Writer):		
	"""
		This class will write on a couchdb document provided a scope
		The scope is the attribute that will be on the couchdb document
	"""
	def __init__(self,**args):
		"""
			@param	uri		host & port reference
			@param	uid		user id involved
			@param	filename	filename (attachment)
			@param	dbname		database name (target)
		"""

		Couch.__init__(self,**args)

	def write(self,**params):
		"""
			write a given attribute to a document database
			@param	label	scope of the row repair|broken|fixed|stats
			@param	row	row to be written
		"""
		
		# document = self.dbase.get(self.uid)
		document = cloudant.document.Document(self.dbase,self.uid) #.get(self.uid)
		if document.exists() is False :
			document = self.dbase.create_document({"_id":self.uid})
		label = params['label']
		row	= params['row']
		if label not in document :
			document[label] = []
		document[label].append(row)
		document.save()
		# self.dbase.bulk_docs([document])
		# self.dbase.save_doc(document)
			
	def archive(self,params=None):
		"""
		This function will archive the document onto itself. 		
		"""
		# document = self.dbase.all_docs(self.uid,include_docs=True)
		document = cloudant.document.Document(self.dbase,self.filename)
		document.fetch()
		content = {}
		# _doc = {}
		for id in document:
			if  id not in ['_id','_rev','_attachments'] :
				content[id] = document[id]
				del document[id]
				
		content = json.dumps(content)	
		# document= _doc
		now = str(datetime.today())
		
		name = '-'.join([document['_id'] , now,'.json'])			
		# self.dbase.bulk_docs([document])
		# self.dbase.put_attachment(document,content,name,'application/json')
		document.put_attachment(self.dbase,name,'application/json',content)
		document.save()