Iterator.py 1.2 KB
#!/usr/bin/python
# -*- coding: utf-8 -*-

#Autor: Erickson Silva <erickson.silva@lavid.ufpb.br> <ericksonsilva@live.com>

from StringAux import *

class Iterator(object):

	# inicializacao das variaveis
	def __init__(self):	
		self.count = -1
		
	def load(self, lista):
		self.__list = list(lista);
		self.size = len(lista)

	def reset(self):
		self.count = -1

	def getSize(self):
		return self.size

	def getCount(self):
		return self.count

	def getToken(self, i):
		if(i == "+"):
			return self.__list[self.count+1]

		elif(i == "-"):
			return self.__list[self.count-1]

		else:
			return self.__list[self.count]

	def getAtualW(self):
		return remover_acentos(self.getToken(0)[0].upper().encode('utf-8'))

	def getAtualT(self):
		return self.getToken(0)[1].upper().encode('utf-8')

	def getProxW(self):
		return remover_acentos(self.getToken("+")[0].upper().encode('utf-8'))

	def getProxT(self):
		return self.getToken("+")[1].upper().encode('utf-8')		

	def getAntW(self):
		return remover_acentos(self.getToken("-")[0].upper().encode('utf-8'))

	def getAntT(self):
		return self.getToken("-")[1].upper().encode('utf-8')

	def hasNext(self):
		if(self.count < self.size-1):
			self.count += 1
			return True
		return False