Corrige função getToken() no Iterator

Erickson Silva
1 parent 2e84c19e
Showing 1 changed file with 3 additions and 9 deletions Show diff stats
src/new/Iterator.py
@@ -25,15 +25,9 @@ class Iterator(object):
 	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 getToken(self, i=None):
+		if(i != None): return self.__list[self.count+(i)]
+		return self.__list[self.count]
  
 	def getAtualW(self):
 		return self.getToken(0)[0].upper()
...	...	@@ -25,15 +25,9 @@ class Iterator(object):
25	25	def getCount(self):
26	26	return self.count
27	27
28		- def getToken(self, i):
29		- if(i == "+"):
30		- return self.__list[self.count+1]
31		-
32		- elif(i == "-"):
33		- return self.__list[self.count-1]
34		-
35		- else:
36		- return self.__list[self.count]
	28	+ def getToken(self, i=None):
	29	+ if(i != None): return self.__list[self.count+(i)]
	30	+ return self.__list[self.count]
37	31
38	32	def getAtualW(self):
39	33	return self.getToken(0)[0].upper()
...	...