from ldtp import *
from ldtputils import *

def selectFolder(windowName,fldr,dest=''):
	if waittillguiexist(windowName) == 0:
		return 0
	if gettreetablerowindex ( windowName, 'ttblMailFolderTree', fldr) == -1:
		click(windowName, 'btnNew')
		waittillguiexist ('dlgCreatefolder')	
	        settextvalue('dlgCreatefolder', 'txtFoldername',fldr)
		selectrowpartialmatch ('dlgCreatefolder', 'ttblMailFolderTree', dest)
		log('User Value Entered', 'info')
		click('dlgCreatefolder', 'btnCreate')
		if guiexist('dlgEvolutionError'):
			click('dlgEvolutionError', 'btnOK')
			click('dlgCreatefolder', 'btnCancel')
			print 'Folder Name already exist'
			log('Folder Already exists', 'error')
	selectrowpartialmatch (windowName, 'ttblMailFolderTree', fldr)
	#waittillguiexist ('frmEvolution-'+_fldr+'*')
	log('Required folder selected', 'info')
	return 1

def verifyFolderExist (folderName):	
	try:
		if selectrowpartialmatch ('frm*Evolution', 'ttblMailFolderTree', folderName) == 1:
			waittillguiexist ('frm*Evolution-'+folderName+'*')
			log('Verify succeeded', 'pass')
			return 1
	except:
		msg = 'verify Failed'
		log(msg, 'error')
		raise LdtpExecutionError (msg)

def copyTo (fromFldr,tofldr):
		windowname = 'dlgSelectfolder'
try:
		if selectrowpartialmatch ('frm*Evolution', 'ttblMailFolderTree', fromFldr) == 1:
			waittillguiexist ('frm*Evolution-'+fromFldr+'*')
			log('From folder selected', 'info')
			selectmenuitem('frm*Evolution', 'mnuFolder;mnuCopyFolderTo')
			if waittillguiexist (windowName) == 0:
				msg = 'Select Folder did not arise'
				log (msg, 'cause')
				raise LdtpExecutionError (msg)
				
			if selectFolder(windowname,tofldr) == 1:
				log('Destionation folder selected')
			else:
				log('Unable to get hold of destination folder', 'error')
                        click(windowname, 'btnCopy')
                if guiexist ('dlgEvolutionError') == 1:
				msg = 'Evolution is offline'
				log (msg, 'cause')
				click('dlgEvolutionError', 'btnOK')
				raise LdtpExecutionError (msg)
		else:
			print 'Unable to find the source folder'
			msg = 'Unable to find the source folder'
			log(msg, 'cause')
			raise LdtpExecutionError (msg)
		print fromFldr+ ' has been copied to '+ tofldr
except :
		msg = 'Cannot copy the folder'
		log(msg, 'error')
		raise LdtpExecutionError (msg)

def moveTo (fromFldr,tofldr):
	windowname = 'dlgSelectfolder'
	try:
		if selectrowpartialmatch ('frm*Evolution', 'ttblMailFolderTree', fromFldr) == 1:
			waittillguiexist ('frm*Evolution-'+fromFldr+'*')
			log('From folder selected', 'info')
			selectmenuitem('frm*Evolution', 'mnuFolder;mnuMoveFolderTo')
			if waittillguiexist (windowName) == 0:
				msg = 'Select Folder did not arise'
				log (msg, 'cause')
				raise LdtpExecutionError (msg)
			if selectFolder(windowName,tofldr) == 1:
				log('Destionation folder selected')
			else:
				msg = 'Unable to get hold of destination folder'
				log(msg, 'error')
				raise LdtpExecutionError (msg)
			click(windowname, 'btnMove')
			if guiexist ('dlgEvolutionError') == 1:
				msg = 'Evolution is offline'
				log (msg,'cause')
				click('dlgEvolutionError', 'btnOK')
				raise LdtpExecutionError (msg)
		else:
			print 'Unable to find the source folder'
			msg = 'Unable to find the source folder'
			log(msg, 'cause')
			raise LdtpExecutionError (msg)
	except :
		msg = 'Cannot copy the folder'
		log(msg, 'error')
		raise LdtpExecutionError (msg)

def selectAll (fldrname):	
	try:
		#remap('evolution','frmEvolution-*')
		if selectrowpartialmatch ('frm*Evolution', 'ttblMailFolderTree', fldrname) == 1:
			waittillguiexist ('frm*Evolution-'+fldrname+'*')
			log('From folder selected', 'info')
			if selectmenuitem('frm*Evolution', 'mnuFolder;mnuSelectAllMessages') == 1:
				print 'All Mails have been selected'
				log('All items have been selected', 'info')	
				return 1
			else:
				print 'Unable to select all mails'
				return 0
		else:
			print 'Unable to find the folder'
			log('Unable to find the folder', 'cause')
			return 0
	except :
		print 'Cannot select all items in the folder'
		msg = 'Cannot select all the items'
		log(msg, 'error')
		raise LdtpExecutionError (msg)
	
def markAllRead(fldrname):	
	try:
		selectAll(fldrname)
		if selectmenuitem('frm*Evolution', 'mnuFolder;mnuMarkMessagesasRead') == 1:
			print 'All messages has been marked read'
			log('All items have been selected', 'info')	
		else:
			print 'Unable to select the mails'
			log('Unable to select the menu Markas;Read', 'error')
	except :
		print 'Cannot set all items in the folder as read'
		msg = 'Cannot mark as read'
		log(msg, 'error')
		raise LdtpExecutionError(msg)
	


def insertFollowup (followupflag, duedate, time, progress):
	try:
		windowName = 'dlgFlagtoFollowUp'
		waittillguiexist (windowName)
		settextvalue (windowName, 'txtFlag', followupflag)
		settextvalue (windowName, 'txtTextDateEntry', duedate)
		settextvalue (windowName, 'txt1', time)
		if progress == 'completed':
			check(windowName, 'chkCompleted')
		elif progress == 'not started':
			uncheck(windowName, 'chkCompleted')
		click(windowName, 'btnOK')
		log('Inserted the followup details')
		print 'Follow up details entered'
	except:
		msg = 'Unable to enter the given details'
		log(msg, 'error')
		print 'Unable to enter the follow up details'
		raise LdtpExecutionError (msg)

def markRead (subject, fldr=''):
	try:
		if fldr != '':
			selectrowpartialmatch ('frm*Evolution', 'ttblMessageFolderTree', fldr)
			waittillguiexist ('frm*Evolution-'+fldr+'*')
		selectrowpartialmatch ('frm*Evolution', 'ttblMessages', subject)
		try:
			selectmenuitem ('frm*Evolution', 'mnuMessage;mnuMarkas;mnuRead')
		except:
			log ('Already in Read state', 'info')
	except:
		msg = 'Unable to mark message as read'
		log (msg, 'error')
		raise LdtpExecutionError (msg)
	
def markUnread (subject, fldr=''):
	try:
		if fldr != '':
			selectrowpartialmatch ('frm*Evolution', 'ttblMessageFolderTree', fldr)
			waittillguiexist ('frm*Evolution-'+fldr+'*')
		selectrowpartialmatch ('frm*Evolution', 'ttblMessages', subject)
		try:
			selectmenuitem ('frm*Evolution', 'mnuMessage;mnuMarkas;mnuUnread')
		except:
			log ('Already in Unread state', 'info')
	except:
		msg = 'Unable to mark message as Unread'
		log (msg, 'error')
		raise LdtpExecutionError (msg)

def markJunk (subject, fldr=''):
	try:
		if fldr != '':
			selectrowpartialmatch ('frm*Evolution', 'ttblMessageFolderTree', fldr)
			waittillguiexist ('frm*Evolution-'+fldr+'*')
		selectrowpartialmatch ('frm*Evolution', 'ttblMessages', subject)
		try:
			selectmenuitem ('frm*Evolution', 'mnuMessage;mnuMarkas;mnuJunk')
		except:
			log ('Already in Junk state', 'info')
	except:
		msg = 'Unable to mark message as Junk'
		log (msg, 'error')
		raise LdtpExecutionError (msg)

def markUnjunk (subject, fldr=''):
	try:
		if fldr != '':
			selectrowpartialmatch ('frm*Evolution', 'ttblMessageFolderTree', fldr)
			waittillguiexist ('frm**Evolution-'+fldr+'*')
		selectrowpartialmatch ('frm*Evolution', 'ttblMessages', subject)
		try:
			selectmenuitem ('frm*Evolution', 'mnuMessage;mnuMarkas;mnuNotJunk')
		except:
			log ('Already in Not Junk state', 'info')
	except:
            msg ='Unable to mark message as not junk'
            log ( msg, 'error')
            raise LdtpExecutionError (msg)